Add files via upload

backend/db/database.py

@@ -0,0 +1,53 @@
+"""
+NeuroSploit v3 - Database Configuration
+"""
+from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine, async_sessionmaker
+from sqlalchemy.orm import DeclarativeBase
+from backend.config import settings
+
+
+class Base(DeclarativeBase):
+    """Base class for all models"""
+    pass
+
+
+# Create async engine
+engine = create_async_engine(
+    settings.DATABASE_URL,
+    echo=settings.DEBUG,
+    future=True
+)
+
+# Create async session factory
+async_session_maker = async_sessionmaker(
+    engine,
+    class_=AsyncSession,
+    expire_on_commit=False
+)
+
+# Alias for background tasks
+async_session_factory = async_session_maker
+
+
+async def get_db() -> AsyncSession:
+    """Dependency to get database session"""
+    async with async_session_maker() as session:
+        try:
+            yield session
+            await session.commit()
+        except Exception:
+            await session.rollback()
+            raise
+        finally:
+            await session.close()
+
+
+async def init_db():
+    """Initialize database tables"""
+    async with engine.begin() as conn:
+        await conn.run_sync(Base.metadata.create_all)
+
+
+async def close_db():
+    """Close database connection"""
+    await engine.dispose()