69

这是 Python 2.5，也是GAE，这并不重要。

我有以下代码。我正在装饰 bar 中的 foo() 方法，使用dec_check该类作为装饰器。

class dec_check(object):

  def __init__(self, f):
    self.func = f

  def __call__(self):
    print 'In dec_check.__init__()'
    self.func()

class bar(object):

  @dec_check
  def foo(self):
    print 'In bar.foo()'

b = bar()
b.foo()

执行此操作时，我希望看到：

In dec_check.__init__()
In bar.foo()

但我得到TypeError: foo() takes exactly 1 argument (0 given)as .foo()，作为一个对象方法，self作为一个参数。我猜问题是bar当我执行装饰器代码时，实例实际上并不存在。

那么如何将实例传递bar给装饰器类呢？

4

3 回答 3

94

你需要把装饰器变成一个描述符——或者通过确保它的（元）类有一个__get__方法，或者更简单地，通过使用装饰器函数而不是装饰器类（因为函数已经是描述符）。例如：

def dec_check(f):
  def deco(self):
    print 'In deco'
    f(self)
  return deco

class bar(object):
  @dec_check
  def foo(self):
    print 'in bar.foo'

b = bar()
b.foo()

这打印

In deco
in bar.foo

如预期的。

于 2010-03-02T18:47:35.367 回答
53

当功能足够时，Alex 的答案就足够了。但是，当您需要一个类时，您可以通过将以下方法添加到装饰器类来使其工作。

def __get__(self, obj, objtype):
    """Support instance methods."""
    import functools
    return functools.partial(self.__call__, obj)

要理解这一点，您需要了解描述符协议。描述符协议是将事物绑定到实例的机制。它由__get__、__set__和组成__delete__，当从实例字典中获取、设置或删除事物时会调用它们。

在这种情况下，当事物从实例中获取时，我们将其__call__方法的第一个参数绑定到实例，使用部分。这在构造类时对成员函数自动完成，但对于像这样的合成成员函数，我们需要显式地完成。

于 2010-07-21T04:28:13.913 回答
6

如果你想把装饰器写成一个类，你可以这样做：

from functools import update_wrapper, partial

class MyDecorator(object):
    def __init__(self, func):
        update_wrapper(self, func)
        self.func = func

    def __get__(self, obj, objtype):
        """Support instance methods."""
        return partial(self.__call__, obj)

    def __call__(self, obj, *args, **kwargs):
        print('Logic here')
        return self.func(obj, *args, **kwargs)

my_decorator = MyDecorator

class MyClass(object):
     @my_decorator
     def my_method(self):
         pass
于 2017-07-27T21:57:04.100 回答