data_types.py

"""Core data types, enums, and constants for the JRPG game."""

from __future__ import annotations

from dataclasses import dataclass, field

from enum import Enum, auto

from typing import Optional, List, Dict, Any

class WeaponType(Enum):

 SWORD = "Sword"

 SPEAR = "Spear"

 AXE = "Axe"

 DAGGER = "Dagger"

 BOW = "Bow"

 STAFF = "Staff"

class Element(Enum):

 FIRE = "Fire"

 ICE = "Ice"

 LIGHTNING = "Lightning"

 WIND = "Wind"

 EARTH = "Earth"

 LIGHT = "Light"

 DARK = "Dark"

 NONE = "None"

class JobClass(Enum):

 WARRIOR = "Warrior"

 KNIGHT = "Knight"

 PALADIN = "Paladin"

 BERSERKER = "Berserker"

 ASSASSIN = "Assassin"

 RANGER = "Ranger"

 HUNTER = "Hunter"

 SPEARMAN = "Spearman"

 DRAGOON = "Dragoon"

 MONK = "Monk"

 CLERIC = "Cleric"

 PRIEST = "Priest"

 FIRE_MAGE = "Fire Mage"

 ICE_MAGE = "Ice Mage"

 STORM_MAGE = "Storm Mage"

 WIND_MAGE = "Wind Mage"

 EARTH_MAGE = "Earth Mage"

 DARK_MAGE = "Dark Mage"

 LIGHT_MAGE = "Light Mage"

 ARCANE_SAGE = "Arcane Sage"

class SkillType(Enum):

 PHYSICAL = "Physical"

 MAGICAL = "Magical"

 HEAL = "Heal"

 BUFF = "Buff"

 DEBUFF = "Debuff"

 SPECIAL = "Special"

class SkillTier(Enum):

 BASIC = "Basic"

 INTERMEDIATE = "Intermediate"

 ULTIMATE = "Ultimate"

class SkillTarget(Enum):

 SINGLE_ENEMY = "SingleEnemy"

 ALL_ENEMIES = "AllEnemies"

 SINGLE_ALLY = "SingleAlly"

 ALL_ALLIES = "AllAllies"

 SELF = "Self"

 RANDOM_ENEMY = "RandomEnemy"

class StatusEffectType(Enum):

 # Damage over time

 POISON = "Poison"

 BURN = "Burn"

 BLEED = "Bleed"

 VENOM = "Venom"

 CURSE = "Curse"

 # Action restriction

 SLEEP = "Sleep"

 STUN = "Stun"

 FREEZE = "Freeze"

 PARALYZE = "Paralyze"

 PETRIFY = "Petrify"

 # Debuffs

 ATTACK_DOWN = "Attack Down"

 DEFENSE_DOWN = "Defense Down"

 MAGIC_DOWN = "Magic Down"

 SPEED_DOWN = "Speed Down"

 ACCURACY_DOWN = "Accuracy Down"

 # Buff blockers

 SILENCE = "Silence"

 SKILL_SEAL = "Skill Seal"

 ITEM_SEAL = "Item Seal"

 HEAL_BLOCK = "Heal Block"

 MANA_BURN = "Mana Burn"

 # Persistent effects

 REGEN = "Regen"

 MANA_REGEN = "Mana Regen"

 SHIELD = "Shield"

 REFLECT = "Reflect"

 COUNTER = "Counter"

 # Special

 CONFUSION = "Confusion"

 CHARM = "Charm"

 FEAR = "Fear"

 BLIND = "Blind"

 WEAKNESS_MARK = "Weakness Mark"

 # Enhancements

 BERSERK = "Berserk"

 HASTE = "Haste"

 FOCUS = "Focus"

 GUARD_UP = "Guard Up"

 MAGIC_BOOST = "Magic Boost"

 # Advanced

 DOOM = "Doom"

 TIME_STOP = "Time Stop"

 CURSE_MARK = "Curse Mark"

 BLOOD_LINK = "Blood Link"

 SOUL_DRAIN = "Soul Drain"

 # Defend

 DEFENDING = "Defending"

class AIType(Enum):

 AGGRESSIVE = "Aggressive"

 DEFENSIVE = "Defensive"

 BALANCED = "Balanced"

 SUPPORT = "Support"

 RANDOM = "Random"

 BERSERKER = "Berserker"

 TACTICAL = "Tactical"

class ItemType(Enum):

 RECOVERY = "Recovery"

 STATUS_CURE = "Status Cure"

 BUFF = "Buff"

 OFFENSIVE = "Offensive"

 ADVANCED = "Advanced"

 REVIVAL = "Revival"

@dataclass

class SkillEffect:

 status: Optional[StatusEffectType] = None

 duration: int = 0

 chance: float = 0.0

 stat_modifier: float = 1.0

 heal_percent: float = 0.0

 shield_amount: int = 0

@dataclass

class Skill:

 id: int

 name: str

 skill_type: SkillType

 power: float

 mp_cost: int

 accuracy: float

 element: Element

 hits: int

 target: SkillTarget

 tier: SkillTier

 description: str

 effect: Optional[SkillEffect] = None

 jobs: List[JobClass] = field(default_factory=list)

@dataclass

class StatusEffect:

 effect_type: StatusEffectType

 duration: int

 power: float = 1.0

 source_name: str = ""

 def tick(self) -> bool:

 """Returns True if effect is still active after tick."""

 if self.duration > 0:

 self.duration -= 1

 return self.duration > 0 or self.duration == -1 # -1 = permanent until removed

@dataclass

class Item:

 id: int

 name: str

 item_type: ItemType

 description: str

 effect_value: int

 target: SkillTarget

 element: Optional[Element] = None

 status_cure: Optional[List[StatusEffectType]] = None

 stat_buff: Optional[Dict[str, float]] = None

 buff_duration: int = 0

@dataclass

class Stats:

 hp: int

 mp: int

 patk: int

 matk: int

 pdef: int

 mdef: int

 spd: int

 eva: float

 acc: float

 crit: float

 def copy(self) -> 'Stats':

 return Stats(

 hp=self.hp, mp=self.mp, patk=self.patk, matk=self.matk,

 pdef=self.pdef, mdef=self.mdef, spd=self.spd,

 eva=self.eva, acc=self.acc, crit=self.crit

 )

@dataclass

class GrowthRates:

 hp: float

 mp: float

 patk: float

 matk: float

 pdef: float

 mdef: float

 spd: float

 eva: float

 acc: float

 crit: float

RARITY_COLORS = {

 3: "★★★",

 4: "★★★★",

 5: "★★★★★"

}