datus-sqlalchemy 0.1.2__tar.gz → 0.1.3__tar.gz

{datus_sqlalchemy-0.1.2 → datus_sqlalchemy-0.1.3}/.gitignore

@@ -135,3 +135,6 @@ dmypy.json
 # OS
 .DS_Store
 Thumbs.db
+
+
+.omc

{datus_sqlalchemy-0.1.2 → datus_sqlalchemy-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datus-sqlalchemy
-Version: 0.1.2
+Version: 0.1.3
 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-agent>0.2.1
+Requires-Dist: datus-agent>0.2.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.2 → datus_sqlalchemy-0.1.3}/datus_sqlalchemy/connector.py

@@ -51,7 +51,7 @@ class SQLAlchemyConnector(BaseSqlConnector):
         # Auto-detect dialect from connection string if not provided
         if not dialect:
             prefix = connection_string.split(":")[0] if isinstance(connection_string, str) else "unknown"
-            dialect = DBType.MYSQL if prefix == "mysql+pymysql" else prefix
+            dialect = "mysql" if prefix == "mysql+pymysql" else prefix
 
         config = ConnectionConfig(timeout_seconds=timeout_seconds)
         super().__init__(config, dialect)
@@ -257,8 +257,10 @@ class SQLAlchemyConnector(BaseSqlConnector):
                 rows = result.fetchall()
                 return [row._asdict() for row in rows]
         except DatusException:
+            self._safe_rollback()
             raise
         except Exception as e:
+            self._safe_rollback()
             raise self._handle_exception(e, sql, "query") from e
 
     @override
@@ -392,7 +394,7 @@ class SQLAlchemyConnector(BaseSqlConnector):
                 conn.commit()
 
             # Update context if applicable
-            if self.dialect != DBType.SQLITE.value:
+            if self.dialect != "sqlite":
                 context = parse_context_switch(sql=sql, dialect=self.dialect)
                 if context:
                     if catalog := context.get("catalog_name"):

{datus_sqlalchemy-0.1.2 → datus_sqlalchemy-0.1.3}/pyproject.toml

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