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

feat: fixed type hints

Browse Source
This commit is contained in:
Hazel Noack 2025-06-12 13:57:08 +02:00
parent 27a04b0e2e
commit de2ab0b8c0
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pycountry_wrapper/country.py
View File

@ -2,6 +2,7 @@ from __future__ import annotations
from typing import Optional from typing import Optional
from functools import wraps from functools import wraps
import pycountry import pycountry
import pycountry.db
from . import config from . import config
@ -22,7 +23,7 @@ class Country:
Empty objects return for every attribute None Empty objects return for every attribute None
""" """
def __init__(self, country: Optional[str] = None, pycountry_object = None) -> None: def __init__(self, country: Optional[str] = None, pycountry_object: Optional[pycountry.db.Country] = 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
@ -31,11 +32,11 @@ class Country:
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")
self.pycountry_object = pycountry_object self.pycountry_object: pycountry.db.Country = pycountry_object
@classmethod @classmethod
def _search_pycountry_object(cls, country: Optional[str], is_fallback: bool = False): def _search_pycountry_object(cls, country: Optional[str], is_fallback: bool = False) -> Optional[pycountry.db.Country]:
# fallback to configured country if necessary # fallback to configured country if necessary
if country is None: if country is None:
if is_fallback: if is_fallback: