summaryrefslogtreecommitdiff
path: root/src/home/bot/lang.py
blob: 2f10358c72ba468e7956df8749b9bfcfc86fb32c (plain)
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
import logging

from typing import Union, Optional

logger = logging.getLogger(__name__)


class LangStrings(dict):
    _lang: Optional[str]

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self._lang = None

    def setlang(self, lang: str):
        self._lang = lang

    def __missing__(self, key):
        logger.warning(f'key {key} is missing in language {self._lang}')
        return '{%s}' % key

    def __setitem__(self, key, value):
        raise NotImplementedError(f'setting translation strings this way is prohibited (was trying to set {key}={value})')


class LangPack:
    strings: dict[str, LangStrings[str, str]]
    default_lang: str

    def __init__(self):
        self.strings = {}
        self.default_lang = 'en'

    def ru(self, **kwargs) -> None:
        self.set(kwargs, 'ru')

    def en(self, **kwargs) -> None:
        self.set(kwargs, 'en')

    def set(self,
            strings: Union[LangStrings, dict],
            lang: str) -> None:

        if isinstance(strings, dict) and not isinstance(strings, LangStrings):
            strings = LangStrings(**strings)
            strings.setlang(lang)

        if lang not in self.strings:
            self.strings[lang] = strings
        else:
            self.strings[lang].update(strings)

    def all(self, key):
        result = []
        for strings in self.strings.values():
            result.append(strings[key])
        return result

    @property
    def languages(self) -> list[str]:
        return list(self.strings.keys())

    def get(self, key: str, lang: str, *args) -> str:
        return self.strings[lang][key] % args

    def __call__(self, *args, **kwargs):
        return self.strings[self.default_lang][args[0]]

    def __getitem__(self, key):
        return self.strings[self.default_lang][key]

    def __setitem__(self, key, value):
        raise NotImplementedError('setting translation strings this way is prohibited')

    def __contains__(self, key):
        return key in self.strings[self.default_lang]