Decorator example code

decorator_example1.py

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


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


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


americano = decorator_function(add_water)
latte = decorator_function(add_milk)

print("# Americano Recipe")
americano()
print("# Latte Recipe")
latte()


# The code above is output as follows.
# # Americano Recipe
# Add espresso
# Add Water
# # Latte Recipe
# Add espresso
# Add Milk