迭代器协议

2.2 版中的新功能。

有两个函数专门用于处理迭代器。

int PyIter_Check(PyObject *o)

如果对象 o 支持迭代器协议，则返回 true。

在旧式类的情况下，此函数可能会返回误报，因为这些类总是定义一个 tp_iternext 槽，其中包含调用 next() 方法或引发 TypeError。

PyObject *PyIter_Next(PyObject *o)

返回迭代 o 的下一个值。该对象必须是一个迭代器（由调用者来检查）。如果没有剩余值，则返回 NULL 且未设置异常。如果检索项目时发生错误，则返回 NULL 并传递异常。

要编写一个遍历迭代器的循环，C 代码应如下所示：

PyObject *iterator = PyObject_GetIter(obj);
PyObject *item;

if (iterator == NULL) {
    /* propagate error */
}

while ((item = PyIter_Next(iterator))) {
    /* do something with item */
    ...
    /* release reference when done */
    Py_DECREF(item);
}

Py_DECREF(iterator);

if (PyErr_Occurred()) {
    /* propagate error */
}
else {
    /* continue doing useful work */
}

迭代器协议 — Python 文档

迭代器协议