From fc8ee44a65e789274e32d9f9aa639bf627d3a095 Mon Sep 17 00:00:00 2001
From: a <4184070+MrCurtis@users.noreply.github.com>
Date: Sun, 28 Jan 2024 21:47:21 +0000
Subject: [PATCH] Add exclude edge check

---
 src/fk_graph/graph.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/fk_graph/graph.py b/src/fk_graph/graph.py
index a7bf87f..a25dfbd 100644
--- 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 = [