clickzetta-dbutils 1.0.6__py3-none-any.whl → 1.0.7__py3-none-any.whl

clickzetta_dbutils/db_utils.py

@@ -255,6 +255,7 @@ class DatabaseConnectionManager:
                     query=query_params
                 )
             elif conn_info.magicToken:
+                # Use magic token for authentication, do not require username and password
                 query_params["magic_token"] = conn_info.magicToken
                 url = URL.create(
                     drivername="clickzetta",
@@ -287,6 +288,8 @@ def get_active_engine(
         workspace: Optional[str] = None,
         schema: Optional[str] = None,
         options: Optional[Dict[str, str]] = None,
+        query: Optional[Dict[str, str]] = None,
+        driver: Optional[str] = None,
         *args, **kwargs
 ) -> Engine:
     """
@@ -298,6 +301,9 @@ def get_active_engine(
         workspace (str, optional): Workspace name. Default is 'default'.
         schema (str, optional): Schema name for the connection. Default is 'public'.
         options (dict, optional): Additional connection options.
+        query (dict, optional): Additional query parameters for SQLAlchemy url.
+        driver (str, optional): Driver name for the connection.
+        *args: Additional arguments for SQLAlchemy engine.
 
     Returns:
         SQLAlchemy Engine instance
@@ -312,5 +318,9 @@ def get_active_engine(
         manager.use_vcluster(vcluster)
     if options:
         manager.use_options(options)
+    if query:
+        manager.use_query(query)
+    if driver:
+        manager.use_driver(driver)
 
     return manager.build(*args, **kwargs)

clickzetta_dbutils/version.py

@@ -1 +1 @@
-__version__ = "1.0.6"
+__version__ = "1.0.7"

{clickzetta_dbutils-1.0.6.dist-info → clickzetta_dbutils-1.0.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: clickzetta-dbutils
-Version: 1.0.6
+Version: 1.0.7
 Summary: clickzetta dbutils
 Author-email: "lin.zhang" <lin.zhang@clickzetta.com>
 Project-URL: documentation, https://www.yunqi.tech/
@@ -16,6 +16,8 @@ Provides-Extra: dev
 Requires-Dist: pytest==8.2.1; extra == "dev"
 Requires-Dist: build; extra == "dev"
 Requires-Dist: pytest-xdist; extra == "dev"
+Provides-Extra: cliclzetta-legecy-sdk
+Requires-Dist: clickzetta-connector; extra == "cliclzetta-legecy-sdk"
+Requires-Dist: clickzetta-sqlalchemy; extra == "cliclzetta-legecy-sdk"
 Provides-Extra: cliclzetta
-Requires-Dist: clickzetta-connector; extra == "cliclzetta"
-Requires-Dist: clickzetta-sqlalchemy; extra == "cliclzetta"
+Requires-Dist: clickzetta-connector-python; extra == "cliclzetta"

clickzetta_dbutils-1.0.7.dist-info/RECORD

@@ -0,0 +1,7 @@
+clickzetta_dbutils/__init__.py,sha256=Q_6kas0RvZ0767qlaA_xGESmXxm0dks1YQ8DqCX8LV0,290
+clickzetta_dbutils/db_utils.py,sha256=Vv00OcjoBZdC8z1ucEtekEwqPhu95GkgazBdoRBd-bI,10643
+clickzetta_dbutils/version.py,sha256=98rTj2jvgnCZhTIeHpbml7e2xuhIqt_9BJNSWkWEQb8,21
+clickzetta_dbutils-1.0.7.dist-info/METADATA,sha256=IPehhi4jMgaS7VZfqMhSwTkYkAbnLXHKf708o4537jo,852
+clickzetta_dbutils-1.0.7.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+clickzetta_dbutils-1.0.7.dist-info/top_level.txt,sha256=8o5KqMSg9pxnPNejHjMaqZV2vEDvwvsz2GdChZI0N6I,19
+clickzetta_dbutils-1.0.7.dist-info/RECORD,,

clickzetta_dbutils-1.0.6.dist-info/RECORD

@@ -1,7 +0,0 @@
-clickzetta_dbutils/__init__.py,sha256=Q_6kas0RvZ0767qlaA_xGESmXxm0dks1YQ8DqCX8LV0,290
-clickzetta_dbutils/db_utils.py,sha256=1iVaiYIQosukrQBAbgjNXJJW51p7k1ihQKBIcl_MvQM,10166
-clickzetta_dbutils/version.py,sha256=fCDDAyG3nMZcE_hvt1RHdxkiFN3DfNSyU_k-rLUDrpE,21
-clickzetta_dbutils-1.0.6.dist-info/METADATA,sha256=mpfK67NkOGCBN_wu5NcTvMLLTsVGlLxmnNdcX3KTiFw,726
-clickzetta_dbutils-1.0.6.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-clickzetta_dbutils-1.0.6.dist-info/top_level.txt,sha256=8o5KqMSg9pxnPNejHjMaqZV2vEDvwvsz2GdChZI0N6I,19
-clickzetta_dbutils-1.0.6.dist-info/RECORD,,