Decorator example code

decorator_example2.py

def decorator_function(original_function):
    def add_espresso():
        print("Add espresso")
        return original_function()

    return add_espresso


@decorator_function
def add_water():
    print("Add Water")


@decorator_function
def add_milk():
    print("Add Milk")


print("# Americano Recipe")
add_water()
print("# Latte Recipe")
add_milk()