mirror of
https://github.com/KimigaiiWuyi/GenshinUID.git
synced 2025-05-06 11:43:45 +08:00
552 lines
18 KiB
Python
552 lines
18 KiB
Python
import json
|
||
import time
|
||
import asyncio
|
||
import threading
|
||
from copy import deepcopy
|
||
from typing import Dict, List, Union, Literal, Optional
|
||
|
||
import aiofiles
|
||
from httpx import ReadTimeout
|
||
from gsuid_core.utils.api.enka.models import EnkaData
|
||
from gsuid_core.utils.api.enka.request import get_enka_info
|
||
|
||
from ..utils.message import UID_HINT
|
||
from .mono.Character import Character
|
||
from ..utils.api.cv.request import _CvApi
|
||
from .draw_normal import get_artifact_score_data
|
||
from ..genshinuid_config.gs_config import gsconfig
|
||
from ..utils.resource.RESOURCE_PATH import PLAYER_PATH
|
||
from ..utils.ambr_to_minigg import convert_ambr_to_weapon
|
||
from ..utils.map.GS_MAP_PATH import (
|
||
icon2Name,
|
||
propId2Name,
|
||
skillId2Name,
|
||
artifact2attr,
|
||
avatarId2Name,
|
||
talentId2Name,
|
||
weaponHash2Name,
|
||
weaponHash2Type,
|
||
artifactId2Piece,
|
||
avatarName2Element,
|
||
)
|
||
|
||
# from gsuid_core.utils.api.minigg.request import get_weapon_info
|
||
|
||
|
||
PROP_ATTR_MAP = {
|
||
'Anemo': '44',
|
||
'Cryo': '46',
|
||
'Dendro': '43',
|
||
'Electro': '41',
|
||
'Geo': '45',
|
||
'Hydro': '42',
|
||
'Pyro': '40',
|
||
}
|
||
AT_MAP = {
|
||
'maxHp': 'maxHP',
|
||
'atk': 'maxATK',
|
||
'def': 'maxDEF',
|
||
'elementalMastery': 'elementalMastery',
|
||
'energyRecharge': 'energyRecharge',
|
||
'healingBonus': 'healingBonus',
|
||
'critRate': 'critRate',
|
||
'critDamage': 'critDMG',
|
||
}
|
||
|
||
ENKA_API: List[Literal['enka', 'microgg']] = ['enka', 'microgg']
|
||
is_enable_akasha = gsconfig.get_config('EnableAkasha').data
|
||
ARTIFACT_DATA = {
|
||
'data': {
|
||
'flower': [],
|
||
'plume': [],
|
||
'sands': [],
|
||
'goblet': [],
|
||
'circlet': [],
|
||
},
|
||
'tag': {
|
||
'flower': [],
|
||
'plume': [],
|
||
'sands': [],
|
||
'goblet': [],
|
||
'circlet': [],
|
||
},
|
||
}
|
||
|
||
|
||
async def switch_api():
|
||
global ENKA_API
|
||
if ENKA_API[0] == 'enka':
|
||
ENKA_API = ['microgg', 'enka']
|
||
else:
|
||
ENKA_API = ['enka', 'microgg']
|
||
return f'切换成功!当前api为{ENKA_API[0]}!'
|
||
|
||
|
||
async def enka_to_dict(
|
||
uid: str, enka_data: Optional[EnkaData] = None
|
||
) -> Union[List[dict], str]:
|
||
'''
|
||
:说明:
|
||
访问enkaAPI并转换为genshinUID的数据Json。
|
||
:参数:
|
||
* ``uid: str``: 玩家uid。
|
||
* ``enka_data: Optional[dict] = None``: 来自enka的dict, 可留空。
|
||
:返回:
|
||
* ``刷新完成提示语: str``: 包含刷新成功的角色列表。
|
||
'''
|
||
if '未找到绑定的UID' in uid:
|
||
return UID_HINT
|
||
if enka_data:
|
||
pass
|
||
else:
|
||
try:
|
||
enka_data = await get_enka_info(uid, ENKA_API[0])
|
||
except ReadTimeout:
|
||
return '网络不太稳定...'
|
||
if isinstance(enka_data, str):
|
||
return []
|
||
if isinstance(enka_data, dict):
|
||
if 'playerInfo' not in enka_data:
|
||
im = (
|
||
'服务器正在维护或者关闭中...\n'
|
||
f'检查{ENKA_API[0]}是否可以访问\n'
|
||
'如可以访问,尝试[切换api]或上报Bug!'
|
||
)
|
||
return im
|
||
elif enka_data is None:
|
||
return []
|
||
|
||
now = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time()))
|
||
playerInfo = enka_data['playerInfo']
|
||
path = PLAYER_PATH / str(uid)
|
||
path.mkdir(parents=True, exist_ok=True)
|
||
|
||
# 保存基本玩家信息
|
||
async with aiofiles.open(
|
||
path / f'{uid}.json', 'w', encoding='UTF-8'
|
||
) as file:
|
||
await file.write(json.dumps(playerInfo, indent=4, ensure_ascii=False))
|
||
|
||
# 保存原始数据
|
||
async with aiofiles.open(
|
||
path / 'rawData.json', 'w', encoding='UTF-8'
|
||
) as file:
|
||
await file.write(json.dumps(enka_data, indent=4, ensure_ascii=False))
|
||
|
||
if (
|
||
enka_data is None
|
||
or enka_data == {}
|
||
or 'avatarInfoList' not in enka_data
|
||
or not enka_data['avatarInfoList']
|
||
):
|
||
return f'UID{uid}刷新失败！未打开角色展柜或角色展柜未展示角色!'
|
||
|
||
char_dict_list = []
|
||
|
||
# 确认是否存在圣遗物列表
|
||
all_artifacts_path = path / 'artifacts.json'
|
||
if not all_artifacts_path.exists():
|
||
all_artifacts_data = deepcopy(ARTIFACT_DATA)
|
||
else:
|
||
async with aiofiles.open(
|
||
all_artifacts_path, 'r', encoding='UTF-8'
|
||
) as file:
|
||
all_artifacts_data = json.loads(await file.read())
|
||
|
||
for char in enka_data['avatarInfoList']:
|
||
# 处理基本信息
|
||
char_data = {}
|
||
avatarId = char['avatarId']
|
||
char_data['playerUid'] = str(uid)
|
||
char_data['playerName'] = enka_data['playerInfo']['nickname']
|
||
char_data['avatarId'] = avatarId
|
||
avatarName = avatarId2Name[str(char['avatarId'])]
|
||
char_data['avatarName'] = avatarId2Name[str(char['avatarId'])]
|
||
char_data['avatarFetter'] = char['fetterInfo']['expLevel']
|
||
char_data['avatarLevel'] = char['propMap']['4001']['val']
|
||
|
||
try:
|
||
char_data['avatarElement'] = avatarName2Element[
|
||
char_data['avatarName']
|
||
]
|
||
except KeyError:
|
||
check = skillId2Name['Name'][
|
||
str(list(char['skillLevelMap'].keys())[2])
|
||
]
|
||
if '风' in check:
|
||
char_data['avatarElement'] = 'Anemo'
|
||
elif '雷' in check:
|
||
char_data['avatarElement'] = 'Electro'
|
||
elif '岩' in check:
|
||
char_data['avatarElement'] = 'Geo'
|
||
elif '草' in check:
|
||
char_data['avatarElement'] = 'Dendro'
|
||
elif '冰' in check:
|
||
char_data['avatarElement'] = 'Cryo'
|
||
elif '水' in check:
|
||
char_data['avatarElement'] = 'Hydro'
|
||
else:
|
||
char_data['avatarElement'] = 'Pyro'
|
||
|
||
char_data['dataTime'] = now
|
||
|
||
char_data['avatarSkill'] = []
|
||
# 处理天赋
|
||
for skill in char['skillLevelMap']:
|
||
skill_temp = {}
|
||
skill_temp['skillId'] = skill
|
||
skill_temp['skillName'] = skillId2Name['Name'][
|
||
skill_temp['skillId']
|
||
]
|
||
skill_temp['skillLevel'] = char['skillLevelMap'][skill]
|
||
skill_temp['skillIcon'] = skillId2Name['Icon'][
|
||
skill_temp['skillId']
|
||
]
|
||
char_data['avatarSkill'].append(skill_temp)
|
||
|
||
if char_data['avatarName'] in ['神里绫华', '安柏']:
|
||
char_data['avatarSkill'][0], char_data['avatarSkill'][-1] = (
|
||
char_data['avatarSkill'][-1],
|
||
char_data['avatarSkill'][0],
|
||
)
|
||
char_data['avatarSkill'][2], char_data['avatarSkill'][-1] = (
|
||
char_data['avatarSkill'][-1],
|
||
char_data['avatarSkill'][2],
|
||
)
|
||
char_data['avatarEnName'] = char_data['avatarSkill'][1][
|
||
'skillIcon'
|
||
].split('_')[-2]
|
||
elif char_data['avatarName'] in ['旅行者']:
|
||
char_data['avatarSkill'][0], char_data['avatarSkill'][-1] = (
|
||
char_data['avatarSkill'][-1],
|
||
char_data['avatarSkill'][0],
|
||
)
|
||
char_data['avatarSkill'][1], char_data['avatarSkill'][-1] = (
|
||
char_data['avatarSkill'][-1],
|
||
char_data['avatarSkill'][1],
|
||
)
|
||
char_data['avatarEnName'] = str(avatarId)
|
||
else:
|
||
char_data['avatarEnName'] = char_data['avatarSkill'][-1][
|
||
'skillIcon'
|
||
].split('_')[-2]
|
||
|
||
# 处理命座
|
||
talent_temp = []
|
||
if 'talentIdList' in char:
|
||
for index, talent in enumerate(char['talentIdList']):
|
||
talentTemp = {}
|
||
talentTemp['talentId'] = char['talentIdList'][index]
|
||
talentTemp['talentName'] = talentId2Name['Name'][str(talent)]
|
||
talentTemp['talentIcon'] = talentId2Name['Icon'][str(talent)]
|
||
talent_temp.append(talentTemp)
|
||
char_data['talentList'] = talent_temp
|
||
|
||
# 处理属性
|
||
fight_prop = {}
|
||
# 血量
|
||
fight_prop['hp'] = char['fightPropMap']['2000']
|
||
fight_prop['baseHp'] = char['fightPropMap']['1']
|
||
fight_prop['addHp'] = (
|
||
char['fightPropMap']['2000'] - char['fightPropMap']['1']
|
||
)
|
||
# 攻击力
|
||
fight_prop['atk'] = char['fightPropMap']['2001']
|
||
fight_prop['baseAtk'] = char['fightPropMap']['4']
|
||
fight_prop['addAtk'] = (
|
||
char['fightPropMap']['2001'] - char['fightPropMap']['4']
|
||
)
|
||
# 防御力
|
||
fight_prop['def'] = char['fightPropMap']['2002']
|
||
fight_prop['baseDef'] = char['fightPropMap']['7']
|
||
fight_prop['addDef'] = (
|
||
char['fightPropMap']['2002'] - char['fightPropMap']['7']
|
||
)
|
||
# 元素精通
|
||
fight_prop['elementalMastery'] = char['fightPropMap']['28']
|
||
# 暴击率
|
||
fight_prop['critRate'] = char['fightPropMap']['20']
|
||
# 暴击伤害
|
||
fight_prop['critDmg'] = char['fightPropMap']['22']
|
||
# 充能效率
|
||
fight_prop['energyRecharge'] = char['fightPropMap']['23']
|
||
# 治疗&受治疗
|
||
fight_prop['healBonus'] = char['fightPropMap']['26']
|
||
fight_prop['healedBonus'] = char['fightPropMap']['27']
|
||
# 物理伤害加成 & 抗性
|
||
fight_prop['physicalDmgSub'] = char['fightPropMap']['29']
|
||
fight_prop['physicalDmgBonus'] = char['fightPropMap']['30']
|
||
# 伤害加成
|
||
fight_prop['dmgBonus'] = char['fightPropMap'][
|
||
PROP_ATTR_MAP[char_data['avatarElement']]
|
||
]
|
||
|
||
char_data['avatarFightProp'] = fight_prop
|
||
|
||
# 处理武器
|
||
weapon_info = {}
|
||
weapon_data = char['equipList'][-1]
|
||
weapon_info['itemId'] = weapon_data['itemId']
|
||
weapon_info['nameTextMapHash'] = weapon_data['flat']['nameTextMapHash']
|
||
weapon_info['weaponIcon'] = weapon_data['flat']['icon']
|
||
weapon_info['weaponType'] = weaponHash2Type[
|
||
str(weapon_info['nameTextMapHash'])
|
||
]
|
||
weapon_info['weaponName'] = weaponHash2Name[
|
||
str(weapon_info['nameTextMapHash'])
|
||
]
|
||
weapon_info['weaponStar'] = weapon_data['flat']['rankLevel']
|
||
# 防止未精炼
|
||
if 'promoteLevel' in weapon_data['weapon']:
|
||
weapon_info['promoteLevel'] = weapon_data['weapon']['promoteLevel']
|
||
else:
|
||
weapon_info['promoteLevel'] = 0
|
||
weapon_info['weaponLevel'] = weapon_data['weapon']['level']
|
||
if 'affixMap' in weapon_data['weapon']:
|
||
weapon_info['weaponAffix'] = (
|
||
list(weapon_data['weapon']['affixMap'].values())[0] + 1
|
||
)
|
||
else:
|
||
weapon_info['weaponAffix'] = 1
|
||
weapon_info['weaponStats'] = []
|
||
|
||
for k in weapon_data['flat']['weaponStats']:
|
||
weapon_prop_temp = {}
|
||
weapon_prop_temp['appendPropId'] = k['appendPropId']
|
||
weapon_prop_temp['statName'] = propId2Name[k['appendPropId']]
|
||
weapon_prop_temp['statValue'] = k['statValue']
|
||
weapon_info['weaponStats'].append(weapon_prop_temp)
|
||
|
||
# 武器特效，须请求API
|
||
'''
|
||
try:
|
||
effect_raw = await get_weapon_info(weapon_info['weaponName'])
|
||
except: # noqa
|
||
'''
|
||
effect_raw = await convert_ambr_to_weapon(weapon_info['itemId'])
|
||
|
||
if (
|
||
not isinstance(effect_raw, List)
|
||
and not isinstance(effect_raw, int)
|
||
and effect_raw is not None
|
||
and int(effect_raw['rarity']) > 2
|
||
):
|
||
effect = effect_raw[f'r{weapon_info["weaponAffix"]}'][
|
||
'description'
|
||
]
|
||
else:
|
||
effect = '无特效。'
|
||
|
||
weapon_info['weaponEffect'] = effect
|
||
char_data['weaponInfo'] = weapon_info
|
||
|
||
# 处理圣遗物
|
||
artifacts_info = []
|
||
artifacts_data = char['equipList'][:-1]
|
||
artifact_set_list = []
|
||
|
||
for artifact in artifacts_data:
|
||
artifact_temp = {}
|
||
artifact_temp['itemId'] = artifact['itemId']
|
||
artifact_temp['nameTextMapHash'] = artifact['flat'][
|
||
'nameTextMapHash'
|
||
]
|
||
artifact_temp['icon'] = artifact['flat']['icon']
|
||
artifact_temp['aritifactName'] = icon2Name[
|
||
artifact['flat']['icon']
|
||
]
|
||
artifact_temp['aritifactSetsName'] = artifact2attr[
|
||
artifact_temp['aritifactName']
|
||
]
|
||
artifact_set_list.append(artifact_temp['aritifactSetsName'])
|
||
artifact_temp['aritifactSetPiece'] = artifactId2Piece[
|
||
artifact_temp['icon'].split('_')[-1]
|
||
][0]
|
||
|
||
artifact_temp['aritifactPieceName'] = artifactId2Piece[
|
||
artifact_temp['icon'].split('_')[-1]
|
||
][1]
|
||
|
||
artifact_temp['aritifactStar'] = artifact['flat']['rankLevel']
|
||
artifact_temp['aritifactLevel'] = (
|
||
artifact['reliquary']['level'] - 1
|
||
)
|
||
|
||
artifact_temp['reliquaryMainstat'] = artifact['flat'][
|
||
'reliquaryMainstat'
|
||
]
|
||
artifact_temp['reliquaryMainstat']['statName'] = propId2Name[
|
||
artifact_temp['reliquaryMainstat']['mainPropId']
|
||
]
|
||
|
||
if 'reliquarySubstats' in artifact['flat']:
|
||
artifact_temp['reliquarySubstats'] = artifact['flat'][
|
||
'reliquarySubstats'
|
||
]
|
||
else:
|
||
artifact_temp['reliquarySubstats'] = []
|
||
for sub in artifact_temp['reliquarySubstats']:
|
||
sub['statName'] = propId2Name[sub['appendPropId']]
|
||
|
||
# 加入单个圣遗物部件
|
||
artifacts_info.append(artifact_temp)
|
||
|
||
# 保存原始数据
|
||
async with aiofiles.open(
|
||
path / 'artifacts.json', 'w', encoding='UTF-8'
|
||
) as file:
|
||
await file.write(
|
||
json.dumps(all_artifacts_data, indent=4, ensure_ascii=False)
|
||
)
|
||
|
||
equipSetList = set(artifact_set_list)
|
||
char_data['equipSets'] = {'type': '', 'set': ''}
|
||
char_data['equipList'] = artifacts_info
|
||
for equip in equipSetList:
|
||
if artifact_set_list.count(equip) >= 4:
|
||
char_data['equipSets']['type'] = '4'
|
||
char_data['equipSets']['set'] = equip
|
||
break
|
||
elif artifact_set_list.count(equip) == 1:
|
||
pass
|
||
elif artifact_set_list.count(equip) >= 2:
|
||
char_data['equipSets']['type'] += '2'
|
||
char_data['equipSets']['set'] += '|' + equip
|
||
|
||
if char_data['equipSets']['set'].startswith('|'):
|
||
char_data['equipSets']['set'] = char_data['equipSets']['set'][1:]
|
||
|
||
threading.Thread(
|
||
target=lambda: asyncio.run(
|
||
_get_data(
|
||
artifacts_info,
|
||
all_artifacts_data,
|
||
avatarId,
|
||
char_data,
|
||
)
|
||
),
|
||
daemon=True,
|
||
).start()
|
||
|
||
char_dict_list.append(char_data)
|
||
async with aiofiles.open(
|
||
path / '{}.json'.format(avatarName), 'w', encoding='UTF-8'
|
||
) as file:
|
||
await file.write(
|
||
json.dumps(char_data, indent=4, ensure_ascii=False)
|
||
)
|
||
|
||
if is_enable_akasha:
|
||
threading.Thread(
|
||
target=lambda: asyncio.run(_restore_cv_data(uid, now)),
|
||
daemon=True,
|
||
).start()
|
||
|
||
return char_dict_list
|
||
|
||
|
||
async def _restore_cv_data(uid: str, now: str):
|
||
cv_api = _CvApi()
|
||
data = await cv_api.get_rank_data(uid)
|
||
path = PLAYER_PATH / str(uid) / 'rank.json'
|
||
if path.exists():
|
||
async with aiofiles.open(path, 'r', encoding='UTF-8') as file:
|
||
rank_data = json.loads(await file.read())
|
||
else:
|
||
rank_data = {}
|
||
if not isinstance(data, int):
|
||
data1, data2 = data[0], data[1]
|
||
for i in data1['data']:
|
||
for j in data2['data']:
|
||
if i['_id'] == j['_id']:
|
||
_value = 0
|
||
stats = {'critValue': j['critValue']}
|
||
for k in j['stats']:
|
||
if k in AT_MAP:
|
||
_k = AT_MAP[k]
|
||
elif k.endswith('DamageBonus'):
|
||
if j['stats'][k]['value'] > 0:
|
||
_value = j['stats'][k]['value']
|
||
continue
|
||
else:
|
||
continue
|
||
else:
|
||
_k = k
|
||
stats[_k] = j['stats'][k]['value']
|
||
stats['DMG'] = _value
|
||
|
||
break
|
||
|
||
cal = i['calculations']
|
||
cal['fit']['stats'] = stats
|
||
rank_data[i['characterId']] = {
|
||
'calculations': cal,
|
||
'time': now,
|
||
}
|
||
async with aiofiles.open(path, 'w', encoding='UTF-8') as file:
|
||
await file.write(
|
||
json.dumps(rank_data, indent=4, ensure_ascii=False)
|
||
)
|
||
|
||
|
||
async def enka_to_data(
|
||
uid: str, enka_data: Optional[EnkaData] = None
|
||
) -> Union[dict, str]:
|
||
raw_data = await enka_to_dict(uid, enka_data)
|
||
if isinstance(raw_data, str):
|
||
return raw_data
|
||
char_name_list = []
|
||
char_name_list_str = ''
|
||
for char_data in raw_data:
|
||
char_name_list.append(char_data['avatarName'])
|
||
char_name_list_str = ','.join(char_name_list)
|
||
return f'UID{uid}刷新完成！\n本次缓存：{char_name_list_str}'
|
||
|
||
|
||
async def _get_data(
|
||
artifacts_info: List,
|
||
all_artifacts_data: Dict,
|
||
avatarId: int,
|
||
char_data: Dict,
|
||
):
|
||
for _artifact in artifacts_info:
|
||
await input_artifacts_data(
|
||
deepcopy(_artifact), all_artifacts_data, avatarId, char_data
|
||
)
|
||
|
||
|
||
async def input_artifacts_data(
|
||
artifact_temp: Dict,
|
||
all_artifacts_data: Dict,
|
||
avatarId: int,
|
||
char_data: Dict,
|
||
):
|
||
artifact_temp = await get_artifact_score_data(
|
||
artifact_temp, Character(char_data)
|
||
)
|
||
# 加入圣遗物数据列表
|
||
if (
|
||
artifact_temp
|
||
not in all_artifacts_data['data'][artifact_temp['aritifactSetPiece']]
|
||
):
|
||
all_artifacts_data['data'][artifact_temp['aritifactSetPiece']].append(
|
||
artifact_temp
|
||
)
|
||
all_artifacts_data['tag'][artifact_temp['aritifactSetPiece']].append(
|
||
[avatarId]
|
||
)
|
||
elif (
|
||
avatarId
|
||
not in all_artifacts_data['tag'][artifact_temp['aritifactSetPiece']][
|
||
all_artifacts_data['data'][
|
||
artifact_temp['aritifactSetPiece']
|
||
].index(artifact_temp)
|
||
]
|
||
):
|
||
all_artifacts_data['tag'][artifact_temp['aritifactSetPiece']][
|
||
all_artifacts_data['data'][
|
||
artifact_temp['aritifactSetPiece']
|
||
].index(artifact_temp)
|
||
].append(avatarId)
|
||
return all_artifacts_data
|