CaiYouHui后端fastapi实现

database.py 371B

123456789101112131415
  1. # app/dependencies/database.py
  2. from sqlalchemy.orm import Session
  3. from contextlib import contextmanager
  4. from typing import Generator
  5. # ✅ 正确导入方式
  6. from app.database import SessionLocal
  7. def get_async_db() -> Generator[Session, None, None]:
  8. """数据库会话依赖注入"""
  9. db = SessionLocal()
  10. try:
  11. yield db
  12. finally:
  13. db.close()