diff options
| author | Raymond Hettinger <python@rcn.com> | 2009-02-14 04:21:49 (GMT) |
|---|---|---|
| committer | Raymond Hettinger <python@rcn.com> | 2009-02-14 04:21:49 (GMT) |
| commit | 9e8dbbcdcd462efc8f50008c7182dd399de89097 (patch) | |
| tree | 205c44bc862e68369139148a4309c72d2598ca4b /Modules | |
| parent | 8c20189ce4bb26b18306a7b5fc05db62cf6e9a11 (diff) | |
| download | cpython-9e8dbbcdcd462efc8f50008c7182dd399de89097.zip cpython-9e8dbbcdcd462efc8f50008c7182dd399de89097.tar.gz cpython-9e8dbbcdcd462efc8f50008c7182dd399de89097.tar.bz2 | |
Add keyword argument support to itertools.count().
Diffstat (limited to 'Modules')
| -rw-r--r-- | Modules/itertoolsmodule.c | 11 |
1 files changed, 5 insertions, 6 deletions
diff --git a/Modules/itertoolsmodule.c b/Modules/itertoolsmodule.c index 75c6e7e..a44d2ae 100644 --- a/Modules/itertoolsmodule.c +++ b/Modules/itertoolsmodule.c @@ -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\ |