迭代器协议

迭代器有两个函数。

int PyIter_Check(PyObject \o*)

如果对象 o 支持迭代器协议则返回真值。 此函数总是会成功执行。

PyObject* PyIter_Next(PyObject \o*)

Return value: New reference.

返回迭代 o 的下一个值。 对象必须是一个迭代器(这应由调用者来判断)。 如果没有余下的值,则返回 NULL 并且不设置异常。 如果在获取条目时发生了错误,则返回 NULL 并且传递异常。

要为迭代器编写一个一个循环,C代码应该看起来像这样

  1. PyObject *iterator = PyObject_GetIter(obj);
  2. PyObject *item;
  3. if (iterator == NULL) {
  4. /* propagate error */
  5. }
  6. while ((item = PyIter_Next(iterator))) {
  7. /* do something with item */
  8. ...
  9. /* release reference when done */
  10. Py_DECREF(item);
  11. }
  12. Py_DECREF(iterator);
  13. if (PyErr_Occurred()) {
  14. /* propagate error */
  15. }
  16. else {
  17. /* continue doing useful work */
  18. }