PamGram/model/apihelper/playercards.py

import os
from typing import Union, Optional

import httpx
import ujson

from model.base import GameItem
from model.game.artifact import ArtifactInfo
from model.game.character import CharacterInfo, CharacterValueInfo
from model.game.fetter import FetterInfo
from model.game.skill import Skill
from model.game.talent import Talent
from model.game.weapon import WeaponInfo
from .helpers import get_headers


class PlayerCardsAPI:
    UI_URL = "https://enka.shinshin.moe/ui/"

    def __init__(self):
        self.client = httpx.AsyncClient(headers=get_headers())
        project_path = os.path.dirname(__file__)
        characters_map_file = os.path.join(project_path, "metadata", "CharactersMap.json")
        name_text_map_hash_file = os.path.join(project_path, "metadata", "NameTextMapHash.json")
        reliquary_name_map_file = os.path.join(project_path, "metadata", "ReliquaryNameMap.json")
        with open(characters_map_file, "r", encoding="utf-8") as f:
            self._characters_map_json: dict = ujson.load(f)
        with open(name_text_map_hash_file, "r", encoding="utf-8") as f:
            self._name_text_map_hash_json: dict = ujson.load(f)
        with open(reliquary_name_map_file, "r", encoding="utf-8") as f:
            self._reliquary_name_map_json: dict = ujson.load(f)

    def get_characters_name(self, item_id: Union[int, str]) -> str:
        if isinstance(item_id, int):
            item_id = str(item_id)
        characters = self.get_characters(item_id)
        name_text_map_hash = characters.get("NameTextMapHash", "-1")
        return self.get_text(str(name_text_map_hash))

    def get_characters(self, item_id: Union[int, str]) -> dict:
        if isinstance(item_id, int):
            item_id = str(item_id)
        return self._characters_map_json.get(item_id, {})

    def get_text(self, hash_value: Union[int, str]) -> str:
        if isinstance(hash_value, int):
            hash_value = str(hash_value)
        return self._name_text_map_hash_json.get(hash_value, "")

    def get_reliquary_name(self, reliquary: str) -> str:
        return self._reliquary_name_map_json[reliquary]

    async def get_data(self, uid: Union[str, int]):
        url = f"https://enka.shinshin.moe/u/{uid}/__data.json"
        response = await self.client.get(url)
        return response

    def data_handler(self, avatar_data: dict, avatar_id: int) -> CharacterInfo:
        artifact_list = []

        weapon_info: Optional[WeaponInfo] = None

        equip_list = avatar_data["equipList"]  # 圣遗物和武器相关
        fetter_info = avatar_data["fetterInfo"]  # 好感等级
        fight_prop_map = avatar_data["fightPropMap"]  # 属性
        # inherent_proud_skill_list = avatar_data["inherentProudSkillList"]  # 不知道
        prop_map = avatar_data["propMap"]  # 角色等级 其他信息
        # proud_skill_extra_level_map = avatar_data["proudSkillExtraLevelMap"]  # 不知道
        # skill_depot_id = avatar_data["skillDepotId"]  # 不知道
        skill_level_map = avatar_data["skillLevelMap"]  # 技能等级

        # 角色等级
        character_level = prop_map['4001']['val']

        # 角色姓名
        character_name = self.get_characters_name(avatar_id)
        characters_data = self.get_characters(avatar_id)

        # 圣遗物和武器
        for equip in equip_list:
            if "reliquary" in equip:  # 圣遗物
                flat = equip["flat"]
                reliquary = equip["reliquary"]
                reliquary_main_stat = flat["reliquaryMainstat"]
                reliquary_sub_stats = flat['reliquarySubstats']
                sub_item = []
                for reliquary_sub in reliquary_sub_stats:
                    sub_item.append(GameItem(name=self.get_reliquary_name(reliquary_sub["appendPropId"]),
                                             item_type=reliquary_sub["appendPropId"], value=reliquary_sub["statValue"]))
                main_item = GameItem(name=self.get_reliquary_name(reliquary_main_stat["mainPropId"]),
                                     item_type=reliquary_main_stat["mainPropId"],
                                     value=reliquary_main_stat["statValue"])
                name = self.get_text(flat["nameTextMapHash"])
                artifact_list.append(ArtifactInfo(item_id=equip["itemId"], name=name, star=flat["rankLevel"],
                                                  level=reliquary["level"] - 1, main_item=main_item, sub_item=sub_item))
            if "weapon" in equip:  # 武器
                flat = equip["flat"]
                weapon_data = equip["weapon"]
                # 防止未精炼
                if 'promoteLevel' in weapon_data:
                    weapon_level = weapon_data['promoteLevel'] - 1
                else:
                    weapon_level = 0
                if 'affixMap' in weapon_data:
                    affix = list(weapon_data['affixMap'].values())[0] + 1
                else:

                    affix = 1
                reliquary_main_stat = flat["weaponStats"][0]
                reliquary_sub_stats = flat['weaponStats'][1]
                sub_item = GameItem(name=self.get_reliquary_name(reliquary_main_stat["appendPropId"]),
                                    item_type=reliquary_sub_stats["appendPropId"],
                                    value=reliquary_sub_stats["statValue"])
                main_item = GameItem(name=self.get_reliquary_name(reliquary_main_stat["appendPropId"]),
                                     item_type=reliquary_main_stat["appendPropId"],
                                     value=reliquary_main_stat["statValue"])
                weapon_name = self.get_text(flat["nameTextMapHash"])
                weapon_info = WeaponInfo(item_id=equip["itemId"], name=weapon_name, star=flat["rankLevel"],
                                         level=weapon_level, main_item=main_item, sub_item=sub_item, affix=affix)

        # 好感度
        fetter = FetterInfo(fetter_info["expLevel"])

        # 基础数值处理
        for i in range(40, 47):
            if fight_prop_map[str(i)] > 0:
                dmg_bonus = fight_prop_map[str(i)]
                break
        else:
            dmg_bonus = 0

        base_value = CharacterValueInfo(fight_prop_map["2000"], fight_prop_map["1"], fight_prop_map["2001"],
                                        fight_prop_map["4"], fight_prop_map["2002"], fight_prop_map["7"],
                                        fight_prop_map["28"], fight_prop_map["20"], fight_prop_map["22"],
                                        fight_prop_map["23"], fight_prop_map["26"], fight_prop_map["27"],
                                        fight_prop_map["29"], fight_prop_map["30"], dmg_bonus)

        # 技能处理
        skill_list = []
        skills = characters_data["Skills"]
        for skill_id in skill_level_map:
            skill_list.append(Skill(skill_id, name=skill_level_map[skill_id], icon=skills[skill_id]))

        # 命座处理
        talent_list = []
        consts = characters_data["Consts"]
        if 'talentIdList' in avatar_data:
            talent_id_list = avatar_data["talentIdList"]
            for index, value in enumerate(talent_id_list):
                talent_list.append(Talent(talent_id_list[index], icon=consts[index]))

        element = characters_data["Element"]
        icon = characters_data["SideIconName"]
        character_info = CharacterInfo(character_name, element, character_level, fetter, base_value, weapon_info,
                                       artifact_list, skill_list, talent_list, icon)
        return character_info