121 lines
4.0 KiB
Python
121 lines
4.0 KiB
Python
import re
|
|
from os.path import splitext, exists, dirname, join
|
|
from os import listdir, makedirs
|
|
import json
|
|
|
|
|
|
def replace_fallback(str, fallback):
|
|
"""
|
|
替换字符串中的回退字符
|
|
:param str: 原始字符串
|
|
:param fallback: 回退字典
|
|
:return: 替换后的字符串
|
|
"""
|
|
for key, value in fallback.items():
|
|
str = str.replace(key, value)
|
|
return str
|
|
|
|
|
|
def patch_ks_file(file_path, string):
|
|
"""
|
|
替换.ks文件中的字符串
|
|
:param file_path: 文件路径
|
|
:param string: 替换的字符串
|
|
"""
|
|
with open(file_path, 'r', encoding='cp932') as file:
|
|
content = file.read()
|
|
contents = content.split('\n')
|
|
i = 0
|
|
result = []
|
|
texts = []
|
|
insert_loc = 0
|
|
have_text = False
|
|
for line in contents:
|
|
if line.endswith('|'):
|
|
if len(texts):
|
|
if have_text:
|
|
ts = string[i]['message'].replace('\n', '[r]\n').split('\n')
|
|
ts[-1] += '[np]'
|
|
texts = texts[0:insert_loc] + ts + texts[insert_loc:]
|
|
i += 1
|
|
result.append('\n'.join(texts))
|
|
texts = []
|
|
have_text = False
|
|
texts.append(line)
|
|
continue
|
|
if line.startswith('@exlink') or line.startswith('@e_xlink'):
|
|
a = re.search(r'txt="([^"]+)"', line)
|
|
if a:
|
|
texts.append(line.replace(a.group(1), string[i]['message']))
|
|
i += 1
|
|
continue
|
|
if line.startswith('@nm'):
|
|
a = re.search(r't="([^"]+)"', line)
|
|
if a:
|
|
texts.append(line.replace(a.group(1), string[i]['name']))
|
|
continue
|
|
if line.endswith('[r]') or line.endswith('[np]'):
|
|
insert_loc = len(texts)
|
|
have_text = True
|
|
continue
|
|
texts.append(line)
|
|
if len(texts):
|
|
if have_text:
|
|
ts = string[i]['message'].replace('\n', '[r]\n').split('\n')
|
|
ts[-1] += '[np]'
|
|
texts = texts[0:insert_loc] + ts + texts[insert_loc:]
|
|
i += 1
|
|
if i < len(string):
|
|
print('Warning: not all strings are used')
|
|
result.append('\n'.join(texts))
|
|
output = join(dirname(file_path), '../patched', f"{splitext(file_path)[0]}.ks")
|
|
if not exists(dirname(output)):
|
|
makedirs(dirname(output))
|
|
with open(output, 'w', encoding='cp932') as file:
|
|
file.write('\n'.join(result))
|
|
print(f"Patched file saved to: {output}")
|
|
|
|
|
|
def get_ks_files(directory):
|
|
"""
|
|
获取指定目录下的所有.ks文件
|
|
:param directory: 目录路径
|
|
:return: ks文件列表
|
|
"""
|
|
return [f for f in listdir(directory) if splitext(f)[1].lower() == '.ks']
|
|
|
|
|
|
if __name__ == "__main__":
|
|
fallback = {}
|
|
if exists('jis_fallback.json'):
|
|
with open('jis_fallback.json', 'r', encoding='utf-8') as file:
|
|
fallback = json.load(file)
|
|
for f in get_ks_files('.'):
|
|
print(f"Processing file: {f}")
|
|
json_file = f"{splitext(f)[0]}.json"
|
|
with open(json_file, 'r', encoding='utf-8') as file:
|
|
content = json.load(file)
|
|
bcontent = []
|
|
for d in content:
|
|
da = {}
|
|
name = d.get('name')
|
|
if name:
|
|
name = replace_fallback(name, fallback)
|
|
try:
|
|
name.encode('cp932')
|
|
da['name'] = name
|
|
except UnicodeEncodeError as e:
|
|
print(d, name, name[e.start:e.end], 'is not cp932')
|
|
raise e
|
|
message = d['message']
|
|
message = replace_fallback(message, fallback)
|
|
try:
|
|
message.encode('cp932')
|
|
da['message'] = message
|
|
except UnicodeEncodeError as e:
|
|
print(d, message, message[e.start:e.end], 'is not cp932')
|
|
raise e
|
|
bcontent.append(da)
|
|
patch_ks_file(f, bcontent)
|
|
|