class coffee_machine_handler: ingredient_quantity_mapping = { "hot_water": 500, "hot_milk": 500, "ginger_syrup": 100, "sugar_syrup": 100, "tea_leaves_syrup": 100, "green_mixture": 100 } drink_types = { "hot_tea": { "hot_water": 200, "hot_milk": 100, "ginger_syrup": 10, "sugar_syrup": 10, "tea_leaves_syrup": 30 }, "hot_coffee": { "hot_water": 100, "ginger_syrup": 30, "hot_milk": 400, "sugar_syrup": 50, "tea_leaves_syrup": 30 }, "black_tea": { "hot_water": 300, "ginger_syrup": 30, "sugar_syrup": 50, "tea_leaves_syrup": 30 }, "green_tea": { "hot_water": 100, "ginger_syrup": 30, "sugar_syrup": 50, "green_mixture": 30 } } def __init__(self): self.__coffee_machine = CoffeeMachine( outlets=3, ingredients_dict=self.ingredient_quantity_mapping, drink_types=self.drink_types) def make_coffee(self, drink_type): return self.__coffee_machine.make_drink(drink_type=drink_type) def add_ingredients(self, ingredient_name, quantity): return self.__coffee_machine.add_quantity_ingredient( ingredient_name, quantity)
from decimal import Decimal from coffee_machine import CoffeeMachine, CoffeeMachineConfig, CoffeeDrink if __name__ == '__main__': config = CoffeeMachineConfig(water=200, coffee=200, milk=200, drinks=[ CoffeeDrink(id='espresso-1', milk=0, coffee=50, water=10, price=Decimal('3.20'), title='Espresso'), CoffeeDrink(id='latte-1', milk=30, coffee=20, water=10, price=Decimal('2.20'), title='Latte') ]) machine = CoffeeMachine(config) machine.make_drink('espresso-1')