mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-07-09 10:54:11 -04:00
57e9fc9ba1
This changeset copies everything in notifications_utils to turn into a local project module. It includes dependency changes to account for this adjustment and the notifications_utils tests. It also fixes all of the import statements. Signed-off-by: Carlo Costino <carlo.costino@gsa.gov>
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
|