HonkaiStarRailWikiDataParser/res_func/relic.py

from pathlib import Path
from typing import List, Dict

import aiofiles
import ujson

from models.enums import RelicAffix, RelicPosition
from models.relic_affix import RelicAffixAll, SingleRelicAffix
from res_func.client import client
from res_func.url import relic_config, relic_main_affix_config, relic_sub_affix_config

final_datas: List[RelicAffixAll] = []
final_datas_map: Dict[str, RelicAffixAll] = {}


async def fetch_all_relic():
    print("开始获取遗器配置")
    relic_data_req = await client.get(relic_config)
    relic_data = relic_data_req.json()
    for key, value in relic_data.items():
        relic_affix_all = RelicAffixAll(
            id=int(key),
            set_id=value["SetID"],
            type=RelicPosition(value["Type"]),
            rarity=int(value["Rarity"].replace("CombatPowerRelicRarity", "")),
            main_affix_group=value["MainAffixGroup"],
            sub_affix_group=value["SubAffixGroup"],
            max_level=value["MaxLevel"],
            main_affix={},
            sub_affix={},
        )
        final_datas.append(relic_affix_all)
        final_datas_map[key] = relic_affix_all
    print("遗器配置获取完毕")


async def fetch_main_affix():
    print("开始获取遗器主词条配置")
    main_affix_req = await client.get(relic_main_affix_config)
    main_affix_data = main_affix_req.json()
    main_affix_groups_map: Dict[str, Dict[str, SingleRelicAffix]] = {}
    for key, value in main_affix_data.items():
        data: Dict[str, SingleRelicAffix] = {}
        for key2, value2 in value.items():
            data[key2] = SingleRelicAffix(
                id=value2["AffixID"],
                property=RelicAffix(value2["Property"]),
                base_value=value2["BaseValue"]["Value"],
                level_value=value2["LevelAdd"]["Value"],
                is_main=True,
            )
        main_affix_groups_map[key] = data
    for final_data in final_datas:
        final_data.main_affix = main_affix_groups_map[str(final_data.main_affix_group)]
    print("遗器主词条配置获取完毕")


async def fetch_sub_affix():
    print("开始获取遗器副词条配置")
    sub_affix_req = await client.get(relic_sub_affix_config)
    sub_affix_data = sub_affix_req.json()
    sub_affix_groups_map: Dict[str, Dict[str, SingleRelicAffix]] = {}
    for key, value in sub_affix_data.items():
        data: Dict[str, SingleRelicAffix] = {}
        for key2, value2 in value.items():
            data[key2] = SingleRelicAffix(
                id=value2["AffixID"],
                property=RelicAffix(value2["Property"]),
                base_value=value2["BaseValue"]["Value"],
                step_value=value2["StepValue"]["Value"],
                is_main=False,
                max_step=value2["StepNum"],
            )
        sub_affix_groups_map[key] = data
    for final_data in final_datas:
        final_data.sub_affix = sub_affix_groups_map[str(final_data.sub_affix_group)]
    print("遗器副词条配置获取完毕")


async def dump_relic_config(path: Path):
    final_data = [data.dict() for data in final_datas]
    final_data.sort(key=lambda x: x["id"])
    async with aiofiles.open(path, "w", encoding="utf-8") as f:
        await f.write(ujson.dumps(final_data, indent=4, ensure_ascii=False))


async def fetch_relic_config():
    await fetch_all_relic()
    await fetch_main_affix()
    await fetch_sub_affix()
    data_path = Path("data")
    await dump_relic_config(data_path / "relic_config.json")