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

Fix and test weak referencing of itertools.tee objects.

Browse source
This commit is contained in:
Raymond Hettinger 2004-10-17 16:40:14 +00:00
parent 837dd93e3b
commit a9f6092904
2 changed files with 14 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
Modules/itertoolsmodule.c
View file

@ -329,6 +329,7 @@ typedef struct {
PyObject_HEAD
teedataobject *dataobj;
int index;
PyObject *weakreflist;
} teeobject;
static PyTypeObject teedataobject_type;
@ -452,6 +453,7 @@ tee_copy(teeobject *to)
Py_INCREF(to->dataobj);
newto->dataobj = to->dataobj;
newto->index = to->index;
newto->weakreflist = NULL;
return (PyObject *)newto;
}
@ -476,6 +478,7 @@ tee_fromiterable(PyObject *iterable)
goto done;
to->dataobj = (teedataobject *)teedataobject_new(it);
to->index = 0;
to->weakreflist = NULL;
done:
Py_XDECREF(it);
return (PyObject *)to;
@ -494,6 +497,8 @@ tee_new(PyTypeObject *type, PyObject *args, PyObject *kw)
static void
tee_dealloc(teeobject *to)
{
if (to->weakreflist != NULL)
PyObject_ClearWeakRefs((PyObject *) to);
Py_XDECREF(to->dataobj);
PyObject_Del(to);
}
@ -533,7 +538,7 @@ static PyTypeObject tee_type = {
0, /* tp_traverse */
0, /* tp_clear */
0, /* tp_richcompare */
0, /* tp_weaklistoffset */
offsetof(teeobject, weakreflist), /* tp_weaklistoffset */
PyObject_SelfIter, /* tp_iter */
(iternextfunc)tee_next, /* tp_iternext */
tee_methods, /* tp_methods */