from sqlalchemy import create_engine from sqlalchemy.orm import DeclarativeBase, sessionmaker from app.config import settings connect_args={"check_same_thread":False} if settings.DATABASE_URL.startswith("sqlite") else {} engine=create_engine(settings.DATABASE_URL, connect_args=connect_args) SessionLocal=sessionmaker(bind=engine, autocommit=False, autoflush=False) class Base(DeclarativeBase): pass def get_db(): db=SessionLocal() try: yield db finally: db.close()