Python decorator with argument(s)

decorator_args.py

from functools import wraps


def decorator_with_argument(argument=None):

    def actual_decorator(func):
    
        @wraps(func)
        def func_wrapper(*args, **kwargs):
            return do_stuff(func(*args, **kwargs), argument)
        
        return func_wrapper

    return actual_decorator


@decorator_with_argument('blah')
def example_1():
    return 1  # returns do_stuff(1, 'blah')

# note the brackets!
@decorator_with_argument()
def example_2():
    return 1  # returns do_stuff(1, None)