103 lines
3.0 KiB
Python
103 lines
3.0 KiB
Python
import os
|
|
|
|
import json
|
|
|
|
import re
|
|
import sys
|
|
|
|
import polib
|
|
|
|
# extract a translated string from the gettext catalogs and insert it into ftl
|
|
# eg:
|
|
# $ python extract-po-string.py media-check.ftl delete-unused "Delete Unused Media" "", 1
|
|
# $ python extract-po-string.py media-check.ftl delete-unused "%(a)s %(b)s" "%(a)s=val1,%(b)s=val2" 0
|
|
ftl_filename, key, msgid, repls, dry_run = sys.argv[1:]
|
|
|
|
replacements = []
|
|
for repl in repls.split(","):
|
|
replacements.append(repl.split("="))
|
|
|
|
print("Loading catalogs...")
|
|
base = "po/desktop"
|
|
langs = [d for d in os.listdir(base) if d != "anki.pot"]
|
|
cats = []
|
|
for lang in langs:
|
|
po_path = os.path.join(base, lang, "anki.po")
|
|
cat = polib.pofile(po_path)
|
|
cats.append((lang, cat))
|
|
|
|
to_insert = []
|
|
for (lang, cat) in cats:
|
|
for entry in cat:
|
|
if entry.msgid == msgid:
|
|
if entry.msgstr:
|
|
msg = entry.msgstr
|
|
if replacements:
|
|
for (old,new) in replacements:
|
|
msg = msg.replace(old, f"{{${new}}}")
|
|
print(lang, "has", msg)
|
|
to_insert.append((lang, msg))
|
|
break
|
|
elif entry.msgstr_plural and entry.msgstr_plural[0]:
|
|
# convert the dict into a list in the correct order
|
|
plurals = list(entry.msgstr_plural.items())
|
|
plurals.sort()
|
|
# update variables and discard keys
|
|
adjusted = []
|
|
for _k, msg in plurals:
|
|
if replacements:
|
|
for (old,new) in replacements:
|
|
msg = msg.replace(old, f"{{${new}}}")
|
|
adjusted.append(msg)
|
|
print(lang, "has", adjusted)
|
|
if len(adjusted) > 1:
|
|
to_insert.append((lang, adjusted))
|
|
else:
|
|
# language doesn't have plurals
|
|
to_insert.append((lang, adjusted[0]))
|
|
|
|
plurals = json.load(open("plurals.json"))
|
|
|
|
def plural_text(key, lang, translation):
|
|
lang = re.sub("(_|-).*", "", lang)
|
|
|
|
# extract the variable - there should be only one
|
|
var = re.findall(r"{(\$.*?)}", translation[0])
|
|
if not len(var) == 1:
|
|
return None
|
|
var = var[0]
|
|
|
|
buf = f"{key} = {{ {var} ->\n"
|
|
|
|
# for each of the plural forms except the last
|
|
for idx, msg in enumerate(translation[:-1]):
|
|
plural_form = plurals[lang][idx]
|
|
buf += f" [{plural_form}] {msg}\n"
|
|
|
|
# add the catchall
|
|
msg = translation[-1]
|
|
buf += f" *[other] {msg}\n"
|
|
buf += " }\n"
|
|
return buf
|
|
|
|
for lang, translation in to_insert:
|
|
dir = os.path.join("ftl", "core", lang)
|
|
ftl_path = os.path.join(dir, ftl_filename)
|
|
|
|
if dry_run == "1":
|
|
continue
|
|
|
|
if not os.path.exists(dir):
|
|
os.mkdir(dir)
|
|
|
|
# simple, non-plural form?
|
|
if isinstance(translation, str):
|
|
buf = f"{key} = {translation}\n"
|
|
else:
|
|
buf = plural_text(key, lang, translation)
|
|
|
|
if buf:
|
|
open(ftl_path, "a").write(buf)
|
|
|
|
print("done")
|