update

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 import streamlit.components.v1 as components
 
 from py_code_analyzer import CodeFetcher, CodeImportsAnalyzer, ImportsGraphVisualizer
+from utils import time_function
 
 TITLE = "Python Code Analyzer"
 st.set_page_config(page_title=TITLE, layout="wide")
@@ -26,18 +27,22 @@ clicked_ok_button = st.button("OK")
 st.markdown("---")
 
 
+@time_function
 def get_python_files(owner, repo, path, ref=""):
     return CodeFetcher().get_python_files(owner, repo, path, ref=ref)
 
 
+@time_function
 def generate_imports_graph(python_files):
     return CodeImportsAnalyzer(python_files).analyze().generate_imports_graph()
 
 
+@time_function
 def generate_graph_visualization_file(imports_graph):
     ImportsGraphVisualizer().visualize(imports_graph)
 
 
+@time_function
 def read_graph_visualization_file():
     return open("nx.html", "r", encoding="utf-8").read()
 

py_code_analyzer/graph_analyzer.py

@@ -1,16 +1,11 @@
+"""GraphAnalyzer uses some open source graph library for network analysis
+"""
 import networkx as nx
 
 
 class GraphAnalyzer:
-    def __init__(self, is_directed: bool = False, is_multi_edges: bool = False):
-        if not is_directed and not is_multi_edges:
-            self.graph = nx.Graph()
-        elif is_directed and not is_multi_edges:
-            self.graph = nx.DiGraph()
-        elif not is_directed and is_multi_edges:
-            self.graph = nx.MultiGraph()
-        else:
-            self.graph = nx.MultiDiGraph()
+    def __init__(self, is_directed: bool = False):
+        self.graph = nx.DiGraph() if is_directed else nx.Graph()
 
     def add_node(self, node, **kwargs):
         self.graph.add_node(node, **kwargs)

utils.py

@@ -0,0 +1,12 @@
+import time
+
+
+def time_function(f):
+    def wrapper(*args, **kwargs):
+        begin = time.time()
+        result = f(*args, **kwargs)
+        end = time.time()
+        print(f"function {f.__name__} took: {end - begin} seconds to execute.")
+        return result
+
+    return wrapper