如何爲Python屬性實現__iadd__

Question

我試圖創建一個Python屬性，其中就地添加是通過不同的方法處理而不是檢索值，添加另一個值並重新分配。因此，對於一個對象o上的屬性x，

o.x += 5 

應該工作不同於

o.x = o.x + 5 

的o.x值應該是在年底相同，以免混淆人們的預期，但我想使就地添加更高效。 （在現實中，操作時間比簡單的加法更多的時間。）

我的第一個想法是定義在類，

x = property(etc. etc.) 
x.__iadd__ = my_iadd 

但由於property實現__slots__這引發了一個AttributeError，想必？

我的下一次嘗試使用一個描述符對象：

class IAddProp(object): 
    def __init__(self): 
     self._val = 5 

    def __get__(self, obj, type=None): 
     return self._val 

    def __set__(self, obj, value): 
     self._val = value 

    def __iadd__(self, value): 
     print '__iadd__!' 
     self._val += value 
     return self 

class TestObj(object): 
    x = IAddProp() 
    #x.__iadd__ = IAddProp.__iadd__ # doesn't help 

>>> o = TestObj() 
>>> print o.x 
5 
>>> o.x = 10 
>>> print o.x 
10 
>>> o.x += 5 # '__iadd__!' not printed 
>>> print o.x 
15 

正如你所看到的，特殊__iadd__方法不叫。我無法理解這是爲什麼，儘管我猜測對象的__getattr__在某種程度上繞過了它。

我該怎麼做？我沒有得到描述符的重點嗎？我需要一個元類嗎？

Answer 1

__iadd__只會在從__get__返回的值上查找。您需要使__get__（或屬性獲取器）返回一個包含__iadd__的對象（或代理對象）。

@property 
def x(self): 
    proxy = IProxy(self._x) 
    proxy.parent = self 
    return proxy 

class IProxy(int, object): 
    def __iadd__(self, val): 
     self.parent.iadd_x(val) 
     return self.parent.x 

Answer 2

激勵你，這裏有一個低於理想這是最好的我已經成功地拿出這麼遠的解決方案：

class IAddObj(object): 
    def __init__(self): 
     self._val = 5 

    def __str__(self): 
     return str(self._val) 

    def __iadd__(self, value): 
     print '__iadd__!' 
     self._val += value 
     return self._val 

class TestObj2(object): 
    def __init__(self): 
     self._x = IAddObj() 

    @property 
    def x(self): 
     return self._x 

    @x.setter 
    def x(self, value): 
     self._x._val = value 

>>> o = TestObj2() 
>>> print o.x 
5 
>>> o.x = 10 
>>> print o.x 
10 
>>> o.x += 5 
__iadd__! 
>>> print o.x 
15 
>>> print o.x + 5 # doesn't work unless __add__ is also implemented 
TypeError: unsupported operand type(s) for +: 'IAddObj' and 'int' 

最大的缺點是，你必須實現全如果您希望屬性的行爲類似於數字，則可以使用IAddObj上的數字魔術方法的補充。擁有IAddObj繼承自int似乎也不讓它繼承運營商，無論是。

Answer 3

爲什麼不能像下面的例子。基本上這個想法是讓Bar類確保屬性x的存儲值始終是一個Foo對象。

class Foo(object): 
    def __init__(self, val=0): 
     print 'init' 
     self._val = val 

    def __add__(self, x): 
     print 'add' 
     return Foo(self._val + x) 

    def __iadd__(self, x): 
     print 'iadd' 
     self._val += x 
     return self 

    def __unicode__(self): 
     return unicode(self._val) 

class Bar(object): 
    def __init__(self): 
     self._x = Foo() 

    def getx(self): 
     print 'getx' 
     return self._x 

    def setx(self, val): 
     if not isinstance(val, Foo): 
      val = Foo(val) 
     print 'setx' 
     self._x = val 

    x = property(getx, setx) 

obj = Bar() 
print unicode(obj.x) 
obj.x += 5 
obj.x = obj.x + 6 
print unicode(obj.x) 

編輯：擴展示例以顯示如何使用它作爲屬性。我首先誤解了這個問題。

Answer 4

在線路的+=操作者

o.x += 5 

被轉換爲

o.x = o.x.__iadd__(5) 

在右手側上的屬性查找被轉換爲

o.x = IAddProp.__get__(TestObj2.x, o, TestObj2).__iadd__(5) 

作爲你可以n請參閱__iadd__()在返回值上被調用的屬性查找，因此您需要在返回的對象上實現__iadd__()。