database: handle failure on transation commit

zidel · Feb 16, 2024 · 05c439c · 05c439c
1 parent 5a79150
commit 05c439c
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/database.py b/database.py
@@ -29,17 +29,18 @@ def __init__(self, cursor, name):
                 self._name = name
 
             def __enter__(self):
-                log.debug('Transaction start, %s', self._name)
                 self._cursor.execute('begin')
                 return self._cursor
 
             def __exit__(self, type, value, traceback):
                 if type is None and value is None and traceback is None:
-                    self._cursor.execute('commit')
-                    log.debug('Transaction commit, %s', self._name)
+                    try:
+                        self._cursor.execute('commit')
+                    except sqlite3.OperationalError as e:
+                        self._cursor.execute('rollback')
+                        raise
                 else:
                     self._cursor.execute('rollback')
-                    log.debug('Transaction rollback, %s', self._name)
 
         return Transaction(self._cursor(), name)