From b512eef3083cf88975fc5ef778a8a68aab377e0b Mon Sep 17 00:00:00 2001 From: Philipp Horstenkamp Date: Thu, 11 Jan 2024 01:19:24 +0100 Subject: [PATCH] Reverted some changes --- .../utils/networkx/networkx_data.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/src/aki_prj23_transparenzregister/utils/networkx/networkx_data.py b/src/aki_prj23_transparenzregister/utils/networkx/networkx_data.py index 9249469..7168eba 100644 --- a/src/aki_prj23_transparenzregister/utils/networkx/networkx_data.py +++ b/src/aki_prj23_transparenzregister/utils/networkx/networkx_data.py @@ -107,7 +107,7 @@ def get_all_company_relations() -> pd.DataFrame: cache=TTLCache(maxsize=100, ttl=int(os.getenv("PYTHON_CACHE_TTL", "3600"))), key=lambda drop_min_person_links: hash(drop_min_person_links), ) -def get_all_person_relations(drop_min_person_links: bool) -> pd.DataFrame: +def get_all_person_relations() -> pd.DataFrame: """These method makes a Database Request for all Persons and their relations, modifies the ID Column and returns the Result as an DataFrame. Returns: @@ -142,8 +142,7 @@ def get_all_person_relations(drop_min_person_links: bool) -> pd.DataFrame: person_relations["id_person"] = person_relations["id_person"].apply( lambda x: f"p_{x}" ) - if drop_min_person_links: - return person_relations.groupby("id_person").filter(lambda x: len(x) > 1) + return person_relations @@ -366,7 +365,7 @@ def get_all_metrics_from_id(company_id: int) -> pd.Series: company_id (int): Id of the Company. """ # Get Data - person_df = get_all_person_relations(False) + person_df = get_all_person_relations() company_df = get_all_company_relations() # Create Edge and Node List from data @@ -401,7 +400,7 @@ def get_relations_number_from_id(id: str) -> tuple[int, int, int]: """ # Get Data - person_df = get_all_person_relations(False) + person_df = get_all_person_relations() company_df = get_all_company_relations() # Create Edge and Node List from data @@ -442,7 +441,7 @@ def get_relations_until_level_3(id: str) -> tuple[dict, list]: tuple[dict, list]: nodes, edges """ # Get Data - person_df = get_all_person_relations(False) + person_df = get_all_person_relations() company_df = get_all_company_relations() # Create Edge and Node List from data