squirrel-battle/squirrelbattle/entities/items.py

from typing import Optional

from .player import Player
from ..interfaces import Entity, FightingEntity, Map


class Item(Entity):
    """
    A class for items
    """
    held: bool
    held_by: Optional[Player]

    def __init__(self, held: bool = False, held_by: Optional[Player] = None,
                 *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.held = held
        self.held_by = held_by

    def drop(self, y: int, x: int) -> None:
        """
        The item is dropped from the inventory onto the floor
        """
        if self.held:
            self.held_by.inventory.remove(self)
            self.held = False
            self.held_by = None
        self.map.add_entity(self)
        self.move(y, x)

    def hold(self, player: "Player") -> None:
        """
        The item is taken from the floor and put into the inventory
        """
        self.held = True
        self.held_by = player
        self.map.remove_entity(self)
        player.inventory.append(self)

    def save_state(self) -> dict:
        """
        Saves the state of the entity into a dictionary
        """
        d = super().save_state()
        d["held"] = self.held
        return d


class Heart(Item):
    """
    A heart item to return health to the player
    """
    healing: int

    def __init__(self, healing: int = 5, *args, **kwargs):
        super().__init__(name="heart", *args, **kwargs)
        self.healing = healing

    def hold(self, player: "Player") -> None:
        """
        When holding a heart, heal the player and don't put item in inventory.
        """
        player.health = min(player.maxhealth, player.health + self.healing)
        self.map.remove_entity(self)

    def save_state(self) -> dict:
        """
        Saves the state of the header into a dictionary
        """
        d = super().save_state()
        d["healing"] = self.healing
        return d


class Bomb(Item):
    """
    A bomb item intended to deal damage to enemies at long range
    """
    damage: int = 5
    exploding: bool

    def __init__(self, damage: int = 5, exploding: bool = False,
                 *args, **kwargs):
        super().__init__(name="bomb", *args, **kwargs)
        self.damage = damage
        self.exploding = exploding

    def drop(self, x: int, y: int) -> None:
        super().drop(x, y)
        self.exploding = True

    def act(self, m: Map) -> None:
        """
        Special exploding action of the bomb
        """
        if self.exploding:
            for e in m.entities.copy():
                if abs(e.x - self.x) + abs(e.y - self.y) <= 1 and \
                        isinstance(e, FightingEntity):
                    e.take_damage(self, self.damage)

    def save_state(self) -> dict:
        """
        Saves the state of the bomb into a dictionary
        """
        d = super().save_state()
        d["exploding"] = self.exploding
        d["damage"] = self.damage
        return d

class Weapon(Item):
    """
    Non-throwable items that improve player damage
    """
    damage: int

    def __init__(self, damage: int = 3, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.damage = damage

    def save_state(self) -> dict:
        """
        Saves the state of the weapon into a dictionary
        """
        d = super().save_state()
        d["damage"] = self.damage
        return d

class Sword(Weapon) :
    """
    A basic weapon
    """
    def __init__(self, name: int, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.name = "sword"