58 lines
1.7 KiB
Python
58 lines
1.7 KiB
Python
from os import listdir, path
|
|
from typing import Any, Dict, List, Union
|
|
|
|
from google_trans_new import google_translator
|
|
from pyUltroid import udB, LOGS
|
|
from yaml import safe_load
|
|
|
|
language = [udB.get_key("language") or "en"]
|
|
languages = {}
|
|
|
|
Trs = google_translator()
|
|
|
|
strings_folder = path.join(path.dirname(path.realpath(__file__)), "strings")
|
|
|
|
for file in listdir(strings_folder):
|
|
if file.endswith(".yml"):
|
|
code = file[:-4]
|
|
try:
|
|
languages[code] = safe_load(
|
|
open(path.join(strings_folder, file), encoding="UTF-8"),
|
|
)
|
|
except Exception as er:
|
|
LOGS.info(f"Error in {file[:-4]} language file")
|
|
LOGS.exception(er)
|
|
|
|
|
|
def get_string(key: str) -> Any:
|
|
lang = language[0]
|
|
try:
|
|
return languages[lang][key]
|
|
except KeyError:
|
|
try:
|
|
en_ = languages["en"][key]
|
|
tr = Trs.translate(en_, lang_tgt=lang).replace("\ N", "\n")
|
|
if en_.count("{}") != tr.count("{}"):
|
|
tr = en_
|
|
if languages.get(lang):
|
|
languages[lang][key] = tr
|
|
else:
|
|
languages.update({lang: {key: tr}})
|
|
return tr
|
|
except KeyError:
|
|
return f"Warning: could not load any string with the key `{key}`"
|
|
except Exception as er:
|
|
LOGS.exception(er)
|
|
return languages["en"].get(key) or f"Failed to load language string '{key}'"
|
|
|
|
|
|
def get_languages() -> Dict[str, Union[str, List[str]]]:
|
|
return {
|
|
code: {
|
|
"name": languages[code]["name"],
|
|
"natively": languages[code]["natively"],
|
|
"authors": languages[code]["authors"],
|
|
}
|
|
for code in languages
|
|
}
|