clickzetta-semantic-model-generator 1.0.6__tar.gz → 1.0.9__tar.gz

{clickzetta_semantic_model_generator-1.0.6 → clickzetta_semantic_model_generator-1.0.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: clickzetta-semantic-model-generator
-Version: 1.0.6
+Version: 1.0.9
 Summary: Curate a Semantic Model for ClickZetta Lakehouse
 License: Apache Software License; BSD License
 Author: qililiang

{clickzetta_semantic_model_generator-1.0.6 → clickzetta_semantic_model_generator-1.0.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "clickzetta-semantic-model-generator"
-version = "1.0.6"
+version = "1.0.9"
 description = "Curate a Semantic Model for ClickZetta Lakehouse"
 authors = ["qililiang <qililiang@clickzetta.com>"]
 license = "Apache Software License; BSD License"

{clickzetta_semantic_model_generator-1.0.6 → clickzetta_semantic_model_generator-1.0.9}/semantic_model_generator/clickzetta_utils/clickzetta_connector.py

@@ -481,7 +481,7 @@ def _fetch_columns_via_show(
 
     rows: List[pd.DataFrame] = []
     category = _catalog_category(session, workspace)
-    is_shared_catalog = category in {"SHARED", "EXTERNAL"}
+    is_shared_catalog = category == "SHARED"
     catalog = workspace if is_shared_catalog else workspace.upper()
     schema = (
         table_schema or ""
@@ -564,7 +564,7 @@ def get_valid_schemas_tables_columns_df(
     table_names: Optional[List[str]] = None,
 ) -> pd.DataFrame:
     category = _catalog_category(session, workspace)
-    skip_information_schema = category in {"SHARED", "EXTERNAL"}
+    skip_information_schema = category == "SHARED"
 
     result = pd.DataFrame()
     if not skip_information_schema:
@@ -581,11 +581,31 @@ def get_valid_schemas_tables_columns_df(
             result = pd.DataFrame()
 
     if result.empty:
+        tables_for_show = table_names
+        if not tables_for_show:
+            try:
+                schema_identifier = (
+                    f"{workspace}.{table_schema}" if table_schema else workspace
+                )
+                tables_for_show = fetch_tables_views_in_schema(
+                    session=session, schema_name=schema_identifier
+                )
+                tables_for_show = [
+                    table.split(".")[-1]
+                    for table in tables_for_show
+                    if table
+                ]
+            except Exception as exc:  # pragma: no cover - defensive logging
+                logger.debug(
+                    "Unable to enumerate tables for SHOW COLUMNS fallback: {}", exc
+                )
+                tables_for_show = None
+
         result = _fetch_columns_via_show(
             session=session,
             workspace=workspace,
             table_schema=table_schema,
-            table_names=table_names,
+            table_names=tables_for_show,
         )
 
     if result.empty:
@@ -662,7 +682,7 @@ def fetch_tables_views_in_schema(
     workspace = parts[0]
     schema = parts[1] if len(parts) > 1 else ""
     category = _catalog_category(session, workspace)
-    is_shared_catalog = category in {"SHARED", "EXTERNAL"}
+    is_shared_catalog = category == "SHARED"
 
     workspace_token = workspace if is_shared_catalog else workspace.upper()
     schema_token = schema if is_shared_catalog else schema.upper()