| 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() | |