Decorators

Resources

• Practical Decorators: Pycon 2019
• Making your Python decorators even better, with functool.wraps
• How To Test Python Decorators? How To Bypass A Decorator?
• Python Decorator Tutorial with Example (has examples of decorators with arguments)

Decorator with arguments

from functools import partial, wraps

def print_result(func=None, *, prefix=''):
    if func is None:
        return partial(print_result, prefix=prefix)

    @wraps(func)
    def wrapper(*args, **kwargs):
        result = func(*args, **kwargs)
        print(f'{prefix}{result}')
        return result
    return wrapper

# ---------- #

@print_result
def add(a, b):
    return a + b

add(2, 3)  # outputs '5'

@print_result()
def add(a, b):
    return a + b

add(2, 3)  # outputs '5'

@print_result(prefix='The return value is ')
def add(a, b):
    return a + b

add(2, 3)  # outputs 'The return value is 5'