From d445143e28e0c0d062b98492820a96d367556e33 Mon Sep 17 00:00:00 2001 From: Caio Hamamura Date: Tue, 15 Apr 2025 13:19:01 -0300 Subject: [PATCH] fix: handle set_isolation_level gracefully Whenever the sqlachemy tries to use set_isolation_level without it being defined it will cast `object has no attribute 'set_isolation_level'` instead of gracefully handling the error by explicitly raising it is not implemented. --- duckdb_engine/__init__.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/duckdb_engine/__init__.py b/duckdb_engine/__init__.py index e6b1680e2..fff2b968c 100644 --- a/duckdb_engine/__init__.py +++ b/duckdb_engine/__init__.py @@ -330,6 +330,9 @@ def _get_server_version_info(self, connection: "Connection") -> Tuple[int, int]: def get_default_isolation_level(self, connection: "Connection") -> None: raise NotImplementedError() + def set_isolation_level(self, connection: "Connection", level: str) -> None: + raise NotImplementedError() + def do_rollback(self, connection: "Connection") -> None: try: super().do_rollback(connection)