From a154b25415e6d4f6875ccd3a5f23e957e270012a Mon Sep 17 00:00:00 2001
From: Dylan De Faoite <dylanseandefaoite@gmail.com>
Date: Thu, 5 Mar 2026 10:09:14 +0000
Subject: [PATCH] fix(db): missing rollback on execute_batch method

Arguably more important on a batch function to have rollback.
---
 server/db/database.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/server/db/database.py b/server/db/database.py
index 346ef8b..f56f579 100644
--- a/server/db/database.py
+++ b/server/db/database.py
@@ -38,9 +38,13 @@ class PostgresConnector:
             raise
 
     def execute_batch(self, query, values):
-        with self.connection.cursor(cursor_factory=RealDictCursor) as cursor:
-            execute_batch(cursor, query, values)
+        try:
+            with self.connection.cursor(cursor_factory=RealDictCursor) as cursor:
+                execute_batch(cursor, query, values)
             self.connection.commit()
+        except Exception:
+            self.connection.rollback()
+            raise
 
     def close(self):
         if self.connection: