hide _add_edges

py_code_analyzer/code_imports_analyzer.py

@@ -47,12 +47,6 @@ class CodeImportsAnalyzer:
             self._node_visitor.visit(tree)
         return self
 
-    def _add_edges(self, nodes):
-        for first_node, second_node in zip(nodes, nodes[1:]):
-            self.graph_analyzer.add_node(first_node, color="gray")
-            self.graph_analyzer.add_node(second_node, color="gray")
-            self.graph_analyzer.add_edge(first_node, second_node)
-
     def generate_imports_graph(self):
         for python_import in self.python_imports:
             _nodes = python_import["file_path"].split("/")
@@ -65,7 +59,7 @@ class CodeImportsAnalyzer:
                     if len(_nodes) >= 3:
                         _nodes[-2] = _nodes[-2] + "/" + _nodes[-1]
                         del _nodes[-1]
-                    self._add_edges(_nodes)
+                    self.graph_analyzer.add_edges_from_nodes(_nodes)
                 else:
                     self.graph_analyzer.add_node(_nodes[0])
 
@@ -75,13 +69,13 @@ class CodeImportsAnalyzer:
                         if _import["module"] is None:
                             _import_names = _import["name"].split(".")
                             _new_nodes = _import_names + [_nodes[-1]]
-                            self._add_edges(_new_nodes)
+                            self.graph_analyzer.add_edges_from_nodes(_new_nodes)
                         else:
                             _import_names = _import["module"].split(".") + [
                                 _import["name"]
                             ]
                             _new_nodes = _import_names + [_nodes[-1]]
-                            self._add_edges(_new_nodes)
+                            self.graph_analyzer.add_edges_from_nodes(_new_nodes)
 
         return self.graph_analyzer.graph
 

py_code_analyzer/graph_analyzer.py

@@ -12,3 +12,10 @@ class GraphAnalyzer:
 
     def add_edge(self, first_node, second_node):
         self.graph.add_edge(first_node, second_node)
+
+    def add_edges_from_nodes(self, nodes):
+        assert len(nodes) > 1
+        for first_node, second_node in zip(nodes, nodes[1:]):
+            self.graph.add_node(first_node, color="gray")  # set default node color
+            self.graph.add_node(second_node, color="gray")  # set default node color
+            self.graph.add_edge(first_node, second_node)