66 lines
2.3 KiB
Python
66 lines
2.3 KiB
Python
import os
|
|
import re
|
|
import pandas as pd
|
|
|
|
def convert_description_to_multiline(description):
|
|
lines = description.strip().split('\n')
|
|
return ' .description = COMPOUND_STRING(\n' + ''.join(
|
|
f' "{line}\\n"\n' for line in lines
|
|
)[:-1] + ' ),' # 去掉最后一行的 \n 再添加 ),
|
|
|
|
def query_pokemon_info(name, df):
|
|
try:
|
|
result = df.loc[name]
|
|
return {
|
|
'speciesName': result['speciesName'],
|
|
'categoryName': result['categoryName'],
|
|
'description': result['description']
|
|
}
|
|
except KeyError:
|
|
return None
|
|
|
|
def replace_species_info(content, df):
|
|
pattern = r"\[SPECIES_(\w+)\]\s*=\s*\{.*?\n\s*\},?"
|
|
matches = list(re.finditer(pattern, content, re.DOTALL))
|
|
|
|
for match in matches:
|
|
species = match.group(1)
|
|
original_block = match.group(0)
|
|
info = query_pokemon_info(species, df)
|
|
if not info:
|
|
print(f"❌ 未找到 {species}，跳过")
|
|
continue
|
|
|
|
block = original_block
|
|
|
|
# 替换 .speciesName
|
|
block = re.sub(r'\.speciesName\s*=\s*_\(.*?\),', f'.speciesName = _("{info["speciesName"]}"),', block)
|
|
|
|
# 替换 .categoryName
|
|
block = re.sub(r'\.categoryName\s*=\s*_\(.*?\),', f'.categoryName = _("{info["categoryName"]}"),', block)
|
|
|
|
# 替换 .description
|
|
block = re.sub(r'\.description\s*=\s*COMPOUND_STRING\((?:.|\n)*?\),', convert_description_to_multiline(info["description"]), block)
|
|
|
|
content = content.replace(original_block, block)
|
|
|
|
return content
|
|
|
|
if __name__ == "__main__":
|
|
|
|
work_folder = r"D:\code\pokeemerald-expansion\src\data\pokemon\species_info"
|
|
df = pd.read_excel(r'D:\code\pokeemerald-expansion\python_tools\src\图鉴对应翻译表手动调整.xlsx')
|
|
df.set_index('name', inplace=True)
|
|
|
|
for filename in os.listdir(work_folder):
|
|
if re.match(r'^gen_\d+_families\.h$', filename):
|
|
file_path = os.path.join(work_folder, filename)
|
|
print(f"✅ 处理文件：{file_path}")
|
|
with open(file_path, "r", encoding="utf-8") as f:
|
|
content = f.read()
|
|
|
|
new_content = replace_species_info(content, df)
|
|
|
|
with open(file_path, "w", encoding="utf-8") as f:
|
|
f.write(new_content)
|