A converter for turning a function decorator into a method decorator

function_decorator_to_method_decorator.py

import functools


def make_method_decorator(decorator):
    # This function returns a version of @decorator fit for wrapping a method

    def method_wrapper(method):
        @functools.wraps(method)
        def inner_method(self, *args, **kwargs):
            @decorator
            def inner_func(*inner_args, **inner_kwargs):
                return method(self, *inner_args, **inner_kwargs)

            return inner_func(*args, **kwargs)

        return inner_method
    return method_wrapper