45 lines
1.0 KiB
Python
45 lines
1.0 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
from components.base_component import BaseComponent
|
|
from equipment_types import EquipmentType
|
|
|
|
if TYPE_CHECKING:
|
|
from entity import Item
|
|
|
|
|
|
class Equippable(BaseComponent):
|
|
parent: Item
|
|
|
|
def __init__(
|
|
self,
|
|
equipment_type: EquipmentType,
|
|
power_bonus: int = 0,
|
|
defense_bonus: int = 0,
|
|
):
|
|
self.equipment_type = equipment_type
|
|
|
|
self.power_bonus = power_bonus
|
|
self.defense_bonus = defense_bonus
|
|
|
|
|
|
class Dagger(Equippable):
|
|
def __init__(self) -> None:
|
|
super().__init__(EquipmentType.WEAPON, power_bonus=2)
|
|
|
|
|
|
class Sword(Equippable):
|
|
def __init__(self) -> None:
|
|
super().__init__(EquipmentType.WEAPON, power_bonus=4)
|
|
|
|
|
|
class LeatherArmor(Equippable):
|
|
def __init__(self) -> None:
|
|
super().__init__(EquipmentType.ARMOR, defense_bonus=1)
|
|
|
|
|
|
class ChainMail(Equippable):
|
|
def __init__(self) -> None:
|
|
super().__init__(EquipmentType.ARMOR, defense_bonus=3)
|