-
Notifications
You must be signed in to change notification settings - Fork 0
/
tool_umipro2txt.py
124 lines (105 loc) · 5.28 KB
/
tool_umipro2txt.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
import os
import re
import shutil
# 定义源目录和目标目录
source_dir = 'story'
target_dir = 'story_cn'
# 括号类
BRACKET_replaces = {
r"\{n\}": r"", # 删除
r"\{ruby:(.*?):(.*?)\}": r"【{ruby}】【{kanji}】", # ruby注音
r"p:4": r"p:四",
r"([0-9a-zA-Zè,.\?!%&'×÷=+-]{2,})": r"@<{text}@>", # 英文字母
r"(?<=【)[^】]*?(@[<>])+[^】]*?(?=】)": r"@[<>]", # ruby注音内还原
r"\{p:1:(.*?)\}": r"@c900.@[{text}@]@c.", # 红字
r"\{p:2:(.*?)\}": r"@c279.@[{text}@]@c.", # 蓝字
r"\{p:四1:(.*?)\}": r"@c960.@[{text}@]@c.", # 金字
r"\{p:四2:(.*?)\}": r"@c649.{text}@c.", # 紫字
r"\{i:(.*?)\}": r"@{{{text}@}}", # 粗体
r"([“]+)@c900.@\[": r"@c900.@[{text}", # 颜色字外右标点
r"@\]@c.([！？、，。ッ…]+)": r"{text}@]@c.", # 颜色字外右标点
r"@\]@c.([”]+)": r"{text}@]@c.", # 颜色字外右标点
r"\{t\}": r"<n>", # 换行
r"<n> ": r"<n>",
r"\{[abcefgmoy]:.*?:(.*?)\}": r"{text}", # 其他特殊
r"【(.*?)】【(.*?)】": r"@b{ruby}.@<{kanji}@>", # ruby注音
}
# BRACKET_replaces = {
# r"@b(.*?).@<(.*?)@>": r"【{ruby}】【{kanji}】", # ruby注音
# r"@c900.": r"@#红左#", # 红字
# r"@c279.": r"@#蓝左#", # 蓝字
# r"@c960.": r"@#金左#", # 金字
# r"@c649.": r"@#紫左#", # 紫字
# r"@c.": r"@#色右#", # 颜色字
# r"([0-9a-zA-Zè,.\?!%&'×÷=+-]{2,})": r"@<{text}@>", # 英文字母
# r"(?<=【)[^】]*?(@[<>])+[^】]*?(?=】)": r"@[<>]", # ruby注音内还原
# r"【(.*?)】【(.*?)】": r"@b{ruby}.@<{kanji}@>", # ruby注音
# r"@#红左#": r"@c900.", # 红字
# r"@#蓝左#": r"@c279.", # 蓝字
# r"@#金左#": r"@c960.", # 金字
# r"@#紫左#": r"@c649.", # 紫字
# r"@#色右#": r"@c.", # 颜色字
# }
OTHER_replaces = {
r"\{[abcefgmoy]:.*?:(.*?)\}": r"{text}" # 其他特殊
}
# 如果目标目录不存在，则创建它
if not os.path.exists(target_dir):
os.makedirs(target_dir)
# 遍历每个 ep 文件夹
for ep in range(1, 9):
cn_folder = os.path.join(source_dir, f'ep{ep}', 'cn')
# cn_folder = os.path.join(source_dir)
# 确保 cn 文件夹存在
if os.path.exists(cn_folder):
# 遍历 cn 文件夹中的所有 .txt 文件
for file_name in os.listdir(cn_folder):
if file_name.endswith('.txt'):
# 构建目标文件路径
target_file = os.path.join(target_dir, file_name)
# 如果目标文件已存在，则先删除它
if os.path.exists(target_file):
os.remove(target_file)
# 复制文件到目标目录
source_file = os.path.join(cn_folder, file_name)
shutil.copy(source_file, target_file)
# 如果文件名以 _op.txt 结尾，则重命名为 _0.txt
if target_file.endswith('op.txt'):
new_name = target_file.replace('op.txt', '0.txt')
# 如果目标文件名已经存在，则先删除它
if os.path.exists(new_name):
os.remove(new_name)
# 重命名文件
os.rename(target_file, new_name)
target_file = new_name
# 删除每行左右两侧的反引号
with open(target_file, 'r', encoding='utf-8') as f:
lines = [line.rstrip('\n') for line in f.readlines()]
lines = [line.strip('`') for line in lines]
# 用唯一占位符将 lines 列表转换为一个单独的字符串
placeholder = "<n>"
combined_text = placeholder.join(lines)
# 替换
for pattern, replace in BRACKET_replaces.items():
if r"kanji" in replace:
combined_text = re.sub(pattern, lambda m: replace.format(ruby=m.group(1), kanji=m.group(2)), combined_text)
elif r"text" in replace:
combined_text = re.sub(pattern, lambda m: replace.format(text=m.group(1)), combined_text)
elif r"<>" in replace:
combined_text = re.sub(pattern, lambda m: re.sub(replace, '', m.group()), combined_text)
else:
combined_text = re.sub(pattern, replace, combined_text)
previous_combined_text = None
while combined_text != previous_combined_text:
# 保存上一次的 combined_text
previous_combined_text = combined_text
# 处理 OTHER_replaces
for pattern, replace in OTHER_replaces.items():
combined_text = re.sub(pattern, lambda m: replace.format(text=m.group(1)) if len(m.groups()) > 0 else "", combined_text)
# 将处理后的字符串使用占位符分割回原来的行列表
processed_lines = combined_text.split(placeholder)
with open(target_file, 'w', encoding='utf-8') as f:
for line in processed_lines:
if line: # 仅在行不为空时写入
f.write(line + '\n')
print("文件处理完成。")