装饰类以监视属性更改

时间:2013-09-15 12:26:36

标签: python decorator python-decorators

我想让每当其中一个属性发生变化时自动向订阅者发送通知的类。所以如果我要写这段代码:

@ChangeMonitor
class ChangingClass(object):

    def __init__(self, x):
        self.x = x


changer = ChangingClass(5)
print("Going to change x.")
changer.x = 6
print("Going to not change x.")
changer.x = 6
print("End of program")

输出结果为:

Going to change x
Old x = 5, new x = 6
Going to not change x.
End of program.

我的问题是如何实现ChangeMonitor装饰器类。在上面的示例中,我假设它将打印一行指示属性的更改,但出于有用的目的,它可以向订阅的对象发送通知。

1 个答案:

答案 0 :(得分:3)

您必须添加__setattr__() method

def ChangeMonitor(cls):
    _sentinel = object()
    old_setattr = getattr(cls, '__setattr__', None)
    def __setattr__(self, name, value):
        old = getattr(self, name, _sentinel)
        if old not is _sentinel and old != value:
            print "Old {0} = {1!r}, new {0} = {2!r}".format(name, old, value)
        if old_setattr:
            old_setattr(self, name, value)
        else:
            # Old-style class
            self.__dict__[name] = value

    cls.__setattr__ = __setattr__

    return cls

这也应该处理现有的__setattr__挂钩。 _sentinel用于允许None作为旧值。

演示:

>>> changer = ChangingClass(5)
>>> changer.x = 6
Old x = 5, new x = 6
>>> changer.x = 6
>>> # nothing printed
...
>>> changer.x = None
Old x = 6, new x = None
>>> changer.x = 6
Old x = None, new x = 6