From 5f66e671a3168d4dd8c5815851e8538c9c242992 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Simon=20K=C3=BCnzel?= <simonk@fsmpi.rwth-aachen.de>
Date: Sat, 22 Jun 2024 18:59:42 +0200
Subject: [PATCH] Fix a return type

---
 src/api/database/mysql_connector.py    | 2 +-
 src/api/database/postgres_connector.py | 2 +-
 src/api/database/sqlite_connector.py   | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/api/database/mysql_connector.py b/src/api/database/mysql_connector.py
index 9a5ed04..76e54f9 100644
--- a/src/api/database/mysql_connector.py
+++ b/src/api/database/mysql_connector.py
@@ -59,7 +59,7 @@ class _MySqlDbConnection(PythonDbConnection[MySQLConnection, MySQLCursor]):
         self._closed = True
         self._py_connection.close()
     
-    def get_transaction_begin_statements(self, writable: bool, isolation_level: TransactionIsolationLevel) -> PreparedStatement or str:
+    def get_transaction_begin_statements(self, writable: bool, isolation_level: TransactionIsolationLevel) -> list[PreparedStatement or str]:
         if writable:
             return [
                 f"SET SESSION TRANSACTION ISOLATION LEVEL {isolation_level.value}",
diff --git a/src/api/database/postgres_connector.py b/src/api/database/postgres_connector.py
index 7975e82..8962b4a 100644
--- a/src/api/database/postgres_connector.py
+++ b/src/api/database/postgres_connector.py
@@ -90,7 +90,7 @@ source_function: {diagnostic.source_function}
         self._closed = True
         self._py_connection.close()
     
-    def get_transaction_begin_statements(self, writable: bool, isolation_level: TransactionIsolationLevel) -> PreparedStatement or str:
+    def get_transaction_begin_statements(self, writable: bool, isolation_level: TransactionIsolationLevel) -> list[PreparedStatement or str]:
         if writable:
             return [f"START TRANSACTION ISOLATION LEVEL {isolation_level.value}, READ WRITE"]
         else:
diff --git a/src/api/database/sqlite_connector.py b/src/api/database/sqlite_connector.py
index 3f5282e..8d47d25 100644
--- a/src/api/database/sqlite_connector.py
+++ b/src/api/database/sqlite_connector.py
@@ -62,7 +62,7 @@ class SqLiteDbConnection(PythonDbConnection[Connection, Cursor]):
         self._closed = True
         self._py_connection.close()
     
-    def get_transaction_begin_statements(self, writable: bool, isolation_level: TransactionIsolationLevel) -> PreparedStatement or str:
+    def get_transaction_begin_statements(self, writable: bool, isolation_level: TransactionIsolationLevel) -> list[PreparedStatement or str]:
         return ["BEGIN DEFERRED TRANSACTION"]  # Transactions in SQLite are serializable by default
     
     def get_transaction_end_statement(self, commit: bool) -> PreparedStatement or str:
-- 
GitLab