ensured db connection closure in get_db_session method

GBSWHS · Nov 6, 2023 · 59bf219 · 59bf219
1 parent deed2d6
commit 59bf219
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/bus_boarding_api/db/dependencies.py b/bus_boarding_api/db/dependencies.py
@@ -9,6 +9,9 @@ async def get_db_session(request: Request) -> AsyncGenerator[AsyncSession, None]
 
     try:  # noqa: WPS501
         yield session
+    except Exception as e:
+        # logger.error(f"An error occurred: {e}")
+        await session.rollback()
+        raise
     finally:
-        await session.commit()
         await session.close()