39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
import json
|
|
|
|
MILK_TYPES = ["Cream", "Half-and-half", "Whole-milk", "Part-Skim", "Skim", "Non-Dairy"]
|
|
|
|
SYRUP_TYPES = ["Vanilla", "Almond", "Raspberry", "Chocolate"]
|
|
|
|
ALCOHOL_TYPES = ["Whisky", "Rum", "Kahlua", "Aquavit"]
|
|
|
|
ADDITION_HEADER = "Accept-Additions"
|
|
|
|
|
|
class Order(object):
|
|
|
|
def __init__(self, name):
|
|
self.name = name
|
|
self.milk = "None"
|
|
self.syrup = "None"
|
|
self.alcohol = "None"
|
|
print(f"Configuring order - {name}")
|
|
|
|
def configure(self, header):
|
|
additions = []
|
|
for k, v in header.items():
|
|
if ADDITION_HEADER in k:
|
|
additions = v
|
|
print(additions)
|
|
addition_list = additions.split(';')
|
|
for ad in addition_list:
|
|
print(ad)
|
|
if ad.strip() in MILK_TYPES:
|
|
self.milk = ad
|
|
elif ad.strip() in SYRUP_TYPES:
|
|
self.syrup = ad
|
|
elif ad.strip() in ALCOHOL_TYPES:
|
|
self.alcohol = ad
|
|
|
|
def json_dump(self):
|
|
return json.dumps([{'name': self.name}, {'milk': self.milk}, {'syrup': self.syrup}, {'alcohol': self.alcohol}])
|