update

README.md

@@ -27,8 +27,8 @@ input to one GitHub public repo's URL(input)
 
 ## TODOs
 
-- [ ] Build a prototype
-  - [ ] Finish `generate_imports_graph` implementation
+- [x] Build a prototype
+  - [x] Finish `generate_imports_graph` implementation
   - [x] Fetch python files given public GitHub repo url(owner, repo, path, ref)
   - [x] Use `ast` to parse imports among given python files
   - [x] Generate a basic `networkx` graph given python imports

app.py

@@ -1,5 +1,7 @@
 from py_code_analyzer import CodeFetcher, CodeImportsAnalyzer, ImportsGraphVisualizer
 
-python_files = CodeFetcher().get_python_files("cyyeh", "gradio", "gradio")
+python_files = CodeFetcher().get_python_files(
+    "cyyeh", "py-code-analyzer", "py_code_analyzer"
+)
 imports_graph = CodeImportsAnalyzer(python_files).analyze().generate_imports_graph()
 ImportsGraphVisualizer().visualize(imports_graph)

py_code_analyzer/code_imports_analyzer.py

@@ -68,15 +68,18 @@ class CodeImportsAnalyzer:
                     self.imports_graph.add_node(_nodes[0])
 
                 # generate graph based on imported modules in each file
-                for _import in python_import["imports"]:
-                    if _import["module"] is None:
-                        _import_names = _import["name"].split(".")
-                        _new_nodes = _import_names + [_nodes[-1]]
-                        self._add_edges(_new_nodes)
-                    else:
-                        _import_names = _import["module"].split(".") + [_import["name"]]
-                        _new_nodes = _import_names + [_nodes[-1]]
-                        self._add_edges(_new_nodes)
+                if python_import["file_name"] != "__init__.py":
+                    for _import in python_import["imports"]:
+                        if _import["module"] is None:
+                            _import_names = _import["name"].split(".")
+                            _new_nodes = _import_names + [_nodes[-1]]
+                            self._add_edges(_new_nodes)
+                        else:
+                            _import_names = _import["module"].split(".") + [
+                                _import["name"]
+                            ]
+                            _new_nodes = _import_names + [_nodes[-1]]
+                            self._add_edges(_new_nodes)
 
         return self.imports_graph