MaaAssistantArknights/tools/OverseasClients/SortJsonByZHServerOrder.py

# WHAT DOES THIS TOOL DO?
#
# This tool sorts the global json files so that they are in the same order as
# the ZH ones. This way, it is easier to find the attributes, see what is
# missing, etc.
#
# USAGE
#
# 1. Install Python
# 2. Run "$python match_json.py xxx" in the terminal where xxx is the desired
# server name i.e. YoStarJP/YoStarEN/YoStarKR/txwy
# 3. The json file will be updated in-place
#

import json
import os
import sys
from collections import OrderedDict

server_list = {
    "YoStarJP": ["yostarjp", "jp", "日", "日服"],
    "YoStarEN": ["yostaren", "en", "美", "美服", "國際", "國際服"],
    "YoStarKR": ["yostarkr", "kr", "韓", "韓服"],
    "txwy": ["txwy", "繁中", "繁中服"],
}

# Get the server name from argument
try:
    # server_name = YoStarJP
    server_name = None
    for listed_server in server_list:
        server_name = (
            listed_server
            if (not server_name)
            and sys.argv[1].casefold() in server_list[listed_server]
            else server_name
        )
    if server_name not in server_list:
        raise
except Exception:
    server_name = None
    while not server_name:
        print("Enter one and only one server name:", ", ".join(server_list))
        t = input()
        for listed_server in server_list:
            server_name = (
                listed_server
                if (not server_name) and t.casefold() in server_list[listed_server]
                else server_name
            )

cur_dir = os.path.dirname(os.path.abspath(__file__))
proj_dir = os.path.join(cur_dir, "../../")

# File name of the json
# NOTE: You may change this to e.g. recruitment.json
# This is hardcoded because tasks.json is the most likely modified one
try:
    json_name = sys.argv[2]
except IndexError:
    json_name = "tasks.json"

zh_json_file = os.path.join(proj_dir, "resource", json_name)
gl_json_file = os.path.join(
    proj_dir, "resource", "global", server_name, "resource", json_name
)

# For test purpose
# gl_json_file = os.path.join(cur_dir, "test.json")

with open(zh_json_file, "r", encoding="utf-8") as zh_fh:
    zh_json = json.load(zh_fh)

with open(gl_json_file, "r", encoding="utf-8") as gl_fh:
    gl_json = json.load(gl_fh)

# Sort the global json object by the order of the ZH one
# Keys not found in ZH json will be put at the end (should not happen though)
LAST_POSITIONS = 1e10
key_order = {k: v for v, k in enumerate(zh_json)}
gl_json_new = OrderedDict(
    sorted(gl_json.items(), key=lambda i: key_order.get(i[0], LAST_POSITIONS))
)

with open(gl_json_file, "wb") as gl_fh:
    t = json.dumps(gl_json_new, indent=4, ensure_ascii=False)
    gl_fh.write(bytes(t, "utf-8"))
    print(
        "successfully sorted",
        os.path.join("resource", "global", server_name, "resource", json_name),
        "by the order in ZH json file",
    )