dataops-sdk 0.5.2__tar.gz → 0.5.3__tar.gz

{dataops_sdk-0.5.2 → dataops_sdk-0.5.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dataops-sdk
-Version: 0.5.2
+Version: 0.5.3
 Summary: Python SDK for the DataOps platform.
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown

{dataops_sdk-0.5.2 → dataops_sdk-0.5.3}/src/dataops_sdk/__init__.py

@@ -93,4 +93,4 @@ from .services import (  # noqa: F401
 )
 from .session import dataops_base_url_context, request_context
 
-__version__ = "0.5.2"
+__version__ = "0.5.3"

{dataops_sdk-0.5.2 → dataops_sdk-0.5.3}/src/dataops_sdk/runner/utils/dbConn_sync.py

@@ -27,12 +27,12 @@ from .dbConn_base import DatabaseConnBase
 # 数据库类型到驱动程序的映射
 # 用于 build_url 函数构建数据库连接 URL
 DBTYPE_DRIVER: Final[Mapping[str, str]] = {
-    "mysql": "mysqldb",
+    "mysql": "pymysql",
     "postgresql": "psycopg",
     "dm": "dmpython",
     "oracle": "oracledb",
-    "doris": "mysqldb",
-    "mariadb": "mysqldb",
+    "doris": "pymysql",
+    "mariadb": "pymysql",
 }
 
 
@@ -73,7 +73,7 @@ def build_url(
         raise NotImplementedError(f"{db_type} 尚未支持，请联系管理员")
 
     query: dict[str, str] = dict()
-    if db_type in {"mysql", "doris"}:
+    if db_type in {"mysql", "mariadb", "doris"}:
         query["local_infile"] = "True"
         query["ssl_disabled"] = "True"
 
@@ -548,8 +548,8 @@ def create_engine_safe(db_type: str, url: str, **kwargs: Any) -> Engine:
         raise NotImplementedError(f"{db_type} 尚未支持，请联系管理员")
 
     # 根据不同数据库类型调整连接参数
-    if db_type in ("mysql", "doris"):
-        # MySQL 需要添加一些兼容参数
+    if db_type in ("mysql", "mariadb", "doris"):
+        # MySQL 协议族需要添加一些兼容参数
         kwargs.setdefault("pool_pre_ping", True)
         kwargs.setdefault("pool_recycle", 3600)
     elif db_type == "postgresql":

{dataops_sdk-0.5.2 → dataops_sdk-0.5.3}/src/dataops_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dataops-sdk
-Version: 0.5.2
+Version: 0.5.3
 Summary: Python SDK for the DataOps platform.
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown

{dataops_sdk-0.5.2 → dataops_sdk-0.5.3}/src/dataops_sdk.egg-info/SOURCES.txt

@@ -89,6 +89,7 @@ src/dataops_sdk/utils/utils.py
 tests/test_constants.py
 tests/test_datasource_factory.py
 tests/test_db_conn_async.py
+tests/test_db_conn_sync.py
 tests/test_exceptions.py
 tests/test_session.py
 tests/test_utils.py

{dataops_sdk-0.5.2 → dataops_sdk-0.5.3}/tests/test_db_conn_async.py

@@ -66,12 +66,15 @@ def _conn(monkeypatch, session: _Session | None = None) -> DatabaseConnAsync:
 
 def test_build_async_url_drivers():
     mysql = build_async_url("127.0.0.1", 3306, "u", "p", "db", "mysql")
-    assert mysql.drivername == "mysql+mariadbconnector"
+    assert mysql.drivername == "mysql+asyncmy"
     assert mysql.query["local_infile"] == "True"
     assert mysql.query["ssl_disabled"] == "True"
 
     mariadb = build_async_url("127.0.0.1", 3306, "u", "p", "db", "mariadb")
-    assert mariadb.drivername == "mariadb+mariadbconnector"
+    assert mariadb.drivername == "mariadb+asyncmy"
+
+    doris = build_async_url("127.0.0.1", 9030, "u", "p", "db", "doris")
+    assert doris.drivername == "doris+asyncmy"
 
     postgres = build_async_url("127.0.0.1", 5432, "u", "p", "db", "postgresql")
     assert postgres.drivername == "postgresql+psycopg"

dataops_sdk-0.5.3/tests/test_db_conn_sync.py

@@ -0,0 +1,26 @@
+"""DatabaseConn 同步连接 URL 构建单元测试."""
+
+import pytest
+from dataops_sdk.runner.utils.dbConn_sync import DBTYPE_DRIVER, build_url
+
+
+def test_build_url_drivers():
+    mysql = build_url("127.0.0.1", 3306, "u", "p", "db", "mysql")
+    assert mysql.drivername == "mysql+pymysql"
+    assert mysql.query["local_infile"] == "True"
+    assert mysql.query["ssl_disabled"] == "True"
+
+    mariadb = build_url("127.0.0.1", 3306, "u", "p", "db", "mariadb")
+    assert mariadb.drivername == "mariadb+pymysql"
+
+    doris = build_url("127.0.0.1", 9030, "u", "p", "db", "doris")
+    assert doris.drivername == "doris+pymysql"
+
+    postgres = build_url("127.0.0.1", 5432, "u", "p", "db", "postgresql")
+    assert postgres.drivername == "postgresql+psycopg"
+    assert "local_infile" not in postgres.query
+
+    assert DBTYPE_DRIVER["oracle"] == "oracledb"
+
+    with pytest.raises(NotImplementedError):
+        build_url("127.0.0.1", 5236, "u", "p", "db", "unknown")