用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作為第一個參數傳遞，但此參數已由裝飾器刪除。