Fixing the rendering of the graph to not use schema when produing rel…

…ationships
tedivm · Sep 26, 2024 · 0a06110 · 0a06110
1 parent e73b2c8
commit 0a06110
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/paracelsus/transformers/dot.py b/paracelsus/transformers/dot.py
@@ -40,7 +40,7 @@ def __init__(self, metaclass: MetaData, column_sort: str) -> None:
                         )
                         continue
 
-                    edge = pydot.Edge(left_table, table.name)
+                    edge = pydot.Edge(left_table.split('.')[-1], table.name)
                     edge.set_label(column.name)
                     edge.set_dir("both")
 

diff --git a/paracelsus/transformers/mermaid.py b/paracelsus/transformers/mermaid.py
@@ -85,7 +85,7 @@ def _relationships(self, column: Column) -> str:
             else:
                 left_operand = "}o"
 
-            output += f"  {left_table} {left_operand}--{right_operand} {right_table} : {column_name}\n"
+            output += f"  {left_table.split('.')[-1]} {left_operand}--{right_operand} {right_table} : {column_name}\n"
         return output
 
     def __str__(self) -> str: