db-init: Implement database initialization and seeding

Backend/database/__init__.py

@@ -0,0 +1,2 @@
+from .connection import engine, async_session_factory, get_db, get_db_context, init_db, close_db
+from .models import Base, Issue, IssueImage, Classification, IssueEvent, Department, Member, Escalation

Backend/database/init_db.py

@@ -0,0 +1,46 @@
+import asyncio
+import logging
+from sqlalchemy.ext.asyncio import create_async_engine
+from Backend.core.config import settings
+from Backend.database.models import Base
+from Backend.database.seed import seed_data
+
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+async def init_models():
+    """Drops and recreates all tables, then seeds initial data."""
+    logger.info("Initializing database...")
+    
+    
+    database_url = settings.database_url.replace("port=6543", "port=5432").replace("postgresql://", "postgresql+asyncpg://")
+    engine = create_async_engine(
+        database_url, 
+        echo=True,
+        connect_args={
+            "statement_cache_size": 0,
+            "prepared_statement_cache_size": 0,
+        }
+    )
+    
+    async with engine.begin() as conn:
+        logger.info("Dropping existing tables...")
+        
+        
+        
+        
+        logger.info("Creating new tables...")
+        await conn.run_sync(Base.metadata.create_all)
+    
+    logger.info("Schema initialized. Seeding data...")
+    try:
+        await seed_data(engine)
+        logger.info("Seeding completed successfully!")
+    except Exception as e:
+        logger.error(f"Seeding failed: {e}")
+    
+    await engine.dispose()
+    logger.info("Database initialization finished.")
+
+if __name__ == "__main__":
+    asyncio.run(init_models())

Backend/database/seed.py

@@ -0,0 +1,83 @@
+import logging
+import uuid
+from sqlalchemy.ext.asyncio import AsyncEngine, AsyncSession
+from sqlalchemy.orm import sessionmaker
+from Backend.database.models import Department, Member
+
+logger = logging.getLogger(__name__)
+
+async def seed_data(engine: AsyncEngine):
+    async_session = sessionmaker(engine, class_=AsyncSession, expire_on_commit=False)
+    
+    async with async_session() as session:
+        
+        pwd_id = uuid.uuid4()
+        sanitation_id = uuid.uuid4()
+        traffic_id = uuid.uuid4()
+        
+        departments = [
+            Department(
+                id=pwd_id,
+                name="Public Works Department",
+                code="PWD",
+                description="Roads, Potholes, Infrastructure",
+                default_sla_hours=48,
+                escalation_email="pwd_head@city.gov"
+            ),
+            Department(
+                id=sanitation_id,
+                name="Sanitation Department",
+                code="SANITATION",
+                description="Garbage, Cleaning, Waste",
+                default_sla_hours=24,
+                escalation_email="sanitation_head@city.gov"
+            ),
+            Department(
+                id=traffic_id,
+                name="Traffic Department",
+                code="TRAFFIC",
+                description="Signals, Signs, Illegal Parking",
+                default_sla_hours=12,
+                escalation_email="traffic_head@city.gov"
+            )
+        ]
+        
+        for dept in departments:
+            session.add(dept)
+        
+        
+        members = [
+            Member(
+                department_id=pwd_id,
+                name="Ramesh Kumar",
+                email="ramesh.pwd@city.gov",
+                role="officer",
+                city="New Delhi",
+                locality="Connaught Place",
+                max_workload=10
+            ),
+            Member(
+                department_id=sanitation_id,
+                name="Suresh Singh",
+                email="suresh.sanitation@city.gov",
+                role="officer",
+                city="New Delhi",
+                locality="Karol Bagh",
+                max_workload=15
+            ),
+             Member(
+                department_id=traffic_id,
+                name="Priya Sharma",
+                email="priya.traffic@city.gov",
+                role="officer",
+                city="New Delhi",
+                locality="Lajpat Nagar",
+                max_workload=12
+            )
+        ]
+        
+        for member in members:
+            session.add(member)
+            
+        await session.commit()
+        logger.info("Seeded 3 departments and 3 members.")