wehub-resource-sync
51 lines
1.9 KiB
Python
51 lines
1.9 KiB
Python
"""This module is the page for Graph Viz Data Search feature"""
|
|
|
|
# pylint: disable=import-error, line-too-long, unused-variable
|
|
|
|
from database import graph_dtls_query
|
|
from pyvis.network import Network
|
|
|
|
|
|
def generate_graph() -> None:
|
|
"""This function is for generating the Graph Visualization"""
|
|
|
|
graph = Network("900px", "900px", notebook=True, heading="")
|
|
return_vals = graph_dtls_query()
|
|
companies = return_vals.get("Companies")
|
|
for index, row in companies.iterrows(): # type: ignore[union-attr] # might ignore other potential errors
|
|
graph.add_node(
|
|
str(row["CompanySeq"]),
|
|
label=row["name"],
|
|
title=row["name"],
|
|
shape="triangle",
|
|
)
|
|
|
|
sectors = return_vals.get("Sectors")
|
|
for index, row in sectors.iterrows(): # type: ignore[union-attr] # might ignore other potential errors
|
|
graph.add_node(
|
|
str(row["SectorSeq"]),
|
|
label=row["sector_name"],
|
|
shape="square",
|
|
color="red",
|
|
title=row["sector_name"],
|
|
)
|
|
|
|
funds = return_vals.get("Funds")
|
|
for index, row in funds.iterrows(): # type: ignore[union-attr] # might ignore other potential errors
|
|
graph.add_node(
|
|
str(row["NewMFSequence"]),
|
|
label=row["fund_name"],
|
|
color="green",
|
|
title=row["fund_name"],
|
|
)
|
|
|
|
comp_sector_relation = return_vals.get("CompanySectorRelation")
|
|
for index, row in comp_sector_relation.iterrows(): # type: ignore[union-attr] # might ignore other potential errors
|
|
graph.add_edge(str(row["CompanySeq"]), str(row["SectorSeq"]), title="BELONGS")
|
|
|
|
fund_hold_company_relation = return_vals.get("FundsHoldsCompaniesRelation")
|
|
for index, row in fund_hold_company_relation.iterrows(): # type: ignore[union-attr] # might ignore other potential errors
|
|
graph.add_edge(str(row["NewMFSequence"]), str(row["CompanySeq"]), title="HOLDS")
|
|
|
|
graph.show("graph_viz.html")
|