Heroku/hikka/translations.py

119 lines
3.4 KiB
Python
Executable File

# █ █ ▀ █▄▀ ▄▀█ █▀█ ▀
# █▀█ █ █ █ █▀█ █▀▄ █
# © Copyright 2022
# https://t.me/hikariatama
#
# 🔒 Licensed under the GNU AGPLv3
# 🌐 https://www.gnu.org/licenses/agpl-3.0.html
import json
import logging
import os
import requests
from . import utils
logger = logging.getLogger(__name__)
class Translator:
def __init__(self, client, db):
self._client = client
self.db = db
async def init(self) -> bool:
self._data = {}
if not (lang := self.db.get(__name__, "lang", False)):
return False
for language in lang.split(" "):
if utils.check_url(language):
try:
ndata = (await utils.run_sync(requests.get, lang)).json()
except Exception:
logger.exception(f"Unable to decode {lang}")
continue
data = ndata.get("data", ndata)
if any(not isinstance(i, str) for i in data.values()):
logger.exception(
"Translation pack format is not valid (typecheck failed)"
)
continue
self._data.update(data)
continue
possible_pack_path = os.path.join(
utils.get_base_dir(),
f"langpacks/{language}.json",
)
if os.path.isfile(possible_pack_path):
with open(possible_pack_path, "r") as f:
self._data.update(json.load(f))
return True
return True
def getkey(self, key):
return self._data.get(key, False)
def gettext(self, text):
return self.getkey(text) or text
class Strings:
def __init__(self, mod, translator):
self._mod = mod
self._translator = translator
if not translator:
logger.debug(f"Module {mod=} got empty translator {translator=}")
self._base_strings = mod.strings # Back 'em up, bc they will get replaced
def __getitem__(self, key: str) -> str:
return (
self._translator.getkey(f"{self._mod.__module__}.{key}")
if self._translator is not None
else False
) or (
getattr(
self._mod,
next(
(
f"strings_{lang}"
for lang in self._translator.db.get(
__name__,
"lang",
"en",
).split(" ")
if hasattr(self._mod, f"strings_{lang}")
and isinstance(getattr(self._mod, f"strings_{lang}"), dict)
and key in getattr(self._mod, f"strings_{lang}")
),
utils.rand(32),
),
self._base_strings,
)
if self._translator is not None
else self._base_strings
).get(
key,
self._base_strings.get(key, "Unknown strings"),
)
def __call__(
self,
key: str,
_=None, # Compatibility tweak for FTG\GeekTG
) -> str:
return self.__getitem__(key)
def __iter__(self):
return self._base_strings.__iter__()