Merchant inventory is well-saved

squirrelbattle/entities/friendly.py

@@ -23,6 +23,15 @@ class Merchant(FriendlyEntity):
         super().__init__(name=name, *args, **kwargs)
         self.inventory = inventory or []
         self.hazel = hazel
+
+        entity_classes = self.get_all_entity_classes_in_a_dict()
+
+        for i in range(len(self.inventory)):
+            if isinstance(self.inventory[i], dict):
+                item_class = entity_classes[self.inventory[i]["type"]]
+                self.inventory[i] = item_class(**self.inventory[i])
+
+        if not self.inventory:
             for i in range(5):
                 self.inventory.append(choice(Item.get_all_items())())
 
@@ -34,6 +43,14 @@ class Merchant(FriendlyEntity):
         # TODO
         return _("I don't sell any squirrel")
 
+    def save_state(self) -> dict:
+        """
+        We save the inventory of the merchant formatted as JSON
+        """
+        d = super().save_state()
+        d["inventory"] = [item.save_state() for item in self.inventory]
+        return d
+
 
 class Sunflower(FriendlyEntity):
     """

squirrelbattle/entities/player.py

@@ -30,9 +30,9 @@ class Player(FightingEntity):
         self.current_xp = current_xp
         self.max_xp = max_xp
         self.inventory = inventory if inventory else list()
+        entity_classes = self.get_all_entity_classes_in_a_dict()
         for i in range(len(self.inventory)):
             if isinstance(self.inventory[i], dict):
-                entity_classes = self.get_all_entity_classes_in_a_dict()
                 item_class = entity_classes[self.inventory[i]["type"]]
                 self.inventory[i] = item_class(**self.inventory[i])
         self.paths = dict()

squirrelbattle/interfaces.py

@@ -362,7 +362,8 @@ class Entity:
         from squirrelbattle.entities.monsters import Tiger, Hedgehog, Rabbit, \
             TeddyBear
         from squirrelbattle.entities.friendly import Merchant, Sunflower
-        from squirrelbattle.entities.items import BodySnatchPotion, Bomb, Heart
+        from squirrelbattle.entities.items import BodySnatchPotion, Bomb, \
+            Heart, Sword
         return {
             "Tiger": Tiger,
             "Bomb": Bomb,
@@ -374,6 +375,7 @@ class Entity:
             "Player": Player,
             "Merchant": Merchant,
             "Sunflower": Sunflower,
+            "Sword": Sword,
         }
 
     def save_state(self) -> dict: