80 lines
2.6 KiB
Python
80 lines
2.6 KiB
Python
import re
|
|
from os.path import splitext
|
|
from os import listdir
|
|
import json
|
|
|
|
|
|
def extract_dialogue_from_ks(file_path, encoding='cp932'):
|
|
# 存储结果
|
|
dialogues = []
|
|
|
|
try:
|
|
with open(file_path, 'r', encoding=encoding) as file:
|
|
content = file.read()
|
|
contents = content.split('\n')
|
|
block_text = ''
|
|
block_name = ''
|
|
for line in contents:
|
|
if line.endswith('|'):
|
|
if block_text:
|
|
if block_name:
|
|
dialogues.append({
|
|
'name': block_name,
|
|
'message': block_text
|
|
})
|
|
else:
|
|
dialogues.append({
|
|
'message': block_text
|
|
})
|
|
block_text = ''
|
|
block_name = ''
|
|
continue
|
|
if line.startswith('@exlink') or line.startswith('@e_xlink'):
|
|
a = re.search(r'txt="([^"]+)"', line)
|
|
if a:
|
|
dialogues.append({
|
|
'message': a.group(1),
|
|
})
|
|
continue
|
|
if line.startswith('@nm'):
|
|
a = re.search(r't="([^"]+)"', line)
|
|
if a:
|
|
block_name = a.group(1)
|
|
continue
|
|
if not (line.endswith('[r]') or line.endswith('[np]')):
|
|
continue
|
|
block_text += line.replace('\n', '').replace('[r]', '\n').replace('[np]', '')
|
|
if block_text:
|
|
if block_name:
|
|
dialogues.append({
|
|
'name': block_name,
|
|
'message': block_text
|
|
})
|
|
else:
|
|
dialogues.append({
|
|
'message': block_text
|
|
})
|
|
return dialogues
|
|
|
|
except Exception as e:
|
|
print(f"Error processing file: {e}")
|
|
return []
|
|
|
|
|
|
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__":
|
|
for f in get_ks_files('.'):
|
|
print(f"Processing file: {f}")
|
|
result = extract_dialogue_from_ks(f)
|
|
print(f"Extracted {len(result)} dialogues from {f}")
|
|
with open(f"{splitext(f)[0]}.json", 'w', encoding='utf-8') as output_file:
|
|
json.dump(result, output_file, ensure_ascii=False, indent=4)
|