用class-decorator装饰class-method

Question

我正在尝试使用class-decorator装饰类方法。 例如

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作为第一个参数传递，但此参数已由装饰器删除。