A 15min primer on decorators in Python

decorators.py

# DECORATORS
# https://realpython.com/primer-on-python-decorators/

# Calling higher-order functions.
# A function that takes another function and extends its behaviour (without modifying it).
import functools


def my_decorator(func):
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        print("Before")
        res = func(*args, **kwargs)
        print("After")
        return res
    return wrapper


# Verbose way
def say_whee():
    print("Whee!")

say_whee = my_decorator(say_whee)

say_whee()


# Syntactic sugar
@my_decorator
def say_whee2(name):
    print(f"Whee2 {name}!")
    return f"Hi {name}!"


result = say_whee2("Bum")
print(result)