mirror of
https://github.com/PaiGramTeam/HonkaiStarRailWikiDataParser.git
synced 2024-11-16 14:31:24 +00:00
109 lines
4.1 KiB
Python
109 lines
4.1 KiB
Python
from pathlib import Path
|
|
from typing import List, Dict
|
|
|
|
import aiofiles
|
|
import ujson
|
|
|
|
from models.enums import RelicAffix
|
|
from func.fetch_relics import read_relics, dump_relics, all_relics
|
|
from res_func.client import client
|
|
from res_func.url import relic_config, relic_main_affix_config, relic_sub_affix_config, relic_set_config
|
|
from models.relic_affix import RelicAffixAll, SingleRelicAffix
|
|
|
|
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),
|
|
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 fix_set_id(text_map_data: Dict[str, str]):
|
|
print("开始修复遗器套装ID")
|
|
set_req = await client.get(relic_set_config)
|
|
set_data = set_req.json()
|
|
set_data_map: Dict[str, int] = {}
|
|
for key, value in set_data.items():
|
|
set_data_map[text_map_data[str(value["SetName"]["Hash"])]] = value["SetID"]
|
|
await read_relics(Path("data") / "relics.json")
|
|
for relic_set in all_relics:
|
|
if set_id := set_data_map.get(relic_set.name):
|
|
relic_set.id = set_id
|
|
else:
|
|
print(f"套装{relic_set.name}没有找到对应的ID")
|
|
await dump_relics(Path("data") / "relics.json")
|
|
print("遗器套装ID修复完毕")
|
|
|
|
|
|
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(text_map_data: Dict[str, str]):
|
|
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")
|
|
await fix_set_id(text_map_data)
|