Retrieval-based-Voice-Conve.../i18n.py

26 lines
744 B
Python
Raw Normal View History

import locale
import json
import os
def load_language_list(language):
2023-04-16 17:30:50 +08:00
with open(f"./i18n/{language}.json", "r", encoding="utf-8") as f:
language_list = json.load(f)
return language_list
class I18nAuto:
def __init__(self, language=None):
2023-04-25 10:51:20 +08:00
if language in ["Auto", None]:
language = locale.getdefaultlocale()[
0
] # getlocale can't identify the system's language ((None, None))
2023-04-16 17:30:50 +08:00
if not os.path.exists(f"./i18n/{language}.json"):
language = "en_US"
self.language = language
2023-04-13 23:57:27 +08:00
print("Use Language:", language)
self.language_map = load_language_list(language)
def __call__(self, key):
return self.language_map.get(key, key)