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"))),
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