Further clean up, also fixed a tiny bug (wrong variable name).

I move the remove item logic to player.py and have skipped some of
inventory/equiptment information provided by the server.

1 files changed, 6 insertions, 0 deletions

diff --git a/player.py b/player.py
index 6277837..6f602c5 100644
--- a/player.py
+++ b/player.py
@@ -38,6 +38,12 @@ class Player:
                     return item
         return -10 # Not found - bug somewhere!
 
+    def remove_item(self, index, amount):
+        if index in self.inventory:
+            self.inventory[index].amount -= amount
+            if self.inventory[index].amount == 0:
+                del self.inventory[index]
+
     def check_inventory(self, user_tree, sale_tree):
         # Check the inventory state.
         test_node = self.inventory.copy()