sql-glider 0.1.7__py3-none-any.whl → 0.1.9__py3-none-any.whl

{sql_glider-0.1.7.dist-info → sql_glider-0.1.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sql-glider
-Version: 0.1.7
+Version: 0.1.9
 Summary: SQL Utility Toolkit for better understanding, use, and governance of your queries in a native environment.
 Project-URL: Homepage, https://github.com/rycowhi/sql-glider/
 Project-URL: Repository, https://github.com/rycowhi/sql-glider/

{sql_glider-0.1.7.dist-info → sql_glider-0.1.9.dist-info}/RECORD

@@ -1,5 +1,5 @@
 sqlglider/__init__.py,sha256=gDf7s52dMcX7JuCZ1SLawcB1vb3U0yJCohu9RQAATBY,125
-sqlglider/_version.py,sha256=szvPIs2C82UunpzuvVg3MbF4QhzbBYTsVJ8DmPfq6_E,704
+sqlglider/_version.py,sha256=ib8ckvf-NNDfacXd8unW0p5cf-gl57XyQvjoEMc_pvc,704
 sqlglider/cli.py,sha256=9sweHRVLk2iBSzCzT2Gcj8y1g1XKzq26iApQsMaFbx4,51786
 sqlglider/global_models.py,sha256=2vyJXAuXOsXQpE-D3F0ejj7eR9z0nDWFjTkielhzM8k,356
 sqlglider/catalog/__init__.py,sha256=2PqFPyzFXJ14FpSUcBmVK2L-a_ypWQHAbHFHxLDk_LE,814
@@ -11,13 +11,13 @@ sqlglider/dissection/analyzer.py,sha256=-GD3-lTbfBthq1BW6HiDjvJx2y4LDmnUVHIVIb0H
 sqlglider/dissection/formatters.py,sha256=M7gsmTNljRIeLIRv4D0vHvqJVrTqWSpsg7vem83zSzY,7302
 sqlglider/dissection/models.py,sha256=RRD3RIteqbUBY6e-74skKDvMH3qeAUaqA2sFcrjP5GQ,3618
 sqlglider/graph/__init__.py,sha256=4DDdrPM75CmeQWt7wHdBsjCm1s70BHGLYdijIbaUEKY,871
-sqlglider/graph/builder.py,sha256=nIddZjGerklkB8NbhVPx2nsMqFI5gZCrFVKVnxuCgms,11944
+sqlglider/graph/builder.py,sha256=0ASA749b_FkcKMEVWiijAJP1QKt54ICY7VXxUuo3-y0,11953
 sqlglider/graph/merge.py,sha256=uUZlm4BN3S9gRL66Cc2mzhbtuh4SVAv2n4cN4eUEQBU,4077
 sqlglider/graph/models.py,sha256=EYmjv_WzDSNp_WfhJ6H-qBIOkAcoNKS7GRUryfKrHuY,9330
 sqlglider/graph/query.py,sha256=LHU8Cvn7ZPPSEnqdDn2pF8f1_LQjIvNIrZqs8cFlb6U,9433
-sqlglider/graph/serialization.py,sha256=7JJo31rwSlxnDhdqdTJdK4Dr_ZcSYetXfx3_CmndSac,2662
+sqlglider/graph/serialization.py,sha256=vMXn7s35jA499e7l90vNVaJE_3QR_VHf3rEfQ9ZlgTQ,2781
 sqlglider/lineage/__init__.py,sha256=llXMeI5_PIZaiBo8tKk3-wOubF4m_6QBHbn1FtWxT7k,256
-sqlglider/lineage/analyzer.py,sha256=Vfh0g9xVEEUkQ87KZlCcZVPltDJ6Uos67PBtDyQ_i8U,64679
+sqlglider/lineage/analyzer.py,sha256=4VoJAsNRpz8T4sifF9VW4_SkSSU-8GtcBdDORCfPV_0,65581
 sqlglider/lineage/formatters.py,sha256=_Y9wcTX4JXn1vVnZ1xI656g1FF2rMjcAVc-GHjbd9QA,10389
 sqlglider/templating/__init__.py,sha256=g3_wb6rSDI0usq2UUMDpn-J5kVwlAw3NtLdwbxL6UHs,1435
 sqlglider/templating/base.py,sha256=y5bWAW7qXl_4pPyo5KycfHwNVvt1-7slZ63DAsvTE1s,2902
@@ -27,8 +27,8 @@ sqlglider/templating/variables.py,sha256=5593PtLBcOxsnMCSRm2pGAD5I0Y9f__VV3_J_Hf
 sqlglider/utils/__init__.py,sha256=KGp9-UzKz_OFBOTFoSy-g-NXDZsvyWXG_9-1zcC6ePE,276
 sqlglider/utils/config.py,sha256=iNJgSXFw3pmL2MCdvW3SJp4X2T3AQP2QyQuXIXT-6H0,4761
 sqlglider/utils/file_utils.py,sha256=5_ff28E0r1R7emZzsOnRuHd-7zIX6873eyr1SuPEr4E,1093
-sql_glider-0.1.7.dist-info/METADATA,sha256=Ng-ht-OuykiQWWUjD364MekChFKLNFW8qYkEIxcYZuc,28445
-sql_glider-0.1.7.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-sql_glider-0.1.7.dist-info/entry_points.txt,sha256=HDuakHqHS5C0HFKsMIxMYmDU7-BLBGrnIJcYaVRu-s0,251
-sql_glider-0.1.7.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-sql_glider-0.1.7.dist-info/RECORD,,
+sql_glider-0.1.9.dist-info/METADATA,sha256=LOSe-vTRzfkFsnas1cHSFgnRJZL3bO34ujqLcUWMN1Q,28445
+sql_glider-0.1.9.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+sql_glider-0.1.9.dist-info/entry_points.txt,sha256=HDuakHqHS5C0HFKsMIxMYmDU7-BLBGrnIJcYaVRu-s0,251
+sql_glider-0.1.9.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+sql_glider-0.1.9.dist-info/RECORD,,

sqlglider/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.1.7'
-__version_tuple__ = version_tuple = (0, 1, 7)
+__version__ = version = '0.1.9'
+__version_tuple__ = version_tuple = (0, 1, 9)
 
 __commit_id__ = commit_id = None

sqlglider/graph/builder.py

@@ -117,8 +117,8 @@ class GraphBuilder:
                     edge_key = (item.source_name.lower(), item.output_name.lower())
                     if edge_key not in self._edge_set:
                         edge = GraphEdge(
-                            source_node=item.source_name,
-                            target_node=item.output_name,
+                            source_node=item.source_name.lower(),
+                            target_node=item.output_name.lower(),
                             file_path=file_path_str,
                             query_index=query_index,
                         )
@@ -285,7 +285,7 @@ class GraphBuilder:
             return self._node_index_map[key]
 
         node = GraphNode.from_identifier(
-            identifier=identifier,
+            identifier=key,
             file_path=file_path,
             query_index=query_index,
         )

sqlglider/graph/serialization.py

@@ -61,15 +61,15 @@ def to_rustworkx(graph: LineageGraph) -> Tuple[rx.PyDiGraph, Dict[str, int]]:
     rx_graph: rx.PyDiGraph = rx.PyDiGraph()
     node_map: Dict[str, int] = {}
 
-    # Add nodes
+    # Add nodes (use lowercase keys for case-insensitive lookup)
     for node in graph.nodes:
         idx = rx_graph.add_node(node.model_dump())
-        node_map[node.identifier] = idx
+        node_map[node.identifier.lower()] = idx
 
-    # Add edges
+    # Add edges (use lowercase for lookup to match node keys)
     for edge in graph.edges:
-        source_idx = node_map.get(edge.source_node)
-        target_idx = node_map.get(edge.target_node)
+        source_idx = node_map.get(edge.source_node.lower())
+        target_idx = node_map.get(edge.target_node.lower())
         if source_idx is not None and target_idx is not None:
             rx_graph.add_edge(source_idx, target_idx, edge.model_dump())
 

sqlglider/lineage/analyzer.py

@@ -615,6 +615,12 @@ class LineageAnalyzer:
             if isinstance(target, exp.Table):
                 return (self._get_qualified_table_name(target), ObjectType.UNKNOWN)
 
+        # CACHE TABLE
+        elif isinstance(self.expr, exp.Cache):
+            target = self.expr.this
+            if isinstance(target, exp.Table):
+                return (self._get_qualified_table_name(target), ObjectType.TABLE)
+
         # DELETE FROM table
         elif isinstance(self.expr, exp.Delete):
             target = self.expr.this
@@ -706,6 +712,10 @@ class LineageAnalyzer:
         elif isinstance(self.expr, exp.Drop):
             return table_node is self.expr.this
 
+        # For CACHE TABLE, the target is self.expr.this
+        elif isinstance(self.expr, exp.Cache):
+            return table_node is self.expr.this
+
         return False
 
     def _analyze_column_lineage_internal(
@@ -889,6 +899,7 @@ class LineageAnalyzer:
             "Drop": f"DROP {getattr(target_expr, 'kind', '')}".strip(),
             "Alter": "ALTER",
             "Truncate": "TRUNCATE",
+            "Cache": "CACHE TABLE",
             "Command": "COMMAND",
         }
 
@@ -943,6 +954,17 @@ class LineageAnalyzer:
                     ):
                         return (target_name, select_node)
 
+        # Check for CACHE TABLE AS SELECT
+        elif isinstance(self.expr, exp.Cache):
+            target = self.expr.this
+            if isinstance(target, exp.Table):
+                target_name = self._get_qualified_table_name(target)
+                select_node = self.expr.expression
+                if isinstance(
+                    select_node, (exp.Select, exp.Union, exp.Intersect, exp.Except)
+                ):
+                    return (target_name, select_node)
+
         # Check for MERGE statement
         elif isinstance(self.expr, exp.Merge):
             target = self.expr.this