From c85139d8aeb613ac66ea35849df5a31787d1f205 Mon Sep 17 00:00:00 2001 From: Peter Stockings Date: Mon, 23 Jan 2023 19:54:20 +1100 Subject: [PATCH] Rollback failed transaction --- db.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/db.py b/db.py index 377194f..0b7ae9d 100644 --- 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()