database.py 412 B

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