Add exclude edge check

MrCurtis · Jan 28, 2024 · fc8ee44 · fc8ee44
1 parent 9a1a3f7
commit fc8ee44
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/fk_graph/graph.py b/src/fk_graph/graph.py
@@ -110,7 +110,11 @@ def _add_related_rows_to_graph(row, row_node, graph, only_tables):
     for relationship in relationships:
         related_rows = _get_related_rows_for_relationship(row, relationship)
         for related_row in related_rows:
-            if _row_is_from_an_included_table(related_row, only_tables):
+            if (
+                _row_is_from_an_included_table(related_row, only_tables
+                and
+                not exclude_edge(row, related_row)
+            ):
                 related_node = _create_node_from_row(related_row)
                 related.append((related_row, related_node))
     unvisited = [