Rename dungeonbattle to squirrelbattle

squirrelbattle/entities/items.py

@@ -0,0 +1,109 @@
+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