From cc4db16f2da30db38a745c7e5c19db2afa2a3c18 Mon Sep 17 00:00:00 2001 From: Domingo Dirutigliano Date: Fri, 22 Sep 2023 21:08:23 +0200 Subject: [PATCH] fix: enable transaction in db.queries --- backend/utils/sqlite.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/backend/utils/sqlite.py b/backend/utils/sqlite.py index c22c2d0..6bc688a 100644 --- a/backend/utils/sqlite.py +++ b/backend/utils/sqlite.py @@ -64,6 +64,7 @@ class SQLite(): def queries_iter(self, queries: list[tuple[str, ...]]): cur = self.conn.cursor() try: + cur.execute("BEGIN") for query_data in queries: values = [] str_query = None @@ -75,6 +76,10 @@ class SQLite(): if str_query: cur.execute(str_query, values) yield cur.fetchall() + cur.execute("COMMIT") + except Exception as e: + cur.execute("ROLLBACK") + raise e finally: cur.close() try: self.conn.commit()