from deal import Deal
from product import Product
beanName = "Canned Beans"
beanCost = 2
beanQuantity = 10
beansOnSale = True
buyThreeGetOneFreeDealOnBeans = Deal()
cannedBeans = Product(beanName, beanCost, beanQuantity, beansOnSale)
buyThreeGetOneFreeDealOnBeans.dealPrice = cannedBeans.price * 3
buyThreeGetOneFreeDealOnBeans.maxQuantity = 4
cannedBeans.deal = buyThreeGetOneFreeDealOnBeans
def calculateCostAfterDeal(item):
if (item.quantity <= item.deal.maxQuantity):
return (item.deal.dealPrice * item.quantity)
else:
totalCostForItemsThatDoNotQualifyForDeal = item.price * (
item.quantity - item.deal.maxQuantity)
total = totalCostForItemsThatDoNotQualifyForDeal + item.deal.dealPrice
return (total)
def calculateCost(item):
if (item.onSale):
return (calculateCostAfterDeal(item))
else:
return (item.price * item.quantity)
