Merge pull request #390 from Mause/bugfix/fix-bleeding-edge-for-excep…

…tions

fix: fix bleeding edge duckdb for exceptions changes

duckdb_engine/__init__.py

@@ -33,6 +33,8 @@ class DBAPI:
 
     # this is being fixed upstream to add a proper exception hierarchy
     Error = getattr(duckdb, "Error", RuntimeError)
+    TransactionException = getattr(duckdb, "TransactionException", Error)
+    ParserException = getattr(duckdb, "ParserException", Error)
 
     @staticmethod
     def Binary(x: Any) -> Any:
@@ -188,7 +190,7 @@ def get_default_isolation_level(self, connection: "Connection") -> None:
     def do_rollback(self, connection: "Connection") -> None:
         try:
             super().do_rollback(connection)
-        except RuntimeError as e:
+        except DBAPI.TransactionException as e:
             if (
                 e.args[0]
                 != "TransactionContext Error: cannot rollback - no transaction is active"

duckdb_engine/tests/test_basic.py

@@ -261,9 +261,7 @@ def test_binary(session: Session) -> None:
 
 def test_comment_support() -> None:
     "comments not yet supported by duckdb"
-    exc = getattr(duckdb, "StandardException", DBAPI.Error)
-
-    with raises(exc, match="syntax error"):
+    with raises(DBAPI.ParserException, match="syntax error"):
         duckdb.default_connection.execute('comment on sqlite_master is "hello world";')