[nit] use explicit types for country overload

yaph · Jul 25, 2024 · ff92eb4 · ff92eb4
1 parent 89ae766
commit ff92eb4
Showing 1 changed file with 3 additions and 15 deletions.
diff --git a/geonamescache/mappers.py b/geonamescache/mappers.py
@@ -4,7 +4,8 @@
 from geonamescache import GeonamesCache
 
 from . import mappings
-from .types import ContinentCode, CountryFields, CountryNumericFields
+from .types import (ContinentCode, CountryFields, CountryNumericFields,
+                    CountryStringFields)
 
 
 @overload
@@ -16,20 +17,7 @@ def country(
 @overload
 def country(
     from_key: str = "name",
-    to_key: Literal[
-        "capital",
-        "currencycode",
-        "currencyname",
-        "iso",
-        "iso3",
-        "fips",
-        "languages",
-        "name",
-        "neighbours",
-        "phone",
-        "postalcoderegex",
-        "tld",
-    ] = "iso",
+    to_key: CountryStringFields = "iso",
 ) -> Callable[[str], str]: ...