197 lines
6.7 KiB
Python
197 lines
6.7 KiB
Python
import re, os
|
||
import shutil
|
||
import json, zhconv
|
||
from datetime import datetime
|
||
from pathlib import Path
|
||
|
||
src_text_path = Path("text/en_US")
|
||
|
||
|
||
# 复制原始文件的函数
|
||
def initialize_files_lua():
|
||
filelist = Path("filelist.txt")
|
||
with filelist.open("w", encoding="utf-8") as f:
|
||
if src_text_path.exists():
|
||
shutil.rmtree(src_text_path)
|
||
src_text_path.mkdir(parents=True)
|
||
|
||
for lua_file in Path("../code").rglob("*.lua"):
|
||
with lua_file.open("r", encoding="utf-8") as f2:
|
||
content = f2.read()
|
||
match = re.search(r"local function textGen.*?\n(?:.*\n)*?end", content)
|
||
if match:
|
||
shutil.copy(lua_file, src_text_path)
|
||
f.write(f"{lua_file}\n")
|
||
|
||
|
||
# 使用正则匹配
|
||
def export_translation_json(input_file, output_file, blank=False):
|
||
input_file = Path(input_file)
|
||
output_file = Path(output_file)
|
||
if not input_file.exists():
|
||
return
|
||
output_file.parent.mkdir(parents=True, exist_ok=True)
|
||
|
||
transdict = {}
|
||
with input_file.open("r", encoding="utf-8") as f2:
|
||
stringset = set()
|
||
content = re.search(
|
||
r"local function textGen.*?\n(?:.*\n)*?end", f2.read()
|
||
).group(0)
|
||
content = re.sub(r'(?<=[^\\\]]")(,\s*\n?\s*[^\\\[]")', r',\n"', content)
|
||
content = re.sub(r'(?<=[^\\\]]")(\s*\n?\s*[^\\\[]\})', r"}", content)
|
||
content = (
|
||
content.replace("\u0010", "\u2190")
|
||
.replace("\u0011", "\u2191")
|
||
.replace("\u0012", "\u2192")
|
||
.replace("\u0013", "\u2193")
|
||
)
|
||
multiplelines = [
|
||
"".join(p)
|
||
for p in re.findall(
|
||
r'\{[\s\n]*((?:"(?:[^"\\]|\\.)*"\s*,?\s*)+)[\s\n]*\}',
|
||
content,
|
||
)
|
||
]
|
||
multiplelines_single = []
|
||
for ct, line in enumerate(multiplelines):
|
||
newline = re.findall(r'"((?:\\.|[^"\\])*)"', line)
|
||
if "\n".join(newline).replace('\\"', '"') not in stringset:
|
||
transdict[f"{ct}-multiple"] = (
|
||
"" if blank else "\n".join(newline).replace('\\"', '"')
|
||
)
|
||
stringset.add("\n".join(newline).replace('\\"', '"'))
|
||
multiplelines_single.extend(newline)
|
||
|
||
singlelines = re.findall(
|
||
r'(?<=[{\(\s\n])"((?:\\.|[^"\\])*)"(?=[}\),\s\n])', content
|
||
)
|
||
|
||
singlelines_unique = []
|
||
for line in singlelines:
|
||
if line not in multiplelines_single:
|
||
singlelines_unique.append(line)
|
||
|
||
for ct, line in enumerate(singlelines_unique):
|
||
if line not in stringset:
|
||
transdict[f"{ct}-single"] = "" if blank else line.replace('\\"', '"')
|
||
stringset.add(line)
|
||
|
||
with output_file.open("w", encoding="utf-8") as f:
|
||
json.dump(transdict, f, ensure_ascii=False, indent=4)
|
||
|
||
|
||
def convert_zh_to_tw(input_file, output_file):
|
||
with open(input_file, "r", encoding="utf-8") as f:
|
||
content = f.read()
|
||
content_tw = zhconv.convert(content, "zh-tw")
|
||
# 寫入檔案
|
||
with open(output_file, "w", encoding="utf-8") as f:
|
||
f.write(content_tw)
|
||
|
||
|
||
# 从新json中更新旧翻译
|
||
def update_translation_json(old_srctext, new_srctext, old_dsttext, new_dsttext):
|
||
with open(old_srctext, "r", encoding="utf-8") as f:
|
||
old_src = json.load(f)
|
||
with open(new_srctext, "r", encoding="utf-8") as f:
|
||
new_src = json.load(f)
|
||
with open(old_dsttext, "r", encoding="utf-8") as f:
|
||
old_dst = json.load(f)
|
||
|
||
# 首先从old_src, old_dst建立翻译字典
|
||
translation_dict = {}
|
||
for key, item in old_src.items():
|
||
if key in old_dst and old_dst[key] != "":
|
||
translation_dict[item] = old_dst[key]
|
||
# 然后从new_src中获取新的翻译
|
||
dst = {}
|
||
newtext_dict = {}
|
||
notfound_dict = {}
|
||
for key, item in new_src.items():
|
||
if item in translation_dict.keys():
|
||
dst[key] = translation_dict[item]
|
||
else:
|
||
dst[key] = ""
|
||
newtext_dict[key] = item
|
||
|
||
for item in translation_dict:
|
||
if item not in new_src.values():
|
||
notfound_dict[item] = translation_dict[item]
|
||
|
||
# 将翻译写入新json
|
||
with open(new_dsttext, "w", encoding="utf-8") as f:
|
||
json.dump(dst, f, ensure_ascii=False, indent=4)
|
||
|
||
# 将未找到的翻译写入json
|
||
with open(
|
||
new_dsttext.replace(".json", "") + "-newtext.txt", "w+", encoding="utf-8"
|
||
) as f:
|
||
json.dump(newtext_dict, f, ensure_ascii=False, indent=4)
|
||
|
||
with open(
|
||
new_dsttext.replace(".json", "") + "-notfound.txt", "w+", encoding="utf-8"
|
||
) as f:
|
||
json.dump(notfound_dict, f, ensure_ascii=False, indent=4)
|
||
|
||
|
||
# 使用正则匹配
|
||
def import_translation_json(
|
||
input_lua, input_src_json, input_translated_json, output_lua
|
||
):
|
||
input_lua = Path(input_lua)
|
||
output_lua = Path(output_lua)
|
||
if not input_lua.exists():
|
||
return
|
||
output_lua.parent.mkdir(parents=True, exist_ok=True)
|
||
|
||
with open(input_src_json, "r", encoding="utf-8") as f1:
|
||
src = json.load(f1)
|
||
with open(input_translated_json, "r", encoding="utf-8") as f2:
|
||
trans = json.load(f2)
|
||
|
||
with input_lua.open("r", encoding="utf-8") as f2:
|
||
content = f2.read()
|
||
|
||
# 替换控制符
|
||
content = (
|
||
content.replace("\u0010", "\u2190")
|
||
.replace("\u0011", "\u2191")
|
||
.replace("\u0012", "\u2192")
|
||
.replace("\u0013", "\u2193")
|
||
)
|
||
|
||
for key, value in src.items():
|
||
if key not in trans or not trans[key]:
|
||
continue
|
||
|
||
if "multiple" in key:
|
||
# 生成原始匹配的正则
|
||
lines_src = [line.replace('"', '\\"') for line in value.split("\n")]
|
||
lines_trans = [
|
||
re.sub(r'(\\[n\\"]|")', r"\\\1", line)
|
||
for line in trans[key].split("\n")
|
||
]
|
||
|
||
escaped_src = r",\s*".join([re.escape(f'"{l}"') for l in lines_src])
|
||
replacement = "{\n" + ",\n".join([f'"{l}"' for l in lines_trans]) + "\n}"
|
||
|
||
pattern = re.compile(r"\{\s*" + escaped_src + r"\s*\}")
|
||
content = pattern.sub(replacement, content)
|
||
|
||
elif "single" in key:
|
||
src_text = re.escape(value.replace('"', '\\"'))
|
||
trans_text = re.sub(r'(\\[n\\"]|")', r"\\\1", trans[key])
|
||
pattern = re.compile(r'"' + src_text + r'"')
|
||
content = pattern.sub(f'"{trans_text}"', content)
|
||
|
||
with output_lua.open("w", encoding="utf-8") as f:
|
||
f.write(content)
|
||
|
||
# 读取filelist.txt，把生成的文件导入到对应的路径
|
||
with open("filelist.txt", "r", encoding="utf-8") as f:
|
||
content = f.read().split("\n")
|
||
for line in content:
|
||
if input_lua.stem in line:
|
||
shutil.copy(output_lua, Path(line))
|