意外的对象分配

class TrafficData(object):
    def __init__(self):
        self.__data = {}
    def __getitem__(self, epoch):
        if not isinstance(epoch, int):
            raise TypeError()
        return self.__data.setdefault(epoch, ProcessTraffic())
    def __iadd__(self, other):
        for epoch, traffic in other.iteritems():

            # these work
            #existing = self[epoch]
            #existing += traffic

            # this does not
            self[epoch] += traffic # here the exception is thrown

        return self

在上面修剪过的代码中,我不希望进行项分配,但显然有一个项分配发生在标记行上,并引发以下异常:

  File "nethogs2.py", line 130, in __iadd__
    self[epoch] += traffic
TypeError: 'TrafficData' object does not support item assignment

但是,如果我使用前面2行注释掉的行,则不会引发异常。

如我所见,这两个应该以同样的方式表现。 self[epoch] 返回对对象的引用,并通过该对象在适当位置进行了修改 __iadd__ . 我在这里误解了什么?我经常在使用字典时遇到这个问题。

更新0

可能值得指出的是 self.__data 有 阿伊达德 已定义,但未定义 __add__ 如果可能的话,我更愿意在适当的地方修改这个值。我还想避免创建 __setitem__ 方法。

更新1

下面是一个演示这个问题的测试用例,我将上面的代码留给现有的答案。

class Value(object):
    def __init__(self, initial=0):
        self.a = initial
    def __iadd__(self, other):
        self.a += other
        return self
    def __str__(self):
        return str(self.a)

class Blah(object):
    def __init__(self):
        self.__data = {}
    def __getitem__(self, key):
        return self.__data.setdefault(key, Value())

a = Blah()
b = a[1]
b += 1
print a[1]
a[1] += 2
print a[1]