Mypy: Return early with default if locale setting is None

eduvpn · May 9, 2022 · 353c736 · 353c736
1 parent 41234f8
commit 353c736
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/eduvpn/i18n.py b/eduvpn/i18n.py
@@ -40,14 +40,20 @@ def setup(app_variant: ApplicationVariant, prefix: str):
 
 def country() -> str:
     try:
-        return locale.getlocale()[0].replace('_', '-')
+        locale_setting = locale.getlocale()[0]
+        if not locale_setting:
+            return COUNTRY
+        return locale_setting.replace('_', '-')
     except Exception:
         return COUNTRY
 
 
 def language() -> str:
     try:
-        return locale.getlocale()[0].split('_')[0]
+        locale_setting = locale.getlocale()[0]
+        if not locale_setting:
+            return LANGUAGE
+        return locale_setting.split('_')[0]
     except Exception:
         return LANGUAGE