A totally new way to do the __new__ wrapper. This should address the

problem brought up in SF bug #444229.
This commit is contained in:
Guido van Rossum 2001-08-06 16:50:37 +00:00
parent 4590049228
commit 0d231eda52
1 changed files with 45 additions and 34 deletions

View File

@ -938,31 +938,6 @@ add_wrappers(PyTypeObject *type, struct wrapperbase *base, void *wrapped)
return 0; return 0;
} }
static int
add_staticmethodwrappers(PyTypeObject *type, struct wrapperbase *base,
void *wrapped)
{
PyObject *dict = type->tp_defined;
PyObject *sm;
for (; base->name != NULL; base++) {
PyObject *descr;
if (PyDict_GetItemString(dict, base->name))
continue;
descr = PyDescr_NewWrapper(type->ob_type, base, wrapped);
if (descr == NULL)
return -1;
sm = PyStaticMethod_New(descr);
Py_DECREF(descr);
if (sm == NULL)
return -1;
if (PyDict_SetItemString(dict, base->name, sm) < 0)
return -1;
Py_DECREF(sm);
}
return 0;
}
static int static int
add_members(PyTypeObject *type, struct memberlist *memb) add_members(PyTypeObject *type, struct memberlist *memb)
{ {
@ -1197,7 +1172,7 @@ PyType_InitDict(PyTypeObject *type)
} }
/* Initialize the base class */ /* Initialize the base class */
if (base) { if (base && base->tp_dict == NULL) {
if (PyType_InitDict(base) < 0) if (PyType_InitDict(base) < 0)
return -1; return -1;
} }
@ -1853,18 +1828,55 @@ static struct wrapperbase tab_init[] = {
}; };
static PyObject * static PyObject *
wrap_new(PyObject *type, PyObject *args, void *wrapped) tp_new_wrapper(PyObject *self, PyObject *args, PyObject *kwds)
{ {
newfunc new = (newfunc)wrapped; PyTypeObject *type, *subtype;
return new((PyTypeObject *)type, args, NULL); PyObject *arg0, *res;
if (self == NULL || !PyType_Check(self))
Py_FatalError("__new__() called with non-type 'self'");
type = (PyTypeObject *)self;
if (!PyTuple_Check(args) || PyTuple_GET_SIZE(args) < 1) {
PyErr_SetString(PyExc_TypeError,
"T.__new__(): not enough arguments");
return NULL;
}
arg0 = PyTuple_GET_ITEM(args, 0);
if (!PyType_Check(arg0)) {
PyErr_SetString(PyExc_TypeError,
"T.__new__(S): S is not a type object");
return NULL;
}
subtype = (PyTypeObject *)arg0;
if (!PyType_IsSubtype(subtype, type)) {
PyErr_SetString(PyExc_TypeError,
"T.__new__(S): S is not a subtype of T");
return NULL;
}
args = PyTuple_GetSlice(args, 1, PyTuple_GET_SIZE(args));
if (args == NULL)
return NULL;
res = type->tp_new(subtype, args, kwds);
Py_DECREF(args);
return res;
} }
static struct wrapperbase tab_new[] = { static struct PyMethodDef tp_new_methoddef[] = {
{"__new__", (wrapperfunc)wrap_new, {"__new__", (PyCFunction)tp_new_wrapper, METH_KEYWORDS,
"T.__new__() -> an object with type T"}, "T.__new__(S, ...) -> a new object with type S, a subtype of T"},
{0} {0}
}; };
static int
add_tp_new_wrapper(PyTypeObject *type)
{
PyObject *func = PyCFunction_New(tp_new_methoddef, (PyObject *)type);
if (func == NULL)
return -1;
return PyDict_SetItemString(type->tp_defined, "__new__", func);
}
static int static int
add_operators(PyTypeObject *type) add_operators(PyTypeObject *type)
{ {
@ -1955,8 +1967,7 @@ add_operators(PyTypeObject *type)
ADD(type->tp_init, tab_init); ADD(type->tp_init, tab_init);
if (type->tp_new != NULL) if (type->tp_new != NULL)
add_staticmethodwrappers(type, tab_new, add_tp_new_wrapper(type);
(void *)(type->tp_new));
return 0; return 0;
} }