🦺 Validate area input when normalizing locations

VUB-HYDR · Oct 30, 2024 · c083f74 · c083f74
1 parent 58cfef9
commit c083f74
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/Database/scr/normalize_locations.py b/Database/scr/normalize_locations.py
@@ -147,6 +147,13 @@ def normalize_locations(
             try:
                 if area:
                     assert isinstance(area, str), f"Area is not a string: {area}"
+                if re.match(
+                    "(country|location|area|adminarea|admin|admin_area|administrative area|administrative_area|none|null)(\s)*(s)*[0-9]*$",
+                    area,
+                    flags=re.IGNORECASE,
+                ):
+                    self.logger.error(f"Input area={area} type={type(area)} is not a valid area name")
+                    return (None, None, None)
                 if in_country:
                     assert isinstance(in_country, str), f"Country is not a string: {in_country}"
                 assert not (