We can equip items

squirrelbattle/entities/player.py

@@ -4,8 +4,9 @@
 from functools import reduce
 from queue import PriorityQueue
 from random import randint
-from typing import Dict, Tuple
+from typing import Dict, Optional, Tuple
 
+from .items import Item
 from ..interfaces import FightingEntity, InventoryHolder
 
 
@@ -16,12 +17,15 @@ class Player(InventoryHolder, FightingEntity):
     current_xp: int = 0
     max_xp: int = 10
     paths: Dict[Tuple[int, int], Tuple[int, int]]
+    equipped_item: Optional[Item]
+    equipped_armor: Optional[Item]
 
     def __init__(self, name: str = "player", maxhealth: int = 20,
                  strength: int = 5, intelligence: int = 1, charisma: int = 1,
                  dexterity: int = 1, constitution: int = 1, level: int = 1,
                  current_xp: int = 0, max_xp: int = 10, inventory: list = None,
-                 hazel: int = 42, *args, **kwargs) \
+                 hazel: int = 42, equipped_item: Optional[Item] = None,
+                 equipped_armor: Optional[Item] = None, *args, **kwargs) \
             -> None:
         super().__init__(name=name, maxhealth=maxhealth, strength=strength,
                          intelligence=intelligence, charisma=charisma,
@@ -32,6 +36,12 @@ class Player(InventoryHolder, FightingEntity):
         self.inventory = self.translate_inventory(inventory or [])
         self.paths = dict()
         self.hazel = hazel
+        if isinstance(equipped_item, dict):
+            equipped_item = self.dict_to_item(equipped_item)
+        if isinstance(equipped_armor, dict):
+            equipped_armor = self.dict_to_item(equipped_armor)
+        self.equipped_item = equipped_item
+        self.equipped_armor = equipped_armor
 
     def move(self, y: int, x: int) -> None:
         """