Commit 85ac8508 authored by Georg Brandl's avatar Georg Brandl

Correctly unpickle 2.4 exceptions via __setstate__ (patch #1498571)

parent b16e4e78
......@@ -150,6 +150,29 @@ BaseException_reduce(PyBaseExceptionObject *self)
return PyTuple_Pack(2, self->ob_type, self->args);
}
/*
* Needed for backward compatibility, since exceptions used to store
* all their attributes in the __dict__. Code is taken from cPickle's
* load_build function.
*/
static PyObject *
BaseException_setstate(PyObject *self, PyObject *state)
{
PyObject *d_key, *d_value;
Py_ssize_t i = 0;
if (state != Py_None) {
if (!PyDict_Check(state)) {
PyErr_SetString(PyExc_TypeError, "state is not a dictionary");
return NULL;
}
while (PyDict_Next(state, &i, &d_key, &d_value)) {
if (PyObject_SetAttr(self, d_key, d_value) < 0)
return NULL;
}
}
Py_RETURN_NONE;
}
#ifdef Py_USING_UNICODE
/* while this method generates fairly uninspired output, it a least
......@@ -168,6 +191,7 @@ BaseException_unicode(PyBaseExceptionObject *self)
static PyMethodDef BaseException_methods[] = {
{"__reduce__", (PyCFunction)BaseException_reduce, METH_NOARGS },
{"__setstate__", (PyCFunction)BaseException_setstate, METH_O },
#ifdef Py_USING_UNICODE
{"__unicode__", (PyCFunction)BaseException_unicode, METH_NOARGS },
#endif
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment