add auto translator
This commit is contained in:
30
.github/auto_translator.py
vendored
30
.github/auto_translator.py
vendored
@@ -2,18 +2,29 @@ import i18n
|
||||
import json
|
||||
from deep_translator import GoogleTranslator
|
||||
import sys
|
||||
import os
|
||||
|
||||
|
||||
def get_path(lang):
|
||||
if lang == 'en':
|
||||
return f'assets/translations/strings.i18n.json'
|
||||
return f'assets/translations/strings_{lang}.i18n.json'
|
||||
return f'../assets/translations/strings.i18n.json'
|
||||
return f'../assets/translations/strings_{lang}.i18n.json'
|
||||
|
||||
|
||||
def read_translate(lang):
|
||||
pat = get_path(lang)
|
||||
if not os.path.isfile(pat):
|
||||
return {}
|
||||
with open(pat) as f:
|
||||
return json.load(f)
|
||||
|
||||
|
||||
def recursive_translate(src, dst, translator):
|
||||
for sk, sv in src.items():
|
||||
if type(sv) == str and sk not in dst:
|
||||
dst[sk] = translator.translate(sv)
|
||||
if type(sv) == str:
|
||||
if sk not in dst:
|
||||
dst[sk] = translator.translate(sv)
|
||||
print(sk, sv, dst[sk])
|
||||
if not dst[sk]:
|
||||
del dst[sk]
|
||||
else:
|
||||
@@ -26,14 +37,11 @@ if __name__ == "__main__":
|
||||
src = sys.argv[1]
|
||||
dst = sys.argv[2]
|
||||
|
||||
src_file = get_path(src)
|
||||
dst_file = get_path(dst)
|
||||
with open(src_file) as sf:
|
||||
src_pofile = json.load(sf)
|
||||
with open(dst_file) as df:
|
||||
dst_pofile = json.load(df)
|
||||
src_pofile = read_translate(src)
|
||||
dst_pofile = read_translate(dst)
|
||||
|
||||
translator = GoogleTranslator(source=src, target=dst if dst != 'zh' else "zh-CN")
|
||||
recursive_translate(src_pofile, dst_pofile, translator)
|
||||
|
||||
with open(dst_file, 'w') as df:
|
||||
with open(get_path(dst), 'w') as df:
|
||||
json.dump(dst_pofile, df)
|
||||
|
||||
Reference in New Issue
Block a user