From 4ceaf4904bc9e8a067e3686fd46e990c5d5573e4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jakub=20Klupie=C4=87?= <jakklu@st.amu.edu.pl>
Date: Sat, 8 May 2021 22:15:26 +0200
Subject: [PATCH] Add inventory component

---
 survival/components/inventory_component.py | 24 ++++++++++++++++++++++
 1 file changed, 24 insertions(+)
 create mode 100644 survival/components/inventory_component.py

diff --git a/survival/components/inventory_component.py b/survival/components/inventory_component.py
new file mode 100644
index 0000000..18d39f9
--- /dev/null
+++ b/survival/components/inventory_component.py
@@ -0,0 +1,24 @@
+class InventoryComponent:
+    def __init__(self, maxitems):
+        self.maxitems = maxitems
+        self.items = {}
+
+    def addItem(self, item, count):
+        if item not in self.items:
+            self.items[item] = count
+        else:
+            self.items[item] = self.items[item] + count
+        if self.items[item] > self.maxitems:
+            self.items[item] = self.maxitems
+
+    def removeItem(self, item, count):
+        if self.items:
+            self.items[item] = self.items[item] - count
+        if self.items[item] < 0:
+            self.items[item] = 0
+
+    def hasItem(self, item):
+        if self.items[item] != 0:
+            return True
+        else:
+            return False