Files
MaaAssistantArknights/tools/TaskSorter/TaskSorter.py
zzyyyl 6e5731fed3 chore: 增加 TaskSorter 脚本对 tasks.json 进行排序
外服任务内字段顺序也依照国服
2023-10-23 16:22:25 +08:00

82 lines
3.0 KiB
Python

import json
import re
from pathlib import Path
def sort_tasks(res: dict[str, any]):
# 暂时只对 Roguelike 和 Reclamation 任务进行类字典序排序,其他任务保持原有相对顺序
classified_lists = {
"Roguelike...": [],
"Roguelike@...": [],
"Phantom@Roguelike...": [],
"Mizuki@Roguelike...": [],
"Sami@Roguelike...": [],
"...@Roguelike...": [],
"Reclamation...": [],
"Reclamation@...": [],
"...@Reclamation...": []
}
unclassified_list = []
classify_rules: list[tuple[str, list]] = [
(r"^Roguelike@", classified_lists["Roguelike@..."]),
(r"^Roguelike", classified_lists["Roguelike..."]),
(r"^Phantom@Roguelike", classified_lists["Phantom@Roguelike..."]),
(r"^Mizuki@Roguelike", classified_lists["Mizuki@Roguelike..."]),
(r"^Sami@Roguelike", classified_lists["Sami@Roguelike..."]),
(r"^(\w+)@Roguelike", classified_lists["...@Roguelike..."]),
(r"^Reclamation@", classified_lists["Reclamation@..."]),
(r"^Reclamation", classified_lists["Reclamation..."]),
(r"^(\w+)@Reclamation", classified_lists["...@Reclamation..."]),
(r"", unclassified_list)
]
for k in res.keys():
for rule, target_list in classify_rules:
if re.search(rule, k):
target_list.append(k)
break
return {
k: res[k]
for k in unclassified_list + sum([
sorted(target_list)
for target_list in classified_lists.values()], [])}
if __name__ == '__main__':
resource_dir = Path(__file__).parents[2] / "resource"
cn_task_file = resource_dir / "tasks.json"
with open(cn_task_file, "r", encoding="utf8") as f:
cn_tasks = json.load(f)
cn_tasks = sort_tasks(cn_tasks)
order = list(cn_tasks.keys())
with open(cn_task_file, "w", encoding="utf8") as f:
json.dump(cn_tasks, f, ensure_ascii=False, indent=4)
print("CN:", str(len(order)).rjust(4, " "), "tasks")
global_resources = {
"EN": resource_dir / "global/YoStarEN/resource/tasks.json",
"JP": resource_dir / "global/YoStarJP/resource/tasks.json",
"KR": resource_dir / "global/YoStarKR/resource/tasks.json",
"TW": resource_dir / "global/txwy/resource/tasks.json"
}
for server, path in global_resources.items():
with open(path, "r", encoding="utf8") as f:
tasks = json.load(f)
tasks = {
k: {
x: tasks[k][x]
for x in sorted(tasks[k].keys(),
key=lambda x: list(cn_tasks[k].keys()).index(x) if k in cn_tasks and x in cn_tasks[k] else -1)
}
for k in sorted(tasks.keys(), key=lambda k: order.index(k) if k in order else -1)
}
with open(path, "w", encoding="utf8") as f:
json.dump(tasks, f, ensure_ascii=False, indent=4)
print(server + ":", str(len(tasks)).rjust(4, " "), "tasks")