Hazel/pycountry-wrapper
1
0
Fork 0
generated from Hazel/python-project
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: fallback country

Browse Source
This commit is contained in:
Hazel Noack 2025-06-12 15:17:15 +02:00
parent 78d60c9908
commit 62fd09b8de
2 changed files with 38 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
pycountry_wrapper/__main__.py
View File

@ -17,3 +17,13 @@ def cli():
empty_country = EmptyCountry(country="zwx") empty_country = EmptyCountry(country="zwx")
print(type(empty_country)) print(type(empty_country))
print(empty_country) print(empty_country)
print()
normal_country = Country("UK")
print(type(normal_country))
print(normal_country)
print()
fallback_country = Country("zwx")
print(type(fallback_country))
print(fallback_country)

54
pycountry_wrapper/country.py
View File

@ -27,12 +27,12 @@ class Country:
self, self,
country: Optional[str] = None, country: Optional[str] = None,
pycountry_object: Optional[pycountry.db.Country] = None, pycountry_object: Optional[pycountry.db.Country] = None,
force_disable_fallback: bool = False disable_fallback: bool = False
) -> None: ) -> None:
if pycountry_object is None: if pycountry_object is None:
# search for the country string instead if the pycountry_object isn't given # search for the country string instead if the pycountry_object isn't given
# this also implements the optional fallback # this also implements the optional fallback
pycountry_object = self._search_pycountry_object(country=country, force_disable_fallback=force_disable_fallback) pycountry_object = self._search_pycountry_object(country=country, disable_fallback=disable_fallback)
if pycountry_object is None: if pycountry_object is None:
raise EmptyCountryException(f"the country {country} was not found and config.fallback_country isn't set") raise EmptyCountryException(f"the country {country} was not found and config.fallback_country isn't set")
@ -41,35 +41,35 @@ class Country:
@classmethod @classmethod
def _search_pycountry_object(cls, country: Optional[str], is_fallback: bool = False, force_disable_fallback: bool = False) -> Optional[pycountry.db.Country]: def _search_pycountry_object(cls, country: Optional[str], disable_fallback: bool = False) -> Optional[pycountry.db.Country]:
# fallback to configured country if necessary
if country is None and not force_disable_fallback:
if force_disable_fallback or config.fallback_country is None:
return None
country = config.fallback_country
pycountry_object = None pycountry_object = None
# the reason I don't immediately return the result is because then there would be a chance if country is not None:
# I would return None even though a country could be found through fuzzy search # the reason I don't immediately return the result is because then there would be a chance
country = country.strip() # I would return None even though a country could be found through fuzzy search
if len(country) == 2: country = country.strip()
pycountry_object = pycountry.countries.get(alpha_2=country.upper()) if len(country) == 2:
elif len(country) == 3: pycountry_object = pycountry.countries.get(alpha_2=country.upper())
pycountry_object = pycountry.countries.get(alpha_3=country.upper()) elif len(country) == 3:
pycountry_object = pycountry.countries.get(alpha_3=country.upper())
if pycountry_object is not None:
return pycountry_object
# fuzzy search if enabled
if config.allow_fuzzy_search:
# fuzzy search raises lookup error if nothing was found
try:
found_countries = pycountry.countries.search_fuzzy(country)
if len(found_countries):
return found_countries[0]
except LookupError:
pass
if pycountry_object is not None: if pycountry_object is not None:
return pycountry_object return pycountry_object
# fuzzy search if enabled if config.fallback_country is not None and not disable_fallback:
if config.allow_fuzzy_search: return cls._search_pycountry_object(country=config.fallback_country, disable_fallback=True)
# fuzzy search raises lookup error if nothing was found
try:
found_countries = pycountry.countries.search_fuzzy(country)
if len(found_countries):
return found_countries[0]
except LookupError:
pass
@classmethod @classmethod
@ -122,7 +122,7 @@ class EmptyCountry(Country):
""" """
def __new__(cls, country: Optional[str] = None, pycountry_object: Optional[pycountry.db.Country] = None, **kwargs): def __new__(cls, country: Optional[str] = None, pycountry_object: Optional[pycountry.db.Country] = None, **kwargs):
try: try:
return Country(country=country, pycountry_object=pycountry_object, force_disable_fallback=False) return Country(country=country, pycountry_object=pycountry_object, disable_fallback=True)
except EmptyCountryException: except EmptyCountryException:
return super().__new__(cls) return super().__new__(cls)