Python：从另一个类调用装饰器方法

Question

假设我有一个定义装饰器的类

class Foo():

    def decorate(param):        
        def wrapper(func):
            def wrapped(*args, **kwargs):
                # do something
                return func(*args, **kwargs)
            return wrapped
        return wrapper

然后我决定在另一个类中调用该装饰器，如下所示：

class Bar():

    foo = Foo()
    param = something

    @foo.decorate(param)
    def func():
        # do something

这会引发TypeError 异常：

TypeError: decorate() takes 1 positional argument but 2 were given

我无法理解这里出了什么问题。应用相同的装饰器但在同一类中定义 Bar 可以正常工作，但为什么它不适用于不同的类？

Answer 1

你的 decorate() 函数应该是一个实例方法，因此它接受实例本身作为第一个参数，当你调用 - @foo.decorate(param) 时，你试图传递两个参数（因为 foo 是一个实例Foo 类）实例本身，以及导致问题的 param 。

您应该将decorate() 方法转换为接受实例作为第一个参数。示例 -

class Foo():

    def decorate(self, param):        
        def wrapper(func):
            def wrapped(*args, **kwargs):
                # do something
                return func(*args, **kwargs)
            return wrapped
        return wrapper

---

另外，你的 func() 也是一个实例方法，它也需要一个 self 。示例 -

class Bar():

    foo = Foo()
    param = something

    @foo.decorate(param)
    def func(self):
        # do something

Answer 2

您只是忘记包含self；你正在调用Foo().decorate()，一个绑定的方法，所以实例会被传入：

class Foo():
    def decorate(self, param):        

请注意，这同样适用于您的 Bar.func() 函数：

class Bar():

    foo = Foo()
    param = something

    @foo.decorate(param)
    def func(self):
        # do something