country can be None in get_country

This commit is contained in:
Edward Betts 2025-01-20 12:06:46 +00:00
parent f5a8676336
commit 3a022e8736

View file

@ -33,7 +33,7 @@ def get_country(alpha_2: str) -> pycountry.db.Country | None:
flag="\U0001F1FD\U0001F1F0", name="Kosovo", alpha_2="xk" flag="\U0001F1FD\U0001F1F0", name="Kosovo", alpha_2="xk"
) )
country: pycountry.db.Country = None country: pycountry.db.Country | None = None
if len(alpha_2) == 2: if len(alpha_2) == 2:
country = pycountry.countries.get(alpha_2=alpha_2.upper()) country = pycountry.countries.get(alpha_2=alpha_2.upper())
elif len(alpha_2) == 3: elif len(alpha_2) == 3: