From 1fd71e71e14efdcb2714411c9e4f0407c713261f Mon Sep 17 00:00:00 2001 From: Alexis Bruneteau Date: Fri, 7 Nov 2025 03:08:33 +0100 Subject: [PATCH] fix: Add missing get_db function to database.py The main.py was trying to import get_db for blockchain initialization but it was missing from database.py. Added the get_db generator function that creates and properly closes database sessions. --- e-voting-system/backend/database.py | 9 +++++++++ 1 file changed, 9 insertions(+) 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()