U [Mhj@s6dZddlZddlZddlmZmZddlmZddlm Z ej ej ej eddlZddlmZmZddlZddlmZe dd Zd d Zed kr2ed ededZesedededededsededeedededeeedS)z PostgreSQL database setup script for the timesheet application This script creates the necessary tables in your existing PostgreSQL database N) create_enginetext) sessionmaker) load_dotenv)Baseengine) UserCreatec Cstdz(t.}|td}|d}td|W5QRXtdtjjtdtdt ddtd }|}zz| t j }td |d |dkrtd td ddd}t||}td|jd|jdtdntdW5| XtdtdtdtdtdtdWn\tk r}z\sz)check_existing_tables..r z existing tables:z - ZusersZactivity_recordscsg|]}|kr|qSrr)r!tZ our_tablesrrr#esz! Timesheet tables already exist: zBDo you want to continue? This will not drop existing data. (y/N): yzSetup cancelled.rzNo tables found in databasez*Warning: Could not check existing tables: N) rrrrrZfetchallleninputlowerrrr)rrZtablestableZexisting_our_tablesZresponserrr%rcheck_existing_tablesNs(   r+__main__z3PostgreSQL Database Setup for Timesheet Applicationz2==================================================Z DATABASE_URLu3❌ DATABASE_URL not found in environment variableszAPlease create a .env file with your PostgreSQL connection string:zCDATABASE_URL=postgresql://username:password@localhost/your_databaser Z postgresqluE❌ DATABASE_URL does not appear to be a PostgreSQL connection stringzCurrent DATABASE_URL: zGExpected format: postgresql://username:password@localhost/database_namezDatabase URL: )__doc__osrZ sqlalchemyrrZsqlalchemy.ormrZdotenvrpathappenddirnameabspath__file__rZdatabaserrrZschemasrr r+__name__rgetenvZ database_urlr startswithrrrrs<   8%