代理迭代

问题

你构建了一个自定义容器对象，里面包含有列表、元组或其他可迭代对象。你想直接在你的这个新容器对象上执行迭代操作。

解决方案

实际上你只需要定义一个 iter () 方法，将迭代操作代理到容器内部的对象上去。比如：

class Node:
	def __init__(self, value):
		self._value = value
		self._children = []
		
	def __repr__(self):
	return 'Node({!r})'.format(self._value)
	def add_child(self, node):
		self._children.append(node)
	def __iter__(self):
		return iter(self._children)
		
	# Example
	if __name__ == '__main__':
		root = Node(0)
		child1 = Node(1)
		child2 = Node(2)
		root.add_child(child1)
		root.add_child(child2)
		# Outputs Node(1), Node(2)
		for ch in root:
			print(ch)

在上面代码中， iter () 方法只是简单的将迭代请求传递给内部的 children属性。

讨论

Python 的迭代器协议需要 iter () 方法返回一个实现了 next () 方法的迭代
器对象。如果你只是迭代遍历其他容器的内容，你无须担心底层是怎样实现的。你所
要做的只是传递迭代请求既可。

这 里 的 iter() 函 数 的 使 用 简 化 了 代 码， iter(s) 只 是 简 单 的 通 过 调 用
s. iter () 方法来返回对应的迭代器对象，就跟 len(s) 会调用 s. len () 原理
是一样的。