迭代器协议

迭代器有两个函数。

int PyIter_Check(PyObject *o)

返回 true , 如果对象 o 支持迭代器协议的话。

PyObject PyIter_Next(PyObject **o)

Return value: New reference.

Return the next value from the iteration o. The object must be an iterator (it is up to the caller to check this). If there are no remaining values, returns NULL with no exception set. If an error occurs while retrieving the item, returns NULL and passes along the exception.

要为迭代器编写一个一个循环,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. }