datus-sqlalchemy 0.1.7__tar.gz → 0.1.8__tar.gz

{datus_sqlalchemy-0.1.7 → datus_sqlalchemy-0.1.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datus-sqlalchemy
-Version: 0.1.7
+Version: 0.1.8
 Summary: SQLAlchemy base connector for Datus database adapters
 Project-URL: Homepage, https://github.com/Datus-ai/datus-db-adapters
 Project-URL: Repository, https://github.com/Datus-ai/datus-db-adapters
@@ -14,7 +14,7 @@ Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.12
 Requires-Python: >=3.12
-Requires-Dist: datus-db-core>=0.1.3
+Requires-Dist: datus-db-core>=0.1.5
 Requires-Dist: pandas>=2.1.4
 Requires-Dist: pyarrow<19.0.0,>=14.0.0
 Requires-Dist: sqlalchemy>=2.0.23

{datus_sqlalchemy-0.1.7 → datus_sqlalchemy-0.1.8}/datus_sqlalchemy/connector.py

@@ -40,6 +40,32 @@ from datus_db_core import (
 
 logger = get_logger(__name__)
 
+_SQL_PREVIEW_CHARS = 50
+
+
+def _log_sql_exception(event: str, sql: str, exc: Exception) -> None:
+    """Log a SQL failure with a bounded preview instead of the full statement."""
+    statement = sql or ""
+    normalized_sql = " ".join(statement.split())
+    sql_preview = normalized_sql[:_SQL_PREVIEW_CHARS]
+    if len(normalized_sql) > _SQL_PREVIEW_CHARS:
+        sql_preview += "..."
+    driver_error = str(getattr(exc, "orig", None) or exc) or type(exc).__name__
+    if statement:
+        driver_error = driver_error.replace(statement, "<sql>")
+    if normalized_sql and normalized_sql != statement:
+        driver_error = driver_error.replace(normalized_sql, "<sql>")
+    safe_exception = RuntimeError(driver_error)
+    logger.error(
+        "%s; sql_preview=%r; sql_chars=%d; error_type=%s; error=%s",
+        event,
+        sql_preview,
+        len(statement),
+        type(exc).__name__,
+        driver_error,
+        exc_info=(type(safe_exception), safe_exception, exc.__traceback__),
+    )
+
 
 class SQLAlchemyConnector(BaseSqlConnector, MigrationTargetMixin):
     """
@@ -170,6 +196,7 @@ class SQLAlchemyConnector(BaseSqlConnector, MigrationTargetMixin):
         """Map SQLAlchemy exceptions to Datus exceptions."""
         if isinstance(e, DatusDbException):
             return e
+        _log_sql_exception(f"{operation} failed", sql, e)
 
         # Extract error message
         if hasattr(e, "detail") and e.detail:
@@ -767,7 +794,7 @@ class SQLAlchemyConnector(BaseSqlConnector, MigrationTargetMixin):
                         yield ()
                     yield from []
         except Exception as e:
-            raise self._handle_exception(e) from e
+            raise self._handle_exception(e, sql, "streaming query") from e
 
     # ==================== MigrationTargetMixin (generic fallback) ====================
     #

{datus_sqlalchemy-0.1.7 → datus_sqlalchemy-0.1.8}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "datus-sqlalchemy"
-version = "0.1.7"
+version = "0.1.8"
 description = "SQLAlchemy base connector for Datus database adapters"
 readme = "README.md"
 requires-python = ">=3.12"
@@ -18,7 +18,7 @@ classifiers = [
 ]
 
 dependencies = [
-    "datus-db-core>=0.1.3",
+    "datus-db-core>=0.1.5",
     "sqlalchemy>=2.0.23",
     "pyarrow>=14.0.0,<19.0.0",
     "pandas>=2.1.4",

{datus_sqlalchemy-0.1.7 → datus_sqlalchemy-0.1.8}/tests/unit/test_connector_unit.py

@@ -255,13 +255,16 @@ def test_execute_ddl_with_context_params():
         ),
     ],
 )
-def test_handle_exception_classifies_common_failures(exception, expected_code):
+def test_handle_exception_classifies_and_logs_common_failures(exception, expected_code, caplog):
     """SQLAlchemy failures should map to stable Datus error categories."""
     connector = DummySQLAlchemyConnector("sqlite://", dialect="sqlite")
 
     classified = connector._handle_exception(exception, "SELECT 1", "query")
 
     assert classified.code == expected_code
+    assert "query failed; sql_preview='SELECT 1'; sql_chars=8" in caplog.text
+    assert caplog.text.count("SELECT 1") == 1
+    assert caplog.records[-1].exc_info[2] is exception.__traceback__
 
 
 def test_ensure_engine_classifies_connection_creation_failure():