attrition-api-dev / app /db /session.py
rriviere's picture
Upload folder using huggingface_hub
e406ddb verified
Raw History Blame Contribute Delete
326 Bytes
from fastapi import Request
from sqlalchemy.orm import Session
from typing import Generator
def get_db(request: Request) -> Generator[Session, None, None]:
db = request.app.state.SessionLocal()
try:
yield db
finally:
# Garantit la fermeture de la session même en cas d'erreur
db.close()