107 lines
2.6 KiB
Python
107 lines
2.6 KiB
Python
# Copyright: Ankitects Pty Ltd and contributors
|
|
# License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html
|
|
|
|
# pylint: disable=unbalanced-tuple-unpacking
|
|
|
|
import json
|
|
import sys
|
|
from typing import Literal, TypedDict
|
|
|
|
sys.path.append("pylib/anki/_vendor")
|
|
|
|
import stringcase
|
|
|
|
strings_json, outfile = sys.argv[1:]
|
|
with open(strings_json, encoding="utf8") as f:
|
|
modules = json.load(f)
|
|
|
|
|
|
class Variable(TypedDict):
|
|
name: str
|
|
kind: Literal["Any", "Int", "String", "Float"]
|
|
|
|
|
|
def legacy_enum() -> str:
|
|
out = ["class LegacyTranslationEnum:"]
|
|
for module in modules:
|
|
for translation in module["translations"]:
|
|
key = stringcase.constcase(translation["key"])
|
|
value = f'({module["index"]}, {translation["index"]})'
|
|
out.append(f" {key} = {value}")
|
|
|
|
return "\n".join(out) + "\n"
|
|
|
|
|
|
def methods() -> str:
|
|
out = [
|
|
"class GeneratedTranslations:",
|
|
" def _translate(self, module: int, translation: int, args: Dict) -> str:",
|
|
" raise Exception('not implemented')",
|
|
]
|
|
for module in modules:
|
|
for translation in module["translations"]:
|
|
key = translation["key"].replace("-", "_")
|
|
arg_types = get_arg_types(translation["variables"])
|
|
args = get_args(translation["variables"])
|
|
doc = translation["text"]
|
|
out.append(
|
|
f"""
|
|
def {key}(self, {arg_types}) -> str:
|
|
r''' {doc} '''
|
|
return self._translate({module["index"]}, {translation["index"]}, {{{args}}})
|
|
"""
|
|
)
|
|
|
|
return "\n".join(out) + "\n"
|
|
|
|
|
|
def get_arg_types(args: list[Variable]) -> str:
|
|
return ", ".join(
|
|
[f"{stringcase.snakecase(arg['name'])}: {arg_kind(arg)}" for arg in args]
|
|
)
|
|
|
|
|
|
def arg_kind(arg: Variable) -> str:
|
|
if arg["kind"] == "Int":
|
|
return "int"
|
|
elif arg["kind"] == "Any":
|
|
return "FluentVariable"
|
|
elif arg["kind"] == "Float":
|
|
return "float"
|
|
else:
|
|
return "str"
|
|
|
|
|
|
def get_args(args: list[Variable]) -> str:
|
|
return ", ".join(
|
|
[f'"{arg["name"]}": {stringcase.snakecase(arg["name"])}' for arg in args]
|
|
)
|
|
|
|
|
|
out = ""
|
|
|
|
out += legacy_enum()
|
|
out += methods()
|
|
|
|
|
|
with open(outfile, "w", encoding="utf8") as f:
|
|
f.write(
|
|
'''# Copyright: Ankitects Pty Ltd and contributors
|
|
# License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html
|
|
# pylint: skip-file
|
|
|
|
from __future__ import annotations
|
|
|
|
"""
|
|
This file is automatically generated from the *.ftl files.
|
|
"""
|
|
|
|
import enum
|
|
from typing import Dict, Union
|
|
|
|
FluentVariable = Union[str, int, float]
|
|
|
|
'''
|
|
+ out
|
|
)
|