Source code for simaple.gear.gear

from __future__ import annotations

from pydantic import BaseModel, ConfigDict, Field

from simaple.core import ExtendedStat, Stat
from simaple.gear.gear_type import GearType
from simaple.gear.potential import Potential


[docs]class GearMeta(BaseModel, frozen=True):
    id: int
    name: str
    base_stat: Stat
    type: GearType
    req_level: int
    boss_reward: bool = False
    superior_eqp: bool = False
    req_job: int = 0
    set_item_id: int = 0
    joker_to_set_item: bool = False
    max_scroll_chance: int
    exceptional_enhancement: bool = False

[docs]    def show(self) -> str:
        job_string = (
            "["
            + ("] [").join(
                [(" V " if self.req_job & (1 << i) != 0 else "   ") for i in range(5)]
            )
            + "]"
        )

        return f"""
        ===================================
        name: {self.name}
        type: {self.type.name} (type number {self.type})
        req_level: {self.req_level}
                 [WAR] [MAG] [ARC] [THF] [PIR]
        req_job: {job_string}
        """


[docs]class Gear(BaseModel):
    meta: GearMeta
    stat: Stat
    scroll_chance: int
    potential: Potential = Field(default_factory=Potential)
    additional_potential: Potential = Field(default_factory=Potential)

    model_config = ConfigDict(validate_assignment=True, frozen=True)

[docs]    @classmethod
    def create_bare_gear(cls, meta: GearMeta) -> Gear:
        return Gear(
            meta=meta,
            stat=meta.base_stat.model_copy(),
            scroll_chance=meta.max_scroll_chance,
        )

[docs]    def add_stat(self, stat: Stat) -> Gear:
        serailized_gear = self.model_dump()
        serailized_gear["stat"] = Stat.model_validate(serailized_gear["stat"]) + stat

        return Gear.model_validate(serailized_gear)

[docs]    def set_potential(self, potential: Potential) -> Gear:
        serailized_gear = self.model_dump()
        serailized_gear["potential"] = potential.model_copy()

        return Gear.model_validate(serailized_gear)

[docs]    def set_additional_potential(self, additional_potential: Potential) -> Gear:
        serailized_gear = self.model_dump()
        serailized_gear["additional_potential"] = additional_potential.model_copy()

        return Gear.model_validate(serailized_gear)

[docs]    def sum_stat(self) -> Stat:
        return self.sum_extended_stat().stat

[docs]    def sum_extended_stat(self) -> ExtendedStat:
        potential_extended_stats = (
            self.potential.get_extended_stat()
            + self.additional_potential.get_extended_stat()
        )
        return ExtendedStat(stat=self.stat) + potential_extended_stats

[docs]    def show(self) -> str:
        return f"""
        {self.meta.show()}
        ===================================
        Basis Stats

        STR: {self.stat.STR}
        DEX: {self.stat.DEX}
        INT: {self.stat.INT}
        LUK: {self.stat.LUK}
        
        MaxHP: {self.stat.MHP}
        MaxMP: {self.stat.MMP}

        ATT: {self.stat.attack_power}
        MAT: {self.stat.magic_attack}

        ===================================
        """