python - 在 Python 中迭代给定类的对象实例

Question

给定一个保存其对象注册表的类:

class Person(object):
   __registry = []

   def __init__(self, name):
       self.__registry.append(self)
       self.name = name

如何使以下代码工作(不使用 Person.__registry):

for personobject in Person:
    print personobject

在研究过程中，我发现了一个提示，可以使用 __getitem__ 方法来使用 __metaclass__。有什么想法吗？

Answer 1

您可以使用简单的元类使您的类对象可迭代。

class IterRegistry(type):
    def __iter__(cls):
        return iter(cls._registry)

class Person(object):
    __metaclass__ = IterRegistry
    _registry = []

    def __init__(self, name):
        self._registry.append(self)
        self.name = name

(我还将 __registry 更改为 _registry 以便更容易从元类访问)。那么，

>>> p = Person('John')
>>> p2 = Person('Mary')
>>> for personobject in Person:
...     print personobject
...
<person.Person object at 0x70410>
<person.Person object at 0x70250>