检测迭代器是否会被消耗

是否有一种统一的方法来知道迭代是否会消耗可迭代对象？

假设你有一个特定的函数crunch它要求一个可迭代对象作为参数，并多次使用它。就像是：

def crunch (vals):

    for v in vals:
        chomp(v)

    for v in vals:
        yum(v)

（注：将两者合并在一起for循环不是一个选项）。

如果使用不是列表的可迭代对象调用函数，就会出现问题。在下面的调用中yum函数永远不会被执行：

crunch(iter(range(4))

原则上我们可以通过重新定义来解决这个问题crunch函数如下：

def crunch (vals):
    vals = list(vals)

    for v in vals:
        chomp(v)

    for v in vals:
        yum(v)

但这会导致使用两倍的内存，如果调用crunch is:

hugeList = list(longDataStream)
crunch(hugeList)

我们可以通过定义来解决这个问题crunch像这样：

def crunch (vals):
    if type(vals) is not list:
        vals = list(vals)

    for v in vals:
        chomp(v)

    for v in vals:
        yum(v)

但仍然可能存在调用代码将数据存储在以下位置的情况：

• 不能食用
• 不是一个列表

例如：

from collections import deque
hugeDeque = deque(longDataStream)
crunch(hugeDeque)

如果有一个isconsumable谓词，以便我们可以定义crunch像这样：

def crunch (vals):
    if isconsumable(vals):
        vals = list(vals)

    for v in vals:
        chomp(v)

    for v in vals:
        yum(v)

这个问题有解决办法吗？

一种可能性是测试该项目是否是序列，使用isinstance(val, collections.Sequence)。仍然不能完全保证不可消耗性，但我认为这已经是你能得到的最好的了。 Python 序列必须有一个长度，这意味着它至少不能是一个开放式迭代器，并且通常意味着必须提前知道元素，这又意味着它们可以被迭代而不消耗它们。仍然可以编写符合序列协议但不可重复的病态类，但您永远无法处理这些类。

请注意，两者都不是Iterable nor Iterator是适当的选择，因为这些类型不保证长度，因此不能保证迭代是有限的，更不用说可重复了。但是，您可以检查两者Sized and Iterable.

重要的是记录您的函数将对其参数进行两次迭代，从而警告用户必须传入支持此功能的对象。