mirror of
https://github.com/trezor/trezor-firmware.git
synced 2025-01-30 17:21:21 +00:00
84 lines
2.7 KiB
Python
84 lines
2.7 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
import collections
|
|
import json
|
|
|
|
import click
|
|
|
|
from cli import TranslationsDir
|
|
from trezorlib._internal import translations
|
|
from trezorlib.debuglink import LayoutType
|
|
|
|
|
|
HERE = Path(__file__).parent
|
|
|
|
# staging directory for layout-specific translation JSON files
|
|
CROWDIN_DIR = HERE / "crowdin"
|
|
|
|
# layouts with translation support
|
|
ALL_LAYOUTS = frozenset(LayoutType) - {LayoutType.T1}
|
|
|
|
|
|
@click.group()
|
|
def cli() -> None:
|
|
pass
|
|
|
|
|
|
@cli.command()
|
|
def split() -> None:
|
|
"""Split translation files for Crowdin upload.
|
|
|
|
Create a separate JSON file for each language and layout.
|
|
"""
|
|
tdir = TranslationsDir()
|
|
|
|
for lang in tdir.all_languages():
|
|
blob_json = tdir.load_lang(lang)
|
|
for layout_type in ALL_LAYOUTS:
|
|
# extract translations specific to this layout
|
|
layout_specific_translations = {
|
|
key: translations.get_translation(blob_json, key, layout_type)
|
|
for key in blob_json["translations"].keys()
|
|
}
|
|
# create a JSON file with only the "translations" item
|
|
result = {"translations": layout_specific_translations}
|
|
with open(CROWDIN_DIR / f"{lang}_{layout_type.name}.json", "w") as f:
|
|
json.dump(result, f, indent=2, ensure_ascii=False)
|
|
|
|
click.echo(f"Successfully generated layout-specific translation files in '{CROWDIN_DIR}'")
|
|
|
|
|
|
@cli.command()
|
|
def merge() -> None:
|
|
"""Merge back translation files downloaded from Crowdin."""
|
|
tdir = TranslationsDir()
|
|
|
|
for lang in sorted(tdir.all_languages()):
|
|
merged_translations: dict[str, str | dict[str, str]] = collections.defaultdict(dict)
|
|
for layout_type in ALL_LAYOUTS:
|
|
with open(CROWDIN_DIR / f"{lang}_{layout_type.name}.json", "r") as f:
|
|
blob_json = json.load(f)
|
|
|
|
# mapping string name to its translation (for the current layout)
|
|
layout_specific_translations: dict[str, str] = blob_json["translations"]
|
|
for key, value in layout_specific_translations.items():
|
|
merged_translations[key][layout_type.name] = value
|
|
|
|
for key in merged_translations.keys():
|
|
# deduplicate entries if all translations are the same
|
|
unique_translations = set(merged_translations[key].values())
|
|
if len(unique_translations) == 1:
|
|
merged_translations[key] = unique_translations.pop()
|
|
|
|
blob_json = tdir.load_lang(lang)
|
|
blob_json["translations"] = merged_translations
|
|
tdir.save_lang(lang, blob_json)
|
|
click.echo(f"Updated {lang}")
|
|
|
|
click.echo(f"Successfully merged back layout-specific translation files from '{CROWDIN_DIR}'")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
cli()
|