blob: 68df6f6e89f5132d510d9a06af4ccfac125b7d4f [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
12 Return true if the object *o* supports the iterator protocol.
13
14
Georg Brandl60203b42010-10-06 10:11:56 +000015.. c:function:: PyObject* PyIter_Next(PyObject *o)
Georg Brandl54a3faa2008-01-20 09:30:57 +000016
Raymond Hettinger8ee77082013-10-09 22:42:46 -070017 Return the next value from the iteration *o*. The object must be an iterator
18 (it is up to the caller to check this). If there are no remaining values,
Serhiy Storchaka25fc0882019-10-30 12:03:20 +020019 returns ``NULL`` with no exception set. If an error occurs while retrieving
20 the item, returns ``NULL`` and passes along the exception.
Georg Brandl54a3faa2008-01-20 09:30:57 +000021
22To write a loop which iterates over an iterator, the C code should look
23something like this::
24
25 PyObject *iterator = PyObject_GetIter(obj);
26 PyObject *item;
27
28 if (iterator == NULL) {
29 /* propagate error */
30 }
31
William Ayd5c7ed752019-12-24 23:25:56 -050032 while ((item = PyIter_Next(iterator))) {
Georg Brandl54a3faa2008-01-20 09:30:57 +000033 /* do something with item */
34 ...
35 /* release reference when done */
36 Py_DECREF(item);
37 }
38
39 Py_DECREF(iterator);
40
41 if (PyErr_Occurred()) {
42 /* propagate error */
43 }
44 else {
45 /* continue doing useful work */
46 }
Vladimir Matveev037245c2020-10-09 17:15:15 -070047
48
49.. c:type:: PySendResult
50
51 The enum value used to represent different results of :c:func:`PyIter_Send`.
52
Vladimir Matveevcfb0f572020-10-13 10:26:51 -070053 .. versionadded:: 3.10
54
Vladimir Matveev037245c2020-10-09 17:15:15 -070055
56.. c:function:: PySendResult PyIter_Send(PyObject *iter, PyObject *arg, PyObject **presult)
57
58 Sends the *arg* value into the iterator *iter*. Returns:
59
60 - ``PYGEN_RETURN`` if iterator returns. Return value is returned via *presult*.
61 - ``PYGEN_NEXT`` if iterator yields. Yielded value is returned via *presult*.
62 - ``PYGEN_ERROR`` if iterator has raised and exception. *presult* is set to ``NULL``.
Vladimir Matveevcfb0f572020-10-13 10:26:51 -070063
64 .. versionadded:: 3.10