dev-jakklu #5
24
survival/components/inventory_component.py
Normal file
24
survival/components/inventory_component.py
Normal file
@ -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
|
Loading…
Reference in New Issue
Block a user