Должен ли я использовать класс дескриптора или фабрику свойств?

Скажем, у меня есть класс, и я хочу создать много похожих атрибутов "свойства-стиля". Например:

class Foo(object):
    @property
    def foo(self):
        return self._foo

    @foo.setter
    def foo(self, val):
        self.update()
        self._foo = val

    @property
    def bar(self):
        return self._bar

    @bar.setter
    def bar(self, val):
        self.update()
        self._bar = val

    @property
    def baz(self):
        return self._baz

    @baz.setter
    def baz(self, val):
        self.update()
        self._baz = val

    def update(self):
        print "Updating..."

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

def create_property(var):
    def _setter(self, val):
        self.update()
        setattr(self, '_' + var, val)

    def _getter(self):
        return getattr(self, '_' + var)

    return property(_getter, _setter)

class Foo(object):
    foo = create_property("foo")
    bar = create_property("bar")
    baz = create_property("baz")

    def update(self):
        print "Updating..."

В качестве альтернативы я мог бы написать класс дескриптора:

class UpdateOnChange(object):
    def __init__(self):
        self.data = WeakKeyDictionary()

    def __get__(self, instance, owner):
        try:
            return self.data[instance]
        except KeyError:
            raise AttributeError("attribute not set")

    def __set__(self, instance, value):
        instance.update()
        self.data[instance] = value

class Foo(object):
    foo = UpdateOnChange()
    bar = UpdateOnChange()
    baz = UpdateOnChange()

    def update(self):
        print "Updating"

Какой лучше / самый быстрый / самый Pythonic?

1 ответ

Взяв идею дескриптора и смешав ее с метаклассом, как я упоминал в комментариях.

class UpdateOnChange(object):
    def __init__(self, name):
        self.name = name
        self._name = '_' + name

    def __get__(self, instance, owner):
        try:
            return getattr(instance, self._name)
        except AttributeError:
            raise AttributeError('%s has no attribute %s.' % (instance, self.name))

    def __set__(self, instance, value):
        # Assume first assignment is initialization and does not require update.
        if not hasattr(instance, self._name):
            setattr(instance, self._name, value)
        else:
            instance.update()
            setattr(instance, self._name, value)

class UpdateOnChangeMeta(type):

     def __new__(cls, name, bases, attrs):
         for attr_name in attrs:
             if attrs[attr_name] == UpdateOnChange:
                 attrs[attr_name] = UpdateOnChange(attr_name)
         return type.__new__(cls, name, bases, attrs)

class Foo(object):

    __metaclass__ = UpdateOnChangeMeta
    foo = UpdateOnChange
    bar = UpdateOnChange
    baz = UpdateOnChange

    def update(self):
        print "Updating"
Другие вопросы по тегам