blob: 5706777c41db482fb2532ee53250c4ddd810215f [file] [log] [blame]
Stéphane Wirtelcbb64842019-05-17 11:55:34 +02001.. highlight:: c
Georg Brandl54a3faa2008-01-20 09:30:57 +00002
3.. _iterator:
4
5Iterator Protocol
6=================
7
Raymond Hettinger8ee77082013-10-09 22:42:46 -07008There are two functions specifically for working with iterators.
Georg Brandl54a3faa2008-01-20 09:30:57 +00009
Georg Brandl60203b42010-10-06 10:11:56 +000010.. c:function:: int PyIter_Check(PyObject *o)
Georg Brandl54a3faa2008-01-20 09:30:57 +000011
Erlend Egeberg Aaslandcc540012021-02-16 16:05:58 +010012 Return non-zero if the object *o* supports the iterator protocol, and ``0``
13 otherwise. This function always succeeds.
Georg Brandl54a3faa2008-01-20 09:30:57 +000014
15
Georg Brandl60203b42010-10-06 10:11:56 +000016.. c:function:: PyObject* PyIter_Next(PyObject *o)
Georg Brandl54a3faa2008-01-20 09:30:57 +000017
Raymond Hettinger8ee77082013-10-09 22:42:46 -070018 Return the next value from the iteration *o*. The object must be an iterator
19 (it is up to the caller to check this). If there are no remaining values,
Serhiy Storchaka25fc0882019-10-30 12:03:20 +020020 returns ``NULL`` with no exception set. If an error occurs while retrieving
21 the item, returns ``NULL`` and passes along the exception.
Georg Brandl54a3faa2008-01-20 09:30:57 +000022
23To write a loop which iterates over an iterator, the C code should look
24something like this::
25
26 PyObject *iterator = PyObject_GetIter(obj);
27 PyObject *item;
28
29 if (iterator == NULL) {
30 /* propagate error */
31 }
32
William Ayd5c7ed752019-12-24 23:25:56 -050033 while ((item = PyIter_Next(iterator))) {
Georg Brandl54a3faa2008-01-20 09:30:57 +000034 /* do something with item */
35 ...
36 /* release reference when done */
37 Py_DECREF(item);
38 }
39
40 Py_DECREF(iterator);
41
42 if (PyErr_Occurred()) {
43 /* propagate error */
44 }
45 else {
46 /* continue doing useful work */
47 }
Vladimir Matveev037245c2020-10-09 17:15:15 -070048
49
50.. c:type:: PySendResult
51
52 The enum value used to represent different results of :c:func:`PyIter_Send`.
53
Vladimir Matveevcfb0f572020-10-13 10:26:51 -070054 .. versionadded:: 3.10
55
Vladimir Matveev037245c2020-10-09 17:15:15 -070056
57.. c:function:: PySendResult PyIter_Send(PyObject *iter, PyObject *arg, PyObject **presult)
58
59 Sends the *arg* value into the iterator *iter*. Returns:
60
61 - ``PYGEN_RETURN`` if iterator returns. Return value is returned via *presult*.
62 - ``PYGEN_NEXT`` if iterator yields. Yielded value is returned via *presult*.
63 - ``PYGEN_ERROR`` if iterator has raised and exception. *presult* is set to ``NULL``.
Vladimir Matveevcfb0f572020-10-13 10:26:51 -070064
65 .. versionadded:: 3.10