Python decorator (class method)

decorator_method.py

from functools import wraps


def method_decorator(method):

    @wraps(method)
    def method_wrapper(self, *args, **kwargs):
        return do_stuff(method(self, *args, **kwargs))

    return method_wrapper


class Example:
    
    @method_decorator
    def example_method(self):
        return 2  # returns do_stuff(2)