Add inventory component

This commit is contained in:
Jakub Klupieć 2021-05-08 22:15:26 +02:00
parent 99934067de
commit 4ceaf4904b

View 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