mirror of https://github.com/python/cpython
- New builtin function enumerate(x), from PEP 279. Example:
enumerate("abc") is an iterator returning (0,"a"), (1,"b"), (2,"c"). The argument can be an arbitrary iterable object.
This commit is contained in:
parent
17afa13a9f
commit
7dab2426ca
|
@ -91,6 +91,7 @@
|
||||||
#include "tupleobject.h"
|
#include "tupleobject.h"
|
||||||
#include "listobject.h"
|
#include "listobject.h"
|
||||||
#include "dictobject.h"
|
#include "dictobject.h"
|
||||||
|
#include "enumobject.h"
|
||||||
#include "methodobject.h"
|
#include "methodobject.h"
|
||||||
#include "moduleobject.h"
|
#include "moduleobject.h"
|
||||||
#include "funcobject.h"
|
#include "funcobject.h"
|
||||||
|
|
|
@ -0,0 +1,16 @@
|
||||||
|
#ifndef Py_ENUMOBJECT_H
|
||||||
|
#define Py_ENUMOBJECT_H
|
||||||
|
|
||||||
|
/* Enumerate Object */
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
extern "C" {
|
||||||
|
#endif
|
||||||
|
|
||||||
|
extern DL_IMPORT(PyTypeObject) PyEnum_Type;
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#endif /* !Py_ENUMOBJECT_H */
|
|
@ -0,0 +1,118 @@
|
||||||
|
from __future__ import generators
|
||||||
|
import unittest
|
||||||
|
|
||||||
|
import test_support
|
||||||
|
|
||||||
|
seq, res = 'abc', [(0,'a'), (1,'b'), (2,'c')]
|
||||||
|
|
||||||
|
class G:
|
||||||
|
'Sequence using __getitem__'
|
||||||
|
def __init__(self, seqn):
|
||||||
|
self.seqn = seqn
|
||||||
|
def __getitem__(self, i):
|
||||||
|
return self.seqn[i]
|
||||||
|
|
||||||
|
class I:
|
||||||
|
'Sequence using iterator protocol'
|
||||||
|
def __init__(self, seqn):
|
||||||
|
self.seqn = seqn
|
||||||
|
self.i = 0
|
||||||
|
def __iter__(self):
|
||||||
|
return self
|
||||||
|
def next(self):
|
||||||
|
if self.i >= len(self.seqn): raise StopIteration
|
||||||
|
v = self.seqn[self.i]
|
||||||
|
self.i += 1
|
||||||
|
return v
|
||||||
|
|
||||||
|
class Ig:
|
||||||
|
'Sequence using iterator protocol defined with a generator'
|
||||||
|
def __init__(self, seqn):
|
||||||
|
self.seqn = seqn
|
||||||
|
self.i = 0
|
||||||
|
def __iter__(self):
|
||||||
|
for val in self.seqn:
|
||||||
|
yield val
|
||||||
|
|
||||||
|
class X:
|
||||||
|
'Missing __getitem__ and __iter__'
|
||||||
|
def __init__(self, seqn):
|
||||||
|
self.seqn = seqn
|
||||||
|
self.i = 0
|
||||||
|
def next(self):
|
||||||
|
if self.i >= len(self.seqn): raise StopIteration
|
||||||
|
v = self.seqn[self.i]
|
||||||
|
self.i += 1
|
||||||
|
return v
|
||||||
|
|
||||||
|
class E:
|
||||||
|
'Test propagation of exceptions'
|
||||||
|
def __init__(self, seqn):
|
||||||
|
self.seqn = seqn
|
||||||
|
self.i = 0
|
||||||
|
def __iter__(self):
|
||||||
|
return self
|
||||||
|
def next(self):
|
||||||
|
3/0
|
||||||
|
|
||||||
|
class N:
|
||||||
|
'Iterator missing next()'
|
||||||
|
def __init__(self, seqn):
|
||||||
|
self.seqn = seqn
|
||||||
|
self.i = 0
|
||||||
|
def __iter__(self):
|
||||||
|
return self
|
||||||
|
|
||||||
|
class EnumerateTestCase(unittest.TestCase):
|
||||||
|
|
||||||
|
enum = enumerate
|
||||||
|
|
||||||
|
def test_basicfunction(self):
|
||||||
|
self.assertEqual(type(self.enum(seq)), self.enum)
|
||||||
|
e = self.enum(seq)
|
||||||
|
self.assertEqual(iter(e), e)
|
||||||
|
self.assertEqual(list(self.enum(seq)), res)
|
||||||
|
self.enum.__doc__
|
||||||
|
|
||||||
|
def test_getitemseqn(self):
|
||||||
|
self.assertEqual(list(self.enum(G(seq))), res)
|
||||||
|
e = self.enum(G(''))
|
||||||
|
self.assertRaises(StopIteration, e.next)
|
||||||
|
|
||||||
|
def test_iteratorseqn(self):
|
||||||
|
self.assertEqual(list(self.enum(I(seq))), res)
|
||||||
|
e = self.enum(I(''))
|
||||||
|
self.assertRaises(StopIteration, e.next)
|
||||||
|
|
||||||
|
def test_iteratorgenerator(self):
|
||||||
|
self.assertEqual(list(self.enum(Ig(seq))), res)
|
||||||
|
e = self.enum(Ig(''))
|
||||||
|
self.assertRaises(StopIteration, e.next)
|
||||||
|
|
||||||
|
def test_noniterable(self):
|
||||||
|
self.assertRaises(TypeError, self.enum, X(seq))
|
||||||
|
|
||||||
|
def test_illformediterable(self):
|
||||||
|
self.assertRaises(TypeError, list, self.enum(N(seq)))
|
||||||
|
|
||||||
|
def test_exception_propagation(self):
|
||||||
|
self.assertRaises(ZeroDivisionError, list, self.enum(E(seq)))
|
||||||
|
|
||||||
|
class MyEnum(enumerate):
|
||||||
|
pass
|
||||||
|
|
||||||
|
class SubclassTestCase(EnumerateTestCase):
|
||||||
|
|
||||||
|
enum = MyEnum
|
||||||
|
|
||||||
|
def suite():
|
||||||
|
suite = unittest.TestSuite()
|
||||||
|
suite.addTest(unittest.makeSuite(EnumerateTestCase))
|
||||||
|
suite.addTest(unittest.makeSuite(SubclassTestCase))
|
||||||
|
return suite
|
||||||
|
|
||||||
|
def test_main():
|
||||||
|
test_support.run_suite(suite())
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
test_main()
|
|
@ -257,6 +257,7 @@ OBJECT_OBJS= \
|
||||||
Objects/cobject.o \
|
Objects/cobject.o \
|
||||||
Objects/complexobject.o \
|
Objects/complexobject.o \
|
||||||
Objects/descrobject.o \
|
Objects/descrobject.o \
|
||||||
|
Objects/enumobject.o \
|
||||||
Objects/fileobject.o \
|
Objects/fileobject.o \
|
||||||
Objects/floatobject.o \
|
Objects/floatobject.o \
|
||||||
Objects/frameobject.o \
|
Objects/frameobject.o \
|
||||||
|
@ -443,6 +444,7 @@ PYTHON_HEADERS= \
|
||||||
Include/complexobject.h \
|
Include/complexobject.h \
|
||||||
Include/descrobject.h \
|
Include/descrobject.h \
|
||||||
Include/dictobject.h \
|
Include/dictobject.h \
|
||||||
|
Include/enumobject.h \
|
||||||
Include/fileobject.h \
|
Include/fileobject.h \
|
||||||
Include/floatobject.h \
|
Include/floatobject.h \
|
||||||
Include/funcobject.h \
|
Include/funcobject.h \
|
||||||
|
|
|
@ -6,6 +6,10 @@ Type/class unification and new-style classes
|
||||||
|
|
||||||
Core and builtins
|
Core and builtins
|
||||||
|
|
||||||
|
- New builtin function enumerate(x), from PEP 279. Example:
|
||||||
|
enumerate("abc") is an iterator returning (0,"a"), (1,"b"), (2,"c").
|
||||||
|
The argument can be an arbitrary iterable object.
|
||||||
|
|
||||||
- The assert statement no longer tests __debug__ at runtime. This means
|
- The assert statement no longer tests __debug__ at runtime. This means
|
||||||
that assert statements cannot be disabled by assigning a false value
|
that assert statements cannot be disabled by assigning a false value
|
||||||
to __debug__.
|
to __debug__.
|
||||||
|
|
|
@ -0,0 +1,139 @@
|
||||||
|
/* enumerate object */
|
||||||
|
|
||||||
|
#include "Python.h"
|
||||||
|
|
||||||
|
typedef struct {
|
||||||
|
PyObject_HEAD
|
||||||
|
long en_index; /* current index of enumeration */
|
||||||
|
PyObject* en_sit; /* secondary iterator of enumeration */
|
||||||
|
} enumobject;
|
||||||
|
|
||||||
|
PyTypeObject PyEnum_Type;
|
||||||
|
|
||||||
|
static PyObject *
|
||||||
|
enum_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
|
||||||
|
{
|
||||||
|
enumobject *en;
|
||||||
|
PyObject *seq = NULL;
|
||||||
|
static char *kwlist[] = {"sequence", 0};
|
||||||
|
|
||||||
|
if (!PyArg_ParseTupleAndKeywords(args, kwds, "O:enumerate", kwlist,
|
||||||
|
&seq))
|
||||||
|
return NULL;
|
||||||
|
|
||||||
|
en = (enumobject *)type->tp_alloc(type, 0);
|
||||||
|
if (en == NULL)
|
||||||
|
return NULL;
|
||||||
|
en->en_index = 0;
|
||||||
|
en->en_sit = PyObject_GetIter(seq);
|
||||||
|
if (en->en_sit == NULL) {
|
||||||
|
Py_DECREF(en);
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
return (PyObject *)en;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void
|
||||||
|
enum_dealloc(enumobject *en)
|
||||||
|
{
|
||||||
|
PyObject_GC_UnTrack(en);
|
||||||
|
Py_XDECREF(en->en_sit);
|
||||||
|
en->ob_type->tp_free(en);
|
||||||
|
}
|
||||||
|
|
||||||
|
static int
|
||||||
|
enum_traverse(enumobject *en, visitproc visit, void *arg)
|
||||||
|
{
|
||||||
|
if (en->en_sit)
|
||||||
|
return visit(en->en_sit, arg);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static PyObject *
|
||||||
|
enum_next(enumobject *en)
|
||||||
|
{
|
||||||
|
PyObject *result;
|
||||||
|
PyObject *next_index;
|
||||||
|
|
||||||
|
PyObject *next_item = PyIter_Next(en->en_sit);
|
||||||
|
if (next_item == NULL)
|
||||||
|
return NULL;
|
||||||
|
|
||||||
|
result = PyTuple_New(2);
|
||||||
|
if (result == NULL) {
|
||||||
|
Py_DECREF(next_item);
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
next_index = PyInt_FromLong(en->en_index++);
|
||||||
|
if (next_index == NULL) {
|
||||||
|
Py_DECREF(next_item);
|
||||||
|
Py_DECREF(result);
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
PyTuple_SET_ITEM(result, 0, next_index);
|
||||||
|
PyTuple_SET_ITEM(result, 1, next_item);
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
static PyObject *
|
||||||
|
enum_getiter(PyObject *en)
|
||||||
|
{
|
||||||
|
Py_INCREF(en);
|
||||||
|
return en;
|
||||||
|
}
|
||||||
|
|
||||||
|
static PyMethodDef enum_methods[] = {
|
||||||
|
{"next", (PyCFunction)enum_next, METH_NOARGS,
|
||||||
|
"return the next (index, value) pair, or raise StopIteration"},
|
||||||
|
{NULL, NULL} /* sentinel */
|
||||||
|
};
|
||||||
|
|
||||||
|
static char enum_doc[] =
|
||||||
|
"enumerate(iterable) -> create an enumerating-iterator";
|
||||||
|
|
||||||
|
PyTypeObject PyEnum_Type = {
|
||||||
|
PyObject_HEAD_INIT(&PyType_Type)
|
||||||
|
0, /* ob_size */
|
||||||
|
"enumerate", /* tp_name */
|
||||||
|
sizeof(enumobject), /* tp_basicsize */
|
||||||
|
0, /* tp_itemsize */
|
||||||
|
/* methods */
|
||||||
|
(destructor)enum_dealloc, /* tp_dealloc */
|
||||||
|
0, /* tp_print */
|
||||||
|
0, /* tp_getattr */
|
||||||
|
0, /* tp_setattr */
|
||||||
|
0, /* tp_compare */
|
||||||
|
0, /* tp_repr */
|
||||||
|
0, /* tp_as_number */
|
||||||
|
0, /* tp_as_sequence */
|
||||||
|
0, /* tp_as_mapping */
|
||||||
|
0, /* tp_hash */
|
||||||
|
0, /* tp_call */
|
||||||
|
0, /* tp_str */
|
||||||
|
PyObject_GenericGetAttr, /* tp_getattro */
|
||||||
|
0, /* tp_setattro */
|
||||||
|
0, /* tp_as_buffer */
|
||||||
|
Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC |
|
||||||
|
Py_TPFLAGS_BASETYPE, /* tp_flags */
|
||||||
|
enum_doc, /* tp_doc */
|
||||||
|
(traverseproc)enum_traverse, /* tp_traverse */
|
||||||
|
0, /* tp_clear */
|
||||||
|
0, /* tp_richcompare */
|
||||||
|
0, /* tp_weaklistoffset */
|
||||||
|
(getiterfunc)enum_getiter, /* tp_iter */
|
||||||
|
(iternextfunc)enum_next, /* tp_iternext */
|
||||||
|
enum_methods, /* tp_methods */
|
||||||
|
0, /* tp_members */
|
||||||
|
0, /* tp_getset */
|
||||||
|
0, /* tp_base */
|
||||||
|
0, /* tp_dict */
|
||||||
|
0, /* tp_descr_get */
|
||||||
|
0, /* tp_descr_set */
|
||||||
|
0, /* tp_dictoffset */
|
||||||
|
0, /* tp_init */
|
||||||
|
PyType_GenericAlloc, /* tp_alloc */
|
||||||
|
enum_new, /* tp_new */
|
||||||
|
PyObject_GC_Del, /* tp_free */
|
||||||
|
};
|
|
@ -1864,6 +1864,7 @@ _PyBuiltin_Init(void)
|
||||||
SETBUILTIN("complex", &PyComplex_Type);
|
SETBUILTIN("complex", &PyComplex_Type);
|
||||||
#endif
|
#endif
|
||||||
SETBUILTIN("dict", &PyDict_Type);
|
SETBUILTIN("dict", &PyDict_Type);
|
||||||
|
SETBUILTIN("enumerate", &PyEnum_Type);
|
||||||
SETBUILTIN("float", &PyFloat_Type);
|
SETBUILTIN("float", &PyFloat_Type);
|
||||||
SETBUILTIN("property", &PyProperty_Type);
|
SETBUILTIN("property", &PyProperty_Type);
|
||||||
SETBUILTIN("int", &PyInt_Type);
|
SETBUILTIN("int", &PyInt_Type);
|
||||||
|
|
Loading…
Reference in New Issue