34 lines
937 B
Python
34 lines
937 B
Python
class InventoryComponent:
|
|
def __init__(self, maxitems=100):
|
|
self.maxitems = maxitems
|
|
self.items = {}
|
|
|
|
def add_item(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 remove_item(self, item, count):
|
|
if item in self.items:
|
|
self.items[item] = self.items[item] - count
|
|
if self.items[item] < 0:
|
|
self.items[item] = 0
|
|
|
|
def count(self, item):
|
|
return self.items[item]
|
|
|
|
def has_item(self, item):
|
|
return item in self.items and self.items[item] != 0
|
|
|
|
def total_items_count(self):
|
|
total = 0
|
|
for item, value in self.items.items():
|
|
total += value
|
|
return total
|
|
|
|
def clear(self):
|
|
self.items = {}
|