StarRailDamageCal/starrail_damage_cal/model.py

from typing import List

from msgspec import Struct, field


class MihomoAvatarSkill(Struct):
    skillId: int
    skillName: str
    skillEffect: str
    skillAttackType: str
    skillLevel: int


class MihomoAvatarExtraAbility(Struct):
    extraAbilityId: int
    extraAbilityLevel: int


class AttributeBounsStatusAdd(Struct):
    property_: str = field(name="property")
    name: str
    value: float


class MihomoAvatarAttributeBonus(Struct):
    attributeBonusId: int
    attributeBonusLevel: int
    statusAdd: AttributeBounsStatusAdd


class RelicMainAffix(Struct):
    AffixID: int
    Property: str
    Name: str
    Value: float


class RelicSubAffix(Struct):
    SubAffixID: int
    Property: str
    Name: str
    Cnt: int
    Step: int
    Value: float


class Relic(Struct):
    relicId: int
    relicName: str
    SetId: int
    SetName: str
    Type: int
    MainAffix: RelicMainAffix
    SubAffixList: List[RelicSubAffix]
    Level: int = 0


class AvatarBaseAttributes(Struct):
    hp: float
    attack: float
    defence: float
    CriticalChanceBase: float
    CriticalDamageBase: float
    speed: float
    BaseAggro: float


class EquipmentBaseAttributes(Struct):
    hp: float
    attack: float
    defence: float


class AvatarEquipmentInfo(Struct):
    equipmentID: int
    equipmentName: str
    equipmentLevel: int
    equipmentPromotion: int
    equipmentRank: int
    equipmentRarity: int
    baseAttributes: EquipmentBaseAttributes


class RankData(Struct):
    rankId: int
    rankName: str


class MihomoCharacter(Struct):
    uid: str
    nickName: str
    avatarId: int
    avatarName: str
    avatarElement: str
    avatarRarity: str
    avatarLevel: int
    avatarPromotion: int
    avatarSkill: List[MihomoAvatarSkill]
    avatarExtraAbility: List[MihomoAvatarExtraAbility]
    avatarAttributeBonus: List[MihomoAvatarAttributeBonus]
    RelicInfo: List[Relic]
    avatarEnName: str
    baseAttributes: AvatarBaseAttributes
    equipmentInfo: AvatarEquipmentInfo
    rank: int
    rankList: List[RankData]