MibooGram/metadata/scripts/metadatas.py
Karako 4c702515a0
Enhance AssetsService
Co-authored-by: xtaodada <xtao@xtaolink.cn>
2022-10-07 13:02:49 +08:00

66 lines
2.4 KiB
Python

import ujson as json
from aiofiles import open as async_open
from httpx import AsyncClient, URL
from utils.const import AMBR_HOST, PROJECT_ROOT
__all__ = ['update_metadata_from_ambr', 'update_metadata_from_github']
client = AsyncClient()
async def update_metadata_from_ambr(overwrite: bool = True):
result = []
targets = ['material', 'weapon', 'avatar', 'reliquary']
for target in targets:
path = PROJECT_ROOT.joinpath(f'metadata/data/{target}.json')
if not overwrite and path.exists():
continue
url = AMBR_HOST.join(f"v2/chs/{target}")
path.parent.mkdir(parents=True, exist_ok=True)
response = await client.get(url)
json_data = json.loads(response.text)['data']['items']
async with async_open(path, mode='w', encoding='utf-8') as file:
data = json.dumps(json_data, ensure_ascii=False)
await file.write(data)
result.append(json_data)
return result
async def update_metadata_from_github(overwrite: bool = True):
path = PROJECT_ROOT.joinpath('metadata/data/namecard.json')
if not overwrite and path.exists():
return
host = URL("https://raw.fastgit.org/Dimbreath/GenshinData/master/")
text_map_url = host.join("TextMap/TextMapCHS.json")
material_url = host.join("ExcelBinOutput/MaterialExcelConfigData.json")
text_map_json_data = json.loads((await client.get(text_map_url)).text)
material_json_data = json.loads((await client.get(material_url)).text)
data = {}
for namecard_data in filter(lambda x: x.get('materialType', None) == 'MATERIAL_NAMECARD', material_json_data):
name = text_map_json_data[str(namecard_data['nameTextMapHash'])]
icon = namecard_data['icon']
navbar = namecard_data['picPath'][0]
banner = namecard_data['picPath'][1]
rank = namecard_data['rankLevel']
description = text_map_json_data[str(namecard_data['descTextMapHash'])].replace('\\n', '\n')
data.update({
str(namecard_data['id']): {
"id": namecard_data['id'],
"name": name,
"rank": rank,
"icon": icon,
"navbar": navbar,
"profile": banner,
"description": description,
}
})
async with async_open(path, mode='w', encoding='utf-8') as file:
data = json.dumps(data, ensure_ascii=False)
await file.write(data)
return data