Simplify DB init - ignore 'already exists' if database works

start.sh

@@ -17,10 +17,9 @@ fi
 echo "Initializing database..."
 python -c "
 from web_app import create_app
-from web_app.models import db
+from web_app.models import db, User
 from config import Config
 import sqlalchemy
-from sqlalchemy import text
 
 print(f'SECRET_KEY set: {bool(Config.SECRET_KEY)}')
 print(f'IS_PRODUCTION: {Config.IS_PRODUCTION}')
@@ -30,50 +29,28 @@ print(f'Database URI: {Config.SQLALCHEMY_DATABASE_URI[:50]}...')
 app = create_app()
 with app.app_context():
     try:
-        # Check if users table exists using inspector
-        inspector = sqlalchemy.inspect(db.engine)
-        tables = inspector.get_table_names()
-        
-        if 'users' in tables:
-            from web_app.models import User
-            user_count = User.query.count()
-            print(f'✅ Database already initialized ({user_count} users, {len(tables)} tables)')
-        else:
-            if tables:
-                print(f'Incomplete database found: {tables}')
-                print('Manually dropping all tables and constraints...')
-                with db.engine.connect() as conn:
-                    # Drop each table with CASCADE
-                    for table in tables:
-                        try:
-                            conn.execute(text(f'DROP TABLE IF EXISTS {table} CASCADE'))
-                            conn.commit()
-                            print(f'  Dropped: {table}')
-                        except Exception as e:
-                            print(f'  Error dropping {table}: {e}')
-                    
-                    # Drop orphaned constraints if any
-                    result = conn.execute(text(\"\"\"
-                        SELECT constraint_name, table_name 
-                        FROM information_schema.table_constraints 
-                        WHERE constraint_schema = 'public'
-                    \"\"\"))
-                    constraints = result.fetchall()
-                    for constraint_name, table_name in constraints:
-                        try:
-                            conn.execute(text(f'ALTER TABLE {table_name} DROP CONSTRAINT IF EXISTS {constraint_name} CASCADE'))
-                            conn.commit()
-                        except:
-                            pass
-                print('✅ Cleanup complete')
-            else:
-                print('Empty database - creating schema...')
-            
+        # Try to query users table - if this works, DB is fine
+        user_count = User.query.count()
+        print(f'✅ Database working ({user_count} users)')
+    except Exception as check_error:
+        print(f'Database needs setup: {str(check_error)[:100]}')
+        try:
+            # Try to create tables
             db.create_all()
-            print('✅ Database schema created successfully!')
-    except Exception as e:
-        print(f'Database error: {str(e)[:200]}')
-        raise
+            print('✅ Database schema created!')
+        except Exception as create_error:
+            error_msg = str(create_error)
+            # If error is just 'already exists', check if database actually works
+            if 'already exists' in error_msg.lower():
+                print('⚠️  Create returned already exists error, testing database...')
+                try:
+                    user_count = User.query.count()
+                    print(f'✅ Database is functional ({user_count} users) - ignoring duplicate error')
+                except Exception as test_error:
+                    print(f'❌ Database not functional: {test_error}')
+                    raise
+            else:
+                raise
 "
 
 echo "Starting gunicorn server..."