145 lines
3.8 KiB
Python
145 lines
3.8 KiB
Python
from __future__ import annotations
|
|
|
|
import json
|
|
import logging
|
|
import os
|
|
import gettext
|
|
|
|
from pathlib import Path
|
|
from typing import List, Dict, Any, TYPE_CHECKING, Tuple
|
|
from .exceptions import TranslationError
|
|
|
|
if TYPE_CHECKING:
|
|
_: Any
|
|
|
|
|
|
class LanguageDefinitions:
|
|
_languages = 'languages.json'
|
|
_cyrillic = 'cyrillic.json'
|
|
|
|
def __init__(self):
|
|
self._mappings = self._get_language_mappings()
|
|
self._cyrillic_languages = self._get_cyrillic_languages()
|
|
|
|
def is_cyrillic(self, language: str) -> bool:
|
|
return language in self._cyrillic_languages
|
|
|
|
def _get_language_mappings(self) -> List[Dict[str, str]]:
|
|
locales_dir = Translation.get_locales_dir()
|
|
languages = Path.joinpath(locales_dir, self._languages)
|
|
|
|
with open(languages, 'r') as fp:
|
|
return json.load(fp)
|
|
|
|
def get_language(self, abbr: str) -> str:
|
|
for entry in self._mappings:
|
|
if entry['abbr'] == abbr:
|
|
return entry['lang']
|
|
|
|
raise ValueError(f'No language with abbreviation "{abbr}" found')
|
|
|
|
def _get_cyrillic_languages(self) -> List[str]:
|
|
locales_dir = Translation.get_locales_dir()
|
|
languages = Path.joinpath(locales_dir, self._cyrillic)
|
|
|
|
with open(languages, 'r') as fp:
|
|
data = json.load(fp)
|
|
return data['languages']
|
|
|
|
|
|
class DeferredTranslation:
|
|
def __init__(self, message: str):
|
|
self.message = message
|
|
|
|
def __len__(self) -> int:
|
|
return len(self.message)
|
|
|
|
def __str__(self) -> str:
|
|
translate = _
|
|
if translate is DeferredTranslation:
|
|
return self.message
|
|
return translate(self.message)
|
|
|
|
def __lt__(self, other) -> bool:
|
|
return self.message < other
|
|
|
|
def __gt__(self, other) -> bool:
|
|
return self.message > other
|
|
|
|
def __add__(self, other) -> DeferredTranslation:
|
|
if isinstance(other, str):
|
|
other = DeferredTranslation(other)
|
|
|
|
concat = self.message + other.message
|
|
return DeferredTranslation(concat)
|
|
|
|
def format(self, *args) -> str:
|
|
return self.message.format(*args)
|
|
|
|
@classmethod
|
|
def install(cls):
|
|
import builtins
|
|
builtins._ = cls
|
|
|
|
|
|
class Translation:
|
|
def __init__(self, locales_dir):
|
|
self._languages = {}
|
|
|
|
for names in self._get_translation_lang():
|
|
try:
|
|
self._languages[names[0]] = gettext.translation('base', localedir=locales_dir, languages=names)
|
|
except FileNotFoundError as error:
|
|
raise TranslationError(f"Could not locate language file for '{names}': {error}")
|
|
|
|
def activate(self, name):
|
|
if language := self._languages.get(name, None):
|
|
languages = LanguageDefinitions()
|
|
|
|
if languages.is_cyrillic(name):
|
|
self._set_font('UniCyr_8x16')
|
|
else:
|
|
# this will reset a possible previously set font to a default font
|
|
self._set_font('')
|
|
|
|
language.install()
|
|
else:
|
|
raise ValueError(f'Language not supported: {name}')
|
|
|
|
def _set_font(self, font: str):
|
|
from archinstall import SysCommand, log
|
|
try:
|
|
log(f'Setting new font: {font}', level=logging.DEBUG)
|
|
SysCommand(f'setfont {font}')
|
|
except Exception:
|
|
log(f'Unable to set font {font}', level=logging.ERROR)
|
|
|
|
@classmethod
|
|
def load_nationalization(cls) -> Translation:
|
|
locales_dir = cls.get_locales_dir()
|
|
return Translation(locales_dir)
|
|
|
|
@classmethod
|
|
def get_locales_dir(cls) -> Path:
|
|
cur_path = Path(__file__).parent.parent
|
|
locales_dir = Path.joinpath(cur_path, 'locales')
|
|
return locales_dir
|
|
|
|
@classmethod
|
|
def _defined_languages(cls) -> List[str]:
|
|
locales_dir = cls.get_locales_dir()
|
|
filenames = os.listdir(locales_dir)
|
|
return list(filter(lambda x: len(x) == 2, filenames))
|
|
|
|
@classmethod
|
|
def _get_translation_lang(cls) -> List[Tuple[str, str]]:
|
|
def_languages = cls._defined_languages()
|
|
languages = LanguageDefinitions()
|
|
return [(languages.get_language(lang), lang) for lang in def_languages]
|
|
|
|
@classmethod
|
|
def get_available_lang(cls) -> List[str]:
|
|
def_languages = cls._defined_languages()
|
|
languages = LanguageDefinitions()
|
|
return [languages.get_language(lang) for lang in def_languages]
|