问题:构建了一个自定义的容器对象,其内部持有列表、元组或其他可迭代对象。我们想让自己的新容器能够完成迭代操作。
解决方案:一般来说,我们所要做的就是定义一个__iter__()
方法,将迭代请求委托到对象内部持有的容器上
class Node:
def __init__(self, value):
self._value = value
self._children = []
def __repr__(self):
return 'Node({!r})'.format(self._value)
def add_children(self, node):
self._children.append(node)
def __iter__(self):
""" 将类实例中的迭代请求,转发给对象内部持有的_children属性上(列表) """
return iter(self._children)
# 示例
root = Node(0)
child1 = Node(1)
child2 = Node(2)
root.add_children(child1)
root.add_children(child2)
# 1、for会请求到__iter__()方法;2、迭代实例root, 实际上是在迭代root._children属性
for ch in root:
print(ch)
Outs:
Node(1)
Node(2)
总结:
1、Python的迭代协议要求__iter__()
返回一个特殊的迭代器对象,由该对象实现的__next__()
方法来完成实际的迭代;
2、如果要做的只是迭代另一个容器中的内容,所要做的就是转发迭代请求。