From 04d88c8f2ffc16e010b2f26dbdc206eac0153ef9 Mon Sep 17 00:00:00 2001
From: Brandon Rodriguez <brodriguez8774@gmail.com>
Date: Sun, 17 Nov 2019 05:34:52 -0500
Subject: [PATCH] Add handling for directed classes to "disconnect_nodes" edge
 function

---
 resources/graphs/basic_graph/components.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/resources/graphs/basic_graph/components.py b/resources/graphs/basic_graph/components.py
index cbeec67..e12c82d 100644
--- a/resources/graphs/basic_graph/components.py
+++ b/resources/graphs/basic_graph/components.py
@@ -211,6 +211,10 @@ class BasicEdge():
         # Get connected nodes.
         try:
             node_1, node_2 = self.get_nodes()
+
+            # Handle for dict, if classes are of Directed type.
+            if isinstance(node_1, str):
+                node_1, node_2 = self.get_nodes().values()
         except ValueError:
             # No nodes connected. Return None.
             return None
-- 
GitLab