File size: 2,192 Bytes
618430a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 |
import os
import sys
from glob import glob
from typing import Any, Dict, List, Union
from pyUltroid import *
from pyUltroid.fns.tools import translate
try:
from yaml import safe_load
except ModuleNotFoundError:
from pyUltroid.fns.tools import safe_load
ULTConfig.lang = udB.get_key("language") or os.getenv("LANGUAGE", "en")
languages = {}
PATH = "strings/strings/{}.yml"
def load(file):
if not file.endswith(".yml"):
return
elif not os.path.exists(file):
file = PATH.format("en")
code = file.split("/")[-1].split("\\")[-1][:-4]
try:
languages[code] = safe_load(
open(file, encoding="UTF-8"),
)
except Exception as er:
LOGS.info(f"Error in {file[:-4]} language file")
LOGS.exception(er)
load(PATH.format(ULTConfig.lang))
def get_string(key: str, _res: bool = True) -> Any:
lang = ULTConfig.lang or "en"
try:
return languages[lang][key]
except KeyError:
try:
en_ = languages["en"][key]
tr = 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:
if not _res:
return
return f"Warning: could not load any string with the key `{key}`"
except TypeError:
pass
except Exception as er:
LOGS.exception(er)
if not _res:
return None
return languages["en"].get(key) or f"Failed to load language string '{key}'"
def get_help(key):
doc = get_string(f"help_{key}", _res=False)
if doc:
return get_string("cmda") + doc
def get_languages() -> Dict[str, Union[str, List[str]]]:
for file in glob("strings/strings/*yml"):
load(file)
return {
code: {
"name": languages[code]["name"],
"natively": languages[code]["natively"],
"authors": languages[code]["authors"],
}
for code in languages
}
|