added check for bidirectional edge adding

smulhern18 · May 24, 2018 · May 24, 2018 · May 24, 2018 · Jun 26, 2018 · Jun 26, 2018
commit 7e87fdc1968e5fb490b2de8231c32fc2b0b4589a
diff --git a/pyvis/network.py b/pyvis/network.py
@@ -323,15 +323,17 @@ def add_edge(self, source, to, **options):
         assert to in self.get_nodes(), \
             "non existent node '" + str(to) + "'"
 
-        for e in self.edges:
-            frm = e['from']
-            dest = e['to']
-            if (
-                (source == dest and to == frm) or
-                (source == frm and to == dest)
-            ):
-                # edge already exists
-                edge_exists = True
+        # we only check existing edge for undirected graphs
+        if not self.directed:
+            for e in self.edges:
+                frm = e['from']
+                dest = e['to']
+                if (
+                    (source == dest and to == frm) or
+                    (source == frm and to == dest)
+                ):
+                    # edge already exists
+                    edge_exists = True
 
         if not edge_exists:
             e = Edge(source, to, self.directed, **options)