将 python 的 dict 功能扩展到用户创建的类

Question

是否可以向用户创建的类添加字典功能？

IE：

class Foo(object):
    def __init__(self, x, y):
       self.x
       self.y
    def __dict__(self):
       return {'x': self.x, 'y': self.y}

f = Foo()
dict(f) <-- throws TypeError: iteration over non-sequence

Answer 1

dict 构造函数需要映射或可迭代的键/值对作为参数，因此您的类需要实现映射协议或可迭代。

下面是如何使用后一种方法的示例：

class Foo(object):
    def __init__(self, x, y):
       self.x = x
       self.y = y
    def __iter__(self):
       return vars(self).iteritems()

示例用法：

>>> dict(Foo(2, 3))
{'x': 2, 'y': 3}

不过，我不知道这有多有用。您可以只执行

>>> vars(Foo(2, 3))
{'x': 2, 'y': 3}

以下操作，而无需在您的类上实现 __iter__() 。