cleanup Ellipsis and NotImplemented strings after we're done

This commit is contained in:
Benjamin Peterson 2012-03-16 18:45:31 -05:00
parent 020bf28200
commit e80b29b5b6
1 changed files with 8 additions and 2 deletions

View File

@ -2815,18 +2815,24 @@ static int
save_ellipsis(PicklerObject *self, PyObject *obj) save_ellipsis(PicklerObject *self, PyObject *obj)
{ {
PyObject *str = PyUnicode_FromString("Ellipsis"); PyObject *str = PyUnicode_FromString("Ellipsis");
int res;
if (str == NULL) if (str == NULL)
return -1; return -1;
return save_global(self, Py_Ellipsis, str); res = save_global(self, Py_Ellipsis, str);
Py_DECREF(str);
return res;
} }
static int static int
save_notimplemented(PicklerObject *self, PyObject *obj) save_notimplemented(PicklerObject *self, PyObject *obj)
{ {
PyObject *str = PyUnicode_FromString("NotImplemented"); PyObject *str = PyUnicode_FromString("NotImplemented");
int res;
if (str == NULL) if (str == NULL)
return -1; return -1;
return save_global(self, Py_NotImplemented, str); res = save_global(self, Py_NotImplemented, str);
Py_DECREF(str);
return res;
} }
static int static int