class Inventory(object):
def __init__(self):
self.items = {}
self.descs = {}
def Add(self, item, amt):
if item in self.items:
self.items[item] += amt
else:
self.items[item] = amt
def Remove(self, item, amt):
if item in self.items and self.items[item] >= amt:
self.items[item] -= amt
if self.items[item] == 0:
del self.items[item]
return True
else:
return False
def GetDescs(self):
data = open('itemlist')
datalist = data.readlines()
for line in range(0, len(datalist), 3):
label = datalist[line].rstrip("\n")
desc = datalist[line + 1].rstrip("\n")
self.itemdescs[label] = desc
data.close()
def HasItem(self, item):
if item in self.items:
return True
else:
return False
def HasNumOfItem(self, item, amt):
if item in self.items and self.items[item] >= amt:
return True
else:
return False