mirror of
https://github.com/Redume/Shirino.git
synced 2025-07-04 03:00:55 +01:00
35 lines
1.2 KiB
Python
35 lines
1.2 KiB
Python
from pathlib import Path
|
|
|
|
import yaml
|
|
|
|
|
|
class I18n:
|
|
"""Load every YAML file in i18n/locales and let you pull out a single-language dict."""
|
|
|
|
def __init__(self, locales_dir: str = "i18n/locales", default_lang: str = "en"):
|
|
base_path = Path(__file__).parent.parent
|
|
self.locales_dir = base_path / locales_dir
|
|
self.default_lang = default_lang
|
|
self.translations: dict[str, dict] = {}
|
|
self._load_translations()
|
|
|
|
def _load_translations(self) -> None:
|
|
for file in self.locales_dir.glob("*.yaml"):
|
|
lang = file.stem.lower()
|
|
with file.open(encoding="utf-8") as f:
|
|
self.translations[lang] = yaml.safe_load(f)
|
|
|
|
def get_locale(self, lang: str | None = None) -> dict:
|
|
"""Return the whole dictionary for one language (fallback → default_lang)."""
|
|
lang = (lang or self.default_lang).lower()[:2]
|
|
lang_dict = self.translations.get(
|
|
lang, self.translations.get(self.default_lang, {})
|
|
)
|
|
|
|
fallback_dict = self.translations.get(self.default_lang, {})
|
|
merged_dict = {
|
|
key: lang_dict.get(key, fallback_dict.get(key, key))
|
|
for key in set(lang_dict) | set(fallback_dict)
|
|
}
|
|
|
|
return merged_dict
|