Reverted some changes

This commit is contained in:
Philipp Horstenkamp 2024-01-11 01:19:24 +01:00
parent 1785114ec1
commit b512eef308
Signed by: Philipp
GPG Key ID: DD53EAC36AFB61B4

View File

@ -107,7 +107,7 @@ def get_all_company_relations() -> pd.DataFrame:
cache=TTLCache(maxsize=100, ttl=int(os.getenv("PYTHON_CACHE_TTL", "3600"))), cache=TTLCache(maxsize=100, ttl=int(os.getenv("PYTHON_CACHE_TTL", "3600"))),
key=lambda drop_min_person_links: hash(drop_min_person_links), 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. """These method makes a Database Request for all Persons and their relations, modifies the ID Column and returns the Result as an DataFrame.
Returns: 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( person_relations["id_person"] = person_relations["id_person"].apply(
lambda x: f"p_{x}" 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 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. company_id (int): Id of the Company.
""" """
# Get Data # Get Data
person_df = get_all_person_relations(False) person_df = get_all_person_relations()
company_df = get_all_company_relations() company_df = get_all_company_relations()
# Create Edge and Node List from data # 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 # Get Data
person_df = get_all_person_relations(False) person_df = get_all_person_relations()
company_df = get_all_company_relations() company_df = get_all_company_relations()
# Create Edge and Node List from data # 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 tuple[dict, list]: nodes, edges
""" """
# Get Data # Get Data
person_df = get_all_person_relations(False) person_df = get_all_person_relations()
company_df = get_all_company_relations() company_df = get_all_company_relations()
# Create Edge and Node List from data # Create Edge and Node List from data