From d54c1cd17dfd08bd65534b10c51c2187fa497eb3 Mon Sep 17 00:00:00 2001 From: TrisNol Date: Sun, 17 Sep 2023 19:25:30 +0200 Subject: [PATCH] test(data-extraction): Adapt to extended enum --- src/aki_prj23_transparenzregister/models/company.py | 2 ++ .../utils/data_extraction/unternehmensregister/transform.py | 4 +++- 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/src/aki_prj23_transparenzregister/models/company.py b/src/aki_prj23_transparenzregister/models/company.py index 0668b5f..66f671a 100644 --- a/src/aki_prj23_transparenzregister/models/company.py +++ b/src/aki_prj23_transparenzregister/models/company.py @@ -27,6 +27,8 @@ class RelationshipRoleEnum(str, MultiValueEnum): STAENDIGER_VERTRETER = "Ständige(r) Vertreter(in)" SONSTIGER_VERTRETER = "Sonstige(r) Vertreter(in)", "Sonstiger Vertreter" GESCHAEFTSLEITER = "Geschäftsleiter(in)", "Geschäftsleiter" + ZWEIGNIEDERLASSUNG = "Zweigniederlassung" + HAUPTNIEDERLASSUNG = "Hauptniederlassung" @dataclass 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 2f599a4..8f5a51c 100644 --- a/src/aki_prj23_transparenzregister/utils/data_extraction/unternehmensregister/transform.py +++ b/src/aki_prj23_transparenzregister/utils/data_extraction/unternehmensregister/transform.py @@ -120,7 +120,9 @@ def parse_stakeholder(data: dict) -> CompanyRelationship | None: if "Organisation" in data["Beteiligter"]: return CompanyToCompanyRelationship( **{ - "role": data["Rolle"]["Rollenbezeichnung"]["content"], + "role": RelationshipRoleEnum( + data["Rolle"]["Rollenbezeichnung"]["content"] + ), "description": data["Beteiligter"]["Organisation"]["Bezeichnung"][ "Bezeichnung_Aktuell" ],