sql-glider 0.1.26__tar.gz → 0.1.27__tar.gz

{sql_glider-0.1.26 → sql_glider-0.1.27}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sql-glider
-Version: 0.1.26
+Version: 0.1.27
 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/
@@ -30,6 +30,7 @@ Provides-Extra: databricks
 Requires-Dist: databricks-sdk>=0.20.0; extra == 'databricks'
 Provides-Extra: plotly
 Requires-Dist: plotly>=5.0.0; extra == 'plotly'
+Requires-Dist: pygraphviz>=1.11; extra == 'plotly'
 Description-Content-Type: text/markdown
 
 # SQL Glider

{sql_glider-0.1.26 → sql_glider-0.1.27}/docs/docs/graph-lineage.md

@@ -329,6 +329,20 @@ The `plotly` format outputs a JSON figure specification that can be loaded into
     pip install sql-glider[plotly]
     ```
 
+!!! tip "Better Layout with Graphviz"
+
+    For complex graphs with many nodes and edges, installing [Graphviz](https://graphviz.org/download/) on your system enables an optimized layout algorithm that minimizes edge crossings. Without Graphviz, a simpler layout is used that may result in overlapping edges.
+
+    ```bash
+    # Install Graphviz first (system package)
+    # Windows: choco install graphviz
+    # macOS: brew install graphviz
+    # Linux: apt install graphviz graphviz-dev
+
+    # Then install sql-glider with plotly extra
+    pip install sql-glider[plotly]
+    ```
+
 **Example usage with Dash:**
 
 ```python

{sql_glider-0.1.26 → sql_glider-0.1.27}/pyproject.toml

