Update db.py

This commit is contained in:
Timothy J. Baek 2024-07-03 23:32:46 -07:00
parent 864646094e
commit 37a5d2c06b

View File

@ -53,8 +53,19 @@ if "sqlite" in SQLALCHEMY_DATABASE_URL:
) )
else: else:
engine = create_engine(SQLALCHEMY_DATABASE_URL, pool_pre_ping=True) engine = create_engine(SQLALCHEMY_DATABASE_URL, pool_pre_ping=True)
SessionLocal = sessionmaker( SessionLocal = sessionmaker(
autocommit=False, autoflush=False, bind=engine, expire_on_commit=False autocommit=False, autoflush=False, bind=engine, expire_on_commit=False
) )
Base = declarative_base() Base = declarative_base()
Session = scoped_session(SessionLocal) Session = scoped_session(SessionLocal)
# Dependency
def get_db():
db = SessionLocal()
try:
yield db
finally:
db.close()