mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-10-10 00:43:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 18

Add keyword argument support to itertools.count().

Browse source
This commit is contained in:
Raymond Hettinger 2009-02-14 04:21:49 +00:00
parent 8c20189ce4
commit 9e8dbbcdcd
3 changed files with 9 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

11
Modules/itertoolsmodule.c
View file

@ -2916,11 +2916,10 @@ count_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
Py_ssize_t cnt = 0;
PyObject *long_cnt = NULL;
PyObject *long_step = NULL;
static char *kwlist[] = {"start", "step", 0};
if (type == &count_type && !_PyArg_NoKeywords("count()", kwds))
return NULL;
if (!PyArg_UnpackTuple(args, "count", 0, 2, &long_cnt, &long_step))
if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO:count",
kwlist, &long_cnt, &long_step))
return NULL;
if (long_cnt != NULL && !PyNumber_Check(long_cnt) ||
@ -3027,10 +3026,10 @@ count_repr(countobject *lz)
}
PyDoc_STRVAR(count_doc,
"count([firstval[, step]]) --> count object\n\
"count([start[, step]]) --> count object\n\
\n\
Return a count object whose .__next__() method returns consecutive\n\
integers starting from zero or, if specified, from firstval.\n\
integers starting from zero or, if specified, from start.\n\
If step is specified, counts by that interval. Equivalent to:\n\n\
def count(firstval=0, step=1):\n\
x = firstval\n\