GrowthOps_OS / app /database.py
Corin1998's picture
Update app/database.py
edf212d verified
raw
history blame contribute delete
700 Bytes
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker, DeclarativeBase
from .config import Settings, build_database_url
settings = Settings()
db_url = build_database_url(settings)
class Base(DeclarativeBase):
pass
engine_kwargs = {"echo": False, "pool_pre_ping": True}
if db_url.startswith("sqlite"):
# Uvicorn / FastAPI でスレッド跨ぎが起きても問題ないように
engine_kwargs["connect_args"] = {"check_same_thread": False}
engine = create_engine(db_url, **engine_kwargs)
SessionLocal = sessionmaker(bind=engine, autoflush=False, autocommit=False)
def get_db():
db = SessionLocal()
try:
yield db
finally:
db.close()