GenshinUID/enkaToData/dataToMap.py
2022-06-01 22:23:22 +08:00

115 lines
4.1 KiB
Python

import asyncio
from pathlib import Path
import httpx
import json
R_PATH = Path(__file__).parents[0]
MAP_PATH = R_PATH / 'map'
DATA_PATH = R_PATH / 'data'
verison = '2.7.0'
avatarName2Element_fileName = f'avatarName2Element_mapping_{verison}.json'
weaponHash2Name_fileName = f'weaponHash2Name_mapping_{verison}.json'
weaponHash2Type_fileName = f'weaponHash2Type_mapping_{verison}.json'
skillId2Name_fileName = f'skillId2Name_mapping_{verison}.json'
talentId2Name_fileName = f'talentId2Name_mapping_{verison}.json'
avatarId2Name_fileName = f'avatarId2Name_mapping_{verison}.json'
with open(DATA_PATH / 'textMap.json', "r", encoding='UTF-8') as f:
raw_data = json.load(f)
async def avatarId2NameJson() -> None:
with open(DATA_PATH / 'AvatarExcelConfigData.json', "r", encoding='UTF-8') as f:
avatar_data = json.load(f)
temp = {}
for i in avatar_data:
temp[str(i['id'])] = raw_data[str(i['nameTextMapHash'])]
with open(MAP_PATH / avatarId2Name_fileName,'w', encoding='UTF-8') as file:
json.dump(temp, file, ensure_ascii=False)
async def avatarName2ElementJson() -> None:
with open(MAP_PATH / avatarId2Name_fileName, "r", encoding='UTF-8') as f:
avatarId2Name = json.load(f)
temp = {}
elementMap = {'':'Anemo','':'Geo','':'Dendro','':'Pyro','':'Hydro','':'Cryo','':'Electro'}
for i in list(avatarId2Name.values()):
data = json.loads(httpx.get('https://info.minigg.cn/characters?query={}'.format(i)).text)
if 'errcode' in data:
pass
else:
temp[i] = elementMap[data['element']]
with open(MAP_PATH / avatarName2Element_fileName,'w', encoding='UTF-8') as file:
json.dump(temp, file, ensure_ascii=False)
async def weaponHash2NameJson() -> None:
with open(DATA_PATH / 'WeaponExcelConfigData.json', "r", encoding='UTF-8') as f:
weapon_data = json.load(f)
temp = {}
for i in weapon_data:
temp[str(i['nameTextMapHash'])] = raw_data[str(i['nameTextMapHash'])]
with open(MAP_PATH / weaponHash2Name_fileName,'w', encoding='UTF-8') as file:
json.dump(temp, file, ensure_ascii=False)
async def weaponHash2NameJson() -> None:
with open(DATA_PATH / 'WeaponExcelConfigData.json', "r", encoding='UTF-8') as f:
weapon_data = json.load(f)
temp = {}
for i in weapon_data:
if i['weaponType'] == 'WEAPON_POLE':
weaponType = "长柄武器"
elif i['weaponType'] == 'WEAPON_BOW':
weaponType = ""
elif i['weaponType'] == 'WEAPON_SWORD_ONE_HAND':
weaponType = "单手剑"
elif i['weaponType'] == 'WEAPON_CLAYMORE':
weaponType = "双手剑"
elif i['weaponType'] == 'WEAPON_CATALYST':
weaponType = "法器"
else:
weaponType = ""
temp[str(i['nameTextMapHash'])] = weaponType
with open(MAP_PATH / weaponHash2Type_fileName,'w', encoding='UTF-8') as file:
json.dump(temp, file, ensure_ascii=False)
async def skillId2NameJson() -> None:
with open(DATA_PATH / 'AvatarSkillExcelConfigData.json', "r", encoding='UTF-8') as f:
skill_data = json.load(f)
temp = {'Name':{},'Icon':{}}
for i in skill_data:
temp['Name'][str(i['id'])] = raw_data[str(i['nameTextMapHash'])]
temp['Icon'][str(i['id'])] = i['skillIcon']
with open(MAP_PATH / skillId2Name_fileName,'w', encoding='UTF-8') as file:
json.dump(temp, file, ensure_ascii=False)
async def talentId2NameJson() -> None:
with open(DATA_PATH / 'AvatarTalentExcelConfigData.json', "r", encoding='UTF-8') as f:
talent_data = json.load(f)
temp = {'Name':{},'Icon':{}}
for i in talent_data:
temp['Name'][str(i['talentId'])] = raw_data[str(i['nameTextMapHash'])]
temp['Icon'][str(i['talentId'])] = i['icon']
with open(MAP_PATH / talentId2Name_fileName,'w', encoding='UTF-8') as file:
json.dump(temp, file, ensure_ascii=False)
async def main():
await avatarId2NameJson()
await avatarName2ElementJson()
await weaponHash2NameJson()
await skillId2NameJson()
await talentId2NameJson()
await weaponHash2NameJson()
asyncio.run(main())