blob: ac201d58909260911c52b41acc9fbc4d4b7c0569 [file] [log] [blame]
Guido van Rossumf70e43a1991-02-19 12:39:46 +00001
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002/* Type object implementation */
3
Guido van Rossumc0b618a1997-05-02 03:12:38 +00004#include "Python.h"
Tim Peters6d6c1a32001-08-02 04:15:00 +00005#include "structmember.h"
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00006
Guido van Rossum6f799372001-09-20 20:46:19 +00007static PyMemberDef type_members[] = {
Tim Peters6d6c1a32001-08-02 04:15:00 +00008 {"__basicsize__", T_INT, offsetof(PyTypeObject,tp_basicsize),READONLY},
9 {"__itemsize__", T_INT, offsetof(PyTypeObject, tp_itemsize), READONLY},
10 {"__flags__", T_LONG, offsetof(PyTypeObject, tp_flags), READONLY},
11 {"__doc__", T_STRING, offsetof(PyTypeObject, tp_doc), READONLY},
Guido van Rossum9676b222001-08-17 20:32:36 +000012 {"__weakrefoffset__", T_LONG,
Tim Peters6d6c1a32001-08-02 04:15:00 +000013 offsetof(PyTypeObject, tp_weaklistoffset), READONLY},
14 {"__base__", T_OBJECT, offsetof(PyTypeObject, tp_base), READONLY},
15 {"__dictoffset__", T_LONG,
16 offsetof(PyTypeObject, tp_dictoffset), READONLY},
17 {"__bases__", T_OBJECT, offsetof(PyTypeObject, tp_bases), READONLY},
18 {"__mro__", T_OBJECT, offsetof(PyTypeObject, tp_mro), READONLY},
19 {0}
20};
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000021
Guido van Rossumc0b618a1997-05-02 03:12:38 +000022static PyObject *
Guido van Rossumc3542212001-08-16 09:18:56 +000023type_name(PyTypeObject *type, void *context)
24{
25 char *s;
26
27 s = strrchr(type->tp_name, '.');
28 if (s == NULL)
29 s = type->tp_name;
30 else
31 s++;
32 return PyString_FromString(s);
33}
34
35static PyObject *
Tim Peters6d6c1a32001-08-02 04:15:00 +000036type_module(PyTypeObject *type, void *context)
Guido van Rossum29ca26e1995-01-07 11:58:15 +000037{
Guido van Rossumc3542212001-08-16 09:18:56 +000038 PyObject *mod;
39 char *s;
40
41 s = strrchr(type->tp_name, '.');
42 if (s != NULL)
43 return PyString_FromStringAndSize(type->tp_name,
44 (int)(s - type->tp_name));
45 if (!(type->tp_flags & Py_TPFLAGS_HEAPTYPE))
46 return PyString_FromString("__builtin__");
47 mod = PyDict_GetItemString(type->tp_defined, "__module__");
48 if (mod != NULL && PyString_Check(mod)) {
49 Py_INCREF(mod);
50 return mod;
51 }
52 PyErr_SetString(PyExc_AttributeError, "__module__");
53 return NULL;
Tim Peters6d6c1a32001-08-02 04:15:00 +000054}
55
Guido van Rossum3926a632001-09-25 16:25:58 +000056static int
57type_set_module(PyTypeObject *type, PyObject *value, void *context)
58{
59 if (!(type->tp_flags & Py_TPFLAGS_DYNAMICTYPE) ||
60 strrchr(type->tp_name, '.')) {
61 PyErr_Format(PyExc_TypeError,
62 "can't set %s.__module__", type->tp_name);
63 return -1;
64 }
65 if (!value) {
66 PyErr_Format(PyExc_TypeError,
67 "can't delete %s.__module__", type->tp_name);
68 return -1;
69 }
70 return PyDict_SetItemString(type->tp_dict, "__module__", value);
71}
72
Tim Peters6d6c1a32001-08-02 04:15:00 +000073static PyObject *
74type_dict(PyTypeObject *type, void *context)
75{
76 if (type->tp_dict == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +000077 Py_INCREF(Py_None);
78 return Py_None;
Guido van Rossum29ca26e1995-01-07 11:58:15 +000079 }
Tim Peters6d6c1a32001-08-02 04:15:00 +000080 if (type->tp_flags & Py_TPFLAGS_DYNAMICTYPE) {
81 Py_INCREF(type->tp_dict);
82 return type->tp_dict;
83 }
84 return PyDictProxy_New(type->tp_dict);
Guido van Rossum29ca26e1995-01-07 11:58:15 +000085}
86
Tim Peters6d6c1a32001-08-02 04:15:00 +000087static PyObject *
88type_defined(PyTypeObject *type, void *context)
89{
90 if (type->tp_defined == NULL) {
91 Py_INCREF(Py_None);
92 return Py_None;
93 }
94 if (type->tp_flags & Py_TPFLAGS_DYNAMICTYPE) {
95 Py_INCREF(type->tp_defined);
96 return type->tp_defined;
97 }
98 return PyDictProxy_New(type->tp_defined);
99}
100
101static PyObject *
102type_dynamic(PyTypeObject *type, void *context)
103{
104 PyObject *res;
105
106 res = (type->tp_flags & Py_TPFLAGS_DYNAMICTYPE) ? Py_True : Py_False;
107 Py_INCREF(res);
108 return res;
109}
110
Guido van Rossum32d34c82001-09-20 21:45:26 +0000111PyGetSetDef type_getsets[] = {
Guido van Rossumc3542212001-08-16 09:18:56 +0000112 {"__name__", (getter)type_name, NULL, NULL},
Guido van Rossum3926a632001-09-25 16:25:58 +0000113 {"__module__", (getter)type_module, (setter)type_set_module, NULL},
Tim Peters6d6c1a32001-08-02 04:15:00 +0000114 {"__dict__", (getter)type_dict, NULL, NULL},
115 {"__defined__", (getter)type_defined, NULL, NULL},
116 {"__dynamic__", (getter)type_dynamic, NULL, NULL},
117 {0}
118};
119
Martin v. Löwis0163d6d2001-06-09 07:34:05 +0000120static int
121type_compare(PyObject *v, PyObject *w)
122{
123 /* This is called with type objects only. So we
124 can just compare the addresses. */
125 Py_uintptr_t vv = (Py_uintptr_t)v;
126 Py_uintptr_t ww = (Py_uintptr_t)w;
127 return (vv < ww) ? -1 : (vv > ww) ? 1 : 0;
128}
129
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000130static PyObject *
Tim Peters6d6c1a32001-08-02 04:15:00 +0000131type_repr(PyTypeObject *type)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000132{
Barry Warsaw7ce36942001-08-24 18:34:26 +0000133 PyObject *mod, *name, *rtn;
Guido van Rossuma4cb7882001-09-25 03:56:29 +0000134 char *kind;
Guido van Rossumc3542212001-08-16 09:18:56 +0000135
136 mod = type_module(type, NULL);
137 if (mod == NULL)
138 PyErr_Clear();
139 else if (!PyString_Check(mod)) {
140 Py_DECREF(mod);
141 mod = NULL;
142 }
143 name = type_name(type, NULL);
144 if (name == NULL)
145 return NULL;
Barry Warsaw7ce36942001-08-24 18:34:26 +0000146
Guido van Rossuma4cb7882001-09-25 03:56:29 +0000147 if (type->tp_flags & Py_TPFLAGS_HEAPTYPE)
148 kind = "class";
149 else
150 kind = "type";
151
Barry Warsaw7ce36942001-08-24 18:34:26 +0000152 if (mod != NULL && strcmp(PyString_AS_STRING(mod), "__builtin__")) {
Guido van Rossuma4cb7882001-09-25 03:56:29 +0000153 rtn = PyString_FromFormat("<%s '%s.%s'>",
154 kind,
Barry Warsaw7ce36942001-08-24 18:34:26 +0000155 PyString_AS_STRING(mod),
156 PyString_AS_STRING(name));
157 }
Guido van Rossumc3542212001-08-16 09:18:56 +0000158 else
Guido van Rossuma4cb7882001-09-25 03:56:29 +0000159 rtn = PyString_FromFormat("<%s '%s'>", kind, type->tp_name);
Barry Warsaw7ce36942001-08-24 18:34:26 +0000160
Guido van Rossumc3542212001-08-16 09:18:56 +0000161 Py_XDECREF(mod);
162 Py_DECREF(name);
Barry Warsaw7ce36942001-08-24 18:34:26 +0000163 return rtn;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000164}
165
Tim Peters6d6c1a32001-08-02 04:15:00 +0000166static PyObject *
167type_call(PyTypeObject *type, PyObject *args, PyObject *kwds)
168{
169 PyObject *obj;
170
171 if (type->tp_new == NULL) {
172 PyErr_Format(PyExc_TypeError,
173 "cannot create '%.100s' instances",
174 type->tp_name);
175 return NULL;
176 }
177
Tim Peters3f996e72001-09-13 19:18:27 +0000178 obj = type->tp_new(type, args, kwds);
Tim Peters6d6c1a32001-08-02 04:15:00 +0000179 if (obj != NULL) {
180 type = obj->ob_type;
181 if (type->tp_init != NULL &&
182 type->tp_init(obj, args, kwds) < 0) {
183 Py_DECREF(obj);
184 obj = NULL;
185 }
186 }
187 return obj;
188}
189
190PyObject *
191PyType_GenericAlloc(PyTypeObject *type, int nitems)
192{
Guido van Rossum6fb3fde2001-08-30 20:00:07 +0000193#define PTRSIZE (sizeof(PyObject *))
194
Tim Peters6d6c1a32001-08-02 04:15:00 +0000195 int size;
Tim Peters6d6c1a32001-08-02 04:15:00 +0000196 PyObject *obj;
197
198 /* Inline PyObject_New() so we can zero the memory */
199 size = _PyObject_VAR_SIZE(type, nitems);
Guido van Rossum6fb3fde2001-08-30 20:00:07 +0000200 /* Round up size, if necessary, so we fully zero out __dict__ */
201 if (type->tp_itemsize % PTRSIZE != 0) {
202 size += PTRSIZE - 1;
203 size /= PTRSIZE;
204 size *= PTRSIZE;
205 }
Neil Schemenauerc806c882001-08-29 23:54:54 +0000206 if (PyType_IS_GC(type)) {
207 obj = _PyObject_GC_Malloc(type, nitems);
208 }
209 else {
210 obj = PyObject_MALLOC(size);
211 }
212 if (obj == NULL)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000213 return PyErr_NoMemory();
Neil Schemenauerc806c882001-08-29 23:54:54 +0000214 memset(obj, '\0', size);
Tim Peters6d6c1a32001-08-02 04:15:00 +0000215 if (type->tp_flags & Py_TPFLAGS_HEAPTYPE)
216 Py_INCREF(type);
217 if (type->tp_itemsize == 0)
218 PyObject_INIT(obj, type);
219 else
220 (void) PyObject_INIT_VAR((PyVarObject *)obj, type, nitems);
221 if (PyType_IS_GC(type))
Neil Schemenauerc806c882001-08-29 23:54:54 +0000222 _PyObject_GC_TRACK(obj);
Tim Peters6d6c1a32001-08-02 04:15:00 +0000223 return obj;
224}
225
226PyObject *
227PyType_GenericNew(PyTypeObject *type, PyObject *args, PyObject *kwds)
228{
229 return type->tp_alloc(type, 0);
230}
231
232/* Helper for subtyping */
233
234static void
235subtype_dealloc(PyObject *self)
236{
Tim Peters6d6c1a32001-08-02 04:15:00 +0000237 PyTypeObject *type, *base;
238 destructor f;
239
240 /* This exists so we can DECREF self->ob_type */
241
242 /* Find the nearest base with a different tp_dealloc */
243 type = self->ob_type;
244 base = type->tp_base;
245 while ((f = base->tp_dealloc) == subtype_dealloc) {
246 base = base->tp_base;
247 assert(base);
248 }
249
250 /* If we added a dict, DECREF it */
Guido van Rossum6fb3fde2001-08-30 20:00:07 +0000251 if (type->tp_dictoffset && !base->tp_dictoffset) {
252 PyObject **dictptr = _PyObject_GetDictPtr(self);
253 if (dictptr != NULL) {
254 PyObject *dict = *dictptr;
255 if (dict != NULL) {
256 Py_DECREF(dict);
257 *dictptr = NULL;
258 }
Tim Peters6d6c1a32001-08-02 04:15:00 +0000259 }
260 }
261
Guido van Rossum9676b222001-08-17 20:32:36 +0000262 /* If we added weaklist, we clear it */
Guido van Rossum6fb3fde2001-08-30 20:00:07 +0000263 if (type->tp_weaklistoffset && !base->tp_weaklistoffset)
Guido van Rossum9676b222001-08-17 20:32:36 +0000264 PyObject_ClearWeakRefs(self);
265
Tim Peters6d6c1a32001-08-02 04:15:00 +0000266 /* Finalize GC if the base doesn't do GC and we do */
267 if (PyType_IS_GC(type) && !PyType_IS_GC(base))
268 PyObject_GC_Fini(self);
269
270 /* Call the base tp_dealloc() */
271 assert(f);
272 f(self);
273
274 /* Can't reference self beyond this point */
275 if (type->tp_flags & Py_TPFLAGS_HEAPTYPE) {
276 Py_DECREF(type);
277 }
278}
279
280staticforward void override_slots(PyTypeObject *type, PyObject *dict);
281staticforward PyTypeObject *solid_base(PyTypeObject *type);
282
283typedef struct {
284 PyTypeObject type;
285 PyNumberMethods as_number;
286 PySequenceMethods as_sequence;
287 PyMappingMethods as_mapping;
288 PyBufferProcs as_buffer;
289 PyObject *name, *slots;
Guido van Rossum6f799372001-09-20 20:46:19 +0000290 PyMemberDef members[1];
Tim Peters6d6c1a32001-08-02 04:15:00 +0000291} etype;
292
293/* type test with subclassing support */
294
295int
296PyType_IsSubtype(PyTypeObject *a, PyTypeObject *b)
297{
298 PyObject *mro;
299
Guido van Rossum9478d072001-09-07 18:52:13 +0000300 if (!(a->tp_flags & Py_TPFLAGS_HAVE_CLASS))
301 return b == a || b == &PyBaseObject_Type;
302
Tim Peters6d6c1a32001-08-02 04:15:00 +0000303 mro = a->tp_mro;
304 if (mro != NULL) {
305 /* Deal with multiple inheritance without recursion
306 by walking the MRO tuple */
307 int i, n;
308 assert(PyTuple_Check(mro));
309 n = PyTuple_GET_SIZE(mro);
310 for (i = 0; i < n; i++) {
311 if (PyTuple_GET_ITEM(mro, i) == (PyObject *)b)
312 return 1;
313 }
314 return 0;
315 }
316 else {
317 /* a is not completely initilized yet; follow tp_base */
318 do {
319 if (a == b)
320 return 1;
321 a = a->tp_base;
322 } while (a != NULL);
323 return b == &PyBaseObject_Type;
324 }
325}
326
Guido van Rossumf21c6be2001-09-14 17:51:50 +0000327/* Internal routines to do a method lookup in the type
Guido van Rossum60718732001-08-28 17:47:51 +0000328 without looking in the instance dictionary
329 (so we can't use PyObject_GetAttr) but still binding
330 it to the instance. The arguments are the object,
331 the method name as a C string, and the address of a
Guido van Rossumf21c6be2001-09-14 17:51:50 +0000332 static variable used to cache the interned Python string.
333
334 Two variants:
335
336 - lookup_maybe() returns NULL without raising an exception
337 when the _PyType_Lookup() call fails;
338
339 - lookup_method() always raises an exception upon errors.
340*/
Guido van Rossum60718732001-08-28 17:47:51 +0000341
342static PyObject *
Guido van Rossumf21c6be2001-09-14 17:51:50 +0000343lookup_maybe(PyObject *self, char *attrstr, PyObject **attrobj)
Guido van Rossum60718732001-08-28 17:47:51 +0000344{
345 PyObject *res;
346
347 if (*attrobj == NULL) {
348 *attrobj = PyString_InternFromString(attrstr);
349 if (*attrobj == NULL)
350 return NULL;
351 }
352 res = _PyType_Lookup(self->ob_type, *attrobj);
Guido van Rossumf21c6be2001-09-14 17:51:50 +0000353 if (res != NULL) {
Guido van Rossum60718732001-08-28 17:47:51 +0000354 descrgetfunc f;
355 if ((f = res->ob_type->tp_descr_get) == NULL)
356 Py_INCREF(res);
357 else
358 res = f(res, self, (PyObject *)(self->ob_type));
359 }
360 return res;
361}
362
Guido van Rossumf21c6be2001-09-14 17:51:50 +0000363static PyObject *
364lookup_method(PyObject *self, char *attrstr, PyObject **attrobj)
365{
366 PyObject *res = lookup_maybe(self, attrstr, attrobj);
367 if (res == NULL && !PyErr_Occurred())
368 PyErr_SetObject(PyExc_AttributeError, *attrobj);
369 return res;
370}
371
Guido van Rossum2730b132001-08-28 18:22:14 +0000372/* A variation of PyObject_CallMethod that uses lookup_method()
373 instead of PyObject_GetAttrString(). This uses the same convention
374 as lookup_method to cache the interned name string object. */
375
376PyObject *
377call_method(PyObject *o, char *name, PyObject **nameobj, char *format, ...)
378{
379 va_list va;
380 PyObject *args, *func = 0, *retval;
381 PyObject *dummy_str = NULL;
382 va_start(va, format);
383
Guido van Rossumf21c6be2001-09-14 17:51:50 +0000384 func = lookup_maybe(o, name, &dummy_str);
385 if (func == NULL) {
386 va_end(va);
387 if (!PyErr_Occurred())
388 PyErr_SetObject(PyExc_AttributeError, dummy_str);
389 Py_XDECREF(dummy_str);
390 return NULL;
391 }
392 Py_DECREF(dummy_str);
393
394 if (format && *format)
395 args = Py_VaBuildValue(format, va);
396 else
397 args = PyTuple_New(0);
398
399 va_end(va);
400
401 if (args == NULL)
402 return NULL;
403
404 assert(PyTuple_Check(args));
405 retval = PyObject_Call(func, args, NULL);
406
407 Py_DECREF(args);
408 Py_DECREF(func);
409
410 return retval;
411}
412
413/* Clone of call_method() that returns NotImplemented when the lookup fails. */
414
415PyObject *
416call_maybe(PyObject *o, char *name, PyObject **nameobj, char *format, ...)
417{
418 va_list va;
419 PyObject *args, *func = 0, *retval;
420 PyObject *dummy_str = NULL;
421 va_start(va, format);
422
423 func = lookup_maybe(o, name, &dummy_str);
Guido van Rossum2730b132001-08-28 18:22:14 +0000424 Py_XDECREF(dummy_str);
425 if (func == NULL) {
426 va_end(va);
Guido van Rossumf21c6be2001-09-14 17:51:50 +0000427 if (!PyErr_Occurred()) {
428 Py_INCREF(Py_NotImplemented);
429 return Py_NotImplemented;
430 }
Guido van Rossum717ce002001-09-14 16:58:08 +0000431 return NULL;
Guido van Rossum2730b132001-08-28 18:22:14 +0000432 }
433
434 if (format && *format)
435 args = Py_VaBuildValue(format, va);
436 else
437 args = PyTuple_New(0);
438
439 va_end(va);
440
Guido van Rossum717ce002001-09-14 16:58:08 +0000441 if (args == NULL)
Guido van Rossum2730b132001-08-28 18:22:14 +0000442 return NULL;
443
Guido van Rossum717ce002001-09-14 16:58:08 +0000444 assert(PyTuple_Check(args));
445 retval = PyObject_Call(func, args, NULL);
Guido van Rossum2730b132001-08-28 18:22:14 +0000446
447 Py_DECREF(args);
448 Py_DECREF(func);
449
450 return retval;
451}
452
Tim Peters6d6c1a32001-08-02 04:15:00 +0000453/* Method resolution order algorithm from "Putting Metaclasses to Work"
454 by Forman and Danforth (Addison-Wesley 1999). */
455
456static int
457conservative_merge(PyObject *left, PyObject *right)
458{
459 int left_size;
460 int right_size;
461 int i, j, r, ok;
462 PyObject *temp, *rr;
463
464 assert(PyList_Check(left));
465 assert(PyList_Check(right));
466
467 again:
468 left_size = PyList_GET_SIZE(left);
469 right_size = PyList_GET_SIZE(right);
470 for (i = 0; i < left_size; i++) {
471 for (j = 0; j < right_size; j++) {
472 if (PyList_GET_ITEM(left, i) ==
473 PyList_GET_ITEM(right, j)) {
474 /* found a merge point */
475 temp = PyList_New(0);
476 if (temp == NULL)
477 return -1;
478 for (r = 0; r < j; r++) {
479 rr = PyList_GET_ITEM(right, r);
480 ok = PySequence_Contains(left, rr);
481 if (ok < 0) {
482 Py_DECREF(temp);
483 return -1;
484 }
485 if (!ok) {
486 ok = PyList_Append(temp, rr);
487 if (ok < 0) {
488 Py_DECREF(temp);
489 return -1;
490 }
491 }
492 }
493 ok = PyList_SetSlice(left, i, i, temp);
494 Py_DECREF(temp);
495 if (ok < 0)
496 return -1;
497 ok = PyList_SetSlice(right, 0, j+1, NULL);
498 if (ok < 0)
499 return -1;
500 goto again;
501 }
502 }
503 }
504 return PyList_SetSlice(left, left_size, left_size, right);
505}
506
507static int
508serious_order_disagreements(PyObject *left, PyObject *right)
509{
510 return 0; /* XXX later -- for now, we cheat: "don't do that" */
511}
512
513static PyObject *
514mro_implementation(PyTypeObject *type)
515{
516 int i, n, ok;
517 PyObject *bases, *result;
518
519 bases = type->tp_bases;
520 n = PyTuple_GET_SIZE(bases);
521 result = Py_BuildValue("[O]", (PyObject *)type);
522 if (result == NULL)
523 return NULL;
524 for (i = 0; i < n; i++) {
525 PyTypeObject *base =
526 (PyTypeObject *) PyTuple_GET_ITEM(bases, i);
527 PyObject *parentMRO = PySequence_List(base->tp_mro);
528 if (parentMRO == NULL) {
529 Py_DECREF(result);
530 return NULL;
531 }
532 if (serious_order_disagreements(result, parentMRO)) {
533 Py_DECREF(result);
534 return NULL;
535 }
536 ok = conservative_merge(result, parentMRO);
537 Py_DECREF(parentMRO);
538 if (ok < 0) {
539 Py_DECREF(result);
540 return NULL;
541 }
542 }
543 return result;
544}
545
546static PyObject *
Martin v. Löwise3eb1f22001-08-16 13:15:00 +0000547mro_external(PyObject *self)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000548{
549 PyTypeObject *type = (PyTypeObject *)self;
550
Tim Peters6d6c1a32001-08-02 04:15:00 +0000551 return mro_implementation(type);
552}
553
554static int
555mro_internal(PyTypeObject *type)
556{
557 PyObject *mro, *result, *tuple;
558
559 if (type->ob_type == &PyType_Type) {
560 result = mro_implementation(type);
561 }
562 else {
Guido van Rossum60718732001-08-28 17:47:51 +0000563 static PyObject *mro_str;
564 mro = lookup_method((PyObject *)type, "mro", &mro_str);
Tim Peters6d6c1a32001-08-02 04:15:00 +0000565 if (mro == NULL)
566 return -1;
567 result = PyObject_CallObject(mro, NULL);
568 Py_DECREF(mro);
569 }
570 if (result == NULL)
571 return -1;
572 tuple = PySequence_Tuple(result);
573 Py_DECREF(result);
574 type->tp_mro = tuple;
575 return 0;
576}
577
578
579/* Calculate the best base amongst multiple base classes.
580 This is the first one that's on the path to the "solid base". */
581
582static PyTypeObject *
583best_base(PyObject *bases)
584{
585 int i, n;
586 PyTypeObject *base, *winner, *candidate, *base_i;
587
588 assert(PyTuple_Check(bases));
589 n = PyTuple_GET_SIZE(bases);
590 assert(n > 0);
591 base = (PyTypeObject *)PyTuple_GET_ITEM(bases, 0);
592 winner = &PyBaseObject_Type;
593 for (i = 0; i < n; i++) {
594 base_i = (PyTypeObject *)PyTuple_GET_ITEM(bases, i);
595 if (!PyType_Check((PyObject *)base_i)) {
596 PyErr_SetString(
597 PyExc_TypeError,
598 "bases must be types");
599 return NULL;
600 }
601 if (base_i->tp_dict == NULL) {
Guido van Rossum528b7eb2001-08-07 17:24:28 +0000602 if (PyType_Ready(base_i) < 0)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000603 return NULL;
604 }
605 candidate = solid_base(base_i);
606 if (PyType_IsSubtype(winner, candidate))
607 ;
608 else if (PyType_IsSubtype(candidate, winner)) {
609 winner = candidate;
610 base = base_i;
611 }
612 else {
613 PyErr_SetString(
614 PyExc_TypeError,
615 "multiple bases have "
616 "instance lay-out conflict");
617 return NULL;
618 }
619 }
620 assert(base != NULL);
621 return base;
622}
623
624static int
625extra_ivars(PyTypeObject *type, PyTypeObject *base)
626{
Neil Schemenauerc806c882001-08-29 23:54:54 +0000627 size_t t_size = type->tp_basicsize;
628 size_t b_size = base->tp_basicsize;
Tim Peters6d6c1a32001-08-02 04:15:00 +0000629
Guido van Rossum9676b222001-08-17 20:32:36 +0000630 assert(t_size >= b_size); /* Else type smaller than base! */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000631 if (type->tp_itemsize || base->tp_itemsize) {
632 /* If itemsize is involved, stricter rules */
633 return t_size != b_size ||
634 type->tp_itemsize != base->tp_itemsize;
635 }
Guido van Rossum9676b222001-08-17 20:32:36 +0000636 if (type->tp_weaklistoffset && base->tp_weaklistoffset == 0 &&
637 type->tp_weaklistoffset + sizeof(PyObject *) == t_size)
638 t_size -= sizeof(PyObject *);
639 if (type->tp_dictoffset && base->tp_dictoffset == 0 &&
640 type->tp_dictoffset + sizeof(PyObject *) == t_size)
641 t_size -= sizeof(PyObject *);
642
643 return t_size != b_size;
Tim Peters6d6c1a32001-08-02 04:15:00 +0000644}
645
646static PyTypeObject *
647solid_base(PyTypeObject *type)
648{
649 PyTypeObject *base;
650
651 if (type->tp_base)
652 base = solid_base(type->tp_base);
653 else
654 base = &PyBaseObject_Type;
655 if (extra_ivars(type, base))
656 return type;
657 else
658 return base;
659}
660
661staticforward void object_dealloc(PyObject *);
662staticforward int object_init(PyObject *, PyObject *, PyObject *);
663
664static PyObject *
Guido van Rossum6fb3fde2001-08-30 20:00:07 +0000665subtype_dict(PyObject *obj, void *context)
666{
667 PyObject **dictptr = _PyObject_GetDictPtr(obj);
668 PyObject *dict;
669
670 if (dictptr == NULL) {
671 PyErr_SetString(PyExc_AttributeError,
672 "This object has no __dict__");
673 return NULL;
674 }
675 dict = *dictptr;
Guido van Rossum3926a632001-09-25 16:25:58 +0000676 if (dict == NULL)
677 *dictptr = dict = PyDict_New();
678 Py_XINCREF(dict);
679 return dict;
Guido van Rossum6fb3fde2001-08-30 20:00:07 +0000680}
681
Guido van Rossum32d34c82001-09-20 21:45:26 +0000682PyGetSetDef subtype_getsets[] = {
Guido van Rossum6fb3fde2001-08-30 20:00:07 +0000683 {"__dict__", subtype_dict, NULL, NULL},
684 {0},
685};
686
687static PyObject *
Tim Peters6d6c1a32001-08-02 04:15:00 +0000688type_new(PyTypeObject *metatype, PyObject *args, PyObject *kwds)
689{
690 PyObject *name, *bases, *dict;
691 static char *kwlist[] = {"name", "bases", "dict", 0};
692 PyObject *slots, *tmp;
Guido van Rossum8d32c8b2001-08-17 11:18:38 +0000693 PyTypeObject *type, *base, *tmptype, *winner;
Tim Peters6d6c1a32001-08-02 04:15:00 +0000694 etype *et;
Guido van Rossum6f799372001-09-20 20:46:19 +0000695 PyMemberDef *mp;
Guido van Rossum9676b222001-08-17 20:32:36 +0000696 int i, nbases, nslots, slotoffset, dynamic, add_dict, add_weak;
Tim Peters6d6c1a32001-08-02 04:15:00 +0000697
Guido van Rossum8d32c8b2001-08-17 11:18:38 +0000698 /* Special case: type(x) should return x->ob_type */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000699 if (metatype == &PyType_Type &&
700 PyTuple_Check(args) && PyTuple_GET_SIZE(args) == 1 &&
701 (kwds == NULL || (PyDict_Check(kwds) && PyDict_Size(kwds) == 0))) {
Tim Peters6d6c1a32001-08-02 04:15:00 +0000702 PyObject *x = PyTuple_GET_ITEM(args, 0);
703 Py_INCREF(x->ob_type);
704 return (PyObject *) x->ob_type;
705 }
706
Guido van Rossum8d32c8b2001-08-17 11:18:38 +0000707 /* Check arguments: (name, bases, dict) */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000708 if (!PyArg_ParseTupleAndKeywords(args, kwds, "SO!O!:type", kwlist,
709 &name,
710 &PyTuple_Type, &bases,
711 &PyDict_Type, &dict))
712 return NULL;
713
714 /* Determine the proper metatype to deal with this,
715 and check for metatype conflicts while we're at it.
716 Note that if some other metatype wins to contract,
717 it's possible that its instances are not types. */
718 nbases = PyTuple_GET_SIZE(bases);
Guido van Rossum8d32c8b2001-08-17 11:18:38 +0000719 winner = metatype;
Tim Peters6d6c1a32001-08-02 04:15:00 +0000720 for (i = 0; i < nbases; i++) {
721 tmp = PyTuple_GET_ITEM(bases, i);
722 tmptype = tmp->ob_type;
Guido van Rossum8d32c8b2001-08-17 11:18:38 +0000723 if (PyType_IsSubtype(winner, tmptype))
Tim Peters6d6c1a32001-08-02 04:15:00 +0000724 continue;
Guido van Rossum8d32c8b2001-08-17 11:18:38 +0000725 if (PyType_IsSubtype(tmptype, winner)) {
726 winner = tmptype;
Tim Peters6d6c1a32001-08-02 04:15:00 +0000727 continue;
728 }
729 PyErr_SetString(PyExc_TypeError,
730 "metatype conflict among bases");
731 return NULL;
732 }
Guido van Rossum8d32c8b2001-08-17 11:18:38 +0000733 if (winner != metatype) {
734 if (winner->tp_new != type_new) /* Pass it to the winner */
735 return winner->tp_new(winner, args, kwds);
736 metatype = winner;
737 }
Tim Peters6d6c1a32001-08-02 04:15:00 +0000738
739 /* Adjust for empty tuple bases */
740 if (nbases == 0) {
741 bases = Py_BuildValue("(O)", &PyBaseObject_Type);
742 if (bases == NULL)
743 return NULL;
744 nbases = 1;
745 }
746 else
747 Py_INCREF(bases);
748
749 /* XXX From here until type is allocated, "return NULL" leaks bases! */
750
751 /* Calculate best base, and check that all bases are type objects */
752 base = best_base(bases);
753 if (base == NULL)
754 return NULL;
755 if (!PyType_HasFeature(base, Py_TPFLAGS_BASETYPE)) {
756 PyErr_Format(PyExc_TypeError,
757 "type '%.100s' is not an acceptable base type",
758 base->tp_name);
759 return NULL;
760 }
761
Guido van Rossum1a493502001-08-17 16:47:50 +0000762 /* Should this be a dynamic class (i.e. modifiable __dict__)?
763 Look in two places for a variable named __dynamic__:
764 1) in the class dict
765 2) in the module dict (globals)
766 The first variable that is an int >= 0 is used.
767 Otherwise, a default is calculated from the base classes:
768 if any base class is dynamic, this class is dynamic; otherwise
769 it is static. */
770 dynamic = -1; /* Not yet determined */
771 /* Look in the class */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000772 tmp = PyDict_GetItemString(dict, "__dynamic__");
773 if (tmp != NULL) {
Guido van Rossum1a493502001-08-17 16:47:50 +0000774 dynamic = PyInt_AsLong(tmp);
Tim Peters6d6c1a32001-08-02 04:15:00 +0000775 if (dynamic < 0)
Guido van Rossum1a493502001-08-17 16:47:50 +0000776 PyErr_Clear();
Tim Peters6d6c1a32001-08-02 04:15:00 +0000777 }
Guido van Rossum1a493502001-08-17 16:47:50 +0000778 if (dynamic < 0) {
779 /* Look in the module globals */
780 tmp = PyEval_GetGlobals();
781 if (tmp != NULL) {
782 tmp = PyDict_GetItemString(tmp, "__dynamic__");
783 if (tmp != NULL) {
784 dynamic = PyInt_AsLong(tmp);
785 if (dynamic < 0)
786 PyErr_Clear();
787 }
788 }
789 }
790 if (dynamic < 0) {
791 /* Make a new class dynamic if any of its bases is
792 dynamic. This is not always the same as inheriting
793 the __dynamic__ class attribute! */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000794 dynamic = 0;
795 for (i = 0; i < nbases; i++) {
Guido van Rossum1a493502001-08-17 16:47:50 +0000796 tmptype = (PyTypeObject *)
797 PyTuple_GET_ITEM(bases, i);
798 if (tmptype->tp_flags &
799 Py_TPFLAGS_DYNAMICTYPE) {
Tim Peters6d6c1a32001-08-02 04:15:00 +0000800 dynamic = 1;
801 break;
802 }
803 }
804 }
805
806 /* Check for a __slots__ sequence variable in dict, and count it */
807 slots = PyDict_GetItemString(dict, "__slots__");
808 nslots = 0;
Guido van Rossum9676b222001-08-17 20:32:36 +0000809 add_dict = 0;
810 add_weak = 0;
Tim Peters6d6c1a32001-08-02 04:15:00 +0000811 if (slots != NULL) {
812 /* Make it into a tuple */
813 if (PyString_Check(slots))
814 slots = Py_BuildValue("(O)", slots);
815 else
816 slots = PySequence_Tuple(slots);
817 if (slots == NULL)
818 return NULL;
819 nslots = PyTuple_GET_SIZE(slots);
Guido van Rossumc4141872001-08-30 04:43:35 +0000820 if (nslots > 0 && base->tp_itemsize != 0) {
821 PyErr_Format(PyExc_TypeError,
822 "nonempty __slots__ "
823 "not supported for subtype of '%s'",
824 base->tp_name);
825 return NULL;
826 }
Tim Peters6d6c1a32001-08-02 04:15:00 +0000827 for (i = 0; i < nslots; i++) {
828 if (!PyString_Check(PyTuple_GET_ITEM(slots, i))) {
829 PyErr_SetString(PyExc_TypeError,
830 "__slots__ must be a sequence of strings");
831 Py_DECREF(slots);
832 return NULL;
833 }
Guido van Rossum9676b222001-08-17 20:32:36 +0000834 /* XXX Check against null bytes in name */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000835 }
836 }
837 if (slots == NULL && base->tp_dictoffset == 0 &&
838 (base->tp_setattro == PyObject_GenericSetAttr ||
Guido van Rossum9676b222001-08-17 20:32:36 +0000839 base->tp_setattro == NULL)) {
Guido van Rossum9676b222001-08-17 20:32:36 +0000840 add_dict++;
841 }
Guido van Rossumc4141872001-08-30 04:43:35 +0000842 if (slots == NULL && base->tp_weaklistoffset == 0 &&
843 base->tp_itemsize == 0) {
Guido van Rossum9676b222001-08-17 20:32:36 +0000844 nslots++;
845 add_weak++;
846 }
Tim Peters6d6c1a32001-08-02 04:15:00 +0000847
848 /* XXX From here until type is safely allocated,
849 "return NULL" may leak slots! */
850
851 /* Allocate the type object */
852 type = (PyTypeObject *)metatype->tp_alloc(metatype, nslots);
853 if (type == NULL)
854 return NULL;
855
856 /* Keep name and slots alive in the extended type object */
857 et = (etype *)type;
858 Py_INCREF(name);
859 et->name = name;
860 et->slots = slots;
861
Guido van Rossumdc91b992001-08-08 22:26:22 +0000862 /* Initialize tp_flags */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000863 type->tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HEAPTYPE |
864 Py_TPFLAGS_BASETYPE;
865 if (dynamic)
866 type->tp_flags |= Py_TPFLAGS_DYNAMICTYPE;
Guido van Rossumdc91b992001-08-08 22:26:22 +0000867
868 /* It's a new-style number unless it specifically inherits any
869 old-style numeric behavior */
870 if ((base->tp_flags & Py_TPFLAGS_CHECKTYPES) ||
871 (base->tp_as_number == NULL))
872 type->tp_flags |= Py_TPFLAGS_CHECKTYPES;
873
874 /* Initialize essential fields */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000875 type->tp_as_number = &et->as_number;
876 type->tp_as_sequence = &et->as_sequence;
877 type->tp_as_mapping = &et->as_mapping;
878 type->tp_as_buffer = &et->as_buffer;
879 type->tp_name = PyString_AS_STRING(name);
880
881 /* Set tp_base and tp_bases */
882 type->tp_bases = bases;
883 Py_INCREF(base);
884 type->tp_base = base;
885
886 /* Initialize tp_defined from passed-in dict */
887 type->tp_defined = dict = PyDict_Copy(dict);
888 if (dict == NULL) {
889 Py_DECREF(type);
890 return NULL;
891 }
892
Guido van Rossumc3542212001-08-16 09:18:56 +0000893 /* Set __module__ in the dict */
894 if (PyDict_GetItemString(dict, "__module__") == NULL) {
895 tmp = PyEval_GetGlobals();
896 if (tmp != NULL) {
897 tmp = PyDict_GetItemString(tmp, "__name__");
898 if (tmp != NULL) {
899 if (PyDict_SetItemString(dict, "__module__",
900 tmp) < 0)
901 return NULL;
902 }
903 }
904 }
905
Tim Peters6d6c1a32001-08-02 04:15:00 +0000906 /* Special-case __new__: if it's a plain function,
907 make it a static function */
908 tmp = PyDict_GetItemString(dict, "__new__");
909 if (tmp != NULL && PyFunction_Check(tmp)) {
910 tmp = PyStaticMethod_New(tmp);
911 if (tmp == NULL) {
912 Py_DECREF(type);
913 return NULL;
914 }
915 PyDict_SetItemString(dict, "__new__", tmp);
916 Py_DECREF(tmp);
917 }
918
919 /* Add descriptors for custom slots from __slots__, or for __dict__ */
920 mp = et->members;
Neil Schemenauerc806c882001-08-29 23:54:54 +0000921 slotoffset = base->tp_basicsize;
Tim Peters6d6c1a32001-08-02 04:15:00 +0000922 if (slots != NULL) {
923 for (i = 0; i < nslots; i++, mp++) {
924 mp->name = PyString_AS_STRING(
925 PyTuple_GET_ITEM(slots, i));
926 mp->type = T_OBJECT;
927 mp->offset = slotoffset;
Guido van Rossum9676b222001-08-17 20:32:36 +0000928 if (base->tp_weaklistoffset == 0 &&
929 strcmp(mp->name, "__weakref__") == 0)
930 type->tp_weaklistoffset = slotoffset;
Tim Peters6d6c1a32001-08-02 04:15:00 +0000931 slotoffset += sizeof(PyObject *);
932 }
933 }
Guido van Rossum9676b222001-08-17 20:32:36 +0000934 else {
935 if (add_dict) {
Guido van Rossum6fb3fde2001-08-30 20:00:07 +0000936 if (base->tp_itemsize)
Tim Peters017cb2c2001-08-30 20:07:55 +0000937 type->tp_dictoffset = -(long)sizeof(PyObject *);
Guido van Rossum6fb3fde2001-08-30 20:00:07 +0000938 else
939 type->tp_dictoffset = slotoffset;
Guido van Rossum9676b222001-08-17 20:32:36 +0000940 slotoffset += sizeof(PyObject *);
Guido van Rossum6fb3fde2001-08-30 20:00:07 +0000941 type->tp_getset = subtype_getsets;
Guido van Rossum9676b222001-08-17 20:32:36 +0000942 }
943 if (add_weak) {
Guido van Rossum6fb3fde2001-08-30 20:00:07 +0000944 assert(!base->tp_itemsize);
Guido van Rossum9676b222001-08-17 20:32:36 +0000945 type->tp_weaklistoffset = slotoffset;
946 mp->name = "__weakref__";
947 mp->type = T_OBJECT;
948 mp->offset = slotoffset;
Tim Peters26f68f52001-09-18 00:23:33 +0000949 mp->flags = READONLY;
Guido van Rossum9676b222001-08-17 20:32:36 +0000950 mp++;
951 slotoffset += sizeof(PyObject *);
952 }
Tim Peters6d6c1a32001-08-02 04:15:00 +0000953 }
954 type->tp_basicsize = slotoffset;
Guido van Rossum6fb3fde2001-08-30 20:00:07 +0000955 type->tp_itemsize = base->tp_itemsize;
Guido van Rossum13d52f02001-08-10 21:24:08 +0000956 type->tp_members = et->members;
Tim Peters6d6c1a32001-08-02 04:15:00 +0000957
958 /* Special case some slots */
959 if (type->tp_dictoffset != 0 || nslots > 0) {
960 if (base->tp_getattr == NULL && base->tp_getattro == NULL)
961 type->tp_getattro = PyObject_GenericGetAttr;
962 if (base->tp_setattr == NULL && base->tp_setattro == NULL)
963 type->tp_setattro = PyObject_GenericSetAttr;
964 }
965 type->tp_dealloc = subtype_dealloc;
966
967 /* Always override allocation strategy to use regular heap */
968 type->tp_alloc = PyType_GenericAlloc;
969 type->tp_free = _PyObject_Del;
970
971 /* Initialize the rest */
Guido van Rossum528b7eb2001-08-07 17:24:28 +0000972 if (PyType_Ready(type) < 0) {
Tim Peters6d6c1a32001-08-02 04:15:00 +0000973 Py_DECREF(type);
974 return NULL;
975 }
976
977 /* Override slots that deserve it */
Guido van Rossum8e248182001-08-12 05:17:56 +0000978 if (!PyType_HasFeature(type, Py_TPFLAGS_DYNAMICTYPE))
979 override_slots(type, type->tp_defined);
Guido van Rossumf040ede2001-08-07 16:40:56 +0000980
Tim Peters6d6c1a32001-08-02 04:15:00 +0000981 return (PyObject *)type;
982}
983
984/* Internal API to look for a name through the MRO.
985 This returns a borrowed reference, and doesn't set an exception! */
986PyObject *
987_PyType_Lookup(PyTypeObject *type, PyObject *name)
988{
989 int i, n;
990 PyObject *mro, *res, *dict;
991
992 /* For static types, look in tp_dict */
993 if (!(type->tp_flags & Py_TPFLAGS_DYNAMICTYPE)) {
994 dict = type->tp_dict;
995 assert(dict && PyDict_Check(dict));
996 return PyDict_GetItem(dict, name);
997 }
998
999 /* For dynamic types, look in tp_defined of types in MRO */
1000 mro = type->tp_mro;
1001 assert(PyTuple_Check(mro));
1002 n = PyTuple_GET_SIZE(mro);
1003 for (i = 0; i < n; i++) {
1004 type = (PyTypeObject *) PyTuple_GET_ITEM(mro, i);
1005 assert(PyType_Check(type));
1006 dict = type->tp_defined;
1007 assert(dict && PyDict_Check(dict));
1008 res = PyDict_GetItem(dict, name);
1009 if (res != NULL)
1010 return res;
1011 }
1012 return NULL;
1013}
1014
1015/* This is similar to PyObject_GenericGetAttr(),
1016 but uses _PyType_Lookup() instead of just looking in type->tp_dict. */
1017static PyObject *
1018type_getattro(PyTypeObject *type, PyObject *name)
1019{
1020 PyTypeObject *metatype = type->ob_type;
1021 PyObject *descr, *res;
1022 descrgetfunc f;
1023
1024 /* Initialize this type (we'll assume the metatype is initialized) */
1025 if (type->tp_dict == NULL) {
Guido van Rossum528b7eb2001-08-07 17:24:28 +00001026 if (PyType_Ready(type) < 0)
Tim Peters6d6c1a32001-08-02 04:15:00 +00001027 return NULL;
1028 }
1029
1030 /* Get a descriptor from the metatype */
1031 descr = _PyType_Lookup(metatype, name);
1032 f = NULL;
1033 if (descr != NULL) {
1034 f = descr->ob_type->tp_descr_get;
1035 if (f != NULL && PyDescr_IsData(descr))
1036 return f(descr,
1037 (PyObject *)type, (PyObject *)metatype);
1038 }
1039
1040 /* Look in tp_defined of this type and its bases */
1041 res = _PyType_Lookup(type, name);
1042 if (res != NULL) {
1043 f = res->ob_type->tp_descr_get;
1044 if (f != NULL)
1045 return f(res, (PyObject *)NULL, (PyObject *)type);
1046 Py_INCREF(res);
1047 return res;
1048 }
1049
1050 /* Use the descriptor from the metatype */
1051 if (f != NULL) {
1052 res = f(descr, (PyObject *)type, (PyObject *)metatype);
1053 return res;
1054 }
1055 if (descr != NULL) {
1056 Py_INCREF(descr);
1057 return descr;
1058 }
1059
1060 /* Give up */
1061 PyErr_Format(PyExc_AttributeError,
1062 "type object '%.50s' has no attribute '%.400s'",
1063 type->tp_name, PyString_AS_STRING(name));
1064 return NULL;
1065}
1066
1067static int
1068type_setattro(PyTypeObject *type, PyObject *name, PyObject *value)
1069{
1070 if (type->tp_flags & Py_TPFLAGS_DYNAMICTYPE)
1071 return PyObject_GenericSetAttr((PyObject *)type, name, value);
1072 PyErr_SetString(PyExc_TypeError, "can't set type attributes");
1073 return -1;
1074}
1075
1076static void
1077type_dealloc(PyTypeObject *type)
1078{
1079 etype *et;
1080
1081 /* Assert this is a heap-allocated type object */
1082 assert(type->tp_flags & Py_TPFLAGS_HEAPTYPE);
1083 et = (etype *)type;
1084 Py_XDECREF(type->tp_base);
1085 Py_XDECREF(type->tp_dict);
1086 Py_XDECREF(type->tp_bases);
1087 Py_XDECREF(type->tp_mro);
1088 Py_XDECREF(type->tp_defined);
1089 /* XXX more? */
1090 Py_XDECREF(et->name);
1091 Py_XDECREF(et->slots);
1092 type->ob_type->tp_free((PyObject *)type);
1093}
1094
1095static PyMethodDef type_methods[] = {
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00001096 {"mro", (PyCFunction)mro_external, METH_NOARGS,
Tim Peters6d6c1a32001-08-02 04:15:00 +00001097 "mro() -> list\nreturn a type's method resolution order"},
1098 {0}
1099};
1100
1101static char type_doc[] =
1102"type(object) -> the object's type\n"
1103"type(name, bases, dict) -> a new type";
1104
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001105PyTypeObject PyType_Type = {
1106 PyObject_HEAD_INIT(&PyType_Type)
Tim Peters6d6c1a32001-08-02 04:15:00 +00001107 0, /* ob_size */
1108 "type", /* tp_name */
1109 sizeof(etype), /* tp_basicsize */
Guido van Rossum6f799372001-09-20 20:46:19 +00001110 sizeof(PyMemberDef), /* tp_itemsize */
Tim Peters6d6c1a32001-08-02 04:15:00 +00001111 (destructor)type_dealloc, /* tp_dealloc */
1112 0, /* tp_print */
1113 0, /* tp_getattr */
1114 0, /* tp_setattr */
1115 type_compare, /* tp_compare */
1116 (reprfunc)type_repr, /* tp_repr */
1117 0, /* tp_as_number */
1118 0, /* tp_as_sequence */
1119 0, /* tp_as_mapping */
1120 (hashfunc)_Py_HashPointer, /* tp_hash */
1121 (ternaryfunc)type_call, /* tp_call */
1122 0, /* tp_str */
1123 (getattrofunc)type_getattro, /* tp_getattro */
1124 (setattrofunc)type_setattro, /* tp_setattro */
1125 0, /* tp_as_buffer */
1126 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
1127 type_doc, /* tp_doc */
1128 0, /* tp_traverse */
1129 0, /* tp_clear */
1130 0, /* tp_richcompare */
1131 0, /* tp_weaklistoffset */
1132 0, /* tp_iter */
1133 0, /* tp_iternext */
1134 type_methods, /* tp_methods */
1135 type_members, /* tp_members */
1136 type_getsets, /* tp_getset */
1137 0, /* tp_base */
1138 0, /* tp_dict */
1139 0, /* tp_descr_get */
1140 0, /* tp_descr_set */
1141 offsetof(PyTypeObject, tp_dict), /* tp_dictoffset */
1142 0, /* tp_init */
1143 0, /* tp_alloc */
1144 type_new, /* tp_new */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001145};
Tim Peters6d6c1a32001-08-02 04:15:00 +00001146
1147
1148/* The base type of all types (eventually)... except itself. */
1149
1150static int
1151object_init(PyObject *self, PyObject *args, PyObject *kwds)
1152{
1153 return 0;
1154}
1155
1156static void
1157object_dealloc(PyObject *self)
1158{
1159 self->ob_type->tp_free(self);
1160}
1161
Guido van Rossum8e248182001-08-12 05:17:56 +00001162static PyObject *
1163object_repr(PyObject *self)
1164{
Guido van Rossum76e69632001-08-16 18:52:43 +00001165 PyTypeObject *type;
Barry Warsaw7ce36942001-08-24 18:34:26 +00001166 PyObject *mod, *name, *rtn;
Guido van Rossum8e248182001-08-12 05:17:56 +00001167
Guido van Rossum76e69632001-08-16 18:52:43 +00001168 type = self->ob_type;
1169 mod = type_module(type, NULL);
1170 if (mod == NULL)
1171 PyErr_Clear();
1172 else if (!PyString_Check(mod)) {
1173 Py_DECREF(mod);
1174 mod = NULL;
1175 }
1176 name = type_name(type, NULL);
1177 if (name == NULL)
1178 return NULL;
1179 if (mod != NULL && strcmp(PyString_AS_STRING(mod), "__builtin__"))
Guido van Rossumff0e6d62001-09-24 16:03:59 +00001180 rtn = PyString_FromFormat("<%s.%s object at %p>",
Barry Warsaw7ce36942001-08-24 18:34:26 +00001181 PyString_AS_STRING(mod),
1182 PyString_AS_STRING(name),
1183 self);
Guido van Rossum76e69632001-08-16 18:52:43 +00001184 else
Guido van Rossumff0e6d62001-09-24 16:03:59 +00001185 rtn = PyString_FromFormat("<%s object at %p>",
Barry Warsaw7ce36942001-08-24 18:34:26 +00001186 type->tp_name, self);
Guido van Rossum76e69632001-08-16 18:52:43 +00001187 Py_XDECREF(mod);
1188 Py_DECREF(name);
Barry Warsaw7ce36942001-08-24 18:34:26 +00001189 return rtn;
Guido van Rossum8e248182001-08-12 05:17:56 +00001190}
1191
Guido van Rossumb8f63662001-08-15 23:57:02 +00001192static PyObject *
1193object_str(PyObject *self)
1194{
1195 unaryfunc f;
1196
1197 f = self->ob_type->tp_repr;
1198 if (f == NULL)
1199 f = object_repr;
1200 return f(self);
1201}
1202
Guido van Rossum8e248182001-08-12 05:17:56 +00001203static long
1204object_hash(PyObject *self)
1205{
1206 return _Py_HashPointer(self);
1207}
Guido van Rossum8e248182001-08-12 05:17:56 +00001208
Tim Peters6d6c1a32001-08-02 04:15:00 +00001209static void
1210object_free(PyObject *self)
1211{
1212 PyObject_Del(self);
1213}
1214
Guido van Rossum5c294fb2001-09-25 03:43:42 +00001215static PyObject *
1216object_get_class(PyObject *self, void *closure)
1217{
1218 Py_INCREF(self->ob_type);
1219 return (PyObject *)(self->ob_type);
1220}
1221
1222static int
1223equiv_structs(PyTypeObject *a, PyTypeObject *b)
1224{
1225 return a == b ||
1226 (a != NULL &&
1227 b != NULL &&
1228 a->tp_basicsize == b->tp_basicsize &&
1229 a->tp_itemsize == b->tp_itemsize &&
1230 a->tp_dictoffset == b->tp_dictoffset &&
1231 a->tp_weaklistoffset == b->tp_weaklistoffset &&
1232 ((a->tp_flags & Py_TPFLAGS_HAVE_GC) ==
1233 (b->tp_flags & Py_TPFLAGS_HAVE_GC)));
1234}
1235
1236static int
1237same_slots_added(PyTypeObject *a, PyTypeObject *b)
1238{
1239 PyTypeObject *base = a->tp_base;
1240 int size;
1241
1242 if (base != b->tp_base)
1243 return 0;
1244 if (equiv_structs(a, base) && equiv_structs(b, base))
1245 return 1;
1246 size = base->tp_basicsize;
1247 if (a->tp_dictoffset == size && b->tp_dictoffset == size)
1248 size += sizeof(PyObject *);
1249 if (a->tp_weaklistoffset == size && b->tp_weaklistoffset == size)
1250 size += sizeof(PyObject *);
1251 return size == a->tp_basicsize && size == b->tp_basicsize;
1252}
1253
1254static int
1255object_set_class(PyObject *self, PyObject *value, void *closure)
1256{
1257 PyTypeObject *old = self->ob_type;
1258 PyTypeObject *new, *newbase, *oldbase;
1259
1260 if (!PyType_Check(value)) {
1261 PyErr_Format(PyExc_TypeError,
1262 "__class__ must be set to new-style class, not '%s' object",
1263 value->ob_type->tp_name);
1264 return -1;
1265 }
1266 new = (PyTypeObject *)value;
1267 newbase = new;
1268 oldbase = old;
1269 while (equiv_structs(newbase, newbase->tp_base))
1270 newbase = newbase->tp_base;
1271 while (equiv_structs(oldbase, oldbase->tp_base))
1272 oldbase = oldbase->tp_base;
1273 if (newbase != oldbase &&
1274 (newbase->tp_base != oldbase->tp_base ||
1275 !same_slots_added(newbase, oldbase))) {
1276 PyErr_Format(PyExc_TypeError,
1277 "__class__ assignment: "
1278 "'%s' object layout differs from '%s'",
1279 new->tp_name,
1280 old->tp_name);
1281 return -1;
1282 }
1283 if (new->tp_flags & Py_TPFLAGS_HEAPTYPE) {
1284 Py_INCREF(new);
1285 }
1286 self->ob_type = new;
1287 if (old->tp_flags & Py_TPFLAGS_HEAPTYPE) {
1288 Py_DECREF(old);
1289 }
1290 return 0;
1291}
1292
1293static PyGetSetDef object_getsets[] = {
1294 {"__class__", object_get_class, object_set_class,
1295 "the object's class"},
Tim Peters6d6c1a32001-08-02 04:15:00 +00001296 {0}
1297};
1298
Guido van Rossum3926a632001-09-25 16:25:58 +00001299static PyObject *
1300object_reduce(PyObject *self, PyObject *args)
1301{
1302 /* Call copy_reg._reduce(self) */
1303 static PyObject *copy_reg_str;
1304 PyObject *copy_reg, *res;
1305
1306 if (!copy_reg_str) {
1307 copy_reg_str = PyString_InternFromString("copy_reg");
1308 if (copy_reg_str == NULL)
1309 return NULL;
1310 }
1311 copy_reg = PyImport_Import(copy_reg_str);
1312 if (!copy_reg)
1313 return NULL;
1314 res = PyEval_CallMethod(copy_reg, "_reduce", "(O)", self);
1315 Py_DECREF(copy_reg);
1316 return res;
1317}
1318
1319static PyMethodDef object_methods[] = {
1320 {"__reduce__", object_reduce, METH_NOARGS, "helper for pickle"},
1321 {0}
1322};
1323
Tim Peters6d6c1a32001-08-02 04:15:00 +00001324PyTypeObject PyBaseObject_Type = {
1325 PyObject_HEAD_INIT(&PyType_Type)
1326 0, /* ob_size */
1327 "object", /* tp_name */
1328 sizeof(PyObject), /* tp_basicsize */
1329 0, /* tp_itemsize */
1330 (destructor)object_dealloc, /* tp_dealloc */
1331 0, /* tp_print */
1332 0, /* tp_getattr */
1333 0, /* tp_setattr */
1334 0, /* tp_compare */
Guido van Rossumb8f63662001-08-15 23:57:02 +00001335 object_repr, /* tp_repr */
Tim Peters6d6c1a32001-08-02 04:15:00 +00001336 0, /* tp_as_number */
1337 0, /* tp_as_sequence */
1338 0, /* tp_as_mapping */
Guido van Rossumb8f63662001-08-15 23:57:02 +00001339 object_hash, /* tp_hash */
Tim Peters6d6c1a32001-08-02 04:15:00 +00001340 0, /* tp_call */
Guido van Rossumb8f63662001-08-15 23:57:02 +00001341 object_str, /* tp_str */
Tim Peters6d6c1a32001-08-02 04:15:00 +00001342 PyObject_GenericGetAttr, /* tp_getattro */
Guido van Rossum13d52f02001-08-10 21:24:08 +00001343 PyObject_GenericSetAttr, /* tp_setattro */
Tim Peters6d6c1a32001-08-02 04:15:00 +00001344 0, /* tp_as_buffer */
1345 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
1346 "The most base type", /* tp_doc */
1347 0, /* tp_traverse */
1348 0, /* tp_clear */
1349 0, /* tp_richcompare */
1350 0, /* tp_weaklistoffset */
1351 0, /* tp_iter */
1352 0, /* tp_iternext */
Guido van Rossum3926a632001-09-25 16:25:58 +00001353 object_methods, /* tp_methods */
Guido van Rossum5c294fb2001-09-25 03:43:42 +00001354 0, /* tp_members */
1355 object_getsets, /* tp_getset */
Tim Peters6d6c1a32001-08-02 04:15:00 +00001356 0, /* tp_base */
1357 0, /* tp_dict */
1358 0, /* tp_descr_get */
1359 0, /* tp_descr_set */
1360 0, /* tp_dictoffset */
1361 object_init, /* tp_init */
1362 PyType_GenericAlloc, /* tp_alloc */
Guido van Rossumc11e1922001-08-09 19:38:15 +00001363 PyType_GenericNew, /* tp_new */
Tim Peters6d6c1a32001-08-02 04:15:00 +00001364 object_free, /* tp_free */
1365};
1366
1367
1368/* Initialize the __dict__ in a type object */
1369
1370static int
1371add_methods(PyTypeObject *type, PyMethodDef *meth)
1372{
1373 PyObject *dict = type->tp_defined;
1374
1375 for (; meth->ml_name != NULL; meth++) {
1376 PyObject *descr;
1377 if (PyDict_GetItemString(dict, meth->ml_name))
1378 continue;
1379 descr = PyDescr_NewMethod(type, meth);
1380 if (descr == NULL)
1381 return -1;
1382 if (PyDict_SetItemString(dict,meth->ml_name,descr) < 0)
1383 return -1;
1384 Py_DECREF(descr);
1385 }
1386 return 0;
1387}
1388
1389static int
Guido van Rossum6f799372001-09-20 20:46:19 +00001390add_members(PyTypeObject *type, PyMemberDef *memb)
Tim Peters6d6c1a32001-08-02 04:15:00 +00001391{
1392 PyObject *dict = type->tp_defined;
1393
1394 for (; memb->name != NULL; memb++) {
1395 PyObject *descr;
1396 if (PyDict_GetItemString(dict, memb->name))
1397 continue;
1398 descr = PyDescr_NewMember(type, memb);
1399 if (descr == NULL)
1400 return -1;
1401 if (PyDict_SetItemString(dict, memb->name, descr) < 0)
1402 return -1;
1403 Py_DECREF(descr);
1404 }
1405 return 0;
1406}
1407
1408static int
Guido van Rossum32d34c82001-09-20 21:45:26 +00001409add_getset(PyTypeObject *type, PyGetSetDef *gsp)
Tim Peters6d6c1a32001-08-02 04:15:00 +00001410{
1411 PyObject *dict = type->tp_defined;
1412
1413 for (; gsp->name != NULL; gsp++) {
1414 PyObject *descr;
1415 if (PyDict_GetItemString(dict, gsp->name))
1416 continue;
1417 descr = PyDescr_NewGetSet(type, gsp);
1418
1419 if (descr == NULL)
1420 return -1;
1421 if (PyDict_SetItemString(dict, gsp->name, descr) < 0)
1422 return -1;
1423 Py_DECREF(descr);
1424 }
1425 return 0;
1426}
1427
Guido van Rossum13d52f02001-08-10 21:24:08 +00001428static void
1429inherit_special(PyTypeObject *type, PyTypeObject *base)
Tim Peters6d6c1a32001-08-02 04:15:00 +00001430{
1431 int oldsize, newsize;
1432
Guido van Rossum13d52f02001-08-10 21:24:08 +00001433 /* Special flag magic */
1434 if (!type->tp_as_buffer && base->tp_as_buffer) {
1435 type->tp_flags &= ~Py_TPFLAGS_HAVE_GETCHARBUFFER;
1436 type->tp_flags |=
1437 base->tp_flags & Py_TPFLAGS_HAVE_GETCHARBUFFER;
1438 }
1439 if (!type->tp_as_sequence && base->tp_as_sequence) {
1440 type->tp_flags &= ~Py_TPFLAGS_HAVE_SEQUENCE_IN;
1441 type->tp_flags |= base->tp_flags & Py_TPFLAGS_HAVE_SEQUENCE_IN;
1442 }
1443 if ((type->tp_flags & Py_TPFLAGS_HAVE_INPLACEOPS) !=
1444 (base->tp_flags & Py_TPFLAGS_HAVE_INPLACEOPS)) {
1445 if ((!type->tp_as_number && base->tp_as_number) ||
1446 (!type->tp_as_sequence && base->tp_as_sequence)) {
1447 type->tp_flags &= ~Py_TPFLAGS_HAVE_INPLACEOPS;
1448 if (!type->tp_as_number && !type->tp_as_sequence) {
1449 type->tp_flags |= base->tp_flags &
1450 Py_TPFLAGS_HAVE_INPLACEOPS;
1451 }
1452 }
1453 /* Wow */
1454 }
1455 if (!type->tp_as_number && base->tp_as_number) {
1456 type->tp_flags &= ~Py_TPFLAGS_CHECKTYPES;
1457 type->tp_flags |= base->tp_flags & Py_TPFLAGS_CHECKTYPES;
1458 }
1459
1460 /* Copying basicsize is connected to the GC flags */
Neil Schemenauerc806c882001-08-29 23:54:54 +00001461 oldsize = base->tp_basicsize;
1462 newsize = type->tp_basicsize ? type->tp_basicsize : oldsize;
1463 if (!(type->tp_flags & Py_TPFLAGS_HAVE_GC) &&
1464 (base->tp_flags & Py_TPFLAGS_HAVE_GC) &&
Guido van Rossum13d52f02001-08-10 21:24:08 +00001465 (type->tp_flags & Py_TPFLAGS_HAVE_RICHCOMPARE/*GC slots exist*/) &&
1466 (!type->tp_traverse && !type->tp_clear)) {
Neil Schemenauerc806c882001-08-29 23:54:54 +00001467 type->tp_flags |= Py_TPFLAGS_HAVE_GC;
Guido van Rossum13d52f02001-08-10 21:24:08 +00001468 if (type->tp_traverse == NULL)
1469 type->tp_traverse = base->tp_traverse;
1470 if (type->tp_clear == NULL)
1471 type->tp_clear = base->tp_clear;
1472 }
1473 if (type->tp_flags & base->tp_flags & Py_TPFLAGS_HAVE_CLASS) {
1474 if (base != &PyBaseObject_Type ||
1475 (type->tp_flags & Py_TPFLAGS_HEAPTYPE)) {
1476 if (type->tp_new == NULL)
1477 type->tp_new = base->tp_new;
1478 }
1479 }
Neil Schemenauerc806c882001-08-29 23:54:54 +00001480 type->tp_basicsize = newsize;
Guido van Rossum4dd64ab2001-08-14 20:04:48 +00001481
1482 /* Copy other non-function slots */
1483
1484#undef COPYVAL
1485#define COPYVAL(SLOT) \
1486 if (type->SLOT == 0) type->SLOT = base->SLOT
1487
1488 COPYVAL(tp_itemsize);
1489 if (type->tp_flags & base->tp_flags & Py_TPFLAGS_HAVE_WEAKREFS) {
1490 COPYVAL(tp_weaklistoffset);
1491 }
1492 if (type->tp_flags & base->tp_flags & Py_TPFLAGS_HAVE_CLASS) {
1493 COPYVAL(tp_dictoffset);
1494 }
Guido van Rossum13d52f02001-08-10 21:24:08 +00001495}
1496
1497static void
1498inherit_slots(PyTypeObject *type, PyTypeObject *base)
1499{
1500 PyTypeObject *basebase;
1501
1502#undef SLOTDEFINED
Tim Peters6d6c1a32001-08-02 04:15:00 +00001503#undef COPYSLOT
1504#undef COPYNUM
1505#undef COPYSEQ
1506#undef COPYMAP
Guido van Rossum13d52f02001-08-10 21:24:08 +00001507
1508#define SLOTDEFINED(SLOT) \
1509 (base->SLOT != 0 && \
1510 (basebase == NULL || base->SLOT != basebase->SLOT))
1511
Tim Peters6d6c1a32001-08-02 04:15:00 +00001512#define COPYSLOT(SLOT) \
Guido van Rossum13d52f02001-08-10 21:24:08 +00001513 if (!type->SLOT && SLOTDEFINED(SLOT)) type->SLOT = base->SLOT
Tim Peters6d6c1a32001-08-02 04:15:00 +00001514
1515#define COPYNUM(SLOT) COPYSLOT(tp_as_number->SLOT)
1516#define COPYSEQ(SLOT) COPYSLOT(tp_as_sequence->SLOT)
1517#define COPYMAP(SLOT) COPYSLOT(tp_as_mapping->SLOT)
1518
Guido van Rossum13d52f02001-08-10 21:24:08 +00001519 /* This won't inherit indirect slots (from tp_as_number etc.)
1520 if type doesn't provide the space. */
1521
1522 if (type->tp_as_number != NULL && base->tp_as_number != NULL) {
1523 basebase = base->tp_base;
1524 if (basebase->tp_as_number == NULL)
1525 basebase = NULL;
Tim Peters6d6c1a32001-08-02 04:15:00 +00001526 COPYNUM(nb_add);
1527 COPYNUM(nb_subtract);
1528 COPYNUM(nb_multiply);
1529 COPYNUM(nb_divide);
1530 COPYNUM(nb_remainder);
1531 COPYNUM(nb_divmod);
1532 COPYNUM(nb_power);
1533 COPYNUM(nb_negative);
1534 COPYNUM(nb_positive);
1535 COPYNUM(nb_absolute);
1536 COPYNUM(nb_nonzero);
1537 COPYNUM(nb_invert);
1538 COPYNUM(nb_lshift);
1539 COPYNUM(nb_rshift);
1540 COPYNUM(nb_and);
1541 COPYNUM(nb_xor);
1542 COPYNUM(nb_or);
1543 COPYNUM(nb_coerce);
1544 COPYNUM(nb_int);
1545 COPYNUM(nb_long);
1546 COPYNUM(nb_float);
1547 COPYNUM(nb_oct);
1548 COPYNUM(nb_hex);
1549 COPYNUM(nb_inplace_add);
1550 COPYNUM(nb_inplace_subtract);
1551 COPYNUM(nb_inplace_multiply);
1552 COPYNUM(nb_inplace_divide);
1553 COPYNUM(nb_inplace_remainder);
1554 COPYNUM(nb_inplace_power);
1555 COPYNUM(nb_inplace_lshift);
1556 COPYNUM(nb_inplace_rshift);
1557 COPYNUM(nb_inplace_and);
1558 COPYNUM(nb_inplace_xor);
1559 COPYNUM(nb_inplace_or);
Guido van Rossumdc91b992001-08-08 22:26:22 +00001560 if (base->tp_flags & Py_TPFLAGS_CHECKTYPES) {
1561 COPYNUM(nb_true_divide);
1562 COPYNUM(nb_floor_divide);
1563 COPYNUM(nb_inplace_true_divide);
1564 COPYNUM(nb_inplace_floor_divide);
1565 }
Tim Peters6d6c1a32001-08-02 04:15:00 +00001566 }
1567
Guido van Rossum13d52f02001-08-10 21:24:08 +00001568 if (type->tp_as_sequence != NULL && base->tp_as_sequence != NULL) {
1569 basebase = base->tp_base;
1570 if (basebase->tp_as_sequence == NULL)
1571 basebase = NULL;
Tim Peters6d6c1a32001-08-02 04:15:00 +00001572 COPYSEQ(sq_length);
1573 COPYSEQ(sq_concat);
1574 COPYSEQ(sq_repeat);
1575 COPYSEQ(sq_item);
1576 COPYSEQ(sq_slice);
1577 COPYSEQ(sq_ass_item);
1578 COPYSEQ(sq_ass_slice);
1579 COPYSEQ(sq_contains);
1580 COPYSEQ(sq_inplace_concat);
1581 COPYSEQ(sq_inplace_repeat);
1582 }
1583
Guido van Rossum13d52f02001-08-10 21:24:08 +00001584 if (type->tp_as_mapping != NULL && base->tp_as_mapping != NULL) {
1585 basebase = base->tp_base;
1586 if (basebase->tp_as_mapping == NULL)
1587 basebase = NULL;
Tim Peters6d6c1a32001-08-02 04:15:00 +00001588 COPYMAP(mp_length);
1589 COPYMAP(mp_subscript);
1590 COPYMAP(mp_ass_subscript);
1591 }
1592
Guido van Rossum13d52f02001-08-10 21:24:08 +00001593 basebase = base->tp_base;
Tim Peters6d6c1a32001-08-02 04:15:00 +00001594
Tim Peters6d6c1a32001-08-02 04:15:00 +00001595 COPYSLOT(tp_dealloc);
1596 COPYSLOT(tp_print);
1597 if (type->tp_getattr == NULL && type->tp_getattro == NULL) {
1598 type->tp_getattr = base->tp_getattr;
1599 type->tp_getattro = base->tp_getattro;
1600 }
1601 if (type->tp_setattr == NULL && type->tp_setattro == NULL) {
1602 type->tp_setattr = base->tp_setattr;
1603 type->tp_setattro = base->tp_setattro;
1604 }
1605 /* tp_compare see tp_richcompare */
1606 COPYSLOT(tp_repr);
Guido van Rossumb8f63662001-08-15 23:57:02 +00001607 /* tp_hash see tp_richcompare */
Tim Peters6d6c1a32001-08-02 04:15:00 +00001608 COPYSLOT(tp_call);
1609 COPYSLOT(tp_str);
1610 COPYSLOT(tp_as_buffer);
Tim Peters6d6c1a32001-08-02 04:15:00 +00001611 if (type->tp_flags & base->tp_flags & Py_TPFLAGS_HAVE_RICHCOMPARE) {
Guido van Rossumb8f63662001-08-15 23:57:02 +00001612 if (type->tp_compare == NULL &&
1613 type->tp_richcompare == NULL &&
1614 type->tp_hash == NULL)
1615 {
Tim Peters6d6c1a32001-08-02 04:15:00 +00001616 type->tp_compare = base->tp_compare;
1617 type->tp_richcompare = base->tp_richcompare;
Guido van Rossumb8f63662001-08-15 23:57:02 +00001618 type->tp_hash = base->tp_hash;
Tim Peters6d6c1a32001-08-02 04:15:00 +00001619 }
1620 }
1621 else {
1622 COPYSLOT(tp_compare);
1623 }
Tim Peters6d6c1a32001-08-02 04:15:00 +00001624 if (type->tp_flags & base->tp_flags & Py_TPFLAGS_HAVE_ITER) {
1625 COPYSLOT(tp_iter);
1626 COPYSLOT(tp_iternext);
1627 }
1628 if (type->tp_flags & base->tp_flags & Py_TPFLAGS_HAVE_CLASS) {
1629 COPYSLOT(tp_descr_get);
1630 COPYSLOT(tp_descr_set);
1631 COPYSLOT(tp_dictoffset);
1632 COPYSLOT(tp_init);
1633 COPYSLOT(tp_alloc);
Tim Peters6d6c1a32001-08-02 04:15:00 +00001634 COPYSLOT(tp_free);
1635 }
Tim Peters6d6c1a32001-08-02 04:15:00 +00001636}
1637
Guido van Rossum13d52f02001-08-10 21:24:08 +00001638staticforward int add_operators(PyTypeObject *);
1639
Tim Peters6d6c1a32001-08-02 04:15:00 +00001640int
Guido van Rossum528b7eb2001-08-07 17:24:28 +00001641PyType_Ready(PyTypeObject *type)
Tim Peters6d6c1a32001-08-02 04:15:00 +00001642{
1643 PyObject *dict, *bases, *x;
1644 PyTypeObject *base;
1645 int i, n;
1646
Guido van Rossumd614f972001-08-10 17:39:49 +00001647 if (type->tp_flags & Py_TPFLAGS_READY) {
1648 assert(type->tp_dict != NULL);
1649 return 0;
1650 }
1651 assert((type->tp_flags & Py_TPFLAGS_READYING) == 0);
1652 assert(type->tp_dict == NULL);
1653
1654 type->tp_flags |= Py_TPFLAGS_READYING;
Tim Peters6d6c1a32001-08-02 04:15:00 +00001655
1656 /* Initialize tp_base (defaults to BaseObject unless that's us) */
1657 base = type->tp_base;
1658 if (base == NULL && type != &PyBaseObject_Type)
1659 base = type->tp_base = &PyBaseObject_Type;
1660
1661 /* Initialize tp_bases */
1662 bases = type->tp_bases;
1663 if (bases == NULL) {
1664 if (base == NULL)
1665 bases = PyTuple_New(0);
1666 else
1667 bases = Py_BuildValue("(O)", base);
1668 if (bases == NULL)
Guido van Rossumd614f972001-08-10 17:39:49 +00001669 goto error;
Tim Peters6d6c1a32001-08-02 04:15:00 +00001670 type->tp_bases = bases;
1671 }
1672
1673 /* Initialize the base class */
Guido van Rossum0d231ed2001-08-06 16:50:37 +00001674 if (base && base->tp_dict == NULL) {
Guido van Rossum528b7eb2001-08-07 17:24:28 +00001675 if (PyType_Ready(base) < 0)
Guido van Rossumd614f972001-08-10 17:39:49 +00001676 goto error;
Tim Peters6d6c1a32001-08-02 04:15:00 +00001677 }
1678
1679 /* Initialize tp_defined */
1680 dict = type->tp_defined;
1681 if (dict == NULL) {
1682 dict = PyDict_New();
1683 if (dict == NULL)
Guido van Rossumd614f972001-08-10 17:39:49 +00001684 goto error;
Tim Peters6d6c1a32001-08-02 04:15:00 +00001685 type->tp_defined = dict;
1686 }
1687
1688 /* Add type-specific descriptors to tp_defined */
1689 if (add_operators(type) < 0)
Guido van Rossumd614f972001-08-10 17:39:49 +00001690 goto error;
Tim Peters6d6c1a32001-08-02 04:15:00 +00001691 if (type->tp_methods != NULL) {
1692 if (add_methods(type, type->tp_methods) < 0)
Guido van Rossumd614f972001-08-10 17:39:49 +00001693 goto error;
Tim Peters6d6c1a32001-08-02 04:15:00 +00001694 }
1695 if (type->tp_members != NULL) {
1696 if (add_members(type, type->tp_members) < 0)
Guido van Rossumd614f972001-08-10 17:39:49 +00001697 goto error;
Tim Peters6d6c1a32001-08-02 04:15:00 +00001698 }
1699 if (type->tp_getset != NULL) {
1700 if (add_getset(type, type->tp_getset) < 0)
Guido van Rossumd614f972001-08-10 17:39:49 +00001701 goto error;
Tim Peters6d6c1a32001-08-02 04:15:00 +00001702 }
1703
1704 /* Temporarily make tp_dict the same object as tp_defined.
1705 (This is needed to call mro(), and can stay this way for
1706 dynamic types). */
1707 Py_INCREF(type->tp_defined);
1708 type->tp_dict = type->tp_defined;
1709
1710 /* Calculate method resolution order */
1711 if (mro_internal(type) < 0) {
Guido van Rossumd614f972001-08-10 17:39:49 +00001712 goto error;
Tim Peters6d6c1a32001-08-02 04:15:00 +00001713 }
1714
Guido van Rossum13d52f02001-08-10 21:24:08 +00001715 /* Inherit special flags from dominant base */
1716 if (type->tp_base != NULL)
1717 inherit_special(type, type->tp_base);
1718
Tim Peters6d6c1a32001-08-02 04:15:00 +00001719 /* Initialize tp_dict properly */
Guido van Rossum8de86802001-08-12 03:43:35 +00001720 if (PyType_HasFeature(type, Py_TPFLAGS_DYNAMICTYPE)) {
Guido van Rossum8e248182001-08-12 05:17:56 +00001721 /* For a dynamic type, all slots are overridden */
1722 override_slots(type, NULL);
Guido van Rossum8de86802001-08-12 03:43:35 +00001723 }
1724 else {
Tim Peters6d6c1a32001-08-02 04:15:00 +00001725 /* For a static type, tp_dict is the consolidation
Guido van Rossum13d52f02001-08-10 21:24:08 +00001726 of the tp_defined of its bases in MRO. */
Tim Peters6d6c1a32001-08-02 04:15:00 +00001727 Py_DECREF(type->tp_dict);
Guido van Rossum13d52f02001-08-10 21:24:08 +00001728 type->tp_dict = PyDict_Copy(type->tp_defined);
Tim Peters6d6c1a32001-08-02 04:15:00 +00001729 if (type->tp_dict == NULL)
Guido van Rossumd614f972001-08-10 17:39:49 +00001730 goto error;
Tim Peters6d6c1a32001-08-02 04:15:00 +00001731 bases = type->tp_mro;
1732 assert(bases != NULL);
1733 assert(PyTuple_Check(bases));
1734 n = PyTuple_GET_SIZE(bases);
Guido van Rossum13d52f02001-08-10 21:24:08 +00001735 for (i = 1; i < n; i++) {
Tim Peters6d6c1a32001-08-02 04:15:00 +00001736 base = (PyTypeObject *)PyTuple_GET_ITEM(bases, i);
1737 assert(PyType_Check(base));
1738 x = base->tp_defined;
Guido van Rossum13d52f02001-08-10 21:24:08 +00001739 if (x != NULL && PyDict_Merge(type->tp_dict, x, 0) < 0)
Guido van Rossumd614f972001-08-10 17:39:49 +00001740 goto error;
Guido van Rossum13d52f02001-08-10 21:24:08 +00001741 inherit_slots(type, base);
Tim Peters6d6c1a32001-08-02 04:15:00 +00001742 }
1743 }
1744
Guido van Rossum13d52f02001-08-10 21:24:08 +00001745 /* Some more special stuff */
1746 base = type->tp_base;
1747 if (base != NULL) {
1748 if (type->tp_as_number == NULL)
1749 type->tp_as_number = base->tp_as_number;
1750 if (type->tp_as_sequence == NULL)
1751 type->tp_as_sequence = base->tp_as_sequence;
1752 if (type->tp_as_mapping == NULL)
1753 type->tp_as_mapping = base->tp_as_mapping;
1754 }
Tim Peters6d6c1a32001-08-02 04:15:00 +00001755
Guido van Rossum13d52f02001-08-10 21:24:08 +00001756 /* All done -- set the ready flag */
Guido van Rossumd614f972001-08-10 17:39:49 +00001757 assert(type->tp_dict != NULL);
1758 type->tp_flags =
1759 (type->tp_flags & ~Py_TPFLAGS_READYING) | Py_TPFLAGS_READY;
Tim Peters6d6c1a32001-08-02 04:15:00 +00001760 return 0;
Guido van Rossumd614f972001-08-10 17:39:49 +00001761
1762 error:
1763 type->tp_flags &= ~Py_TPFLAGS_READYING;
1764 return -1;
Tim Peters6d6c1a32001-08-02 04:15:00 +00001765}
1766
1767
1768/* Generic wrappers for overloadable 'operators' such as __getitem__ */
1769
1770/* There's a wrapper *function* for each distinct function typedef used
1771 for type object slots (e.g. binaryfunc, ternaryfunc, etc.). There's a
1772 wrapper *table* for each distinct operation (e.g. __len__, __add__).
1773 Most tables have only one entry; the tables for binary operators have two
1774 entries, one regular and one with reversed arguments. */
1775
1776static PyObject *
1777wrap_inquiry(PyObject *self, PyObject *args, void *wrapped)
1778{
1779 inquiry func = (inquiry)wrapped;
1780 int res;
1781
1782 if (!PyArg_ParseTuple(args, ""))
1783 return NULL;
1784 res = (*func)(self);
1785 if (res == -1 && PyErr_Occurred())
1786 return NULL;
1787 return PyInt_FromLong((long)res);
1788}
1789
1790static struct wrapperbase tab_len[] = {
1791 {"__len__", (wrapperfunc)wrap_inquiry, "x.__len__() <==> len(x)"},
1792 {0}
1793};
1794
1795static PyObject *
1796wrap_binaryfunc(PyObject *self, PyObject *args, void *wrapped)
1797{
1798 binaryfunc func = (binaryfunc)wrapped;
1799 PyObject *other;
1800
1801 if (!PyArg_ParseTuple(args, "O", &other))
1802 return NULL;
1803 return (*func)(self, other);
1804}
1805
1806static PyObject *
1807wrap_binaryfunc_r(PyObject *self, PyObject *args, void *wrapped)
1808{
1809 binaryfunc func = (binaryfunc)wrapped;
1810 PyObject *other;
1811
1812 if (!PyArg_ParseTuple(args, "O", &other))
1813 return NULL;
1814 return (*func)(other, self);
1815}
1816
1817#undef BINARY
1818#define BINARY(NAME, OP) \
1819static struct wrapperbase tab_##NAME[] = { \
1820 {"__" #NAME "__", \
1821 (wrapperfunc)wrap_binaryfunc, \
1822 "x.__" #NAME "__(y) <==> " #OP}, \
1823 {"__r" #NAME "__", \
1824 (wrapperfunc)wrap_binaryfunc_r, \
1825 "y.__r" #NAME "__(x) <==> " #OP}, \
1826 {0} \
1827}
1828
1829BINARY(add, "x+y");
1830BINARY(sub, "x-y");
1831BINARY(mul, "x*y");
1832BINARY(div, "x/y");
1833BINARY(mod, "x%y");
1834BINARY(divmod, "divmod(x,y)");
1835BINARY(lshift, "x<<y");
1836BINARY(rshift, "x>>y");
1837BINARY(and, "x&y");
1838BINARY(xor, "x^y");
1839BINARY(or, "x|y");
Guido van Rossum2ed6bf82001-09-27 20:30:07 +00001840
1841static PyObject *
1842wrap_coercefunc(PyObject *self, PyObject *args, void *wrapped)
1843{
1844 coercion func = (coercion)wrapped;
1845 PyObject *other, *res;
1846 int ok;
1847
1848 if (!PyArg_ParseTuple(args, "O", &other))
1849 return NULL;
1850 ok = func(&self, &other);
1851 if (ok < 0)
1852 return NULL;
1853 if (ok > 0) {
1854 Py_INCREF(Py_NotImplemented);
1855 return Py_NotImplemented;
1856 }
1857 res = PyTuple_New(2);
1858 if (res == NULL) {
1859 Py_DECREF(self);
1860 Py_DECREF(other);
1861 return NULL;
1862 }
1863 PyTuple_SET_ITEM(res, 0, self);
1864 PyTuple_SET_ITEM(res, 1, other);
1865 return res;
1866}
1867
1868static struct wrapperbase tab_coerce[] = {
1869 {"__coerce__", (wrapperfunc)wrap_coercefunc,
1870 "x.__coerce__(y) <==> coerce(x, y)"},
1871 {0}
1872};
1873
Guido van Rossum874f15a2001-09-25 21:16:33 +00001874BINARY(floordiv, "x//y");
1875BINARY(truediv, "x/y # true division");
Tim Peters6d6c1a32001-08-02 04:15:00 +00001876
1877static PyObject *
1878wrap_ternaryfunc(PyObject *self, PyObject *args, void *wrapped)
1879{
1880 ternaryfunc func = (ternaryfunc)wrapped;
1881 PyObject *other;
1882 PyObject *third = Py_None;
1883
1884 /* Note: This wrapper only works for __pow__() */
1885
1886 if (!PyArg_ParseTuple(args, "O|O", &other, &third))
1887 return NULL;
1888 return (*func)(self, other, third);
1889}
1890
1891#undef TERNARY
1892#define TERNARY(NAME, OP) \
1893static struct wrapperbase tab_##NAME[] = { \
1894 {"__" #NAME "__", \
1895 (wrapperfunc)wrap_ternaryfunc, \
1896 "x.__" #NAME "__(y, z) <==> " #OP}, \
1897 {"__r" #NAME "__", \
1898 (wrapperfunc)wrap_ternaryfunc, \
1899 "y.__r" #NAME "__(x, z) <==> " #OP}, \
1900 {0} \
1901}
1902
1903TERNARY(pow, "(x**y) % z");
1904
1905#undef UNARY
1906#define UNARY(NAME, OP) \
1907static struct wrapperbase tab_##NAME[] = { \
1908 {"__" #NAME "__", \
1909 (wrapperfunc)wrap_unaryfunc, \
1910 "x.__" #NAME "__() <==> " #OP}, \
1911 {0} \
1912}
1913
1914static PyObject *
1915wrap_unaryfunc(PyObject *self, PyObject *args, void *wrapped)
1916{
1917 unaryfunc func = (unaryfunc)wrapped;
1918
1919 if (!PyArg_ParseTuple(args, ""))
1920 return NULL;
1921 return (*func)(self);
1922}
1923
1924UNARY(neg, "-x");
1925UNARY(pos, "+x");
1926UNARY(abs, "abs(x)");
1927UNARY(nonzero, "x != 0");
1928UNARY(invert, "~x");
1929UNARY(int, "int(x)");
1930UNARY(long, "long(x)");
1931UNARY(float, "float(x)");
1932UNARY(oct, "oct(x)");
1933UNARY(hex, "hex(x)");
1934
1935#undef IBINARY
1936#define IBINARY(NAME, OP) \
1937static struct wrapperbase tab_##NAME[] = { \
1938 {"__" #NAME "__", \
1939 (wrapperfunc)wrap_binaryfunc, \
1940 "x.__" #NAME "__(y) <==> " #OP}, \
1941 {0} \
1942}
1943
1944IBINARY(iadd, "x+=y");
1945IBINARY(isub, "x-=y");
1946IBINARY(imul, "x*=y");
1947IBINARY(idiv, "x/=y");
1948IBINARY(imod, "x%=y");
1949IBINARY(ilshift, "x<<=y");
1950IBINARY(irshift, "x>>=y");
1951IBINARY(iand, "x&=y");
1952IBINARY(ixor, "x^=y");
1953IBINARY(ior, "x|=y");
Guido van Rossum874f15a2001-09-25 21:16:33 +00001954IBINARY(ifloordiv, "x//=y");
1955IBINARY(itruediv, "x/=y # true division");
Tim Peters6d6c1a32001-08-02 04:15:00 +00001956
1957#undef ITERNARY
1958#define ITERNARY(NAME, OP) \
1959static struct wrapperbase tab_##NAME[] = { \
1960 {"__" #NAME "__", \
1961 (wrapperfunc)wrap_ternaryfunc, \
1962 "x.__" #NAME "__(y) <==> " #OP}, \
1963 {0} \
1964}
1965
1966ITERNARY(ipow, "x = (x**y) % z");
1967
1968static struct wrapperbase tab_getitem[] = {
1969 {"__getitem__", (wrapperfunc)wrap_binaryfunc,
1970 "x.__getitem__(y) <==> x[y]"},
1971 {0}
1972};
1973
1974static PyObject *
1975wrap_intargfunc(PyObject *self, PyObject *args, void *wrapped)
1976{
1977 intargfunc func = (intargfunc)wrapped;
1978 int i;
1979
1980 if (!PyArg_ParseTuple(args, "i", &i))
1981 return NULL;
1982 return (*func)(self, i);
1983}
1984
1985static struct wrapperbase tab_mul_int[] = {
1986 {"__mul__", (wrapperfunc)wrap_intargfunc, "x.__mul__(n) <==> x*n"},
1987 {"__rmul__", (wrapperfunc)wrap_intargfunc, "x.__rmul__(n) <==> n*x"},
1988 {0}
1989};
1990
1991static struct wrapperbase tab_concat[] = {
1992 {"__add__", (wrapperfunc)wrap_binaryfunc, "x.__add__(y) <==> x+y"},
1993 {0}
1994};
1995
1996static struct wrapperbase tab_imul_int[] = {
1997 {"__imul__", (wrapperfunc)wrap_intargfunc, "x.__imul__(n) <==> x*=n"},
1998 {0}
1999};
2000
Guido van Rossum5d815f32001-08-17 21:57:47 +00002001static int
2002getindex(PyObject *self, PyObject *arg)
2003{
2004 int i;
2005
2006 i = PyInt_AsLong(arg);
2007 if (i == -1 && PyErr_Occurred())
2008 return -1;
2009 if (i < 0) {
2010 PySequenceMethods *sq = self->ob_type->tp_as_sequence;
2011 if (sq && sq->sq_length) {
2012 int n = (*sq->sq_length)(self);
2013 if (n < 0)
2014 return -1;
2015 i += n;
2016 }
2017 }
2018 return i;
2019}
2020
2021static PyObject *
2022wrap_sq_item(PyObject *self, PyObject *args, void *wrapped)
2023{
2024 intargfunc func = (intargfunc)wrapped;
2025 PyObject *arg;
2026 int i;
2027
2028 if (!PyArg_ParseTuple(args, "O", &arg))
2029 return NULL;
2030 i = getindex(self, arg);
2031 if (i == -1 && PyErr_Occurred())
2032 return NULL;
2033 return (*func)(self, i);
2034}
2035
Tim Peters6d6c1a32001-08-02 04:15:00 +00002036static struct wrapperbase tab_getitem_int[] = {
Guido van Rossum5d815f32001-08-17 21:57:47 +00002037 {"__getitem__", (wrapperfunc)wrap_sq_item,
Tim Peters6d6c1a32001-08-02 04:15:00 +00002038 "x.__getitem__(i) <==> x[i]"},
2039 {0}
2040};
2041
2042static PyObject *
2043wrap_intintargfunc(PyObject *self, PyObject *args, void *wrapped)
2044{
2045 intintargfunc func = (intintargfunc)wrapped;
2046 int i, j;
2047
2048 if (!PyArg_ParseTuple(args, "ii", &i, &j))
2049 return NULL;
2050 return (*func)(self, i, j);
2051}
2052
2053static struct wrapperbase tab_getslice[] = {
2054 {"__getslice__", (wrapperfunc)wrap_intintargfunc,
2055 "x.__getslice__(i, j) <==> x[i:j]"},
2056 {0}
2057};
2058
2059static PyObject *
Guido van Rossum5d815f32001-08-17 21:57:47 +00002060wrap_sq_setitem(PyObject *self, PyObject *args, void *wrapped)
Tim Peters6d6c1a32001-08-02 04:15:00 +00002061{
2062 intobjargproc func = (intobjargproc)wrapped;
2063 int i, res;
Guido van Rossum5d815f32001-08-17 21:57:47 +00002064 PyObject *arg, *value;
Tim Peters6d6c1a32001-08-02 04:15:00 +00002065
Guido van Rossum5d815f32001-08-17 21:57:47 +00002066 if (!PyArg_ParseTuple(args, "OO", &arg, &value))
2067 return NULL;
2068 i = getindex(self, arg);
2069 if (i == -1 && PyErr_Occurred())
Tim Peters6d6c1a32001-08-02 04:15:00 +00002070 return NULL;
2071 res = (*func)(self, i, value);
2072 if (res == -1 && PyErr_Occurred())
2073 return NULL;
2074 Py_INCREF(Py_None);
2075 return Py_None;
2076}
2077
Guido van Rossum2b8d7bd2001-08-02 15:31:58 +00002078static PyObject *
Guido van Rossum5d815f32001-08-17 21:57:47 +00002079wrap_sq_delitem(PyObject *self, PyObject *args, void *wrapped)
Guido van Rossum2b8d7bd2001-08-02 15:31:58 +00002080{
2081 intobjargproc func = (intobjargproc)wrapped;
2082 int i, res;
Guido van Rossum5d815f32001-08-17 21:57:47 +00002083 PyObject *arg;
Guido van Rossum2b8d7bd2001-08-02 15:31:58 +00002084
Guido van Rossum5d815f32001-08-17 21:57:47 +00002085 if (!PyArg_ParseTuple(args, "O", &arg))
2086 return NULL;
2087 i = getindex(self, arg);
2088 if (i == -1 && PyErr_Occurred())
Guido van Rossum2b8d7bd2001-08-02 15:31:58 +00002089 return NULL;
2090 res = (*func)(self, i, NULL);
2091 if (res == -1 && PyErr_Occurred())
2092 return NULL;
2093 Py_INCREF(Py_None);
2094 return Py_None;
2095}
2096
Tim Peters6d6c1a32001-08-02 04:15:00 +00002097static struct wrapperbase tab_setitem_int[] = {
Guido van Rossum5d815f32001-08-17 21:57:47 +00002098 {"__setitem__", (wrapperfunc)wrap_sq_setitem,
Tim Peters6d6c1a32001-08-02 04:15:00 +00002099 "x.__setitem__(i, y) <==> x[i]=y"},
Guido van Rossum5d815f32001-08-17 21:57:47 +00002100 {"__delitem__", (wrapperfunc)wrap_sq_delitem,
Guido van Rossum2b8d7bd2001-08-02 15:31:58 +00002101 "x.__delitem__(y) <==> del x[y]"},
Tim Peters6d6c1a32001-08-02 04:15:00 +00002102 {0}
2103};
2104
2105static PyObject *
2106wrap_intintobjargproc(PyObject *self, PyObject *args, void *wrapped)
2107{
2108 intintobjargproc func = (intintobjargproc)wrapped;
2109 int i, j, res;
2110 PyObject *value;
2111
2112 if (!PyArg_ParseTuple(args, "iiO", &i, &j, &value))
2113 return NULL;
2114 res = (*func)(self, i, j, value);
2115 if (res == -1 && PyErr_Occurred())
2116 return NULL;
2117 Py_INCREF(Py_None);
2118 return Py_None;
2119}
2120
2121static struct wrapperbase tab_setslice[] = {
2122 {"__setslice__", (wrapperfunc)wrap_intintobjargproc,
2123 "x.__setslice__(i, j, y) <==> x[i:j]=y"},
2124 {0}
2125};
2126
2127/* XXX objobjproc is a misnomer; should be objargpred */
2128static PyObject *
2129wrap_objobjproc(PyObject *self, PyObject *args, void *wrapped)
2130{
2131 objobjproc func = (objobjproc)wrapped;
2132 int res;
2133 PyObject *value;
2134
2135 if (!PyArg_ParseTuple(args, "O", &value))
2136 return NULL;
2137 res = (*func)(self, value);
2138 if (res == -1 && PyErr_Occurred())
2139 return NULL;
2140 return PyInt_FromLong((long)res);
2141}
2142
2143static struct wrapperbase tab_contains[] = {
2144 {"__contains__", (wrapperfunc)wrap_objobjproc,
2145 "x.__contains__(y) <==> y in x"},
2146 {0}
2147};
2148
2149static PyObject *
2150wrap_objobjargproc(PyObject *self, PyObject *args, void *wrapped)
2151{
2152 objobjargproc func = (objobjargproc)wrapped;
2153 int res;
2154 PyObject *key, *value;
2155
2156 if (!PyArg_ParseTuple(args, "OO", &key, &value))
2157 return NULL;
2158 res = (*func)(self, key, value);
2159 if (res == -1 && PyErr_Occurred())
2160 return NULL;
2161 Py_INCREF(Py_None);
2162 return Py_None;
2163}
2164
Guido van Rossum2b8d7bd2001-08-02 15:31:58 +00002165static PyObject *
2166wrap_delitem(PyObject *self, PyObject *args, void *wrapped)
2167{
2168 objobjargproc func = (objobjargproc)wrapped;
2169 int res;
2170 PyObject *key;
2171
2172 if (!PyArg_ParseTuple(args, "O", &key))
2173 return NULL;
2174 res = (*func)(self, key, NULL);
2175 if (res == -1 && PyErr_Occurred())
2176 return NULL;
2177 Py_INCREF(Py_None);
2178 return Py_None;
2179}
2180
Tim Peters6d6c1a32001-08-02 04:15:00 +00002181static struct wrapperbase tab_setitem[] = {
2182 {"__setitem__", (wrapperfunc)wrap_objobjargproc,
2183 "x.__setitem__(y, z) <==> x[y]=z"},
Guido van Rossum2b8d7bd2001-08-02 15:31:58 +00002184 {"__delitem__", (wrapperfunc)wrap_delitem,
2185 "x.__delitem__(y) <==> del x[y]"},
Tim Peters6d6c1a32001-08-02 04:15:00 +00002186 {0}
2187};
2188
2189static PyObject *
2190wrap_cmpfunc(PyObject *self, PyObject *args, void *wrapped)
2191{
2192 cmpfunc func = (cmpfunc)wrapped;
2193 int res;
2194 PyObject *other;
2195
2196 if (!PyArg_ParseTuple(args, "O", &other))
2197 return NULL;
Guido van Rossum3d45d8f2001-09-24 18:47:40 +00002198 if (other->ob_type->tp_compare != func &&
2199 !PyType_IsSubtype(other->ob_type, self->ob_type)) {
Guido van Rossumceccae52001-09-18 20:03:57 +00002200 PyErr_Format(
2201 PyExc_TypeError,
2202 "%s.__cmp__(x,y) requires y to be a '%s', not a '%s'",
2203 self->ob_type->tp_name,
2204 self->ob_type->tp_name,
2205 other->ob_type->tp_name);
2206 return NULL;
2207 }
Tim Peters6d6c1a32001-08-02 04:15:00 +00002208 res = (*func)(self, other);
2209 if (PyErr_Occurred())
2210 return NULL;
2211 return PyInt_FromLong((long)res);
2212}
2213
2214static struct wrapperbase tab_cmp[] = {
2215 {"__cmp__", (wrapperfunc)wrap_cmpfunc,
2216 "x.__cmp__(y) <==> cmp(x,y)"},
2217 {0}
2218};
2219
2220static struct wrapperbase tab_repr[] = {
2221 {"__repr__", (wrapperfunc)wrap_unaryfunc,
2222 "x.__repr__() <==> repr(x)"},
2223 {0}
2224};
2225
2226static struct wrapperbase tab_getattr[] = {
Guido van Rossum867a8d22001-09-21 19:29:08 +00002227 {"__getattribute__", (wrapperfunc)wrap_binaryfunc,
2228 "x.__getattribute__('name') <==> x.name"},
Tim Peters6d6c1a32001-08-02 04:15:00 +00002229 {0}
2230};
2231
2232static PyObject *
2233wrap_setattr(PyObject *self, PyObject *args, void *wrapped)
2234{
2235 setattrofunc func = (setattrofunc)wrapped;
2236 int res;
2237 PyObject *name, *value;
2238
2239 if (!PyArg_ParseTuple(args, "OO", &name, &value))
2240 return NULL;
2241 res = (*func)(self, name, value);
2242 if (res < 0)
2243 return NULL;
2244 Py_INCREF(Py_None);
2245 return Py_None;
2246}
2247
2248static PyObject *
2249wrap_delattr(PyObject *self, PyObject *args, void *wrapped)
2250{
2251 setattrofunc func = (setattrofunc)wrapped;
2252 int res;
2253 PyObject *name;
2254
2255 if (!PyArg_ParseTuple(args, "O", &name))
2256 return NULL;
2257 res = (*func)(self, name, NULL);
2258 if (res < 0)
2259 return NULL;
2260 Py_INCREF(Py_None);
2261 return Py_None;
2262}
2263
2264static struct wrapperbase tab_setattr[] = {
2265 {"__setattr__", (wrapperfunc)wrap_setattr,
2266 "x.__setattr__('name', value) <==> x.name = value"},
2267 {"__delattr__", (wrapperfunc)wrap_delattr,
2268 "x.__delattr__('name') <==> del x.name"},
2269 {0}
2270};
2271
2272static PyObject *
2273wrap_hashfunc(PyObject *self, PyObject *args, void *wrapped)
2274{
2275 hashfunc func = (hashfunc)wrapped;
2276 long res;
2277
2278 if (!PyArg_ParseTuple(args, ""))
2279 return NULL;
2280 res = (*func)(self);
2281 if (res == -1 && PyErr_Occurred())
2282 return NULL;
2283 return PyInt_FromLong(res);
2284}
2285
2286static struct wrapperbase tab_hash[] = {
2287 {"__hash__", (wrapperfunc)wrap_hashfunc,
2288 "x.__hash__() <==> hash(x)"},
2289 {0}
2290};
2291
2292static PyObject *
2293wrap_call(PyObject *self, PyObject *args, void *wrapped)
2294{
2295 ternaryfunc func = (ternaryfunc)wrapped;
2296
2297 /* XXX What about keyword arguments? */
2298 return (*func)(self, args, NULL);
2299}
2300
2301static struct wrapperbase tab_call[] = {
2302 {"__call__", (wrapperfunc)wrap_call,
2303 "x.__call__(...) <==> x(...)"},
2304 {0}
2305};
2306
2307static struct wrapperbase tab_str[] = {
2308 {"__str__", (wrapperfunc)wrap_unaryfunc,
2309 "x.__str__() <==> str(x)"},
2310 {0}
2311};
2312
2313static PyObject *
2314wrap_richcmpfunc(PyObject *self, PyObject *args, void *wrapped, int op)
2315{
2316 richcmpfunc func = (richcmpfunc)wrapped;
2317 PyObject *other;
2318
2319 if (!PyArg_ParseTuple(args, "O", &other))
2320 return NULL;
2321 return (*func)(self, other, op);
2322}
2323
2324#undef RICHCMP_WRAPPER
2325#define RICHCMP_WRAPPER(NAME, OP) \
2326static PyObject * \
2327richcmp_##NAME(PyObject *self, PyObject *args, void *wrapped) \
2328{ \
2329 return wrap_richcmpfunc(self, args, wrapped, OP); \
2330}
2331
Jack Jansen8e938b42001-08-08 15:29:49 +00002332RICHCMP_WRAPPER(lt, Py_LT)
2333RICHCMP_WRAPPER(le, Py_LE)
2334RICHCMP_WRAPPER(eq, Py_EQ)
2335RICHCMP_WRAPPER(ne, Py_NE)
2336RICHCMP_WRAPPER(gt, Py_GT)
2337RICHCMP_WRAPPER(ge, Py_GE)
Tim Peters6d6c1a32001-08-02 04:15:00 +00002338
2339#undef RICHCMP_ENTRY
2340#define RICHCMP_ENTRY(NAME, EXPR) \
2341 {"__" #NAME "__", (wrapperfunc)richcmp_##NAME, \
2342 "x.__" #NAME "__(y) <==> " EXPR}
2343
2344static struct wrapperbase tab_richcmp[] = {
2345 RICHCMP_ENTRY(lt, "x<y"),
2346 RICHCMP_ENTRY(le, "x<=y"),
2347 RICHCMP_ENTRY(eq, "x==y"),
2348 RICHCMP_ENTRY(ne, "x!=y"),
2349 RICHCMP_ENTRY(gt, "x>y"),
2350 RICHCMP_ENTRY(ge, "x>=y"),
2351 {0}
2352};
2353
2354static struct wrapperbase tab_iter[] = {
2355 {"__iter__", (wrapperfunc)wrap_unaryfunc, "x.__iter__() <==> iter(x)"},
2356 {0}
2357};
2358
2359static PyObject *
2360wrap_next(PyObject *self, PyObject *args, void *wrapped)
2361{
2362 unaryfunc func = (unaryfunc)wrapped;
2363 PyObject *res;
2364
2365 if (!PyArg_ParseTuple(args, ""))
2366 return NULL;
2367 res = (*func)(self);
2368 if (res == NULL && !PyErr_Occurred())
2369 PyErr_SetNone(PyExc_StopIteration);
2370 return res;
2371}
2372
2373static struct wrapperbase tab_next[] = {
2374 {"next", (wrapperfunc)wrap_next,
2375 "x.next() -> the next value, or raise StopIteration"},
2376 {0}
2377};
2378
2379static PyObject *
2380wrap_descr_get(PyObject *self, PyObject *args, void *wrapped)
2381{
2382 descrgetfunc func = (descrgetfunc)wrapped;
2383 PyObject *obj;
2384 PyObject *type = NULL;
2385
2386 if (!PyArg_ParseTuple(args, "O|O", &obj, &type))
2387 return NULL;
Tim Peters6d6c1a32001-08-02 04:15:00 +00002388 return (*func)(self, obj, type);
2389}
2390
2391static struct wrapperbase tab_descr_get[] = {
2392 {"__get__", (wrapperfunc)wrap_descr_get,
2393 "descr.__get__(obj, type) -> value"},
2394 {0}
2395};
2396
2397static PyObject *
2398wrap_descrsetfunc(PyObject *self, PyObject *args, void *wrapped)
2399{
2400 descrsetfunc func = (descrsetfunc)wrapped;
2401 PyObject *obj, *value;
2402 int ret;
2403
2404 if (!PyArg_ParseTuple(args, "OO", &obj, &value))
2405 return NULL;
2406 ret = (*func)(self, obj, value);
2407 if (ret < 0)
2408 return NULL;
2409 Py_INCREF(Py_None);
2410 return Py_None;
2411}
2412
2413static struct wrapperbase tab_descr_set[] = {
2414 {"__set__", (wrapperfunc)wrap_descrsetfunc,
2415 "descr.__set__(obj, value)"},
2416 {0}
2417};
2418
2419static PyObject *
2420wrap_init(PyObject *self, PyObject *args, void *wrapped)
2421{
2422 initproc func = (initproc)wrapped;
2423
2424 /* XXX What about keyword arguments? */
2425 if (func(self, args, NULL) < 0)
2426 return NULL;
2427 Py_INCREF(Py_None);
2428 return Py_None;
2429}
2430
2431static struct wrapperbase tab_init[] = {
2432 {"__init__", (wrapperfunc)wrap_init,
2433 "x.__init__(...) initializes x; "
2434 "see x.__type__.__doc__ for signature"},
2435 {0}
2436};
2437
2438static PyObject *
Guido van Rossum0d231ed2001-08-06 16:50:37 +00002439tp_new_wrapper(PyObject *self, PyObject *args, PyObject *kwds)
Tim Peters6d6c1a32001-08-02 04:15:00 +00002440{
Barry Warsaw60f01882001-08-22 19:24:42 +00002441 PyTypeObject *type, *subtype, *staticbase;
Guido van Rossum0d231ed2001-08-06 16:50:37 +00002442 PyObject *arg0, *res;
2443
2444 if (self == NULL || !PyType_Check(self))
2445 Py_FatalError("__new__() called with non-type 'self'");
2446 type = (PyTypeObject *)self;
2447 if (!PyTuple_Check(args) || PyTuple_GET_SIZE(args) < 1) {
Barry Warsaw60f01882001-08-22 19:24:42 +00002448 PyErr_Format(PyExc_TypeError,
2449 "%s.__new__(): not enough arguments",
2450 type->tp_name);
Guido van Rossum0d231ed2001-08-06 16:50:37 +00002451 return NULL;
2452 }
2453 arg0 = PyTuple_GET_ITEM(args, 0);
2454 if (!PyType_Check(arg0)) {
Barry Warsaw60f01882001-08-22 19:24:42 +00002455 PyErr_Format(PyExc_TypeError,
2456 "%s.__new__(X): X is not a type object (%s)",
2457 type->tp_name,
2458 arg0->ob_type->tp_name);
Guido van Rossum0d231ed2001-08-06 16:50:37 +00002459 return NULL;
2460 }
2461 subtype = (PyTypeObject *)arg0;
2462 if (!PyType_IsSubtype(subtype, type)) {
Barry Warsaw60f01882001-08-22 19:24:42 +00002463 PyErr_Format(PyExc_TypeError,
2464 "%s.__new__(%s): %s is not a subtype of %s",
2465 type->tp_name,
2466 subtype->tp_name,
2467 subtype->tp_name,
2468 type->tp_name);
Guido van Rossum0d231ed2001-08-06 16:50:37 +00002469 return NULL;
2470 }
Barry Warsaw60f01882001-08-22 19:24:42 +00002471
2472 /* Check that the use doesn't do something silly and unsafe like
2473 object.__new__(dictionary). To do this, we check that the
2474 most derived base that's not a heap type is this type. */
2475 staticbase = subtype;
2476 while (staticbase && (staticbase->tp_flags & Py_TPFLAGS_HEAPTYPE))
2477 staticbase = staticbase->tp_base;
Guido van Rossuma8c60f42001-09-14 19:43:36 +00002478 if (staticbase->tp_new != type->tp_new) {
Barry Warsaw60f01882001-08-22 19:24:42 +00002479 PyErr_Format(PyExc_TypeError,
2480 "%s.__new__(%s) is not safe, use %s.__new__()",
2481 type->tp_name,
2482 subtype->tp_name,
2483 staticbase == NULL ? "?" : staticbase->tp_name);
2484 return NULL;
2485 }
2486
Guido van Rossum0d231ed2001-08-06 16:50:37 +00002487 args = PyTuple_GetSlice(args, 1, PyTuple_GET_SIZE(args));
2488 if (args == NULL)
2489 return NULL;
2490 res = type->tp_new(subtype, args, kwds);
2491 Py_DECREF(args);
2492 return res;
Tim Peters6d6c1a32001-08-02 04:15:00 +00002493}
2494
Guido van Rossum0d231ed2001-08-06 16:50:37 +00002495static struct PyMethodDef tp_new_methoddef[] = {
2496 {"__new__", (PyCFunction)tp_new_wrapper, METH_KEYWORDS,
2497 "T.__new__(S, ...) -> a new object with type S, a subtype of T"},
Tim Peters6d6c1a32001-08-02 04:15:00 +00002498 {0}
2499};
2500
2501static int
Guido van Rossum0d231ed2001-08-06 16:50:37 +00002502add_tp_new_wrapper(PyTypeObject *type)
2503{
Guido van Rossumf040ede2001-08-07 16:40:56 +00002504 PyObject *func;
Guido van Rossum0d231ed2001-08-06 16:50:37 +00002505
Guido van Rossumf040ede2001-08-07 16:40:56 +00002506 if (PyDict_GetItemString(type->tp_defined, "__new__") != NULL)
2507 return 0;
2508 func = PyCFunction_New(tp_new_methoddef, (PyObject *)type);
Guido van Rossum0d231ed2001-08-06 16:50:37 +00002509 if (func == NULL)
2510 return -1;
2511 return PyDict_SetItemString(type->tp_defined, "__new__", func);
2512}
2513
Guido van Rossum13d52f02001-08-10 21:24:08 +00002514static int
2515add_wrappers(PyTypeObject *type, struct wrapperbase *wraps, void *wrapped)
2516{
2517 PyObject *dict = type->tp_defined;
2518
2519 for (; wraps->name != NULL; wraps++) {
2520 PyObject *descr;
2521 if (PyDict_GetItemString(dict, wraps->name))
2522 continue;
2523 descr = PyDescr_NewWrapper(type, wraps, wrapped);
2524 if (descr == NULL)
2525 return -1;
2526 if (PyDict_SetItemString(dict, wraps->name, descr) < 0)
2527 return -1;
2528 Py_DECREF(descr);
2529 }
2530 return 0;
2531}
2532
Guido van Rossum528b7eb2001-08-07 17:24:28 +00002533/* This function is called by PyType_Ready() to populate the type's
Guido van Rossumf040ede2001-08-07 16:40:56 +00002534 dictionary with method descriptors for function slots. For each
2535 function slot (like tp_repr) that's defined in the type, one or
2536 more corresponding descriptors are added in the type's tp_defined
2537 dictionary under the appropriate name (like __repr__). Some
2538 function slots cause more than one descriptor to be added (for
2539 example, the nb_add slot adds both __add__ and __radd__
2540 descriptors) and some function slots compete for the same
2541 descriptor (for example both sq_item and mp_subscript generate a
2542 __getitem__ descriptor). This only adds new descriptors and
2543 doesn't overwrite entries in tp_defined that were previously
2544 defined. The descriptors contain a reference to the C function
2545 they must call, so that it's safe if they are copied into a
2546 subtype's __dict__ and the subtype has a different C function in
2547 its slot -- calling the method defined by the descriptor will call
2548 the C function that was used to create it, rather than the C
2549 function present in the slot when it is called. (This is important
2550 because a subtype may have a C function in the slot that calls the
2551 method from the dictionary, and we want to avoid infinite recursion
2552 here.) */
2553
Guido van Rossum0d231ed2001-08-06 16:50:37 +00002554static int
Tim Peters6d6c1a32001-08-02 04:15:00 +00002555add_operators(PyTypeObject *type)
2556{
2557 PySequenceMethods *sq;
2558 PyMappingMethods *mp;
2559 PyNumberMethods *nb;
2560
2561#undef ADD
2562#define ADD(SLOT, TABLE) \
2563 if (SLOT) { \
2564 if (add_wrappers(type, TABLE, (void *)(SLOT)) < 0) \
2565 return -1; \
2566 }
2567
2568 if ((sq = type->tp_as_sequence) != NULL) {
2569 ADD(sq->sq_length, tab_len);
2570 ADD(sq->sq_concat, tab_concat);
2571 ADD(sq->sq_repeat, tab_mul_int);
2572 ADD(sq->sq_item, tab_getitem_int);
2573 ADD(sq->sq_slice, tab_getslice);
2574 ADD(sq->sq_ass_item, tab_setitem_int);
2575 ADD(sq->sq_ass_slice, tab_setslice);
2576 ADD(sq->sq_contains, tab_contains);
2577 ADD(sq->sq_inplace_concat, tab_iadd);
2578 ADD(sq->sq_inplace_repeat, tab_imul_int);
2579 }
2580
2581 if ((mp = type->tp_as_mapping) != NULL) {
2582 if (sq->sq_length == NULL)
2583 ADD(mp->mp_length, tab_len);
2584 ADD(mp->mp_subscript, tab_getitem);
2585 ADD(mp->mp_ass_subscript, tab_setitem);
2586 }
2587
2588 /* We don't support "old-style numbers" because their binary
2589 operators require that both arguments have the same type;
2590 the wrappers here only work for new-style numbers. */
2591 if ((type->tp_flags & Py_TPFLAGS_CHECKTYPES) &&
2592 (nb = type->tp_as_number) != NULL) {
2593 ADD(nb->nb_add, tab_add);
2594 ADD(nb->nb_subtract, tab_sub);
2595 ADD(nb->nb_multiply, tab_mul);
2596 ADD(nb->nb_divide, tab_div);
2597 ADD(nb->nb_remainder, tab_mod);
2598 ADD(nb->nb_divmod, tab_divmod);
2599 ADD(nb->nb_power, tab_pow);
2600 ADD(nb->nb_negative, tab_neg);
2601 ADD(nb->nb_positive, tab_pos);
2602 ADD(nb->nb_absolute, tab_abs);
2603 ADD(nb->nb_nonzero, tab_nonzero);
2604 ADD(nb->nb_invert, tab_invert);
2605 ADD(nb->nb_lshift, tab_lshift);
2606 ADD(nb->nb_rshift, tab_rshift);
2607 ADD(nb->nb_and, tab_and);
2608 ADD(nb->nb_xor, tab_xor);
2609 ADD(nb->nb_or, tab_or);
Guido van Rossum2ed6bf82001-09-27 20:30:07 +00002610 ADD(nb->nb_coerce, tab_coerce);
Tim Peters6d6c1a32001-08-02 04:15:00 +00002611 ADD(nb->nb_int, tab_int);
2612 ADD(nb->nb_long, tab_long);
2613 ADD(nb->nb_float, tab_float);
2614 ADD(nb->nb_oct, tab_oct);
2615 ADD(nb->nb_hex, tab_hex);
2616 ADD(nb->nb_inplace_add, tab_iadd);
2617 ADD(nb->nb_inplace_subtract, tab_isub);
2618 ADD(nb->nb_inplace_multiply, tab_imul);
2619 ADD(nb->nb_inplace_divide, tab_idiv);
2620 ADD(nb->nb_inplace_remainder, tab_imod);
2621 ADD(nb->nb_inplace_power, tab_ipow);
2622 ADD(nb->nb_inplace_lshift, tab_ilshift);
2623 ADD(nb->nb_inplace_rshift, tab_irshift);
2624 ADD(nb->nb_inplace_and, tab_iand);
2625 ADD(nb->nb_inplace_xor, tab_ixor);
2626 ADD(nb->nb_inplace_or, tab_ior);
Guido van Rossum874f15a2001-09-25 21:16:33 +00002627 if (type->tp_flags & Py_TPFLAGS_CHECKTYPES) {
2628 ADD(nb->nb_floor_divide, tab_floordiv);
2629 ADD(nb->nb_true_divide, tab_truediv);
2630 ADD(nb->nb_inplace_floor_divide, tab_ifloordiv);
2631 ADD(nb->nb_inplace_true_divide, tab_itruediv);
2632 }
Tim Peters6d6c1a32001-08-02 04:15:00 +00002633 }
2634
2635 ADD(type->tp_getattro, tab_getattr);
2636 ADD(type->tp_setattro, tab_setattr);
2637 ADD(type->tp_compare, tab_cmp);
2638 ADD(type->tp_repr, tab_repr);
2639 ADD(type->tp_hash, tab_hash);
2640 ADD(type->tp_call, tab_call);
2641 ADD(type->tp_str, tab_str);
2642 ADD(type->tp_richcompare, tab_richcmp);
2643 ADD(type->tp_iter, tab_iter);
2644 ADD(type->tp_iternext, tab_next);
2645 ADD(type->tp_descr_get, tab_descr_get);
2646 ADD(type->tp_descr_set, tab_descr_set);
2647 ADD(type->tp_init, tab_init);
2648
Guido van Rossumf040ede2001-08-07 16:40:56 +00002649 if (type->tp_new != NULL) {
2650 if (add_tp_new_wrapper(type) < 0)
2651 return -1;
2652 }
Tim Peters6d6c1a32001-08-02 04:15:00 +00002653
2654 return 0;
2655}
2656
Guido van Rossumf040ede2001-08-07 16:40:56 +00002657/* Slot wrappers that call the corresponding __foo__ slot. See comments
2658 below at override_slots() for more explanation. */
Tim Peters6d6c1a32001-08-02 04:15:00 +00002659
Guido van Rossumdc91b992001-08-08 22:26:22 +00002660#define SLOT0(FUNCNAME, OPSTR) \
Tim Peters6d6c1a32001-08-02 04:15:00 +00002661static PyObject * \
Guido van Rossumdc91b992001-08-08 22:26:22 +00002662FUNCNAME(PyObject *self) \
Tim Peters6d6c1a32001-08-02 04:15:00 +00002663{ \
Guido van Rossum5592e4d2001-08-28 18:28:21 +00002664 static PyObject *cache_str; \
Guido van Rossum717ce002001-09-14 16:58:08 +00002665 return call_method(self, OPSTR, &cache_str, "()"); \
Tim Peters6d6c1a32001-08-02 04:15:00 +00002666}
2667
Guido van Rossumdc91b992001-08-08 22:26:22 +00002668#define SLOT1(FUNCNAME, OPSTR, ARG1TYPE, ARGCODES) \
Tim Peters6d6c1a32001-08-02 04:15:00 +00002669static PyObject * \
Guido van Rossumdc91b992001-08-08 22:26:22 +00002670FUNCNAME(PyObject *self, ARG1TYPE arg1) \
Tim Peters6d6c1a32001-08-02 04:15:00 +00002671{ \
Guido van Rossum2730b132001-08-28 18:22:14 +00002672 static PyObject *cache_str; \
Guido van Rossum717ce002001-09-14 16:58:08 +00002673 return call_method(self, OPSTR, &cache_str, "(" ARGCODES ")", arg1); \
Tim Peters6d6c1a32001-08-02 04:15:00 +00002674}
2675
Guido van Rossumdc91b992001-08-08 22:26:22 +00002676
2677#define SLOT1BINFULL(FUNCNAME, TESTFUNC, SLOTNAME, OPSTR, ROPSTR) \
Tim Peters6d6c1a32001-08-02 04:15:00 +00002678static PyObject * \
Guido van Rossumdc91b992001-08-08 22:26:22 +00002679FUNCNAME(PyObject *self, PyObject *other) \
Tim Peters6d6c1a32001-08-02 04:15:00 +00002680{ \
Guido van Rossum2730b132001-08-28 18:22:14 +00002681 static PyObject *cache_str, *rcache_str; \
Guido van Rossumdc91b992001-08-08 22:26:22 +00002682 if (self->ob_type->tp_as_number != NULL && \
2683 self->ob_type->tp_as_number->SLOTNAME == TESTFUNC) { \
2684 PyObject *r; \
Guido van Rossumf21c6be2001-09-14 17:51:50 +00002685 r = call_maybe( \
Guido van Rossum717ce002001-09-14 16:58:08 +00002686 self, OPSTR, &cache_str, "(O)", other); \
Guido van Rossumdc91b992001-08-08 22:26:22 +00002687 if (r != Py_NotImplemented || \
2688 other->ob_type == self->ob_type) \
2689 return r; \
2690 Py_DECREF(r); \
2691 } \
2692 if (other->ob_type->tp_as_number != NULL && \
2693 other->ob_type->tp_as_number->SLOTNAME == TESTFUNC) { \
Guido van Rossumf21c6be2001-09-14 17:51:50 +00002694 return call_maybe( \
Guido van Rossum717ce002001-09-14 16:58:08 +00002695 other, ROPSTR, &rcache_str, "(O)", self); \
Guido van Rossumdc91b992001-08-08 22:26:22 +00002696 } \
2697 Py_INCREF(Py_NotImplemented); \
2698 return Py_NotImplemented; \
2699}
2700
2701#define SLOT1BIN(FUNCNAME, SLOTNAME, OPSTR, ROPSTR) \
2702 SLOT1BINFULL(FUNCNAME, FUNCNAME, SLOTNAME, OPSTR, ROPSTR)
2703
2704#define SLOT2(FUNCNAME, OPSTR, ARG1TYPE, ARG2TYPE, ARGCODES) \
2705static PyObject * \
2706FUNCNAME(PyObject *self, ARG1TYPE arg1, ARG2TYPE arg2) \
2707{ \
Guido van Rossum2730b132001-08-28 18:22:14 +00002708 static PyObject *cache_str; \
Guido van Rossum717ce002001-09-14 16:58:08 +00002709 return call_method(self, OPSTR, &cache_str, \
2710 "(" ARGCODES ")", arg1, arg2); \
Tim Peters6d6c1a32001-08-02 04:15:00 +00002711}
2712
2713static int
2714slot_sq_length(PyObject *self)
2715{
Guido van Rossum2730b132001-08-28 18:22:14 +00002716 static PyObject *len_str;
Guido van Rossum717ce002001-09-14 16:58:08 +00002717 PyObject *res = call_method(self, "__len__", &len_str, "()");
Tim Peters6d6c1a32001-08-02 04:15:00 +00002718
2719 if (res == NULL)
2720 return -1;
2721 return (int)PyInt_AsLong(res);
2722}
2723
Guido van Rossumdc91b992001-08-08 22:26:22 +00002724SLOT1(slot_sq_concat, "__add__", PyObject *, "O")
2725SLOT1(slot_sq_repeat, "__mul__", int, "i")
2726SLOT1(slot_sq_item, "__getitem__", int, "i")
2727SLOT2(slot_sq_slice, "__getslice__", int, int, "ii")
Tim Peters6d6c1a32001-08-02 04:15:00 +00002728
2729static int
2730slot_sq_ass_item(PyObject *self, int index, PyObject *value)
2731{
2732 PyObject *res;
Guido van Rossum2730b132001-08-28 18:22:14 +00002733 static PyObject *delitem_str, *setitem_str;
Tim Peters6d6c1a32001-08-02 04:15:00 +00002734
2735 if (value == NULL)
Guido van Rossum2730b132001-08-28 18:22:14 +00002736 res = call_method(self, "__delitem__", &delitem_str,
Guido van Rossum717ce002001-09-14 16:58:08 +00002737 "(i)", index);
Tim Peters6d6c1a32001-08-02 04:15:00 +00002738 else
Guido van Rossum2730b132001-08-28 18:22:14 +00002739 res = call_method(self, "__setitem__", &setitem_str,
Guido van Rossum717ce002001-09-14 16:58:08 +00002740 "(iO)", index, value);
Tim Peters6d6c1a32001-08-02 04:15:00 +00002741 if (res == NULL)
2742 return -1;
2743 Py_DECREF(res);
2744 return 0;
2745}
2746
2747static int
2748slot_sq_ass_slice(PyObject *self, int i, int j, PyObject *value)
2749{
2750 PyObject *res;
Guido van Rossum2730b132001-08-28 18:22:14 +00002751 static PyObject *delslice_str, *setslice_str;
Tim Peters6d6c1a32001-08-02 04:15:00 +00002752
2753 if (value == NULL)
Guido van Rossum2730b132001-08-28 18:22:14 +00002754 res = call_method(self, "__delslice__", &delslice_str,
Guido van Rossum717ce002001-09-14 16:58:08 +00002755 "(ii)", i, j);
Tim Peters6d6c1a32001-08-02 04:15:00 +00002756 else
Guido van Rossum2730b132001-08-28 18:22:14 +00002757 res = call_method(self, "__setslice__", &setslice_str,
Guido van Rossum717ce002001-09-14 16:58:08 +00002758 "(iiO)", i, j, value);
Tim Peters6d6c1a32001-08-02 04:15:00 +00002759 if (res == NULL)
2760 return -1;
2761 Py_DECREF(res);
2762 return 0;
2763}
2764
2765static int
2766slot_sq_contains(PyObject *self, PyObject *value)
2767{
Guido van Rossumb8f63662001-08-15 23:57:02 +00002768 PyObject *func, *res, *args;
Guido van Rossum60718732001-08-28 17:47:51 +00002769 static PyObject *contains_str;
Tim Peters6d6c1a32001-08-02 04:15:00 +00002770
Guido van Rossum60718732001-08-28 17:47:51 +00002771 func = lookup_method(self, "__contains__", &contains_str);
Guido van Rossumb8f63662001-08-15 23:57:02 +00002772
2773 if (func != NULL) {
2774 args = Py_BuildValue("(O)", value);
2775 if (args == NULL)
2776 res = NULL;
2777 else {
Guido van Rossum717ce002001-09-14 16:58:08 +00002778 res = PyObject_Call(func, args, NULL);
Guido van Rossumb8f63662001-08-15 23:57:02 +00002779 Py_DECREF(args);
2780 }
2781 Py_DECREF(func);
2782 if (res == NULL)
2783 return -1;
2784 return PyObject_IsTrue(res);
2785 }
2786 else {
2787 PyErr_Clear();
Tim Peters16a77ad2001-09-08 04:00:12 +00002788 return _PySequence_IterSearch(self, value,
2789 PY_ITERSEARCH_CONTAINS);
Guido van Rossumb8f63662001-08-15 23:57:02 +00002790 }
Tim Peters6d6c1a32001-08-02 04:15:00 +00002791}
2792
Guido van Rossumdc91b992001-08-08 22:26:22 +00002793SLOT1(slot_sq_inplace_concat, "__iadd__", PyObject *, "O")
2794SLOT1(slot_sq_inplace_repeat, "__imul__", int, "i")
Tim Peters6d6c1a32001-08-02 04:15:00 +00002795
2796#define slot_mp_length slot_sq_length
2797
Guido van Rossumdc91b992001-08-08 22:26:22 +00002798SLOT1(slot_mp_subscript, "__getitem__", PyObject *, "O")
Tim Peters6d6c1a32001-08-02 04:15:00 +00002799
2800static int
2801slot_mp_ass_subscript(PyObject *self, PyObject *key, PyObject *value)
2802{
2803 PyObject *res;
Guido van Rossum2730b132001-08-28 18:22:14 +00002804 static PyObject *delitem_str, *setitem_str;
Tim Peters6d6c1a32001-08-02 04:15:00 +00002805
2806 if (value == NULL)
Guido van Rossum2730b132001-08-28 18:22:14 +00002807 res = call_method(self, "__delitem__", &delitem_str,
Guido van Rossum717ce002001-09-14 16:58:08 +00002808 "(O)", key);
Tim Peters6d6c1a32001-08-02 04:15:00 +00002809 else
Guido van Rossum2730b132001-08-28 18:22:14 +00002810 res = call_method(self, "__setitem__", &setitem_str,
Guido van Rossum717ce002001-09-14 16:58:08 +00002811 "(OO)", key, value);
Tim Peters6d6c1a32001-08-02 04:15:00 +00002812 if (res == NULL)
2813 return -1;
2814 Py_DECREF(res);
2815 return 0;
2816}
2817
Guido van Rossumdc91b992001-08-08 22:26:22 +00002818SLOT1BIN(slot_nb_add, nb_add, "__add__", "__radd__")
2819SLOT1BIN(slot_nb_subtract, nb_subtract, "__sub__", "__rsub__")
2820SLOT1BIN(slot_nb_multiply, nb_multiply, "__mul__", "__rmul__")
2821SLOT1BIN(slot_nb_divide, nb_divide, "__div__", "__rdiv__")
2822SLOT1BIN(slot_nb_remainder, nb_remainder, "__mod__", "__rmod__")
2823SLOT1BIN(slot_nb_divmod, nb_divmod, "__divmod__", "__rdivmod__")
2824
2825staticforward PyObject *slot_nb_power(PyObject *, PyObject *, PyObject *);
2826
2827SLOT1BINFULL(slot_nb_power_binary, slot_nb_power,
2828 nb_power, "__pow__", "__rpow__")
2829
2830static PyObject *
2831slot_nb_power(PyObject *self, PyObject *other, PyObject *modulus)
2832{
Guido van Rossum2730b132001-08-28 18:22:14 +00002833 static PyObject *pow_str;
2834
Guido van Rossumdc91b992001-08-08 22:26:22 +00002835 if (modulus == Py_None)
2836 return slot_nb_power_binary(self, other);
2837 /* Three-arg power doesn't use __rpow__ */
Guido van Rossum2730b132001-08-28 18:22:14 +00002838 return call_method(self, "__pow__", &pow_str,
Guido van Rossum717ce002001-09-14 16:58:08 +00002839 "(OO)", other, modulus);
Guido van Rossumdc91b992001-08-08 22:26:22 +00002840}
2841
2842SLOT0(slot_nb_negative, "__neg__")
2843SLOT0(slot_nb_positive, "__pos__")
2844SLOT0(slot_nb_absolute, "__abs__")
Tim Peters6d6c1a32001-08-02 04:15:00 +00002845
2846static int
2847slot_nb_nonzero(PyObject *self)
2848{
Guido van Rossumb8f63662001-08-15 23:57:02 +00002849 PyObject *func, *res;
Guido van Rossum60718732001-08-28 17:47:51 +00002850 static PyObject *nonzero_str, *len_str;
Tim Peters6d6c1a32001-08-02 04:15:00 +00002851
Guido van Rossum60718732001-08-28 17:47:51 +00002852 func = lookup_method(self, "__nonzero__", &nonzero_str);
Guido van Rossumb8f63662001-08-15 23:57:02 +00002853 if (func == NULL) {
2854 PyErr_Clear();
Guido van Rossum60718732001-08-28 17:47:51 +00002855 func = lookup_method(self, "__len__", &len_str);
Guido van Rossumb8f63662001-08-15 23:57:02 +00002856 }
2857
2858 if (func != NULL) {
Guido van Rossum717ce002001-09-14 16:58:08 +00002859 res = PyObject_CallObject(func, NULL);
Guido van Rossumb8f63662001-08-15 23:57:02 +00002860 Py_DECREF(func);
2861 if (res == NULL)
2862 return -1;
2863 return PyObject_IsTrue(res);
2864 }
2865 else {
2866 PyErr_Clear();
2867 return 1;
2868 }
Tim Peters6d6c1a32001-08-02 04:15:00 +00002869}
2870
Guido van Rossumdc91b992001-08-08 22:26:22 +00002871SLOT0(slot_nb_invert, "__invert__")
2872SLOT1BIN(slot_nb_lshift, nb_lshift, "__lshift__", "__rlshift__")
2873SLOT1BIN(slot_nb_rshift, nb_rshift, "__rshift__", "__rrshift__")
2874SLOT1BIN(slot_nb_and, nb_and, "__and__", "__rand__")
2875SLOT1BIN(slot_nb_xor, nb_xor, "__xor__", "__rxor__")
2876SLOT1BIN(slot_nb_or, nb_or, "__or__", "__ror__")
Guido van Rossum2ed6bf82001-09-27 20:30:07 +00002877
2878static int
2879slot_nb_coerce(PyObject **a, PyObject **b)
2880{
2881 static PyObject *coerce_str;
2882 PyObject *self = *a, *other = *b;
2883
2884 if (self->ob_type->tp_as_number != NULL &&
2885 self->ob_type->tp_as_number->nb_coerce == slot_nb_coerce) {
2886 PyObject *r;
2887 r = call_maybe(
2888 self, "__coerce__", &coerce_str, "(O)", other);
2889 if (r == NULL)
2890 return -1;
2891 if (r == Py_NotImplemented) {
2892 Py_DECREF(r);
2893 return 1;
2894 }
2895 if (!PyTuple_Check(r) || PyTuple_GET_SIZE(r) != 2) {
2896 PyErr_SetString(PyExc_TypeError,
2897 "__coerce__ didn't return a 2-tuple");
2898 Py_DECREF(r);
2899 return -1;
2900 }
2901 *a = PyTuple_GET_ITEM(r, 0);
2902 Py_INCREF(*a);
2903 *b = PyTuple_GET_ITEM(r, 1);
2904 Py_INCREF(*b);
2905 Py_DECREF(r);
2906 return 0;
2907 }
2908 if (other->ob_type->tp_as_number != NULL &&
2909 other->ob_type->tp_as_number->nb_coerce == slot_nb_coerce) {
2910 PyObject *r;
2911 r = call_maybe(
2912 other, "__coerce__", &coerce_str, "(O)", self);
2913 if (r == NULL)
2914 return -1;
2915 if (r == Py_NotImplemented) {
2916 Py_DECREF(r);
2917 return 1;
2918 }
2919 if (!PyTuple_Check(r) || PyTuple_GET_SIZE(r) != 2) {
2920 PyErr_SetString(PyExc_TypeError,
2921 "__coerce__ didn't return a 2-tuple");
2922 Py_DECREF(r);
2923 return -1;
2924 }
2925 *a = PyTuple_GET_ITEM(r, 1);
2926 Py_INCREF(*a);
2927 *b = PyTuple_GET_ITEM(r, 0);
2928 Py_INCREF(*b);
2929 Py_DECREF(r);
2930 return 0;
2931 }
2932 return 1;
2933}
2934
Guido van Rossumdc91b992001-08-08 22:26:22 +00002935SLOT0(slot_nb_int, "__int__")
2936SLOT0(slot_nb_long, "__long__")
2937SLOT0(slot_nb_float, "__float__")
2938SLOT0(slot_nb_oct, "__oct__")
2939SLOT0(slot_nb_hex, "__hex__")
2940SLOT1(slot_nb_inplace_add, "__iadd__", PyObject *, "O")
2941SLOT1(slot_nb_inplace_subtract, "__isub__", PyObject *, "O")
2942SLOT1(slot_nb_inplace_multiply, "__imul__", PyObject *, "O")
2943SLOT1(slot_nb_inplace_divide, "__idiv__", PyObject *, "O")
2944SLOT1(slot_nb_inplace_remainder, "__imod__", PyObject *, "O")
2945SLOT2(slot_nb_inplace_power, "__ipow__", PyObject *, PyObject *, "OO")
2946SLOT1(slot_nb_inplace_lshift, "__ilshift__", PyObject *, "O")
2947SLOT1(slot_nb_inplace_rshift, "__irshift__", PyObject *, "O")
2948SLOT1(slot_nb_inplace_and, "__iand__", PyObject *, "O")
2949SLOT1(slot_nb_inplace_xor, "__ixor__", PyObject *, "O")
2950SLOT1(slot_nb_inplace_or, "__ior__", PyObject *, "O")
2951SLOT1BIN(slot_nb_floor_divide, nb_floor_divide,
2952 "__floordiv__", "__rfloordiv__")
2953SLOT1BIN(slot_nb_true_divide, nb_true_divide, "__truediv__", "__rtruediv__")
2954SLOT1(slot_nb_inplace_floor_divide, "__ifloordiv__", PyObject *, "O")
2955SLOT1(slot_nb_inplace_true_divide, "__itruediv__", PyObject *, "O")
Tim Peters6d6c1a32001-08-02 04:15:00 +00002956
2957static int
Guido van Rossumb8f63662001-08-15 23:57:02 +00002958half_compare(PyObject *self, PyObject *other)
Tim Peters6d6c1a32001-08-02 04:15:00 +00002959{
Guido van Rossumb8f63662001-08-15 23:57:02 +00002960 PyObject *func, *args, *res;
Guido van Rossum60718732001-08-28 17:47:51 +00002961 static PyObject *cmp_str;
Guido van Rossumb8f63662001-08-15 23:57:02 +00002962 int c;
Tim Peters6d6c1a32001-08-02 04:15:00 +00002963
Guido van Rossum60718732001-08-28 17:47:51 +00002964 func = lookup_method(self, "__cmp__", &cmp_str);
Guido van Rossumb8f63662001-08-15 23:57:02 +00002965 if (func == NULL) {
2966 PyErr_Clear();
2967 }
2968 else {
2969 args = Py_BuildValue("(O)", other);
2970 if (args == NULL)
2971 res = NULL;
2972 else {
Guido van Rossum717ce002001-09-14 16:58:08 +00002973 res = PyObject_Call(func, args, NULL);
Guido van Rossumb8f63662001-08-15 23:57:02 +00002974 Py_DECREF(args);
2975 }
2976 if (res != Py_NotImplemented) {
2977 if (res == NULL)
2978 return -2;
2979 c = PyInt_AsLong(res);
2980 Py_DECREF(res);
2981 if (c == -1 && PyErr_Occurred())
2982 return -2;
2983 return (c < 0) ? -1 : (c > 0) ? 1 : 0;
2984 }
2985 Py_DECREF(res);
2986 }
2987 return 2;
Tim Peters6d6c1a32001-08-02 04:15:00 +00002988}
2989
Guido van Rossumab3b0342001-09-18 20:38:53 +00002990/* This slot is published for the benefit of try_3way_compare in object.c */
2991int
2992_PyObject_SlotCompare(PyObject *self, PyObject *other)
Guido van Rossumb8f63662001-08-15 23:57:02 +00002993{
2994 int c;
2995
Guido van Rossumab3b0342001-09-18 20:38:53 +00002996 if (self->ob_type->tp_compare == _PyObject_SlotCompare) {
Guido van Rossumb8f63662001-08-15 23:57:02 +00002997 c = half_compare(self, other);
2998 if (c <= 1)
2999 return c;
3000 }
Guido van Rossumab3b0342001-09-18 20:38:53 +00003001 if (other->ob_type->tp_compare == _PyObject_SlotCompare) {
Guido van Rossumb8f63662001-08-15 23:57:02 +00003002 c = half_compare(other, self);
3003 if (c < -1)
3004 return -2;
3005 if (c <= 1)
3006 return -c;
3007 }
3008 return (void *)self < (void *)other ? -1 :
3009 (void *)self > (void *)other ? 1 : 0;
3010}
3011
3012static PyObject *
3013slot_tp_repr(PyObject *self)
3014{
3015 PyObject *func, *res;
Guido van Rossum60718732001-08-28 17:47:51 +00003016 static PyObject *repr_str;
Guido van Rossumb8f63662001-08-15 23:57:02 +00003017
Guido van Rossum60718732001-08-28 17:47:51 +00003018 func = lookup_method(self, "__repr__", &repr_str);
Guido van Rossumb8f63662001-08-15 23:57:02 +00003019 if (func != NULL) {
3020 res = PyEval_CallObject(func, NULL);
3021 Py_DECREF(func);
3022 return res;
3023 }
Barry Warsaw7ce36942001-08-24 18:34:26 +00003024 PyErr_Clear();
3025 return PyString_FromFormat("<%s object at %p>",
3026 self->ob_type->tp_name, self);
Guido van Rossumb8f63662001-08-15 23:57:02 +00003027}
3028
3029static PyObject *
3030slot_tp_str(PyObject *self)
3031{
3032 PyObject *func, *res;
Guido van Rossum60718732001-08-28 17:47:51 +00003033 static PyObject *str_str;
Guido van Rossumb8f63662001-08-15 23:57:02 +00003034
Guido van Rossum60718732001-08-28 17:47:51 +00003035 func = lookup_method(self, "__str__", &str_str);
Guido van Rossumb8f63662001-08-15 23:57:02 +00003036 if (func != NULL) {
3037 res = PyEval_CallObject(func, NULL);
3038 Py_DECREF(func);
3039 return res;
3040 }
3041 else {
3042 PyErr_Clear();
3043 return slot_tp_repr(self);
3044 }
3045}
Tim Peters6d6c1a32001-08-02 04:15:00 +00003046
3047static long
3048slot_tp_hash(PyObject *self)
3049{
Guido van Rossumb8f63662001-08-15 23:57:02 +00003050 PyObject *func, *res;
Guido van Rossum60718732001-08-28 17:47:51 +00003051 static PyObject *hash_str, *eq_str, *cmp_str;
3052
Tim Peters6d6c1a32001-08-02 04:15:00 +00003053 long h;
3054
Guido van Rossum60718732001-08-28 17:47:51 +00003055 func = lookup_method(self, "__hash__", &hash_str);
Guido van Rossumb8f63662001-08-15 23:57:02 +00003056
3057 if (func != NULL) {
3058 res = PyEval_CallObject(func, NULL);
3059 Py_DECREF(func);
3060 if (res == NULL)
3061 return -1;
3062 h = PyInt_AsLong(res);
3063 }
3064 else {
3065 PyErr_Clear();
Guido van Rossum60718732001-08-28 17:47:51 +00003066 func = lookup_method(self, "__eq__", &eq_str);
Guido van Rossumb8f63662001-08-15 23:57:02 +00003067 if (func == NULL) {
3068 PyErr_Clear();
Guido van Rossum60718732001-08-28 17:47:51 +00003069 func = lookup_method(self, "__cmp__", &cmp_str);
Guido van Rossumb8f63662001-08-15 23:57:02 +00003070 }
3071 if (func != NULL) {
3072 Py_DECREF(func);
3073 PyErr_SetString(PyExc_TypeError, "unhashable type");
3074 return -1;
3075 }
3076 PyErr_Clear();
3077 h = _Py_HashPointer((void *)self);
3078 }
Tim Peters6d6c1a32001-08-02 04:15:00 +00003079 if (h == -1 && !PyErr_Occurred())
3080 h = -2;
3081 return h;
3082}
3083
3084static PyObject *
3085slot_tp_call(PyObject *self, PyObject *args, PyObject *kwds)
3086{
Guido van Rossum60718732001-08-28 17:47:51 +00003087 static PyObject *call_str;
3088 PyObject *meth = lookup_method(self, "__call__", &call_str);
Tim Peters6d6c1a32001-08-02 04:15:00 +00003089 PyObject *res;
3090
3091 if (meth == NULL)
3092 return NULL;
3093 res = PyObject_Call(meth, args, kwds);
3094 Py_DECREF(meth);
3095 return res;
3096}
3097
Tim Peters6d6c1a32001-08-02 04:15:00 +00003098static PyObject *
3099slot_tp_getattro(PyObject *self, PyObject *name)
3100{
3101 PyTypeObject *tp = self->ob_type;
Tim Peters6d6c1a32001-08-02 04:15:00 +00003102 PyObject *getattr;
Guido van Rossum8e248182001-08-12 05:17:56 +00003103 static PyObject *getattr_str = NULL;
Tim Peters6d6c1a32001-08-02 04:15:00 +00003104
Guido van Rossum8e248182001-08-12 05:17:56 +00003105 if (getattr_str == NULL) {
Guido van Rossum867a8d22001-09-21 19:29:08 +00003106 getattr_str = PyString_InternFromString("__getattribute__");
Guido van Rossum8e248182001-08-12 05:17:56 +00003107 if (getattr_str == NULL)
3108 return NULL;
Tim Peters6d6c1a32001-08-02 04:15:00 +00003109 }
Guido van Rossum8e248182001-08-12 05:17:56 +00003110 getattr = _PyType_Lookup(tp, getattr_str);
Guido van Rossumc3542212001-08-16 09:18:56 +00003111 if (getattr == NULL) {
3112 /* Avoid further slowdowns */
3113 if (tp->tp_getattro == slot_tp_getattro)
3114 tp->tp_getattro = PyObject_GenericGetAttr;
Guido van Rossum8e248182001-08-12 05:17:56 +00003115 return PyObject_GenericGetAttr(self, name);
Guido van Rossumc3542212001-08-16 09:18:56 +00003116 }
Tim Peters6d6c1a32001-08-02 04:15:00 +00003117 return PyObject_CallFunction(getattr, "OO", self, name);
3118}
3119
Guido van Rossum19c1cd52001-09-21 21:24:49 +00003120static PyObject *
3121slot_tp_getattr_hook(PyObject *self, PyObject *name)
3122{
3123 PyTypeObject *tp = self->ob_type;
3124 PyObject *getattr, *getattribute, *res;
3125 static PyObject *getattribute_str = NULL;
3126 static PyObject *getattr_str = NULL;
3127
3128 if (getattr_str == NULL) {
3129 getattr_str = PyString_InternFromString("__getattr__");
3130 if (getattr_str == NULL)
3131 return NULL;
3132 }
3133 if (getattribute_str == NULL) {
3134 getattribute_str =
3135 PyString_InternFromString("__getattribute__");
3136 if (getattribute_str == NULL)
3137 return NULL;
3138 }
3139 getattr = _PyType_Lookup(tp, getattr_str);
3140 getattribute = _PyType_Lookup(tp, getattribute_str);
3141 if (getattr == NULL && getattribute == NULL) {
3142 /* Avoid further slowdowns */
3143 if (tp->tp_getattro == slot_tp_getattr_hook)
3144 tp->tp_getattro = PyObject_GenericGetAttr;
3145 return PyObject_GenericGetAttr(self, name);
3146 }
3147 if (getattribute == NULL)
3148 res = PyObject_GenericGetAttr(self, name);
3149 else
3150 res = PyObject_CallFunction(getattribute, "OO", self, name);
Guido van Rossum3926a632001-09-25 16:25:58 +00003151 if (getattr != NULL &&
3152 res == NULL && PyErr_ExceptionMatches(PyExc_AttributeError)) {
Guido van Rossum19c1cd52001-09-21 21:24:49 +00003153 PyErr_Clear();
3154 res = PyObject_CallFunction(getattr, "OO", self, name);
3155 }
3156 return res;
3157}
3158
Tim Peters6d6c1a32001-08-02 04:15:00 +00003159static int
3160slot_tp_setattro(PyObject *self, PyObject *name, PyObject *value)
3161{
3162 PyObject *res;
Guido van Rossum2730b132001-08-28 18:22:14 +00003163 static PyObject *delattr_str, *setattr_str;
Tim Peters6d6c1a32001-08-02 04:15:00 +00003164
3165 if (value == NULL)
Guido van Rossum2730b132001-08-28 18:22:14 +00003166 res = call_method(self, "__delattr__", &delattr_str,
Guido van Rossum717ce002001-09-14 16:58:08 +00003167 "(O)", name);
Tim Peters6d6c1a32001-08-02 04:15:00 +00003168 else
Guido van Rossum2730b132001-08-28 18:22:14 +00003169 res = call_method(self, "__setattr__", &setattr_str,
Guido van Rossum717ce002001-09-14 16:58:08 +00003170 "(OO)", name, value);
Tim Peters6d6c1a32001-08-02 04:15:00 +00003171 if (res == NULL)
3172 return -1;
3173 Py_DECREF(res);
3174 return 0;
3175}
3176
3177/* Map rich comparison operators to their __xx__ namesakes */
3178static char *name_op[] = {
3179 "__lt__",
3180 "__le__",
3181 "__eq__",
3182 "__ne__",
3183 "__gt__",
3184 "__ge__",
3185};
3186
3187static PyObject *
Guido van Rossumb8f63662001-08-15 23:57:02 +00003188half_richcompare(PyObject *self, PyObject *other, int op)
Tim Peters6d6c1a32001-08-02 04:15:00 +00003189{
Guido van Rossumb8f63662001-08-15 23:57:02 +00003190 PyObject *func, *args, *res;
Guido van Rossum60718732001-08-28 17:47:51 +00003191 static PyObject *op_str[6];
Tim Peters6d6c1a32001-08-02 04:15:00 +00003192
Guido van Rossum60718732001-08-28 17:47:51 +00003193 func = lookup_method(self, name_op[op], &op_str[op]);
Guido van Rossumb8f63662001-08-15 23:57:02 +00003194 if (func == NULL) {
3195 PyErr_Clear();
3196 Py_INCREF(Py_NotImplemented);
3197 return Py_NotImplemented;
3198 }
3199 args = Py_BuildValue("(O)", other);
3200 if (args == NULL)
3201 res = NULL;
3202 else {
Guido van Rossum717ce002001-09-14 16:58:08 +00003203 res = PyObject_Call(func, args, NULL);
Guido van Rossumb8f63662001-08-15 23:57:02 +00003204 Py_DECREF(args);
3205 }
3206 Py_DECREF(func);
Tim Peters6d6c1a32001-08-02 04:15:00 +00003207 return res;
3208}
3209
Guido van Rossumb8f63662001-08-15 23:57:02 +00003210/* Map rich comparison operators to their swapped version, e.g. LT --> GT */
3211static int swapped_op[] = {Py_GT, Py_GE, Py_EQ, Py_NE, Py_LT, Py_LE};
3212
3213static PyObject *
3214slot_tp_richcompare(PyObject *self, PyObject *other, int op)
3215{
3216 PyObject *res;
3217
3218 if (self->ob_type->tp_richcompare == slot_tp_richcompare) {
3219 res = half_richcompare(self, other, op);
3220 if (res != Py_NotImplemented)
3221 return res;
3222 Py_DECREF(res);
3223 }
3224 if (other->ob_type->tp_richcompare == slot_tp_richcompare) {
3225 res = half_richcompare(other, self, swapped_op[op]);
3226 if (res != Py_NotImplemented) {
3227 return res;
3228 }
3229 Py_DECREF(res);
3230 }
3231 Py_INCREF(Py_NotImplemented);
3232 return Py_NotImplemented;
3233}
3234
3235static PyObject *
3236slot_tp_iter(PyObject *self)
3237{
3238 PyObject *func, *res;
Guido van Rossum60718732001-08-28 17:47:51 +00003239 static PyObject *iter_str, *getitem_str;
Guido van Rossumb8f63662001-08-15 23:57:02 +00003240
Guido van Rossum60718732001-08-28 17:47:51 +00003241 func = lookup_method(self, "__iter__", &iter_str);
Guido van Rossumb8f63662001-08-15 23:57:02 +00003242 if (func != NULL) {
3243 res = PyObject_CallObject(func, NULL);
3244 Py_DECREF(func);
3245 return res;
3246 }
3247 PyErr_Clear();
Guido van Rossum60718732001-08-28 17:47:51 +00003248 func = lookup_method(self, "__getitem__", &getitem_str);
Guido van Rossumb8f63662001-08-15 23:57:02 +00003249 if (func == NULL) {
3250 PyErr_SetString(PyExc_TypeError, "iter() of non-sequence");
3251 return NULL;
3252 }
3253 Py_DECREF(func);
3254 return PySeqIter_New(self);
3255}
Tim Peters6d6c1a32001-08-02 04:15:00 +00003256
3257static PyObject *
3258slot_tp_iternext(PyObject *self)
3259{
Guido van Rossum2730b132001-08-28 18:22:14 +00003260 static PyObject *next_str;
Guido van Rossum717ce002001-09-14 16:58:08 +00003261 return call_method(self, "next", &next_str, "()");
Tim Peters6d6c1a32001-08-02 04:15:00 +00003262}
3263
Guido van Rossum1a493502001-08-17 16:47:50 +00003264static PyObject *
3265slot_tp_descr_get(PyObject *self, PyObject *obj, PyObject *type)
3266{
3267 PyTypeObject *tp = self->ob_type;
3268 PyObject *get;
3269 static PyObject *get_str = NULL;
3270
3271 if (get_str == NULL) {
3272 get_str = PyString_InternFromString("__get__");
3273 if (get_str == NULL)
3274 return NULL;
3275 }
3276 get = _PyType_Lookup(tp, get_str);
3277 if (get == NULL) {
3278 /* Avoid further slowdowns */
3279 if (tp->tp_descr_get == slot_tp_descr_get)
3280 tp->tp_descr_get = NULL;
3281 Py_INCREF(self);
3282 return self;
3283 }
Guido van Rossum2c252392001-08-24 10:13:31 +00003284 if (obj == NULL)
3285 obj = Py_None;
3286 if (type == NULL)
3287 type = Py_None;
Guido van Rossum1a493502001-08-17 16:47:50 +00003288 return PyObject_CallFunction(get, "OOO", self, obj, type);
3289}
Tim Peters6d6c1a32001-08-02 04:15:00 +00003290
3291static int
3292slot_tp_descr_set(PyObject *self, PyObject *target, PyObject *value)
3293{
Guido van Rossum2c252392001-08-24 10:13:31 +00003294 PyObject *res;
Guido van Rossum2730b132001-08-28 18:22:14 +00003295 static PyObject *del_str, *set_str;
Guido van Rossum2c252392001-08-24 10:13:31 +00003296
3297 if (value == NULL)
Guido van Rossum2730b132001-08-28 18:22:14 +00003298 res = call_method(self, "__del__", &del_str,
Guido van Rossum717ce002001-09-14 16:58:08 +00003299 "(O)", target);
Guido van Rossum2c252392001-08-24 10:13:31 +00003300 else
Guido van Rossum2730b132001-08-28 18:22:14 +00003301 res = call_method(self, "__set__", &set_str,
Guido van Rossum717ce002001-09-14 16:58:08 +00003302 "(OO)", target, value);
Tim Peters6d6c1a32001-08-02 04:15:00 +00003303 if (res == NULL)
3304 return -1;
3305 Py_DECREF(res);
3306 return 0;
3307}
3308
3309static int
3310slot_tp_init(PyObject *self, PyObject *args, PyObject *kwds)
3311{
Guido van Rossum60718732001-08-28 17:47:51 +00003312 static PyObject *init_str;
3313 PyObject *meth = lookup_method(self, "__init__", &init_str);
Tim Peters6d6c1a32001-08-02 04:15:00 +00003314 PyObject *res;
3315
3316 if (meth == NULL)
3317 return -1;
3318 res = PyObject_Call(meth, args, kwds);
3319 Py_DECREF(meth);
3320 if (res == NULL)
3321 return -1;
3322 Py_DECREF(res);
3323 return 0;
3324}
3325
3326static PyObject *
3327slot_tp_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
3328{
3329 PyObject *func = PyObject_GetAttrString((PyObject *)type, "__new__");
3330 PyObject *newargs, *x;
3331 int i, n;
3332
3333 if (func == NULL)
3334 return NULL;
3335 assert(PyTuple_Check(args));
3336 n = PyTuple_GET_SIZE(args);
3337 newargs = PyTuple_New(n+1);
3338 if (newargs == NULL)
3339 return NULL;
3340 Py_INCREF(type);
3341 PyTuple_SET_ITEM(newargs, 0, (PyObject *)type);
3342 for (i = 0; i < n; i++) {
3343 x = PyTuple_GET_ITEM(args, i);
3344 Py_INCREF(x);
3345 PyTuple_SET_ITEM(newargs, i+1, x);
3346 }
3347 x = PyObject_Call(func, newargs, kwds);
3348 Py_DECREF(func);
3349 return x;
3350}
3351
Guido van Rossumf040ede2001-08-07 16:40:56 +00003352/* This is called at the very end of type_new() (even after
Guido van Rossum528b7eb2001-08-07 17:24:28 +00003353 PyType_Ready()) to complete the initialization of dynamic types.
Guido van Rossumf040ede2001-08-07 16:40:56 +00003354 The dict argument is the dictionary argument passed to type_new(),
3355 which is the local namespace of the class statement, in other
3356 words, it contains the methods. For each special method (like
3357 __repr__) defined in the dictionary, the corresponding function
3358 slot in the type object (like tp_repr) is set to a special function
3359 whose name is 'slot_' followed by the slot name and whose signature
3360 is whatever is required for that slot. These slot functions look
3361 up the corresponding method in the type's dictionary and call it.
3362 The slot functions have to take care of the various peculiarities
3363 of the mapping between slots and special methods, such as mapping
3364 one slot to multiple methods (tp_richcompare <--> __le__, __lt__
3365 etc.) or mapping multiple slots to a single method (sq_item,
3366 mp_subscript <--> __getitem__). */
3367
Tim Peters6d6c1a32001-08-02 04:15:00 +00003368static void
3369override_slots(PyTypeObject *type, PyObject *dict)
3370{
3371 PySequenceMethods *sq = type->tp_as_sequence;
3372 PyMappingMethods *mp = type->tp_as_mapping;
3373 PyNumberMethods *nb = type->tp_as_number;
3374
Guido van Rossumdc91b992001-08-08 22:26:22 +00003375#define SQSLOT(OPNAME, SLOTNAME, FUNCNAME) \
Guido van Rossum8e248182001-08-12 05:17:56 +00003376 if (dict == NULL || PyDict_GetItemString(dict, OPNAME)) { \
Guido van Rossumdc91b992001-08-08 22:26:22 +00003377 sq->SLOTNAME = FUNCNAME; \
Tim Peters6d6c1a32001-08-02 04:15:00 +00003378 }
3379
Guido van Rossumdc91b992001-08-08 22:26:22 +00003380#define MPSLOT(OPNAME, SLOTNAME, FUNCNAME) \
Guido van Rossum8e248182001-08-12 05:17:56 +00003381 if (dict == NULL || PyDict_GetItemString(dict, OPNAME)) { \
Guido van Rossumdc91b992001-08-08 22:26:22 +00003382 mp->SLOTNAME = FUNCNAME; \
Tim Peters6d6c1a32001-08-02 04:15:00 +00003383 }
3384
Guido van Rossumdc91b992001-08-08 22:26:22 +00003385#define NBSLOT(OPNAME, SLOTNAME, FUNCNAME) \
Guido van Rossum8e248182001-08-12 05:17:56 +00003386 if (dict == NULL || PyDict_GetItemString(dict, OPNAME)) { \
Guido van Rossumdc91b992001-08-08 22:26:22 +00003387 nb->SLOTNAME = FUNCNAME; \
Tim Peters6d6c1a32001-08-02 04:15:00 +00003388 }
3389
Guido van Rossumdc91b992001-08-08 22:26:22 +00003390#define TPSLOT(OPNAME, SLOTNAME, FUNCNAME) \
Guido van Rossum8e248182001-08-12 05:17:56 +00003391 if (dict == NULL || PyDict_GetItemString(dict, OPNAME)) { \
Guido van Rossumdc91b992001-08-08 22:26:22 +00003392 type->SLOTNAME = FUNCNAME; \
Tim Peters6d6c1a32001-08-02 04:15:00 +00003393 }
3394
Guido van Rossumdc91b992001-08-08 22:26:22 +00003395 SQSLOT("__len__", sq_length, slot_sq_length);
3396 SQSLOT("__add__", sq_concat, slot_sq_concat);
3397 SQSLOT("__mul__", sq_repeat, slot_sq_repeat);
3398 SQSLOT("__getitem__", sq_item, slot_sq_item);
3399 SQSLOT("__getslice__", sq_slice, slot_sq_slice);
3400 SQSLOT("__setitem__", sq_ass_item, slot_sq_ass_item);
3401 SQSLOT("__delitem__", sq_ass_item, slot_sq_ass_item);
3402 SQSLOT("__setslice__", sq_ass_slice, slot_sq_ass_slice);
3403 SQSLOT("__delslice__", sq_ass_slice, slot_sq_ass_slice);
3404 SQSLOT("__contains__", sq_contains, slot_sq_contains);
3405 SQSLOT("__iadd__", sq_inplace_concat, slot_sq_inplace_concat);
3406 SQSLOT("__imul__", sq_inplace_repeat, slot_sq_inplace_repeat);
Tim Peters6d6c1a32001-08-02 04:15:00 +00003407
Guido van Rossumdc91b992001-08-08 22:26:22 +00003408 MPSLOT("__len__", mp_length, slot_mp_length);
3409 MPSLOT("__getitem__", mp_subscript, slot_mp_subscript);
3410 MPSLOT("__setitem__", mp_ass_subscript, slot_mp_ass_subscript);
3411 MPSLOT("__delitem__", mp_ass_subscript, slot_mp_ass_subscript);
Tim Peters6d6c1a32001-08-02 04:15:00 +00003412
Guido van Rossumdc91b992001-08-08 22:26:22 +00003413 NBSLOT("__add__", nb_add, slot_nb_add);
3414 NBSLOT("__sub__", nb_subtract, slot_nb_subtract);
3415 NBSLOT("__mul__", nb_multiply, slot_nb_multiply);
3416 NBSLOT("__div__", nb_divide, slot_nb_divide);
3417 NBSLOT("__mod__", nb_remainder, slot_nb_remainder);
3418 NBSLOT("__divmod__", nb_divmod, slot_nb_divmod);
3419 NBSLOT("__pow__", nb_power, slot_nb_power);
3420 NBSLOT("__neg__", nb_negative, slot_nb_negative);
3421 NBSLOT("__pos__", nb_positive, slot_nb_positive);
3422 NBSLOT("__abs__", nb_absolute, slot_nb_absolute);
3423 NBSLOT("__nonzero__", nb_nonzero, slot_nb_nonzero);
3424 NBSLOT("__invert__", nb_invert, slot_nb_invert);
3425 NBSLOT("__lshift__", nb_lshift, slot_nb_lshift);
3426 NBSLOT("__rshift__", nb_rshift, slot_nb_rshift);
3427 NBSLOT("__and__", nb_and, slot_nb_and);
3428 NBSLOT("__xor__", nb_xor, slot_nb_xor);
3429 NBSLOT("__or__", nb_or, slot_nb_or);
Guido van Rossum2ed6bf82001-09-27 20:30:07 +00003430 NBSLOT("__coerce__", nb_coerce, slot_nb_coerce);
Guido van Rossumdc91b992001-08-08 22:26:22 +00003431 NBSLOT("__int__", nb_int, slot_nb_int);
3432 NBSLOT("__long__", nb_long, slot_nb_long);
3433 NBSLOT("__float__", nb_float, slot_nb_float);
3434 NBSLOT("__oct__", nb_oct, slot_nb_oct);
3435 NBSLOT("__hex__", nb_hex, slot_nb_hex);
3436 NBSLOT("__iadd__", nb_inplace_add, slot_nb_inplace_add);
3437 NBSLOT("__isub__", nb_inplace_subtract, slot_nb_inplace_subtract);
3438 NBSLOT("__imul__", nb_inplace_multiply, slot_nb_inplace_multiply);
3439 NBSLOT("__idiv__", nb_inplace_divide, slot_nb_inplace_divide);
3440 NBSLOT("__imod__", nb_inplace_remainder, slot_nb_inplace_remainder);
3441 NBSLOT("__ipow__", nb_inplace_power, slot_nb_inplace_power);
3442 NBSLOT("__ilshift__", nb_inplace_lshift, slot_nb_inplace_lshift);
3443 NBSLOT("__irshift__", nb_inplace_rshift, slot_nb_inplace_rshift);
3444 NBSLOT("__iand__", nb_inplace_and, slot_nb_inplace_and);
3445 NBSLOT("__ixor__", nb_inplace_xor, slot_nb_inplace_xor);
3446 NBSLOT("__ior__", nb_inplace_or, slot_nb_inplace_or);
3447 NBSLOT("__floordiv__", nb_floor_divide, slot_nb_floor_divide);
3448 NBSLOT("__truediv__", nb_true_divide, slot_nb_true_divide);
3449 NBSLOT("__ifloordiv__", nb_inplace_floor_divide,
3450 slot_nb_inplace_floor_divide);
3451 NBSLOT("__itruediv__", nb_inplace_true_divide,
3452 slot_nb_inplace_true_divide);
Tim Peters6d6c1a32001-08-02 04:15:00 +00003453
Guido van Rossum8e248182001-08-12 05:17:56 +00003454 if (dict == NULL ||
3455 PyDict_GetItemString(dict, "__str__") ||
Tim Peters6d6c1a32001-08-02 04:15:00 +00003456 PyDict_GetItemString(dict, "__repr__"))
3457 type->tp_print = NULL;
3458
Guido van Rossumab3b0342001-09-18 20:38:53 +00003459 TPSLOT("__cmp__", tp_compare, _PyObject_SlotCompare);
Guido van Rossumdc91b992001-08-08 22:26:22 +00003460 TPSLOT("__repr__", tp_repr, slot_tp_repr);
3461 TPSLOT("__hash__", tp_hash, slot_tp_hash);
3462 TPSLOT("__call__", tp_call, slot_tp_call);
3463 TPSLOT("__str__", tp_str, slot_tp_str);
Guido van Rossum867a8d22001-09-21 19:29:08 +00003464 TPSLOT("__getattribute__", tp_getattro, slot_tp_getattro);
Guido van Rossum19c1cd52001-09-21 21:24:49 +00003465 TPSLOT("__getattr__", tp_getattro, slot_tp_getattr_hook);
Guido van Rossumdc91b992001-08-08 22:26:22 +00003466 TPSLOT("__setattr__", tp_setattro, slot_tp_setattro);
3467 TPSLOT("__lt__", tp_richcompare, slot_tp_richcompare);
3468 TPSLOT("__le__", tp_richcompare, slot_tp_richcompare);
3469 TPSLOT("__eq__", tp_richcompare, slot_tp_richcompare);
3470 TPSLOT("__ne__", tp_richcompare, slot_tp_richcompare);
3471 TPSLOT("__gt__", tp_richcompare, slot_tp_richcompare);
3472 TPSLOT("__ge__", tp_richcompare, slot_tp_richcompare);
3473 TPSLOT("__iter__", tp_iter, slot_tp_iter);
3474 TPSLOT("next", tp_iternext, slot_tp_iternext);
3475 TPSLOT("__get__", tp_descr_get, slot_tp_descr_get);
3476 TPSLOT("__set__", tp_descr_set, slot_tp_descr_set);
3477 TPSLOT("__init__", tp_init, slot_tp_init);
3478 TPSLOT("__new__", tp_new, slot_tp_new);
Tim Peters6d6c1a32001-08-02 04:15:00 +00003479}
Guido van Rossum705f0f52001-08-24 16:47:00 +00003480
3481
3482/* Cooperative 'super' */
3483
3484typedef struct {
3485 PyObject_HEAD
Guido van Rossume705ef12001-08-29 15:47:06 +00003486 PyTypeObject *type;
Guido van Rossum705f0f52001-08-24 16:47:00 +00003487 PyObject *obj;
3488} superobject;
3489
Guido van Rossum6f799372001-09-20 20:46:19 +00003490static PyMemberDef super_members[] = {
3491 {"__thisclass__", T_OBJECT, offsetof(superobject, type), READONLY,
3492 "the class invoking super()"},
3493 {"__self__", T_OBJECT, offsetof(superobject, obj), READONLY,
3494 "the instance invoking super(); may be None"},
Guido van Rossum41eb14d2001-08-30 23:13:11 +00003495 {0}
3496};
3497
Guido van Rossum705f0f52001-08-24 16:47:00 +00003498static void
3499super_dealloc(PyObject *self)
3500{
3501 superobject *su = (superobject *)self;
3502
3503 Py_XDECREF(su->obj);
3504 Py_XDECREF(su->type);
3505 self->ob_type->tp_free(self);
3506}
3507
3508static PyObject *
Guido van Rossum41eb14d2001-08-30 23:13:11 +00003509super_repr(PyObject *self)
3510{
3511 superobject *su = (superobject *)self;
3512
3513 if (su->obj)
3514 return PyString_FromFormat(
Guido van Rossuma4cb7882001-09-25 03:56:29 +00003515 "<super: <class '%s'>, <%s object>>",
Guido van Rossum41eb14d2001-08-30 23:13:11 +00003516 su->type ? su->type->tp_name : "NULL",
3517 su->obj->ob_type->tp_name);
3518 else
3519 return PyString_FromFormat(
Guido van Rossuma4cb7882001-09-25 03:56:29 +00003520 "<super: <class '%s'>, NULL>",
Guido van Rossum41eb14d2001-08-30 23:13:11 +00003521 su->type ? su->type->tp_name : "NULL");
3522}
3523
3524static PyObject *
Guido van Rossum705f0f52001-08-24 16:47:00 +00003525super_getattro(PyObject *self, PyObject *name)
3526{
3527 superobject *su = (superobject *)self;
3528
3529 if (su->obj != NULL) {
3530 PyObject *mro, *res, *tmp;
3531 descrgetfunc f;
3532 int i, n;
3533
Guido van Rossume705ef12001-08-29 15:47:06 +00003534 mro = su->obj->ob_type->tp_mro;
Guido van Rossum41eb14d2001-08-30 23:13:11 +00003535 if (mro == NULL)
3536 n = 0;
3537 else {
3538 assert(PyTuple_Check(mro));
3539 n = PyTuple_GET_SIZE(mro);
3540 }
Guido van Rossum705f0f52001-08-24 16:47:00 +00003541 for (i = 0; i < n; i++) {
Guido van Rossume705ef12001-08-29 15:47:06 +00003542 if ((PyObject *)(su->type) == PyTuple_GET_ITEM(mro, i))
Guido van Rossum705f0f52001-08-24 16:47:00 +00003543 break;
3544 }
Guido van Rossume705ef12001-08-29 15:47:06 +00003545 if (i >= n && PyType_Check(su->obj)) {
3546 mro = ((PyTypeObject *)(su->obj))->tp_mro;
Guido van Rossum41eb14d2001-08-30 23:13:11 +00003547 if (mro == NULL)
3548 n = 0;
3549 else {
3550 assert(PyTuple_Check(mro));
3551 n = PyTuple_GET_SIZE(mro);
3552 }
Guido van Rossume705ef12001-08-29 15:47:06 +00003553 for (i = 0; i < n; i++) {
3554 if ((PyObject *)(su->type) ==
3555 PyTuple_GET_ITEM(mro, i))
3556 break;
3557 }
Guido van Rossume705ef12001-08-29 15:47:06 +00003558 }
Guido van Rossum705f0f52001-08-24 16:47:00 +00003559 i++;
3560 res = NULL;
3561 for (; i < n; i++) {
3562 tmp = PyTuple_GET_ITEM(mro, i);
3563 assert(PyType_Check(tmp));
3564 res = PyDict_GetItem(
3565 ((PyTypeObject *)tmp)->tp_defined, name);
3566 if (res != NULL) {
3567 Py_INCREF(res);
3568 f = res->ob_type->tp_descr_get;
3569 if (f != NULL) {
3570 tmp = f(res, su->obj, res);
3571 Py_DECREF(res);
3572 res = tmp;
3573 }
3574 return res;
3575 }
3576 }
3577 }
3578 return PyObject_GenericGetAttr(self, name);
3579}
3580
3581static PyObject *
3582super_descr_get(PyObject *self, PyObject *obj, PyObject *type)
3583{
3584 superobject *su = (superobject *)self;
3585 superobject *new;
3586
3587 if (obj == NULL || obj == Py_None || su->obj != NULL) {
3588 /* Not binding to an object, or already bound */
3589 Py_INCREF(self);
3590 return self;
3591 }
3592 new = (superobject *)PySuper_Type.tp_new(&PySuper_Type, NULL, NULL);
3593 if (new == NULL)
3594 return NULL;
3595 Py_INCREF(su->type);
3596 Py_INCREF(obj);
3597 new->type = su->type;
3598 new->obj = obj;
3599 return (PyObject *)new;
3600}
3601
3602static int
3603super_init(PyObject *self, PyObject *args, PyObject *kwds)
3604{
3605 superobject *su = (superobject *)self;
Guido van Rossume705ef12001-08-29 15:47:06 +00003606 PyTypeObject *type;
3607 PyObject *obj = NULL;
Guido van Rossum705f0f52001-08-24 16:47:00 +00003608
3609 if (!PyArg_ParseTuple(args, "O!|O:super", &PyType_Type, &type, &obj))
3610 return -1;
3611 if (obj == Py_None)
3612 obj = NULL;
Guido van Rossume705ef12001-08-29 15:47:06 +00003613 if (obj != NULL &&
3614 !PyType_IsSubtype(obj->ob_type, type) &&
3615 !(PyType_Check(obj) &&
3616 PyType_IsSubtype((PyTypeObject *)obj, type))) {
Guido van Rossum705f0f52001-08-24 16:47:00 +00003617 PyErr_SetString(PyExc_TypeError,
Guido van Rossume705ef12001-08-29 15:47:06 +00003618 "super(type, obj): "
3619 "obj must be an instance or subtype of type");
Guido van Rossum705f0f52001-08-24 16:47:00 +00003620 return -1;
3621 }
3622 Py_INCREF(type);
3623 Py_XINCREF(obj);
3624 su->type = type;
3625 su->obj = obj;
3626 return 0;
3627}
3628
3629static char super_doc[] =
3630"super(type) -> unbound super object\n"
3631"super(type, obj) -> bound super object; requires isinstance(obj, type)\n"
Guido van Rossume705ef12001-08-29 15:47:06 +00003632"super(type, type2) -> bound super object; requires issubclass(type2, type)\n"
Guido van Rossum705f0f52001-08-24 16:47:00 +00003633"Typical use to call a cooperative superclass method:\n"
3634"class C(B):\n"
3635" def meth(self, arg):\n"
3636" super(C, self).meth(arg)";
3637
3638PyTypeObject PySuper_Type = {
3639 PyObject_HEAD_INIT(&PyType_Type)
3640 0, /* ob_size */
3641 "super", /* tp_name */
3642 sizeof(superobject), /* tp_basicsize */
3643 0, /* tp_itemsize */
3644 /* methods */
3645 super_dealloc, /* tp_dealloc */
3646 0, /* tp_print */
3647 0, /* tp_getattr */
3648 0, /* tp_setattr */
3649 0, /* tp_compare */
Guido van Rossum41eb14d2001-08-30 23:13:11 +00003650 super_repr, /* tp_repr */
Guido van Rossum705f0f52001-08-24 16:47:00 +00003651 0, /* tp_as_number */
3652 0, /* tp_as_sequence */
3653 0, /* tp_as_mapping */
3654 0, /* tp_hash */
3655 0, /* tp_call */
3656 0, /* tp_str */
3657 super_getattro, /* tp_getattro */
3658 0, /* tp_setattro */
3659 0, /* tp_as_buffer */
Guido van Rossum31bcff82001-08-30 04:37:15 +00003660 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
Guido van Rossum705f0f52001-08-24 16:47:00 +00003661 super_doc, /* tp_doc */
3662 0, /* tp_traverse */
3663 0, /* tp_clear */
3664 0, /* tp_richcompare */
3665 0, /* tp_weaklistoffset */
3666 0, /* tp_iter */
3667 0, /* tp_iternext */
3668 0, /* tp_methods */
Guido van Rossum41eb14d2001-08-30 23:13:11 +00003669 super_members, /* tp_members */
Guido van Rossum705f0f52001-08-24 16:47:00 +00003670 0, /* tp_getset */
3671 0, /* tp_base */
3672 0, /* tp_dict */
3673 super_descr_get, /* tp_descr_get */
3674 0, /* tp_descr_set */
3675 0, /* tp_dictoffset */
3676 super_init, /* tp_init */
3677 PyType_GenericAlloc, /* tp_alloc */
3678 PyType_GenericNew, /* tp_new */
3679 _PyObject_Del, /* tp_free */
3680};