From 21e9cbef4b71fd463486096b38b0dc216ce218af Mon Sep 17 00:00:00 2001 From: tcely Date: Thu, 12 Dec 2024 09:03:08 -0500 Subject: [PATCH] Cleanup for _start_transaction_under_autocommit --- tubesync/tubesync/sqlite3/base.py | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/tubesync/tubesync/sqlite3/base.py b/tubesync/tubesync/sqlite3/base.py index d09d68cd..28d0ebc7 100644 --- a/tubesync/tubesync/sqlite3/base.py +++ b/tubesync/tubesync/sqlite3/base.py @@ -5,15 +5,14 @@ class DatabaseWrapper(base.DatabaseWrapper): def _start_transaction_under_autocommit(self): conn_params = self.get_connection_params() - if "transaction_mode" not in conn_params: - self.cursor().execute("BEGIN TRANSACTION") - else: + transaction_modes = frozenset(["DEFERRED", "EXCLUSIVE", "IMMEDIATE"]) + + sql_statement = "BEGIN TRANSACTION" + if "transaction_mode" in conn_params: tm = str(conn_params["transaction_mode"]).upper().strip() - transaction_modes = frozenset(["DEFERRED", "EXCLUSIVE", "IMMEDIATE"]) if tm in transaction_modes: - self.cursor().execute(f"BEGIN {tm} TRANSACTION") - else: - self.cursor().execute("BEGIN TRANSACTION") + sql_statement = f"BEGIN {tm} TRANSACTION" + self.cursor().execute(sql_statement) def init_connection_state(self):