KankenOnline/kanken_online/lang.py

82 lines
2.7 KiB
Python

import json
import os
from pathlib import Path
from typing import Dict, Optional
import click
from flask import Flask, session, current_app
LanguageMapping = Dict[str, str]
def load_language(lang_code: str) -> LanguageMapping:
with open(Path("kanken_online/static/lang", f"{lang_code}.json")) as f:
return json.load(f)
LANGUAGES = {}
def update_languages():
ENGLISH = load_language("en")
JAPANESE = load_language("ja")
global LANGUAGES
LANGUAGES = {
"en": ENGLISH,
"ja": JAPANESE
}
update_languages()
# assert all(all(key in lang for key in EXISTING_STRINGS) for lang in LANGUAGES) # Ensure all strings are mapped for all existing languages
# assert not [key for lang in LANGUAGES for key in lang if ((key in lang) and (key not in EXISTING_STRINGS))]
# assert not any((((key in lang) and (key not in EXISTING_STRINGS)) for key in lang) for lang in LANGUAGES) # Ensure no languages have strings not specified by the main index
def localize(text_id: str, language: Optional[Dict[str, str]] = None) -> str:
if language is None:
preference = session.get("language", "ja")
language = LANGUAGES.get(preference)
assert language is not None
return language.get(text_id, text_id)
@click.command("add-string")
@click.argument("string")
def add_string(string):
"""Add a new string to the translations."""
for path in Path("kanken_online/static/lang").glob("*.json"):
with open(path) as f:
data = json.load(f)
data[string] = ""
with open(path, mode="w") as f:
json.dump(data, f, ensure_ascii=False, indent=0)
@click.command("set-string")
@click.argument("lang")
@click.argument("string")
@click.argument("value")
def set_string(lang, string, value):
"""Set the value of a string for a particular language's translation file."""
path = Path("kanken_online/static/lang", f"{lang}.json")
with open(path) as f:
data = json.load(f)
data[string] = value
with open(path, mode="w") as f:
json.dump(data, f, ensure_ascii=False, indent=0)
@click.command("rename-string")
@click.argument("string")
@click.argument("new_name")
def rename_string(string, new_name):
"""Rename a string in the translations files."""
for path in Path("kanken_online/static/lang").glob("*.json"):
with open(path) as f:
data = json.load(f)
data[new_name] = data[string]
del data[string]
with open(path, mode="w") as f:
json.dump(data, f, ensure_ascii=False, indent=0)
def add_translation_commands(app: Flask):
app.cli.add_command(add_string)
app.cli.add_command(set_string)
app.cli.add_command(rename_string)