diff options
| author | Raymond Hettinger <python@rcn.com> | 2004-05-08 19:49:42 +0000 |
|---|---|---|
| committer | Raymond Hettinger <python@rcn.com> | 2004-05-08 19:49:42 +0000 |
| commit | 9d7c870c6d66c8b2e066001f31ac3037d50f7012 (patch) | |
| tree | c41da1755bdc2af527581c02cf2fd2ed9781ac38 /Modules/itertoolsmodule.c | |
| parent | ffa5a5015a359045d7efb2c4cd0b5738c97b86ce (diff) | |
| download | cpython-git-9d7c870c6d66c8b2e066001f31ac3037d50f7012.tar.gz | |
SF #950057: itertools.chain doesn't "process" exceptions as they occur
Both cycle() and chain() were handling exceptions only when switching
input sources. The patch makes the handle more immediate.
Will backport.
Diffstat (limited to 'Modules/itertoolsmodule.c')
| -rw-r--r-- | Modules/itertoolsmodule.c | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/Modules/itertoolsmodule.c b/Modules/itertoolsmodule.c index 4ce46430df..3515bc6058 100644 --- a/Modules/itertoolsmodule.c +++ b/Modules/itertoolsmodule.c @@ -699,6 +699,12 @@ cycle_next(cycleobject *lz) PyList_Append(lz->saved, item); return item; } + if (PyErr_Occurred()) { + if (PyErr_ExceptionMatches(PyExc_StopIteration)) + PyErr_Clear(); + else + return NULL; + } if (PyList_Size(lz->saved) == 0) return NULL; it = PyObject_GetIter(lz->saved); @@ -1658,6 +1664,12 @@ chain_next(chainobject *lz) item = PyIter_Next(it); if (item != NULL) return item; + if (PyErr_Occurred()) { + if (PyErr_ExceptionMatches(PyExc_StopIteration)) + PyErr_Clear(); + else + return NULL; + } lz->iternum++; } return NULL; |