| 123456789101112131415161718192021222324252627282930 |
- """Database connection and session management (SQLite + SQLAlchemy)."""
- from sqlalchemy import create_engine
- from sqlalchemy.orm import sessionmaker, declarative_base
- from .config import DATABASE_URL
- # SQLite engine (check_same_thread=False for FastAPI async)
- engine = create_engine(
- DATABASE_URL,
- connect_args={"check_same_thread": False},
- echo=False,
- )
- SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
- Base = declarative_base()
- def get_db():
- """Dependency for FastAPI route database sessions."""
- db = SessionLocal()
- try:
- yield db
- finally:
- db.close()
- def init_db():
- """Create all database tables."""
- from . import models # noqa: F401 - ensure models are imported
- Base.metadata.create_all(bind=engine)
|
