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()