PaiGramTeam/PaiGram
3
0
Fork 0
mirror of https://github.com/PaiGramTeam/PaiGram.git synced 2024-11-22 07:07:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

🐛 Fix metadata from ambr

Browse Source
This commit is contained in:
xtaodada 2023-06-06 17:23:15 +08:00
parent ceaf0a1b09
commit 36a61e9986
Signed by: xtaodada
GPG Key ID: 4CBB3F4FA8C85659
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
metadata/scripts/metadatas.py
View File

@ -1,6 +1,6 @@
from base64 import b64decode as parse_token from base64 import b64decode as parse_token
from contextlib import contextmanager from contextlib import contextmanager
from typing import Iterator from typing import Iterator, Dict
import ujson as json import ujson as json
from aiofiles import open as async_open from aiofiles import open as async_open
@ -19,6 +19,17 @@ RESOURCE_DEFAULT_PATH = f"{RESOURCE_REPO}/{RESOURCE_BRANCH}/{RESOURCE_ROOT}/"
client = AsyncClient() client = AsyncClient()
async def fix_metadata_from_ambr(json_data: Dict[str, Dict], data_type: str):
if data_type == "weapon":
need_append_ids = [11304]
need_attr = ["id", "rank", "type", "name", "icon", "route"]
for wid in need_append_ids:
url = AMBR_HOST.join(f"v2/chs/{data_type}/{wid}")
response = await client.get(url)
json_data_ = json.loads(response.text)["data"]
json_data[str(json_data_["id"])] = {k: json_data_[k] for k in need_attr}
async def update_metadata_from_ambr(overwrite: bool = True): async def update_metadata_from_ambr(overwrite: bool = True):
result = [] result = []
targets = ["material", "weapon", "avatar", "reliquary"] targets = ["material", "weapon", "avatar", "reliquary"]
@ -30,6 +41,7 @@ async def update_metadata_from_ambr(overwrite: bool = True):
path.parent.mkdir(parents=True, exist_ok=True) path.parent.mkdir(parents=True, exist_ok=True)
response = await client.get(url) response = await client.get(url)
json_data = json.loads(response.text)["data"]["items"] json_data = json.loads(response.text)["data"]["items"]
await fix_metadata_from_ambr(json_data, target)
async with async_open(path, mode="w", encoding="utf-8") as file: async with async_open(path, mode="w", encoding="utf-8") as file:
data = json.dumps(json_data, ensure_ascii=False) data = json.dumps(json_data, ensure_ascii=False)
await file.write(data) await file.write(data)