Rollback failed transaction

2023-01-23 19:54:20 +11:00
parent 0e5e1a8b76
commit c85139d8ae
1 changed files with 6 additions and 1 deletions
--- a/db.py
+++ b/db.py
@@ -18,7 +18,12 @@ class DataBase():

    def execute(self, query, args=(), one=False, commit=False):
        cur = self.conn.cursor(cursor_factory=RealDictCursor)
-        cur.execute(query, args)
+        try:
+            cur.execute(query, args)
+        except:
+            self.conn.rollback()
+            raise Exception('Rolling back transaction...')
+
        rv = None
        if cur.description is not None:
            rv = cur.fetchall()