summaryrefslogtreecommitdiffstats
path: root/Objects/iterobject.c
diff options
context:
space:
mode:
authorRaymond Hettinger <python@rcn.com>2004-03-18 22:43:10 (GMT)
committerRaymond Hettinger <python@rcn.com>2004-03-18 22:43:10 (GMT)
commit435bf58b7bc9df7dd8683c3fc72b5b7ca1316541 (patch)
tree97f781846e1bf121dc94ca8f18282e5a3d6af8b9 /Objects/iterobject.c
parent1e5809ff02201291df47d94dad843ca32048e4d3 (diff)
downloadcpython-435bf58b7bc9df7dd8683c3fc72b5b7ca1316541.zip
cpython-435bf58b7bc9df7dd8683c3fc72b5b7ca1316541.tar.gz
cpython-435bf58b7bc9df7dd8683c3fc72b5b7ca1316541.tar.bz2
Make iterators length transparent where possible.
Diffstat (limited to 'Objects/iterobject.c')
-rw-r--r--Objects/iterobject.c15
1 files changed, 14 insertions, 1 deletions
diff --git a/Objects/iterobject.c b/Objects/iterobject.c
index abcf8db..a407dd5 100644
--- a/Objects/iterobject.c
+++ b/Objects/iterobject.c
@@ -71,6 +71,19 @@ iter_iternext(PyObject *iterator)
return NULL;
}
+static int
+iter_len(seqiterobject *it)
+{
+ if (it->it_seq)
+ return PyObject_Size(it->it_seq) - it->it_index;
+ return 0;
+}
+
+static PySequenceMethods iter_as_sequence = {
+ (inquiry)iter_len, /* sq_length */
+ 0, /* sq_concat */
+};
+
PyTypeObject PySeqIter_Type = {
PyObject_HEAD_INIT(&PyType_Type)
0, /* ob_size */
@@ -85,7 +98,7 @@ PyTypeObject PySeqIter_Type = {
0, /* tp_compare */
0, /* tp_repr */
0, /* tp_as_number */
- 0, /* tp_as_sequence */
+ &iter_as_sequence, /* tp_as_sequence */
0, /* tp_as_mapping */
0, /* tp_hash */
0, /* tp_call */