def test_sellItemPackageWithoutAutoPurchase(self):
productHandler.initProducts(productData, productPackageData)
productHandler.buyProducts('1', 8)
productHandler.buyProducts('4', 5)
productHandler.sellProductPackage('P1', 2, False)
self.assertEqual(productHandler.getProductInventory('P1'), 0)
self.assertEqual(productHandler.getProductInventory('1'), 0)
self.assertEqual(productHandler.getProductInventory('4'), 1)
def test_buyItem(self):
productHandler.initProducts(productData, productPackageData)
productHandler.buyProducts('1', 15)
self.assertEqual(productHandler.getProductInventory('1'), 15)
def test_sellItemsWithoutAutoPurchase(self):
productHandler.initProducts(productData, productPackageData)
productHandler.buyProducts('2', 15)
productHandler.sellProducts('2', 5, False)
self.assertEqual(productHandler.getProductInventory('2'), 10)
def test_buyItemThatDoesNotExist(self):
productHandler.initProducts(productData, productPackageData)
state1 = productHandler.getProducts()
productHandler.buyProducts('A', 15)
state2 = productHandler.getProducts()
self.assertEqual(state1, state2)
def test_buyPackage(self):
productHandler.initProducts(productData, productPackageData)
state1 = productHandler.getProducts()
productHandler.buyProducts('P1', 1)
state2 = productHandler.getProducts()
self.assertDictEqual(state1, state2)
def main():
choice = ""
autoPurchase = True
# Every forth is for free
discount = 4
with open('products.json') as productData:
with open('productPackages.json') as productPackageData:
productHandler.initProducts(
json.load(productData), json.load(productPackageData))
while True:
print("\n" + tabulate([['S[quantity] [product ID]', 'To sell products, enter S + number of items to sell + SPACE + product ID'],
['I[quantity] [product ID]', 'To buy, enter I + number of items to buy + SPACE + product ID.'], [
'L', 'Enter L to show the current product status.'], ['C', 'Enter C to show the product catalogue.'],
['T', 'Toggle automatic purchasing. Current status is ' + (
'On' if autoPurchase else 'Off') + '. \nThis means that when X number of items are sold 2X are bought.'],
['D[every n:th free]',
'Handle discounts given. Currently every n:th item (of the same product) is given for free. Does not include product packages.'],
['Q', 'Enter Q to quit.']], colalign=("center",), headers=['Operation', 'Description']))
choice = input("\nWhat would you like to do? ")
if len(choice) == 0:
print("You need to provide an input.")
else:
operation = choice[0].upper()
# Quit program
if operation == "Q" and len(choice) == 1:
print("Closing application")
exit(0)
# Show inventory
elif operation == "L" and len(choice) == 1:
res = productHandler.getProducts()
ids = []
names = []
fruits = []
inventory = []
for product in res:
fruitList = []
ids.append(res[product].identifier)
inventory.append(res[product].numInInventory)
# Evaluate if package
if res[product].identifier[0] == 'P':
names.append(res[product].name)
for packageItem in res[product].products:
fruitList.append(
packageItem[0].name + " x" + str(packageItem[1]))
fruits.append(', '.join(fruitList))
else:
names.append("-")
fruits.append(res[product].name)
table = zip(ids, names, fruits, inventory)
print("\n" + tabulate(table, colalign=("right", "left", "left",
"right"), headers=['ID', 'Package Name', 'Fruits', 'Number of items in inventory']))
# Show product catalogue
elif operation == "C" and len(choice) == 1:
res = productHandler.getProducts()
ids = []
names = []
fruits = []
prices = []
for product in res:
fruitList = []
ids.append(res[product].identifier)
prices.append(str(res[product].price) +
' ' + res[product].currency)
# Evaluate if package
if res[product].identifier[0] == 'P':
for packageItem in res[product].products:
fruitList.append(
packageItem[0].name + " x" + str(packageItem[1]))
names.append(res[product].name)
fruits.append(', '.join(fruitList))
else:
names.append("-")
fruits.append(res[product].name)
table = zip(ids, names, fruits, prices)
print("\n" + tabulate(table, colalign=("right", "left", "left",
"right"), headers=['ID', 'Package Name', 'Fruits', 'Price']))
# Change automatic purchase toggle
elif operation == "T" and len(choice) == 1:
autoPurchase = productHandler.toggleAutoPurchase(autoPurchase)
print("Changing automatic purchasing to " +
('On' if autoPurchase else 'Off'))
elif operation == "D" and isinstance(int(choice[1:]), int):
discount = int(choice[1:])
print("Changing discount to " + str(discount) + ". Buy " +
str(discount - 1) + ", get the next for free.")
# Check productID and number of items as both are required to sell and buy
elif(" " not in choice):
print(
"Sorry, you are missing the product ID (or have forgotten to separate it with a SPACE)")
else:
productID = choice.split(' ')[1].upper()
try:
numOfItems = int(choice[1:].split(' ')[0])
except ValueError:
print("Sorry, the number of items is not a number.")
if numOfItems < 0:
print("Sorry, the number of items requested cannot be negative.")
if productID in productHandler.getProducts():
# Sell products
if operation == "S":
if(productID[0] == 'P'):
res = productHandler.sellProductPackage(
productID, numOfItems, autoPurchase)
else:
res = productHandler.sellProducts(
productID, numOfItems, autoPurchase)
productInventory = productHandler.getProductInventory(
productID)
productName = productHandler.getProductName(productID)
totalCost = productHandler.getProductCost(
productID, numOfItems, discount)
totalDiscounted = productHandler.getProductCost(
productID, numOfItems, None) - totalCost
currency = productHandler.getCurrency(
productID)
if(totalDiscounted == 0 and res):
print(str(numOfItems) + " " + productName +
"/s sold: " + str(totalCost) + " " + currency)
elif(totalDiscounted > 0 and res):
print(str(numOfItems) + " " + productName + "/s sold: " +
str(totalCost) + " " + currency + ". Discount earned: " + str(totalDiscounted) + " " + currency)
elif(not res):
print(
"You are trying to sell more items than you currently have in stock. Try again")
if(autoPurchase and res):
print(str(2 * numOfItems) + " " + productName +
"/s added to the inventory as automatic purchasing is ON. Automatic purchasing can be turned off using (T).")
print("Current " + productName +
" inventory: " + str(productInventory))
# Buy products (request delivery)
elif operation == "I":
productHandler.buyProducts(productID, numOfItems)
productName = productHandler.getProductName(productID)
productInventory = productHandler.getProductInventory(
productID)
print(str(numOfItems) + " " + productName +
"/s added to the inventory")
print("Current \u001b[1m" + productName +
"\u001b[0m inventory: " + str(productInventory))
else:
print("The command " + operation +
" is not a valid input.")
else:
print(
"Sorry, the operator or product ID provided is incorrect. Check the product catalogue (C) to get some help.")