db.py 395 B

12345678910111213141516
  1. from sqlalchemy import create_engine
  2. from sqlalchemy.orm import declarative_base, sessionmaker
  3. from app.config import settings
  4. engine = create_engine(settings.database_url, pool_pre_ping=True)
  5. SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
  6. Base = declarative_base()
  7. def get_db():
  8. db = SessionLocal()
  9. try:
  10. yield db
  11. finally:
  12. db.close()