bukang-backend / app /db /session.py
XRyZ's picture
Upload folder using huggingface_hub
01c6926 verified
Raw
History Blame Contribute Delete
384 Bytes
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from app.config import settings
engine = create_engine(
settings.database_url,
pool_pre_ping=True,
pool_recycle=300,
)
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
def get_db():
db = SessionLocal()
try:
yield db
finally:
db.close()