Python decorator with function registration

decorator_with_function_registration.py

def make_registrar():
    registry = {}

    def registrar(func):
        registry[func.__name__] = func

        def wrapper(*args, **kwargs):
            # Note - Modify function here
            return func(*args, **kwargs)
        return wrapper

    registrar.all = registry
    return registrar


reg = make_registrar()


@reg
def f1(a):
    return a


@reg
def f2(a, b):
    return a+b


print(reg.all)
# >>> {'f2': <function f2 at ...>, 'f1': <function f1 at ...>}