diff --git a/src/aki_prj23_transparenzregister/utils/data_extraction/unternehmensregister/transform.py b/src/aki_prj23_transparenzregister/utils/data_extraction/unternehmensregister/transform.py index 610433d..ab60100 100644 --- a/src/aki_prj23_transparenzregister/utils/data_extraction/unternehmensregister/transform.py +++ b/src/aki_prj23_transparenzregister/utils/data_extraction/unternehmensregister/transform.py @@ -223,9 +223,10 @@ def name_from_beteiligung(data: dict) -> str: Returns: str: Company name """ - return data["XJustiz_Daten"]["Grunddaten"]["Verfahrensdaten"]["Beteiligung"][0][ + name = data["XJustiz_Daten"]["Grunddaten"]["Verfahrensdaten"]["Beteiligung"][0][ "Beteiligter" ]["Organisation"]["Bezeichnung"]["Bezeichnung_Aktuell"] + return name.replace('"', "") def map_rechtsform(company_name: str, data: dict) -> CompanyTypeEnum | None: diff --git a/tests/utils/data_extraction/unternehmensregister/transform_test.py b/tests/utils/data_extraction/unternehmensregister/transform_test.py index b690765..20c0a4d 100644 --- a/tests/utils/data_extraction/unternehmensregister/transform_test.py +++ b/tests/utils/data_extraction/unternehmensregister/transform_test.py @@ -304,6 +304,32 @@ def test_name_from_beteiligung() -> None: assert transform.name_from_beteiligung(data) == expected_result +def test_name_from_beteiligung_remove_quotes() -> None: + data = { + "XJustiz_Daten": { + "Grunddaten": { + "Verfahrensdaten": { + "Beteiligung": [ + { + "Beteiligter": { + "Beteiligtennummer": "1", + "Organisation": { + "Bezeichnung": { + "Bezeichnung_Aktuell": '"Siemes Verwaltungs-GmbH"' + }, + }, + } + }, + ] + } + } + } + } + + expected_result = "Siemes Verwaltungs-GmbH" + assert transform.name_from_beteiligung(data) == expected_result + + def test_map_rechtsform() -> None: data = { "XJustiz_Daten": {