clickzetta-semantic-model-generator 1.0.7__py3-none-any.whl → 1.0.10__py3-none-any.whl

{clickzetta_semantic_model_generator-1.0.7.dist-info → clickzetta_semantic_model_generator-1.0.10.dist-info}/METADATA

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

{clickzetta_semantic_model_generator-1.0.7.dist-info → clickzetta_semantic_model_generator-1.0.10.dist-info}/RECORD

@@ -1,5 +1,5 @@
 semantic_model_generator/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-semantic_model_generator/clickzetta_utils/clickzetta_connector.py,sha256=ZGsBNOJ79gDSRqfLaQvtvegX-ouiqTepgIlY_4CWN_0,34512
+semantic_model_generator/clickzetta_utils/clickzetta_connector.py,sha256=Qjh79E7TSfkeGiXITwWnWSXEpVTMKMbynDCr4vln2sc,34299
 semantic_model_generator/clickzetta_utils/env_vars.py,sha256=8cbL6R75c1-aVQ2i1TDr9SiHCUjTrgvXbIRz4MbcmbE,7664
 semantic_model_generator/clickzetta_utils/utils.py,sha256=UBfWy9qOTyut8tL02gOHHbh6Uz8RqRz5Mm2YdKWFN54,4950
 semantic_model_generator/data_processing/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -32,7 +32,7 @@ semantic_model_generator/validate/context_length.py,sha256=HL-GfaRXNcVji1-pAFGXG
 semantic_model_generator/validate/keywords.py,sha256=frZ5HjRXP69K6dYAU5_d86oSp40_3yoLUg1eQwU3oLM,7080
 semantic_model_generator/validate/schema.py,sha256=eL_wl5yscIeczwNBRUKhF_7QqWW2wSGimkgaOhMFsrA,5893
 semantic_model_generator/validate_model.py,sha256=Uq-V-GfPeF2Dy4l9uF5Guv104gDCDGh0Cxz1AJOu5dk,836
-clickzetta_semantic_model_generator-1.0.7.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-clickzetta_semantic_model_generator-1.0.7.dist-info/METADATA,sha256=sfjkgnNn6-Md3p3-m-gMsINMCy9BXCKyzU56Fx49TBY,7816
-clickzetta_semantic_model_generator-1.0.7.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-clickzetta_semantic_model_generator-1.0.7.dist-info/RECORD,,
+clickzetta_semantic_model_generator-1.0.10.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+clickzetta_semantic_model_generator-1.0.10.dist-info/METADATA,sha256=x-Z3VZhV7l0K_1tjpaoXfgFXrCgwIveIdM3BDP4CvNc,7817
+clickzetta_semantic_model_generator-1.0.10.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+clickzetta_semantic_model_generator-1.0.10.dist-info/RECORD,,

semantic_model_generator/clickzetta_utils/clickzetta_connector.py

@@ -481,29 +481,20 @@ 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 ""
-    )
-    if schema and not is_shared_catalog:
-        schema = schema.upper()
+    schema_token = table_schema or ""
+    if schema_token and not is_shared_catalog:
+        schema_token = schema_token.upper()
 
     for table_name in table_names:
-        qualified_parts = [
-            part
-            for part in (
-                catalog,
-                schema,
-                table_name.upper() if not is_shared_catalog else table_name,
-            )
-            if part
-        ]
-        qualified_table = ".".join(qualified_parts)
-        if is_shared_catalog:
-            query = f"SHOW COLUMNS IN SHARE {qualified_table}"
-        else:
-            query = f"SHOW COLUMNS IN {qualified_table}"
+        table_token = str(table_name).strip()
+        if not table_token:
+            continue
+        qualified_table = join_quoted_identifiers(
+            *(part for part in (catalog, schema_token, table_token) if part)
+        )
+        query = f"SHOW COLUMNS IN {qualified_table}"
         try:
             df = session.sql(query).to_pandas()
         except Exception as exc:
@@ -564,7 +555,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:
@@ -591,7 +582,7 @@ def get_valid_schemas_tables_columns_df(
                     session=session, schema_name=schema_identifier
                 )
                 tables_for_show = [
-                    table.split(".")[-1].upper()
+                    table.split(".")[-1]
                     for table in tables_for_show
                     if table
                 ]
@@ -682,7 +673,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()