Issue #7410: deepcopy of itertools.count was resetting the count.

This commit is contained in:
Raymond Hettinger 2009-11-30 21:55:17 +00:00
parent 9aeb9df7da
commit 6c8ee7a333
3 changed files with 28 additions and 1 deletions

View file

@ -3049,6 +3049,22 @@ count_repr(countobject *lz)
lz->long_cnt, lz->long_step);
}
static PyObject *
count_reduce(countobject *lz)
{
if (lz->cnt == PY_SSIZE_T_MAX)
return Py_BuildValue("O(OO)", Py_TYPE(lz), lz->long_cnt, lz->long_step);
return Py_BuildValue("O(n)", Py_TYPE(lz), lz->cnt);
}
PyDoc_STRVAR(count_reduce_doc, "Return state information for pickling.");
static PyMethodDef count_methods[] = {
{"__reduce__", (PyCFunction)count_reduce, METH_NOARGS,
count_reduce_doc},
{NULL, NULL} /* sentinel */
};
PyDoc_STRVAR(count_doc,
"count(start=0, step=1]) --> count object\n\
\n\
@ -3090,7 +3106,7 @@ static PyTypeObject count_type = {
0, /* tp_weaklistoffset */
PyObject_SelfIter, /* tp_iter */
(iternextfunc)count_next, /* tp_iternext */
0, /* tp_methods */
count_methods, /* tp_methods */
0, /* tp_members */
0, /* tp_getset */
0, /* tp_base */