feat!: Represent HSR character stats with a dictionary

enka/clients/hsr.py

@@ -190,8 +190,8 @@ def _post_process_character(self, character: Character) -> None:
  StatType.IMAGINARY_DMG_BOOST: chara_stats["ImaginaryAddedRatio"],
  }
 
- character.stats = [
- Stat(
+ character.stats = {
+ stat_type: Stat(
  type=stat_type,
  value=value,
  name=self._assets.text_map[stat_type.value],
@@ -200,7 +200,7 @@ def _post_process_character(self, character: Character) -> None:
  ),
  )
  for stat_type, value in final_stats.items()
- ]
+ }
 
  def _add_up_character_stats(self, character: Character) -> dict[str, float]: # noqa: C901
  chara_stats = DEFAULT_STATS.copy()

enka/models/hsr/character.py

@@ -145,7 +145,7 @@ class Character(BaseModel):
  rarity: Literal[4, 5] = Field(None)
  element: Element = Field(None)
  path: Path = Field(None)
- stats: list[Stat] = Field(list)
+ stats: dict[StatType, Stat] = Field(default_factory=dict)
 
  @computed_field
  @property
@@ -158,9 +158,13 @@ def max_level(self) -> Literal[20, 30, 40, 50, 60, 70, 80]:
  def highest_dmg_bonus_stat(self) -> Stat:
  """Character's highest damage bonus stat."""
  return max(
- (stat for stat in self.stats if stat.type in DMG_BONUS_PROPS.values()),
+ (stat for stat in self.stats.values() if stat.type in DMG_BONUS_PROPS.values()),
  key=lambda stat: stat.value,
  default=next(
- (stat for stat in self.stats if stat.type.value == DMG_BONUS_PROPS[self.element])
+ (
+ stat
+ for stat in self.stats.values()
+ if stat.type.value == DMG_BONUS_PROPS[self.element]
+ )
  ),
  )

examples/hsr.py

@@ -42,7 +42,7 @@ async def main() -> None: # noqa: C901, D103
  print(stat.name, stat.formatted_value)
 
  print("\nStats:")
- for stat in character.stats:
+ for stat in character.stats.values():
  if stat.value == 0 or stat.type.value in enka.hsr.DMG_BONUS_PROPS.values():
  continue
  print(stat.name, stat.formatted_value)