@@ -35,7 +35,7 @@ dependencies = [
 
 [project.optional-dependencies]
 databricks = ["databricks-sdk>=0.20.0"]
-plotly = ["plotly>=5.0.0"]
+plotly = ["plotly>=5.0.0", "pygraphviz>=1.11"]
 
 [project.urls]
 Homepage = "https://github.com/rycowhi/sql-glider/"

{sql_glider-0.1.26 → sql_glider-0.1.27}/src/sqlglider/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.1.26'
-__version_tuple__ = version_tuple = (0, 1, 26)
+__version__ = version = '0.1.27'
+__version_tuple__ = version_tuple = (0, 1, 27)
 
 __commit_id__ = commit_id = None

{sql_glider-0.1.26 → sql_glider-0.1.27}/src/sqlglider/graph/diagram_formatters.py

@@ -364,7 +364,11 @@ def _compute_layered_layout(
     x_spacing: float = 250.0,
     y_spacing: float = 100.0,
 ) -> dict[str, tuple[float, float]]:
-    """Compute layered layout positions for nodes using topological ordering.
+    """Compute layered layout positions for nodes using Graphviz's dot algorithm.
+
+    Uses Graphviz's dot layout engine (Sugiyama algorithm) which minimizes edge
+    crossings by using the barycenter heuristic to optimize node ordering within
+    each layer.
 
     Positions nodes in layers from left to right based on their dependencies.
     Nodes with no incoming edges are placed in layer 0, their dependents in
@@ -382,6 +386,89 @@ def _compute_layered_layout(
     if not nodes:
         return {}
 
+    try:
+        import pygraphviz  # noqa: F401  # type: ignore[import-not-found]
+
+        return _compute_graphviz_layout(nodes, edges, x_spacing, y_spacing)
+    except ImportError:
+        # Fallback to simple layered layout if pygraphviz not available
+        return _compute_simple_layered_layout(nodes, edges, x_spacing, y_spacing)
+
+
+def _compute_graphviz_layout(
+    nodes: list[str],
+    edges: list[tuple[str, str]],
+    x_spacing: float = 250.0,
+    y_spacing: float = 100.0,
+) -> dict[str, tuple[float, float]]:
+    """Compute layout using Graphviz's dot algorithm with crossing minimization.
+
+    Args:
+        nodes: List of node identifiers
+        edges: List of (source, target) edge tuples
+        x_spacing: Horizontal spacing between layers
+        y_spacing: Vertical spacing between nodes in the same layer
+
+    Returns:
+        Dictionary mapping node identifiers to (x, y) positions
+    """
+    import pygraphviz as pgv  # type: ignore[import-not-found]
+
+    # Create directed graph
+    g = pgv.AGraph(directed=True, rankdir="LR")
+
+    # Set graph attributes for spacing
+    g.graph_attr["ranksep"] = str(x_spacing / 72.0)  # Convert pixels to inches
+    g.graph_attr["nodesep"] = str(y_spacing / 72.0)
+
+    # Add nodes
+    for node in nodes:
+        g.add_node(node)
+
+    # Add edges (only for nodes that exist in our node list)
+    node_set = set(nodes)
+    for src, tgt in edges:
+        if src in node_set and tgt in node_set:
+            g.add_edge(src, tgt)
+
+    # Compute layout using dot algorithm
+    g.layout(prog="dot")
+
+    # Extract positions
+    positions: dict[str, tuple[float, float]] = {}
+    for node in nodes:
+        n = g.get_node(node)
+        # Position is returned as "x,y" string in points (1/72 inch)
+        pos_str = n.attr.get("pos", "0,0")
+        if pos_str:
+            x_str, y_str = pos_str.split(",")
+            # Convert from points to our coordinate system
+            x = float(x_str)
+            y = float(y_str)
+            positions[node] = (x, y)
+
+    return positions
+
+
+def _compute_simple_layered_layout(
+    nodes: list[str],
+    edges: list[tuple[str, str]],
+    x_spacing: float = 250.0,
+    y_spacing: float = 100.0,
+) -> dict[str, tuple[float, float]]:
+    """Fallback simple layered layout using topological ordering.
+
+    Used when pygraphviz is not available. Does not minimize edge crossings.
+
+    Args:
+        nodes: List of node identifiers
+        edges: List of (source, target) edge tuples
+        x_spacing: Horizontal spacing between layers
+        y_spacing: Vertical spacing between nodes in the same layer
+
+    Returns:
+        Dictionary mapping node identifiers to (x, y) positions
+    """
     # Build adjacency structures
     incoming: dict[str, set[str]] = defaultdict(set)
     outgoing: dict[str, set[str]] = defaultdict(set)

{sql_glider-0.1.26 → sql_glider-0.1.27}/uv.lock

@@ -814,6 +814,12 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/c7/21/705964c7812476f378728bdf590ca4b771ec72385c533964653c68e86bdc/pygments-2.19.2-py3-none-any.whl", hash = "sha256:86540386c03d588bb81d44bc3928634ff26449851e99741617ecb9037ee5ec0b", size = 1225217, upload-time = "2025-06-21T13:39:07.939Z" },
 ]
 
+[[package]]
+name = "pygraphviz"
+version = "1.14"
+source = { registry = "https://pypi.org/simple/" }
+sdist = { url = "https://files.pythonhosted.org/packages/66/ca/823d5c74a73d6b8b08e1f5aea12468ef334f0732c65cbb18df2a7f285c87/pygraphviz-1.14.tar.gz", hash = "sha256:c10df02377f4e39b00ae17c862f4ee7e5767317f1c6b2dfd04cea6acc7fc2bea", size = 106003, upload-time = "2024-09-29T18:31:12.471Z" }
+
 [[package]]
 name = "pymdown-extensions"
 version = "10.20.1"
@@ -1161,6 +1167,7 @@ databricks = [
 ]
 plotly = [
     { name = "plotly" },
+    { name = "pygraphviz" },
 ]
 
 [package.dev-dependencies]
@@ -1185,6 +1192,7 @@ requires-dist = [
     { name = "jinja2", specifier = ">=3.0.0" },
     { name = "plotly", marker = "extra == 'plotly'", specifier = ">=5.0.0" },
     { name = "pydantic", specifier = ">=2.0.0" },
+    { name = "pygraphviz", marker = "extra == 'plotly'", specifier = ">=1.11" },
     { name = "rich", specifier = ">=13.0.0" },
     { name = "rustworkx", specifier = ">=0.15.0" },
     { name = "sqlglot", extras = ["rs"], specifier = ">=25.0.0" },