类方法和静态方法的python装饰器？

Question

我正在尝试为一个实例方法编写一个装饰器，如下所示：

from functools import wraps

def plus_decorator(f):
    @wraps(f)
    def wrapper(*args, **kwargs):
        return 1 + f(*args, **kwargs)
    return wrapper

@plus_decorator
def return_i(i):
    return i

class A(object):
    @plus_decorator
    def return_i(self, i):
        return i

    @plus_decorator
    @classmethod
    def return_i_class(cls, i):
        return i

    @plus_decorator
    @staticmethod
    def return_i_static(i):
        return i

if __name__ == '__main__':
    print return_i(1)
    a = A()
    print a.return_i(1)
    print A.return_i_class(1)
    print A.return_i_static(1)

但是，它弹出错误：

AttributeError: 'classmethod' object has no attribute '__module__'

我想知道为什么装饰器不能在classmethod和 上工作staticmethod。我认为装饰器大多只是将它接收到的所有参数传递给包装器，并且只修改结果。如何修改装饰器以使其适用于classmethod和staticmethod？

Answer 1

只需翻转订单。把@classmethodor@staticmethod放在外面，你的装饰器（它使用@wraps，因此需要一个裸函数）在里面。

@classmethod
@plus_decorator
def return_i_class(cls, i):
    return i