refactor

Makefile

@@ -1,3 +1,6 @@
+run:
+	pipenv run python app.py
+
 format:
 	pipenv run black py_code_analyzer && \
 	pipenv run isort py_code_analyzer && \

app.py

@@ -3,6 +3,4 @@ from py_code_analyzer.code_fetcher import get_repository_python_files
 
 python_files = get_repository_python_files("cyyeh", "gradio")
 
-code_analyzer = CodeAnalyzer(python_files)
-code_analyzer.analyze_imports()
-code_analyzer.report_imports()
+CodeAnalyzer(python_files).analyze_imports().report()

py_code_analyzer/code_analyzer.py

@@ -34,6 +34,7 @@ class CodeAnalyzer:
             tree = ast.parse(program)
             self.imports += [{"file_name": python_file["name"], "imports": []}]
             self._node_visitor.visit(tree)
+        return self
 
-    def report_imports(self):
+    def report(self):
         pprint(self.imports)