refactor _execute_sql to fix python <3.11 error and add custom exception

Signed-off-by: Grant Ramsay <seapagan@gmail.com>

Author: seapagan

sqliter/exceptions.py

@@ -139,3 +139,9 @@ class TableDeletionError(SqliterError):
     """Raised when a table cannot be deleted from the database."""
 
     message_template = "Failed to delete the table: '{}'"
+
+
+class SqlExecutionError(SqliterError):
+    """Raised when an SQL execution fails."""
+
+    message_template = "Failed to execute SQL: '{}'"

sqliter/sqliter.py

@@ -21,6 +21,7 @@
     RecordInsertionError,
     RecordNotFoundError,
     RecordUpdateError,
+    SqlExecutionError,
     TableCreationError,
     TableDeletionError,
 )
@@ -273,6 +274,14 @@ def create_table(
             raise TableCreationError(table_name) from exc
 
     def _execute_sql(self, sql: str) -> None:
+        """Execute an SQL statement.
+
+        Args:
+            sql: The SQL statement to execute.
+
+        Raises:
+            SqlExecutionError: If the SQL execution fails.
+        """
         if self.debug:
             self._log_sql(sql, [])
 
@@ -281,8 +290,8 @@ def _execute_sql(self, sql: str) -> None:
                 cursor = conn.cursor()
                 cursor.execute(sql)
                 conn.commit()
-        except sqlite3.Error as exc:
-            raise TableCreationError(sql) from exc
+        except (sqlite3.Error, sqlite3.Warning) as exc:
+            raise SqlExecutionError(sql) from exc
 
     def drop_table(self, model_class: type[BaseDBModel]) -> None:
         """Drop the table associated with the given model class.

tests/test_execute_sql.py

@@ -5,7 +5,7 @@
 import pytest
 
 from sqliter import SqliterDB
-from sqliter.exceptions import TableCreationError
+from sqliter.exceptions import SqlExecutionError
 
 
 class TestExecuteSQL:
@@ -39,7 +39,7 @@ def test_execute_sql_error(self) -> None:
         """Test SQL execution with an error."""
         # Missing closing parenthesis...
         invalid_sql = "CREATE TABLE test (id INTEGER PRIMARY KEY, name TEXT"
-        with pytest.raises(TableCreationError):
+        with pytest.raises(SqlExecutionError):
             self.db._execute_sql(invalid_sql)
 
     @patch("sqliter.sqliter.SqliterDB._log_sql")
@@ -73,7 +73,7 @@ def test_execute_sql_multiple_statements(self) -> None:
         CREATE TABLE test_multi (id INTEGER PRIMARY KEY, name TEXT);
         INSERT INTO test_multi (name) VALUES ('test');
         """
-        with pytest.raises(TableCreationError) as exc_info:
+        with pytest.raises(SqlExecutionError) as exc_info:
             self.db._execute_sql(sql)
 
         assert "You can only execute one statement at a time." in str(
@@ -86,7 +86,7 @@ def test_execute_sql_with_parameters(self) -> None:
             "CREATE TABLE test_params (id INTEGER PRIMARY KEY, name TEXT)"
         )
         sql = "INSERT INTO test_params (name) VALUES (?)"
-        with pytest.raises(TableCreationError):
+        with pytest.raises(SqlExecutionError):
             self.db._execute_sql(
                 sql
             )  # This should fail as _execute_sql doesn't support parameters