在 Python 2.7 中定义类似“属性”的装饰器

Question

我正在尝试编写一个类似于@property 的装饰器，但遇到了一些问题。

class Dec(object):
  def __init__(self, fn):
    self._fn = fn
    self._before = None
  @property
  def before(self)
    return self._before
  @before.setter
  def before(self, fn):
    self._before = fn
  def __call__(self, *args, **kwargs):
    self._before(*args, **kwargs)
    self._fn(*args, **kwargs)

def withbefore(fn):
  return Dec(fn)

它是一个简单的链式装饰器。 @property/@.setter 语法正是我想要克隆的。

这行得通：

@withbefore
def foo():
   ...
@foo.before
def beforefoo():
  ...

但在课堂上却没有：

class Weee(object):
    @withbefore
    def do_stuff(self):
      pass
    @do_stuff.before
    def before_do_stuff(self):
      pass

它会引发导入错误。

TypeError: 'NoneType' object is not callable

如何正确模拟@property/.{setter,getter,deleter}？

Answer 1

真诚地，我认为你最好：

from functools import wraps

def withbefore(fn):
    def dec(bef):
        fn._before_fn = bef
        return bef

    @wraps(fn)
    def _wrapper(*args, **kwargs):
        fn._before_fn(*args, **kwargs)
        return fn(*args, **kwargs)

    _wrapper.before = dec
    return _wrapper

它更紧凑，更 Pythonic，应该适用于所有情况。

Answer 2

实际上，它引发了一个 TypeError。

无论如何，我在使用函数运行装饰器时也遇到了同样的错误。发生这种情况是因为，当您使用 @foo.before 装饰函数时，它将以装饰函数作为参数调用 self._before 函数。由于self._before 是None，所以会报错。

有多种解决方案。我最喜欢的是将不同的默认值设置为 self._before - 一个将设置 self._before 值的函数：

class Dec(object):
  def __init__(self, fn):
    self._fn = fn
    def setbefore(b):
        self._before = b
    self._before = self.default_before = setbefore

当然，这个函数不应该在调用Dec对象的时候调用所以我们改一下__call__方法：

  def __call__(self, *args, **kwargs):
      if self._before != self.default_before:
         self._before(*args, **kwargs)
      self._fn(*args, **kwargs)