sqlh 0.3.2__tar.gz → 0.3.4__tar.gz

{sqlh-0.3.2 → sqlh-0.3.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: sqlh
-Version: 0.3.2
+Version: 0.3.4
 Summary: A lightweight SQL lineage analysis library for tracking table dependencies in data pipelines
 Keywords: sql,lineage,data-pipeline,dag,dependency,database,etl,data-engineering
 Requires-Python: >=3.10

{sqlh-0.3.2 → sqlh-0.3.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "sqlh"
-version = "0.3.2"
+version = "0.3.4"
 
 description = "A lightweight SQL lineage analysis library for tracking table dependencies in data pipelines"
 readme = "README.md"

{sqlh-0.3.2 → sqlh-0.3.4}/sqlh/__init__.py

@@ -14,7 +14,7 @@ from .utils import (
     table_count
 )
 
-__version__ = "0.3.2"
+__version__ = "0.3.4"
 
 __all__ = [
     "split_sql",

{sqlh-0.3.2 → sqlh-0.3.4}/sqlh/core/graph.py

@@ -351,6 +351,8 @@ class DagGraph:
             mermaid_str = f"graph {direction}"
             for _from, _to in self.__edges:
                 mermaid_str += f"\n  {_from} --> {_to}"
+            # 去除花括号
+            mermaid_str = mermaid_str.replace("{", "").replace("}", "")
             return mermaid_str
 
     def to_dict(self) -> dict:

{sqlh-0.3.2 → sqlh-0.3.4}/sqlh/utils.py

@@ -98,8 +98,8 @@ def __build_tables_and_graph(sql_stmt_str: str) -> Tuple[list, list, DagGraph]:
     for sql_stmt in sql_stmt_lst:
         table_info = get_source_target_tables(sql_stmt)
         if table_info:
-            sources = [re.sub(r"`|\"", "", t) for t in table_info["source_tables"]]
-            targets = [re.sub(r"`|\"", "", t) for t in table_info["target_tables"]]
+            sources = [re.sub(r"`|\"|\{|\}", "", t) for t in table_info["source_tables"]]
+            targets = [re.sub(r"`|\"|\{|\}", "", t) for t in table_info["target_tables"]]
             source_tables.update(sources)
             target_tables.update(targets)
             for src in sources: