File size: 634 Bytes
26e1c2e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 | 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()
|