Fix counties on import

code4romania · Feb 19, 2024 · 3c6b341 · 3c6b341
1 parent cad5628
commit 3c6b341
Showing 1 changed file with 87 additions and 0 deletions.
diff --git a/backend/importer/tasks.py b/backend/importer/tasks.py
@@ -22,6 +22,90 @@ def parse_imported_date(date_str: str):
     return datetime_obj_with_tz
 
 
+def map_county(county: str) -> str:
+    county_mapping = {
+        "Alba": "Alba",
+        "Arad": "Arad",
+        "Argeș": "Argeș",
+        "Arges": "Argeș",
+        "Bacău": "Bacău",
+        "Bacau": "Bacău",
+        "Bihor": "Bihor",
+        "Bistrița-Năsăud": "Bistrița-Năsăud",
+        "Bistrita-Nasaud": "Bistrița-Năsăud",
+        "Bistrița-Nasaud": "Bistrița-Năsăud",
+        "Bistrita-Năsaud": "Bistrița-Năsăud",
+        "Bistrita-Nasăud": "Bistrița-Năsăud",
+        "Bistrița-Năsaud": "Bistrița-Năsăud",
+        "Bistrița-Nasăud": "Bistrița-Năsăud",
+        "Bistrita-Năsăud": "Bistrița-Năsăud",
+        "Botoșani": "Botoșani",
+        "Botosani": "Botoșani",
+        "Brașov": "Brașov",
+        "Brasov": "Brașov",
+        "Brăila": "Brăila",
+        "Braila": "Brăila",
+        "București": "București",
+        "Bucuresti": "București",
+        "Buzău": "Buzău",
+        "Buzau": "Buzău",
+        "Caraș-Severin": "Caraș-Severin",
+        "Caras-Severin": "Caraș-Severin",
+        "Călărași": "Călărași",
+        "Calarasi": "Călărași",
+        "Călarasi": "Călărași",
+        "Calărasi": "Călărași",
+        "Calarași": "Călărași",
+        "Călărasi": "Călărași",
+        "Călarași": "Călărași",
+        "Calărași": "Călărași",
+        "Cluj": "Cluj",
+        "Constanța": "Constanța",
+        "Constanta": "Constanța",
+        "Covasna": "Covasna",
+        "Dâmbovița": "Dâmbovița",
+        "Dambovita": "Dâmbovița",
+        "Dâmbovita": "Dâmbovița",
+        "Dambovița": "Dâmbovița",
+        "Dolj": "Dolj",
+        "Galati": "Galați",
+        "Giurgiu": "Giurgiu",
+        "Gorj": "Gorj",
+        "Harghita": "Harghita",
+        "Hunedoara": "Hunedoara",
+        "Ialomița": "Ialomița",
+        "Ialomita": "Ialomița",
+        "Iași": "Iași",
+        "Iasi": "Iași",
+        "Ilfov": "Ilfov",
+        "Maramureș": "Maramureș",
+        "Maramures": "Maramureș",
+        "Mehedinți": "Mehedinți",
+        "Mehedinti": "Mehedinți",
+        "Mureș": "Mureș",
+        "Mures": "Mureș",
+        "Neamț": "Neamț",
+        "Neamt": "Neamț",
+        "Olt": "Olt",
+        "Prahova": "Prahova",
+        "Satu Mare": "Satu Mare",
+        "Sălaj": "Sălaj",
+        "Salaj": "Sălaj",
+        "Sibiu": "Sibiu",
+        "Suceava": "Suceava",
+        "Teleorman": "Teleorman",
+        "Timiș": "Timiș",
+        "Timis": "Timiș",
+        "Tulcea": "Tulcea",
+        "Vaslui": "Vaslui",
+        "Vâlcea": "Vâlcea",
+        "Valcea": "Vâlcea",
+        "Vrancea": "Vrancea",
+    }
+
+    return county_mapping.get(county, county)
+
+
 def ngo_slugs_to_ids(ngo_slugs: str) -> List[int]:
     ngo_slugs = ngo_slugs.split(",")
     ngo_ids = []
@@ -46,6 +130,8 @@ def ngo_slugs_to_ids(ngo_slugs: str) -> List[int]:
             "date_created": parse_imported_date,
             "bank_account": lambda x: x.replace(" ", ""),
             "address": lambda x: x.strip(),
+            "county": map_county,
+            "active_region": map_county,
         },
         "fields_post": {},
     },
@@ -64,6 +150,7 @@ def ngo_slugs_to_ids(ngo_slugs: str) -> List[int]:
         ),
         "fields_mapping": {
             "date_created": parse_imported_date,
+            "county": map_county,
         },
         "fields_post": {},
     },