在映射类型的python中重载+ =

时间:2012-04-05 03:46:10

标签: python dictionary operator-overloading

我想在Python中使用+ =表示法来更新类似dict的对象。我想要与dict.update方法具有相同的行为。这是我的班级(带有“。”访问权限的词典):

class sdict(dict):
    def __getattr__(self, attr):
        return self.get(attr, None)
    __setattr__= dict.__setitem__
    __delattr__= dict.__delitem__

我试过了:

__iadd__ = dict.update

def __iadd__(self, other):
    self.update(other)
    return self

但这些都不起作用。 (第一个销毁原始字典,第二个生成SyntaxError)

更新:
第二个定义确实有效。它对我不起作用,因为我忘了def。第一个不起作用,因为dict.update返回None。

1 个答案:

答案 0 :(得分:12)

我认为你所缺少的只是一个def:

class sdict(dict):
    def __getattr__(self, attr):
        return self.get(attr, None)
    __setattr__= dict.__setitem__
    __delattr__= dict.__delitem__
    def __iadd__(self, other):
        self.update(other)
        return self

>>> a = sdict()
>>> a.b = 3
>>> a
{'b': 3}
>>> a.b
3
>>> a['b']
3
>>> a += {'fred': 3}
>>> a
{'b': 3, 'fred': 3}