| from collections.abc import Generator | |
| from sqlalchemy import create_engine | |
| from sqlalchemy.orm import Session, sessionmaker | |
| from app.config import settings | |
| _is_sqlite = settings.murshid_db_url.startswith("sqlite") | |
| engine = create_engine( | |
| settings.murshid_db_url, | |
| connect_args={"check_same_thread": False} if _is_sqlite else {}, | |
| pool_pre_ping=not _is_sqlite, | |
| pool_recycle=3600 if not _is_sqlite else -1, | |
| ) | |
| SessionLocal = sessionmaker(bind=engine, autocommit=False, autoflush=False) | |
| def get_db() -> Generator[Session, None, None]: | |
| db = SessionLocal() | |
| try: | |
| yield db | |
| finally: | |
| db.close() | |