cyyeh/py-code-analyzer

1"""This file generates a pyvis graph locally given a networkx graph2"""3import networkx as nx4 5from pyvis import Network6 7 8class ImportsGraphVisualizer:9    @classmethod10    def visualize(11        cls,12        imports_graph: nx.Graph,13        heading: str = "",14        directed: bool = True,15        layout: bool = False,  # Use hierarchical if True16        neighborhood_highlight: bool = True,17        select_menu: bool = True,18        width: int = 100,19        height: int = 800,20        show_buttons: bool = False,21        display_html_name: str = "nx.html",22    ):23        _pyvis_network = Network(24            width=f"{width}%",25            height=f"{height}px",26            directed=directed,27            layout=layout,28            heading=heading,29            neighborhood_highlight=neighborhood_highlight,30            select_menu=select_menu,31        )32        _pyvis_network.toggle_hide_edges_on_drag(True)33 34        _pyvis_network.from_nx(imports_graph)35        if show_buttons:36            _pyvis_network.show_buttons()37        _pyvis_network.show(display_html_name)38