diff --git a/e-voting-system/backend/database.py b/e-voting-system/backend/database.py index a3ac03e..0ac977f 100644 --- a/e-voting-system/backend/database.py +++ b/e-voting-system/backend/database.py @@ -23,3 +23,12 @@ def init_db(): """Initialiser la base de données (créer les tables)""" from .models import Base Base.metadata.create_all(bind=engine) + + +def get_db(): + """Dépendance pour obtenir une session de base de données""" + db = SessionLocal() + try: + yield db + finally: + db.close()