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

fix: lookup error for fuzzy search

Browse Source
This commit is contained in:
Hazel Noack 2025-06-12 14:40:23 +02:00
parent 65b0c2b8e6
commit af3fff8559
2 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pycountry_wrapper/__main__.py
View File

@ -5,4 +5,4 @@ from . import Country, EmptyCountry
def cli():
print(f"Running {__name__} version {__version__} from __main__.py")
print(EmptyCountry(country="doesn't exist").name)
print(EmptyCountry(country="zwx").name)

11
pycountry_wrapper/country.py
View File

@ -58,9 +58,14 @@ class Country:
# fuzzy search if enabled
if config.allow_fuzzy_search:
found_countries = pycountry.countries.search_fuzzy(country)
if len(found_countries):
return found_countries[0]
# 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
def search(cls, country: Optional[str]) -> Optional[Country]: