用类装饰器装饰类方法

Question

我正在尝试使用类装饰器来装饰类方法。例如

class MyDecorator(object):
    def __init__(self,param1):
        self.param1 = param1
        # do some action with param1

    def __call__(self,func):
        def wrapped(*args,**kwargs):
             print "in wrapper:"
             func(*args,**kwargs)

        return wrapped

还有我的一些班级：

class A:
    @MyDecorator("Blablabla")
    def func1(arg1,arg2,arg3):
         print (arg1,arg2,arg3)

但是当我做下一个动作时：

a = A()
a.func(1,2,3)

我收到以下错误：

TypeError: func1() takes exactly 3 arguments (4 given)

Answer 1

class A:
    @MyDecorator("Blablabla")
    def func1(self, arg1, arg2, arg3):
        print (arg1,arg2,arg3)

您需要将 self 参数添加到您的函数中。

Answer 2

如果您更喜欢在没有第一个 self 参数的情况下编写 func1 方法，则需要在装饰器中删除此参数：

def __call__(self, func):
    def wrapped(obj, *args, **kwargs):
         # obj is just ignored
         print "in wrapper:"
         func(*args, **kwargs)

这与使用@staticmethod 完全一样：调用a.func1 时，a 作为第一个参数传递，但此参数被装饰器删除。