fix: add none check to conneted_on...

simulate-digital-rail · Apr 18, 2023 · d5e7a59 · d5e7a59
1 parent ed4e588
commit d5e7a59
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/yaramo/node.py b/yaramo/node.py
@@ -58,14 +58,20 @@ def connected_nodes(self):
 
     @property
     def connected_on_head(self):
+        if self.connected_edge_on_head is None:
+            return None
         return self.connected_edge_on_head.get_opposite_node(self)
 
     @property
     def connected_on_left(self):
+        if self.connected_edge_on_left is None:
+            return None
         return self.connected_edge_on_left.get_opposite_node(self)
 
     @property
     def connected_on_right(self):
+        if self.connected_edge_on_right is None:
+            return None
         return self.connected_edge_on_right.get_opposite_node(self)
 
     def set_connection_head_edge(self, edge: "Edge"):