database.py 414 B

123456789101112131415
  1. from sqlalchemy import create_engine
  2. from sqlalchemy.orm import sessionmaker, declarative_base
  3. from app.core.config import settings
  4. engine = create_engine(settings.database_url, echo=settings.debug)
  5. SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
  6. Base = declarative_base()
  7. # 依赖注入
  8. def get_db():
  9. db = SessionLocal()
  10. try:
  11. yield db
  12. finally:
  13. db.close()