mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-20 23:41:17 -05:00
82 lines
2.2 KiB
Python
82 lines
2.2 KiB
Python
|
from functools import lru_cache
|
|||
|
|
|
|||
|
|
from notifications_utils.insensitive_dict import InsensitiveDict
|
|||
|
|
from notifications_utils.sanitise_text import SanitiseASCII
|
|||
|
|
|
|||
|
|
from .data import (
|
|||
|
|
ADDITIONAL_SYNONYMS,
|
|||
|
|
COUNTRIES_AND_TERRITORIES,
|
|||
|
|
EUROPEAN_ISLANDS,
|
|||
|
|
ROYAL_MAIL_EUROPEAN,
|
|||
|
|
UK,
|
|||
|
|
UK_ISLANDS,
|
|||
|
|
WELSH_NAMES,
|
|||
|
|
Postage,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
class CountryMapping(InsensitiveDict):
|
|||
|
|
@staticmethod
|
|||
|
|
@lru_cache(maxsize=2048, typed=False)
|
|||
|
|
def make_key(original_key):
|
|||
|
|
original_key = original_key.replace("&", "and")
|
|||
|
|
original_key = original_key.replace("+", "and")
|
|||
|
|
|
|||
|
|
normalised = "".join(
|
|||
|
|
character.lower()
|
|||
|
|
for character in original_key
|
|||
|
|
if character not in " _-'’,.()"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
if "?" in SanitiseASCII.encode(normalised):
|
|||
|
|
return normalised
|
|||
|
|
|
|||
|
|
return SanitiseASCII.encode(normalised)
|
|||
|
|
|
|||
|
|
def __contains__(self, key):
|
|||
|
|
if any(c.isdigit() for c in key):
|
|||
|
|
# A string with a digit can’t be a country and is probably a
|
|||
|
|
# postcode, so let’s do a little optimisation, skip the
|
|||
|
|
# expensive string manipulation to normalise the key and say
|
|||
|
|
# that there’s no matching country
|
|||
|
|
return False
|
|||
|
|
return super().__contains__(key)
|
|||
|
|
|
|||
|
|
def __getitem__(self, key):
|
|||
|
|
for key_ in (key, f"the {key}", f"yr {key}", f"y {key}"):
|
|||
|
|
if key_ in self:
|
|||
|
|
return super().__getitem__(key_)
|
|||
|
|
|
|||
|
|
raise CountryNotFoundError(f"Not a known country or territory ({key})")
|
|||
|
|
|
|||
|
|
|
|||
|
|
countries = CountryMapping(
|
|||
|
|
dict(
|
|||
|
|
COUNTRIES_AND_TERRITORIES
|
|||
|
|
+ UK_ISLANDS
|
|||
|
|
+ EUROPEAN_ISLANDS
|
|||
|
|
+ WELSH_NAMES
|
|||
|
|
+ ADDITIONAL_SYNONYMS
|
|||
|
|
)
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
class Country:
|
|||
|
|
def __init__(self, given_name):
|
|||
|
|
self.canonical_name = countries[given_name]
|
|||
|
|
|
|||
|
|
def __eq__(self, other):
|
|||
|
|
return self.canonical_name == other.canonical_name
|
|||
|
|
|
|||
|
|
@property
|
|||
|
|
def postage_zone(self):
|
|||
|
|
if self.canonical_name == UK:
|
|||
|
|
return Postage.UK
|
|||
|
|
if self.canonical_name in ROYAL_MAIL_EUROPEAN:
|
|||
|
|
return Postage.EUROPE
|
|||
|
|
return Postage.REST_OF_WORLD
|
|||
|
|
|
|||
|
|
|
|||
|
|
class CountryNotFoundError(KeyError):
|
|||
|
|
pass
|