Update the player inventory when opening the menu/loading the game, fixes #72

squirrelbattle/display/menudisplay.py

@@ -162,6 +162,7 @@ class PlayerInventoryDisplay(MenuDisplay):
     def update(self, game: Game) -> None:
         self.player = game.player
         self.update_menu(game.inventory_menu)
+        game.inventory_menu.update_player(self.player)
         self.store_mode = game.state == GameMode.STORE
         self.chest_mode = game.state == GameMode.CHEST
         self.selected = game.state == GameMode.INVENTORY \

squirrelbattle/game.py

@@ -435,6 +435,7 @@ class Game:
             return
 
         self.player = players[0]
+        self.inventory_menu.update_player(self.player)
         self.map.compute_visibility(self.player.y, self.player.x,
                                     self.player.vision)
         self.display_actions(DisplayActions.UPDATE)