mirror of
https://github.com/PaiGramTeam/StarRailDamageCal.git
synced 2024-11-28 18:19:25 +00:00
105 lines
2.5 KiB
Python
105 lines
2.5 KiB
Python
from typing import List, Union
|
|
|
|
import msgspec
|
|
from msgspec import Struct, field
|
|
|
|
|
|
class DamageInstanceSkill(Struct):
|
|
skillId: int
|
|
skillName: str
|
|
skillEffect: str
|
|
skillAttackType: str
|
|
skillLevel: int
|
|
|
|
|
|
class DamageInstanceRelicSubAffix(Struct):
|
|
SubAffixID: int
|
|
Property: str
|
|
Name: str
|
|
Cnt: int
|
|
Step: int
|
|
Value: float
|
|
|
|
|
|
class DamageInstanceRelicMainAffix(Struct):
|
|
AffixID: int
|
|
Property: str
|
|
Name: str
|
|
Value: float
|
|
|
|
|
|
class DamageInstanceRelic(Struct):
|
|
relicId: int
|
|
relicName: str
|
|
SetId: int
|
|
SetName: str
|
|
Type: int
|
|
MainAffix: DamageInstanceRelicMainAffix
|
|
SubAffixList: Union[List[DamageInstanceRelicSubAffix], None]
|
|
Level: int = 0
|
|
|
|
|
|
class DamageInstanceWeapon(Struct):
|
|
id_: str = field(name="id")
|
|
level: int
|
|
rank: int
|
|
promotion: int
|
|
|
|
|
|
class AttributeBounsStatusAdd(Struct):
|
|
property_: str = field(name="property")
|
|
name: str
|
|
value: float
|
|
|
|
|
|
class DamageInstanceAvatarAttributeBouns(Struct):
|
|
attributeBonusId: int
|
|
attributeBonusLevel: int
|
|
statusAdd: AttributeBounsStatusAdd
|
|
|
|
|
|
class DamageInstanceAvatar(Struct):
|
|
id_: str = field(name="id")
|
|
level: int
|
|
rank: int
|
|
element: str
|
|
promotion: int
|
|
attribute_bonus: Union[List[DamageInstanceAvatarAttributeBouns], None]
|
|
extra_ability: Union[List, None]
|
|
|
|
|
|
class DamageInstance:
|
|
avatar: DamageInstanceAvatar
|
|
weapon: DamageInstanceWeapon
|
|
relic: List[DamageInstanceRelic]
|
|
skill: List[DamageInstanceSkill]
|
|
|
|
def __init__(self, char):
|
|
self.avatar = DamageInstanceAvatar(
|
|
id_=char.char_id,
|
|
level=char.char_level,
|
|
rank=char.char_rank,
|
|
element=char.char_element,
|
|
promotion=char.char_promotion,
|
|
attribute_bonus=msgspec.convert(
|
|
char.attribute_bonus,
|
|
Union[List[DamageInstanceAvatarAttributeBouns], None],
|
|
),
|
|
extra_ability=msgspec.convert(
|
|
char.extra_ability,
|
|
Union[List, None],
|
|
),
|
|
)
|
|
self.weapon = DamageInstanceWeapon(
|
|
id_=char.equipment["equipmentID"],
|
|
level=char.equipment["equipmentLevel"],
|
|
rank=char.equipment["equipmentRank"],
|
|
promotion=char.equipment["equipmentPromotion"],
|
|
)
|
|
self.relic = []
|
|
for relic in char.char_relic:
|
|
self.relic.append(msgspec.convert(relic, DamageInstanceRelic))
|
|
self.skill = []
|
|
for skill in char.char_skill:
|
|
self.skill.append(msgspec.convert(skill, DamageInstanceSkill))
|