Декорирование класса для отслеживания изменений атрибутов

1

Я хочу иметь классы, которые автоматически отправляют уведомления подписчикам всякий раз, когда изменяется один из их атрибутов. Поэтому, если бы я написал этот код:

@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. В приведенном выше примере я предполагаю, что он напечатает строку, указывающую изменения атрибута, но в полезных целях он может отправлять уведомления подписанным объектам.

Теги:
decorator
python-decorators

1 ответ

3
Лучший ответ

Вам нужно добавить метод __setattr__():

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
  • 0
    Я не вижу использования _sentinel или второго оператора if. Если я опущу _sentinel и заменим выражение «if old_setattr» на «old_setattr (self, name, value)», поведение будет таким же. Не могли бы вы объяснить это? В любом случае, спасибо за вашу помощь, вы мне очень помогли!
  • 0
    Если вы замените часовой на None вы не сможете определить, было ли старое значение None заменено новым. Вместо этого он будет неотличим от случая «вообще нет старого значения». Я полагаю, что классы старого стиля (не наследующие от object ) не __setattr__ хука по умолчанию __setattr__ ; Я не в состоянии проверить это прямо сейчас; это может быть излишним, чтобы проверить, был ли найден крюк
Показать ещё 1 комментарий

Ещё вопросы

Сообщество Overcoder
Наверх
Меню