blob: 0e6099c5b7338b20a90da21a74f85f8b76c3c72b [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 Rossum874f15a2001-09-25 21:16:33 +00001840BINARY(floordiv, "x//y");
1841BINARY(truediv, "x/y # true division");
Tim Peters6d6c1a32001-08-02 04:15:00 +00001842
1843static PyObject *
1844wrap_ternaryfunc(PyObject *self, PyObject *args, void *wrapped)
1845{
1846 ternaryfunc func = (ternaryfunc)wrapped;
1847 PyObject *other;
1848 PyObject *third = Py_None;
1849
1850 /* Note: This wrapper only works for __pow__() */
1851
1852 if (!PyArg_ParseTuple(args, "O|O", &other, &third))
1853 return NULL;
1854 return (*func)(self, other, third);
1855}
1856
1857#undef TERNARY
1858#define TERNARY(NAME, OP) \
1859static struct wrapperbase tab_##NAME[] = { \
1860 {"__" #NAME "__", \
1861 (wrapperfunc)wrap_ternaryfunc, \
1862 "x.__" #NAME "__(y, z) <==> " #OP}, \
1863 {"__r" #NAME "__", \
1864 (wrapperfunc)wrap_ternaryfunc, \
1865 "y.__r" #NAME "__(x, z) <==> " #OP}, \
1866 {0} \
1867}
1868
1869TERNARY(pow, "(x**y) % z");
1870
1871#undef UNARY
1872#define UNARY(NAME, OP) \
1873static struct wrapperbase tab_##NAME[] = { \
1874 {"__" #NAME "__", \
1875 (wrapperfunc)wrap_unaryfunc, \
1876 "x.__" #NAME "__() <==> " #OP}, \
1877 {0} \
1878}
1879
1880static PyObject *
1881wrap_unaryfunc(PyObject *self, PyObject *args, void *wrapped)
1882{
1883 unaryfunc func = (unaryfunc)wrapped;
1884
1885 if (!PyArg_ParseTuple(args, ""))
1886 return NULL;
1887 return (*func)(self);
1888}
1889
1890UNARY(neg, "-x");
1891UNARY(pos, "+x");
1892UNARY(abs, "abs(x)");
1893UNARY(nonzero, "x != 0");
1894UNARY(invert, "~x");
1895UNARY(int, "int(x)");
1896UNARY(long, "long(x)");
1897UNARY(float, "float(x)");
1898UNARY(oct, "oct(x)");
1899UNARY(hex, "hex(x)");
1900
1901#undef IBINARY
1902#define IBINARY(NAME, OP) \
1903static struct wrapperbase tab_##NAME[] = { \
1904 {"__" #NAME "__", \
1905 (wrapperfunc)wrap_binaryfunc, \
1906 "x.__" #NAME "__(y) <==> " #OP}, \
1907 {0} \
1908}
1909
1910IBINARY(iadd, "x+=y");
1911IBINARY(isub, "x-=y");
1912IBINARY(imul, "x*=y");
1913IBINARY(idiv, "x/=y");
1914IBINARY(imod, "x%=y");
1915IBINARY(ilshift, "x<<=y");
1916IBINARY(irshift, "x>>=y");
1917IBINARY(iand, "x&=y");
1918IBINARY(ixor, "x^=y");
1919IBINARY(ior, "x|=y");
Guido van Rossum874f15a2001-09-25 21:16:33 +00001920IBINARY(ifloordiv, "x//=y");
1921IBINARY(itruediv, "x/=y # true division");
Tim Peters6d6c1a32001-08-02 04:15:00 +00001922
1923#undef ITERNARY
1924#define ITERNARY(NAME, OP) \
1925static struct wrapperbase tab_##NAME[] = { \
1926 {"__" #NAME "__", \
1927 (wrapperfunc)wrap_ternaryfunc, \
1928 "x.__" #NAME "__(y) <==> " #OP}, \
1929 {0} \
1930}
1931
1932ITERNARY(ipow, "x = (x**y) % z");
1933
1934static struct wrapperbase tab_getitem[] = {
1935 {"__getitem__", (wrapperfunc)wrap_binaryfunc,
1936 "x.__getitem__(y) <==> x[y]"},
1937 {0}
1938};
1939
1940static PyObject *
1941wrap_intargfunc(PyObject *self, PyObject *args, void *wrapped)
1942{
1943 intargfunc func = (intargfunc)wrapped;
1944 int i;
1945
1946 if (!PyArg_ParseTuple(args, "i", &i))
1947 return NULL;
1948 return (*func)(self, i);
1949}
1950
1951static struct wrapperbase tab_mul_int[] = {
1952 {"__mul__", (wrapperfunc)wrap_intargfunc, "x.__mul__(n) <==> x*n"},
1953 {"__rmul__", (wrapperfunc)wrap_intargfunc, "x.__rmul__(n) <==> n*x"},
1954 {0}
1955};
1956
1957static struct wrapperbase tab_concat[] = {
1958 {"__add__", (wrapperfunc)wrap_binaryfunc, "x.__add__(y) <==> x+y"},
1959 {0}
1960};
1961
1962static struct wrapperbase tab_imul_int[] = {
1963 {"__imul__", (wrapperfunc)wrap_intargfunc, "x.__imul__(n) <==> x*=n"},
1964 {0}
1965};
1966
Guido van Rossum5d815f32001-08-17 21:57:47 +00001967static int
1968getindex(PyObject *self, PyObject *arg)
1969{
1970 int i;
1971
1972 i = PyInt_AsLong(arg);
1973 if (i == -1 && PyErr_Occurred())
1974 return -1;
1975 if (i < 0) {
1976 PySequenceMethods *sq = self->ob_type->tp_as_sequence;
1977 if (sq && sq->sq_length) {
1978 int n = (*sq->sq_length)(self);
1979 if (n < 0)
1980 return -1;
1981 i += n;
1982 }
1983 }
1984 return i;
1985}
1986
1987static PyObject *
1988wrap_sq_item(PyObject *self, PyObject *args, void *wrapped)
1989{
1990 intargfunc func = (intargfunc)wrapped;
1991 PyObject *arg;
1992 int i;
1993
1994 if (!PyArg_ParseTuple(args, "O", &arg))
1995 return NULL;
1996 i = getindex(self, arg);
1997 if (i == -1 && PyErr_Occurred())
1998 return NULL;
1999 return (*func)(self, i);
2000}
2001
Tim Peters6d6c1a32001-08-02 04:15:00 +00002002static struct wrapperbase tab_getitem_int[] = {
Guido van Rossum5d815f32001-08-17 21:57:47 +00002003 {"__getitem__", (wrapperfunc)wrap_sq_item,
Tim Peters6d6c1a32001-08-02 04:15:00 +00002004 "x.__getitem__(i) <==> x[i]"},
2005 {0}
2006};
2007
2008static PyObject *
2009wrap_intintargfunc(PyObject *self, PyObject *args, void *wrapped)
2010{
2011 intintargfunc func = (intintargfunc)wrapped;
2012 int i, j;
2013
2014 if (!PyArg_ParseTuple(args, "ii", &i, &j))
2015 return NULL;
2016 return (*func)(self, i, j);
2017}
2018
2019static struct wrapperbase tab_getslice[] = {
2020 {"__getslice__", (wrapperfunc)wrap_intintargfunc,
2021 "x.__getslice__(i, j) <==> x[i:j]"},
2022 {0}
2023};
2024
2025static PyObject *
Guido van Rossum5d815f32001-08-17 21:57:47 +00002026wrap_sq_setitem(PyObject *self, PyObject *args, void *wrapped)
Tim Peters6d6c1a32001-08-02 04:15:00 +00002027{
2028 intobjargproc func = (intobjargproc)wrapped;
2029 int i, res;
Guido van Rossum5d815f32001-08-17 21:57:47 +00002030 PyObject *arg, *value;
Tim Peters6d6c1a32001-08-02 04:15:00 +00002031
Guido van Rossum5d815f32001-08-17 21:57:47 +00002032 if (!PyArg_ParseTuple(args, "OO", &arg, &value))
2033 return NULL;
2034 i = getindex(self, arg);
2035 if (i == -1 && PyErr_Occurred())
Tim Peters6d6c1a32001-08-02 04:15:00 +00002036 return NULL;
2037 res = (*func)(self, i, value);
2038 if (res == -1 && PyErr_Occurred())
2039 return NULL;
2040 Py_INCREF(Py_None);
2041 return Py_None;
2042}
2043
Guido van Rossum2b8d7bd2001-08-02 15:31:58 +00002044static PyObject *
Guido van Rossum5d815f32001-08-17 21:57:47 +00002045wrap_sq_delitem(PyObject *self, PyObject *args, void *wrapped)
Guido van Rossum2b8d7bd2001-08-02 15:31:58 +00002046{
2047 intobjargproc func = (intobjargproc)wrapped;
2048 int i, res;
Guido van Rossum5d815f32001-08-17 21:57:47 +00002049 PyObject *arg;
Guido van Rossum2b8d7bd2001-08-02 15:31:58 +00002050
Guido van Rossum5d815f32001-08-17 21:57:47 +00002051 if (!PyArg_ParseTuple(args, "O", &arg))
2052 return NULL;
2053 i = getindex(self, arg);
2054 if (i == -1 && PyErr_Occurred())
Guido van Rossum2b8d7bd2001-08-02 15:31:58 +00002055 return NULL;
2056 res = (*func)(self, i, NULL);
2057 if (res == -1 && PyErr_Occurred())
2058 return NULL;
2059 Py_INCREF(Py_None);
2060 return Py_None;
2061}
2062
Tim Peters6d6c1a32001-08-02 04:15:00 +00002063static struct wrapperbase tab_setitem_int[] = {
Guido van Rossum5d815f32001-08-17 21:57:47 +00002064 {"__setitem__", (wrapperfunc)wrap_sq_setitem,
Tim Peters6d6c1a32001-08-02 04:15:00 +00002065 "x.__setitem__(i, y) <==> x[i]=y"},
Guido van Rossum5d815f32001-08-17 21:57:47 +00002066 {"__delitem__", (wrapperfunc)wrap_sq_delitem,
Guido van Rossum2b8d7bd2001-08-02 15:31:58 +00002067 "x.__delitem__(y) <==> del x[y]"},
Tim Peters6d6c1a32001-08-02 04:15:00 +00002068 {0}
2069};
2070
2071static PyObject *
2072wrap_intintobjargproc(PyObject *self, PyObject *args, void *wrapped)
2073{
2074 intintobjargproc func = (intintobjargproc)wrapped;
2075 int i, j, res;
2076 PyObject *value;
2077
2078 if (!PyArg_ParseTuple(args, "iiO", &i, &j, &value))
2079 return NULL;
2080 res = (*func)(self, i, j, value);
2081 if (res == -1 && PyErr_Occurred())
2082 return NULL;
2083 Py_INCREF(Py_None);
2084 return Py_None;
2085}
2086
2087static struct wrapperbase tab_setslice[] = {
2088 {"__setslice__", (wrapperfunc)wrap_intintobjargproc,
2089 "x.__setslice__(i, j, y) <==> x[i:j]=y"},
2090 {0}
2091};
2092
2093/* XXX objobjproc is a misnomer; should be objargpred */
2094static PyObject *
2095wrap_objobjproc(PyObject *self, PyObject *args, void *wrapped)
2096{
2097 objobjproc func = (objobjproc)wrapped;
2098 int res;
2099 PyObject *value;
2100
2101 if (!PyArg_ParseTuple(args, "O", &value))
2102 return NULL;
2103 res = (*func)(self, value);
2104 if (res == -1 && PyErr_Occurred())
2105 return NULL;
2106 return PyInt_FromLong((long)res);
2107}
2108
2109static struct wrapperbase tab_contains[] = {
2110 {"__contains__", (wrapperfunc)wrap_objobjproc,
2111 "x.__contains__(y) <==> y in x"},
2112 {0}
2113};
2114
2115static PyObject *
2116wrap_objobjargproc(PyObject *self, PyObject *args, void *wrapped)
2117{
2118 objobjargproc func = (objobjargproc)wrapped;
2119 int res;
2120 PyObject *key, *value;
2121
2122 if (!PyArg_ParseTuple(args, "OO", &key, &value))
2123 return NULL;
2124 res = (*func)(self, key, value);
2125 if (res == -1 && PyErr_Occurred())
2126 return NULL;
2127 Py_INCREF(Py_None);
2128 return Py_None;
2129}
2130
Guido van Rossum2b8d7bd2001-08-02 15:31:58 +00002131static PyObject *
2132wrap_delitem(PyObject *self, PyObject *args, void *wrapped)
2133{
2134 objobjargproc func = (objobjargproc)wrapped;
2135 int res;
2136 PyObject *key;
2137
2138 if (!PyArg_ParseTuple(args, "O", &key))
2139 return NULL;
2140 res = (*func)(self, key, NULL);
2141 if (res == -1 && PyErr_Occurred())
2142 return NULL;
2143 Py_INCREF(Py_None);
2144 return Py_None;
2145}
2146
Tim Peters6d6c1a32001-08-02 04:15:00 +00002147static struct wrapperbase tab_setitem[] = {
2148 {"__setitem__", (wrapperfunc)wrap_objobjargproc,
2149 "x.__setitem__(y, z) <==> x[y]=z"},
Guido van Rossum2b8d7bd2001-08-02 15:31:58 +00002150 {"__delitem__", (wrapperfunc)wrap_delitem,
2151 "x.__delitem__(y) <==> del x[y]"},
Tim Peters6d6c1a32001-08-02 04:15:00 +00002152 {0}
2153};
2154
2155static PyObject *
2156wrap_cmpfunc(PyObject *self, PyObject *args, void *wrapped)
2157{
2158 cmpfunc func = (cmpfunc)wrapped;
2159 int res;
2160 PyObject *other;
2161
2162 if (!PyArg_ParseTuple(args, "O", &other))
2163 return NULL;
Guido van Rossum3d45d8f2001-09-24 18:47:40 +00002164 if (other->ob_type->tp_compare != func &&
2165 !PyType_IsSubtype(other->ob_type, self->ob_type)) {
Guido van Rossumceccae52001-09-18 20:03:57 +00002166 PyErr_Format(
2167 PyExc_TypeError,
2168 "%s.__cmp__(x,y) requires y to be a '%s', not a '%s'",
2169 self->ob_type->tp_name,
2170 self->ob_type->tp_name,
2171 other->ob_type->tp_name);
2172 return NULL;
2173 }
Tim Peters6d6c1a32001-08-02 04:15:00 +00002174 res = (*func)(self, other);
2175 if (PyErr_Occurred())
2176 return NULL;
2177 return PyInt_FromLong((long)res);
2178}
2179
2180static struct wrapperbase tab_cmp[] = {
2181 {"__cmp__", (wrapperfunc)wrap_cmpfunc,
2182 "x.__cmp__(y) <==> cmp(x,y)"},
2183 {0}
2184};
2185
2186static struct wrapperbase tab_repr[] = {
2187 {"__repr__", (wrapperfunc)wrap_unaryfunc,
2188 "x.__repr__() <==> repr(x)"},
2189 {0}
2190};
2191
2192static struct wrapperbase tab_getattr[] = {
Guido van Rossum867a8d22001-09-21 19:29:08 +00002193 {"__getattribute__", (wrapperfunc)wrap_binaryfunc,
2194 "x.__getattribute__('name') <==> x.name"},
Tim Peters6d6c1a32001-08-02 04:15:00 +00002195 {0}
2196};
2197
2198static PyObject *
2199wrap_setattr(PyObject *self, PyObject *args, void *wrapped)
2200{
2201 setattrofunc func = (setattrofunc)wrapped;
2202 int res;
2203 PyObject *name, *value;
2204
2205 if (!PyArg_ParseTuple(args, "OO", &name, &value))
2206 return NULL;
2207 res = (*func)(self, name, value);
2208 if (res < 0)
2209 return NULL;
2210 Py_INCREF(Py_None);
2211 return Py_None;
2212}
2213
2214static PyObject *
2215wrap_delattr(PyObject *self, PyObject *args, void *wrapped)
2216{
2217 setattrofunc func = (setattrofunc)wrapped;
2218 int res;
2219 PyObject *name;
2220
2221 if (!PyArg_ParseTuple(args, "O", &name))
2222 return NULL;
2223 res = (*func)(self, name, NULL);
2224 if (res < 0)
2225 return NULL;
2226 Py_INCREF(Py_None);
2227 return Py_None;
2228}
2229
2230static struct wrapperbase tab_setattr[] = {
2231 {"__setattr__", (wrapperfunc)wrap_setattr,
2232 "x.__setattr__('name', value) <==> x.name = value"},
2233 {"__delattr__", (wrapperfunc)wrap_delattr,
2234 "x.__delattr__('name') <==> del x.name"},
2235 {0}
2236};
2237
2238static PyObject *
2239wrap_hashfunc(PyObject *self, PyObject *args, void *wrapped)
2240{
2241 hashfunc func = (hashfunc)wrapped;
2242 long res;
2243
2244 if (!PyArg_ParseTuple(args, ""))
2245 return NULL;
2246 res = (*func)(self);
2247 if (res == -1 && PyErr_Occurred())
2248 return NULL;
2249 return PyInt_FromLong(res);
2250}
2251
2252static struct wrapperbase tab_hash[] = {
2253 {"__hash__", (wrapperfunc)wrap_hashfunc,
2254 "x.__hash__() <==> hash(x)"},
2255 {0}
2256};
2257
2258static PyObject *
2259wrap_call(PyObject *self, PyObject *args, void *wrapped)
2260{
2261 ternaryfunc func = (ternaryfunc)wrapped;
2262
2263 /* XXX What about keyword arguments? */
2264 return (*func)(self, args, NULL);
2265}
2266
2267static struct wrapperbase tab_call[] = {
2268 {"__call__", (wrapperfunc)wrap_call,
2269 "x.__call__(...) <==> x(...)"},
2270 {0}
2271};
2272
2273static struct wrapperbase tab_str[] = {
2274 {"__str__", (wrapperfunc)wrap_unaryfunc,
2275 "x.__str__() <==> str(x)"},
2276 {0}
2277};
2278
2279static PyObject *
2280wrap_richcmpfunc(PyObject *self, PyObject *args, void *wrapped, int op)
2281{
2282 richcmpfunc func = (richcmpfunc)wrapped;
2283 PyObject *other;
2284
2285 if (!PyArg_ParseTuple(args, "O", &other))
2286 return NULL;
2287 return (*func)(self, other, op);
2288}
2289
2290#undef RICHCMP_WRAPPER
2291#define RICHCMP_WRAPPER(NAME, OP) \
2292static PyObject * \
2293richcmp_##NAME(PyObject *self, PyObject *args, void *wrapped) \
2294{ \
2295 return wrap_richcmpfunc(self, args, wrapped, OP); \
2296}
2297
Jack Jansen8e938b42001-08-08 15:29:49 +00002298RICHCMP_WRAPPER(lt, Py_LT)
2299RICHCMP_WRAPPER(le, Py_LE)
2300RICHCMP_WRAPPER(eq, Py_EQ)
2301RICHCMP_WRAPPER(ne, Py_NE)
2302RICHCMP_WRAPPER(gt, Py_GT)
2303RICHCMP_WRAPPER(ge, Py_GE)
Tim Peters6d6c1a32001-08-02 04:15:00 +00002304
2305#undef RICHCMP_ENTRY
2306#define RICHCMP_ENTRY(NAME, EXPR) \
2307 {"__" #NAME "__", (wrapperfunc)richcmp_##NAME, \
2308 "x.__" #NAME "__(y) <==> " EXPR}
2309
2310static struct wrapperbase tab_richcmp[] = {
2311 RICHCMP_ENTRY(lt, "x<y"),
2312 RICHCMP_ENTRY(le, "x<=y"),
2313 RICHCMP_ENTRY(eq, "x==y"),
2314 RICHCMP_ENTRY(ne, "x!=y"),
2315 RICHCMP_ENTRY(gt, "x>y"),
2316 RICHCMP_ENTRY(ge, "x>=y"),
2317 {0}
2318};
2319
2320static struct wrapperbase tab_iter[] = {
2321 {"__iter__", (wrapperfunc)wrap_unaryfunc, "x.__iter__() <==> iter(x)"},
2322 {0}
2323};
2324
2325static PyObject *
2326wrap_next(PyObject *self, PyObject *args, void *wrapped)
2327{
2328 unaryfunc func = (unaryfunc)wrapped;
2329 PyObject *res;
2330
2331 if (!PyArg_ParseTuple(args, ""))
2332 return NULL;
2333 res = (*func)(self);
2334 if (res == NULL && !PyErr_Occurred())
2335 PyErr_SetNone(PyExc_StopIteration);
2336 return res;
2337}
2338
2339static struct wrapperbase tab_next[] = {
2340 {"next", (wrapperfunc)wrap_next,
2341 "x.next() -> the next value, or raise StopIteration"},
2342 {0}
2343};
2344
2345static PyObject *
2346wrap_descr_get(PyObject *self, PyObject *args, void *wrapped)
2347{
2348 descrgetfunc func = (descrgetfunc)wrapped;
2349 PyObject *obj;
2350 PyObject *type = NULL;
2351
2352 if (!PyArg_ParseTuple(args, "O|O", &obj, &type))
2353 return NULL;
Tim Peters6d6c1a32001-08-02 04:15:00 +00002354 return (*func)(self, obj, type);
2355}
2356
2357static struct wrapperbase tab_descr_get[] = {
2358 {"__get__", (wrapperfunc)wrap_descr_get,
2359 "descr.__get__(obj, type) -> value"},
2360 {0}
2361};
2362
2363static PyObject *
2364wrap_descrsetfunc(PyObject *self, PyObject *args, void *wrapped)
2365{
2366 descrsetfunc func = (descrsetfunc)wrapped;
2367 PyObject *obj, *value;
2368 int ret;
2369
2370 if (!PyArg_ParseTuple(args, "OO", &obj, &value))
2371 return NULL;
2372 ret = (*func)(self, obj, value);
2373 if (ret < 0)
2374 return NULL;
2375 Py_INCREF(Py_None);
2376 return Py_None;
2377}
2378
2379static struct wrapperbase tab_descr_set[] = {
2380 {"__set__", (wrapperfunc)wrap_descrsetfunc,
2381 "descr.__set__(obj, value)"},
2382 {0}
2383};
2384
2385static PyObject *
2386wrap_init(PyObject *self, PyObject *args, void *wrapped)
2387{
2388 initproc func = (initproc)wrapped;
2389
2390 /* XXX What about keyword arguments? */
2391 if (func(self, args, NULL) < 0)
2392 return NULL;
2393 Py_INCREF(Py_None);
2394 return Py_None;
2395}
2396
2397static struct wrapperbase tab_init[] = {
2398 {"__init__", (wrapperfunc)wrap_init,
2399 "x.__init__(...) initializes x; "
2400 "see x.__type__.__doc__ for signature"},
2401 {0}
2402};
2403
2404static PyObject *
Guido van Rossum0d231ed2001-08-06 16:50:37 +00002405tp_new_wrapper(PyObject *self, PyObject *args, PyObject *kwds)
Tim Peters6d6c1a32001-08-02 04:15:00 +00002406{
Barry Warsaw60f01882001-08-22 19:24:42 +00002407 PyTypeObject *type, *subtype, *staticbase;
Guido van Rossum0d231ed2001-08-06 16:50:37 +00002408 PyObject *arg0, *res;
2409
2410 if (self == NULL || !PyType_Check(self))
2411 Py_FatalError("__new__() called with non-type 'self'");
2412 type = (PyTypeObject *)self;
2413 if (!PyTuple_Check(args) || PyTuple_GET_SIZE(args) < 1) {
Barry Warsaw60f01882001-08-22 19:24:42 +00002414 PyErr_Format(PyExc_TypeError,
2415 "%s.__new__(): not enough arguments",
2416 type->tp_name);
Guido van Rossum0d231ed2001-08-06 16:50:37 +00002417 return NULL;
2418 }
2419 arg0 = PyTuple_GET_ITEM(args, 0);
2420 if (!PyType_Check(arg0)) {
Barry Warsaw60f01882001-08-22 19:24:42 +00002421 PyErr_Format(PyExc_TypeError,
2422 "%s.__new__(X): X is not a type object (%s)",
2423 type->tp_name,
2424 arg0->ob_type->tp_name);
Guido van Rossum0d231ed2001-08-06 16:50:37 +00002425 return NULL;
2426 }
2427 subtype = (PyTypeObject *)arg0;
2428 if (!PyType_IsSubtype(subtype, type)) {
Barry Warsaw60f01882001-08-22 19:24:42 +00002429 PyErr_Format(PyExc_TypeError,
2430 "%s.__new__(%s): %s is not a subtype of %s",
2431 type->tp_name,
2432 subtype->tp_name,
2433 subtype->tp_name,
2434 type->tp_name);
Guido van Rossum0d231ed2001-08-06 16:50:37 +00002435 return NULL;
2436 }
Barry Warsaw60f01882001-08-22 19:24:42 +00002437
2438 /* Check that the use doesn't do something silly and unsafe like
2439 object.__new__(dictionary). To do this, we check that the
2440 most derived base that's not a heap type is this type. */
2441 staticbase = subtype;
2442 while (staticbase && (staticbase->tp_flags & Py_TPFLAGS_HEAPTYPE))
2443 staticbase = staticbase->tp_base;
Guido van Rossuma8c60f42001-09-14 19:43:36 +00002444 if (staticbase->tp_new != type->tp_new) {
Barry Warsaw60f01882001-08-22 19:24:42 +00002445 PyErr_Format(PyExc_TypeError,
2446 "%s.__new__(%s) is not safe, use %s.__new__()",
2447 type->tp_name,
2448 subtype->tp_name,
2449 staticbase == NULL ? "?" : staticbase->tp_name);
2450 return NULL;
2451 }
2452
Guido van Rossum0d231ed2001-08-06 16:50:37 +00002453 args = PyTuple_GetSlice(args, 1, PyTuple_GET_SIZE(args));
2454 if (args == NULL)
2455 return NULL;
2456 res = type->tp_new(subtype, args, kwds);
2457 Py_DECREF(args);
2458 return res;
Tim Peters6d6c1a32001-08-02 04:15:00 +00002459}
2460
Guido van Rossum0d231ed2001-08-06 16:50:37 +00002461static struct PyMethodDef tp_new_methoddef[] = {
2462 {"__new__", (PyCFunction)tp_new_wrapper, METH_KEYWORDS,
2463 "T.__new__(S, ...) -> a new object with type S, a subtype of T"},
Tim Peters6d6c1a32001-08-02 04:15:00 +00002464 {0}
2465};
2466
2467static int
Guido van Rossum0d231ed2001-08-06 16:50:37 +00002468add_tp_new_wrapper(PyTypeObject *type)
2469{
Guido van Rossumf040ede2001-08-07 16:40:56 +00002470 PyObject *func;
Guido van Rossum0d231ed2001-08-06 16:50:37 +00002471
Guido van Rossumf040ede2001-08-07 16:40:56 +00002472 if (PyDict_GetItemString(type->tp_defined, "__new__") != NULL)
2473 return 0;
2474 func = PyCFunction_New(tp_new_methoddef, (PyObject *)type);
Guido van Rossum0d231ed2001-08-06 16:50:37 +00002475 if (func == NULL)
2476 return -1;
2477 return PyDict_SetItemString(type->tp_defined, "__new__", func);
2478}
2479
Guido van Rossum13d52f02001-08-10 21:24:08 +00002480static int
2481add_wrappers(PyTypeObject *type, struct wrapperbase *wraps, void *wrapped)
2482{
2483 PyObject *dict = type->tp_defined;
2484
2485 for (; wraps->name != NULL; wraps++) {
2486 PyObject *descr;
2487 if (PyDict_GetItemString(dict, wraps->name))
2488 continue;
2489 descr = PyDescr_NewWrapper(type, wraps, wrapped);
2490 if (descr == NULL)
2491 return -1;
2492 if (PyDict_SetItemString(dict, wraps->name, descr) < 0)
2493 return -1;
2494 Py_DECREF(descr);
2495 }
2496 return 0;
2497}
2498
Guido van Rossum528b7eb2001-08-07 17:24:28 +00002499/* This function is called by PyType_Ready() to populate the type's
Guido van Rossumf040ede2001-08-07 16:40:56 +00002500 dictionary with method descriptors for function slots. For each
2501 function slot (like tp_repr) that's defined in the type, one or
2502 more corresponding descriptors are added in the type's tp_defined
2503 dictionary under the appropriate name (like __repr__). Some
2504 function slots cause more than one descriptor to be added (for
2505 example, the nb_add slot adds both __add__ and __radd__
2506 descriptors) and some function slots compete for the same
2507 descriptor (for example both sq_item and mp_subscript generate a
2508 __getitem__ descriptor). This only adds new descriptors and
2509 doesn't overwrite entries in tp_defined that were previously
2510 defined. The descriptors contain a reference to the C function
2511 they must call, so that it's safe if they are copied into a
2512 subtype's __dict__ and the subtype has a different C function in
2513 its slot -- calling the method defined by the descriptor will call
2514 the C function that was used to create it, rather than the C
2515 function present in the slot when it is called. (This is important
2516 because a subtype may have a C function in the slot that calls the
2517 method from the dictionary, and we want to avoid infinite recursion
2518 here.) */
2519
Guido van Rossum0d231ed2001-08-06 16:50:37 +00002520static int
Tim Peters6d6c1a32001-08-02 04:15:00 +00002521add_operators(PyTypeObject *type)
2522{
2523 PySequenceMethods *sq;
2524 PyMappingMethods *mp;
2525 PyNumberMethods *nb;
2526
2527#undef ADD
2528#define ADD(SLOT, TABLE) \
2529 if (SLOT) { \
2530 if (add_wrappers(type, TABLE, (void *)(SLOT)) < 0) \
2531 return -1; \
2532 }
2533
2534 if ((sq = type->tp_as_sequence) != NULL) {
2535 ADD(sq->sq_length, tab_len);
2536 ADD(sq->sq_concat, tab_concat);
2537 ADD(sq->sq_repeat, tab_mul_int);
2538 ADD(sq->sq_item, tab_getitem_int);
2539 ADD(sq->sq_slice, tab_getslice);
2540 ADD(sq->sq_ass_item, tab_setitem_int);
2541 ADD(sq->sq_ass_slice, tab_setslice);
2542 ADD(sq->sq_contains, tab_contains);
2543 ADD(sq->sq_inplace_concat, tab_iadd);
2544 ADD(sq->sq_inplace_repeat, tab_imul_int);
2545 }
2546
2547 if ((mp = type->tp_as_mapping) != NULL) {
2548 if (sq->sq_length == NULL)
2549 ADD(mp->mp_length, tab_len);
2550 ADD(mp->mp_subscript, tab_getitem);
2551 ADD(mp->mp_ass_subscript, tab_setitem);
2552 }
2553
2554 /* We don't support "old-style numbers" because their binary
2555 operators require that both arguments have the same type;
2556 the wrappers here only work for new-style numbers. */
2557 if ((type->tp_flags & Py_TPFLAGS_CHECKTYPES) &&
2558 (nb = type->tp_as_number) != NULL) {
2559 ADD(nb->nb_add, tab_add);
2560 ADD(nb->nb_subtract, tab_sub);
2561 ADD(nb->nb_multiply, tab_mul);
2562 ADD(nb->nb_divide, tab_div);
2563 ADD(nb->nb_remainder, tab_mod);
2564 ADD(nb->nb_divmod, tab_divmod);
2565 ADD(nb->nb_power, tab_pow);
2566 ADD(nb->nb_negative, tab_neg);
2567 ADD(nb->nb_positive, tab_pos);
2568 ADD(nb->nb_absolute, tab_abs);
2569 ADD(nb->nb_nonzero, tab_nonzero);
2570 ADD(nb->nb_invert, tab_invert);
2571 ADD(nb->nb_lshift, tab_lshift);
2572 ADD(nb->nb_rshift, tab_rshift);
2573 ADD(nb->nb_and, tab_and);
2574 ADD(nb->nb_xor, tab_xor);
2575 ADD(nb->nb_or, tab_or);
2576 /* We don't support coerce() -- see above comment */
2577 ADD(nb->nb_int, tab_int);
2578 ADD(nb->nb_long, tab_long);
2579 ADD(nb->nb_float, tab_float);
2580 ADD(nb->nb_oct, tab_oct);
2581 ADD(nb->nb_hex, tab_hex);
2582 ADD(nb->nb_inplace_add, tab_iadd);
2583 ADD(nb->nb_inplace_subtract, tab_isub);
2584 ADD(nb->nb_inplace_multiply, tab_imul);
2585 ADD(nb->nb_inplace_divide, tab_idiv);
2586 ADD(nb->nb_inplace_remainder, tab_imod);
2587 ADD(nb->nb_inplace_power, tab_ipow);
2588 ADD(nb->nb_inplace_lshift, tab_ilshift);
2589 ADD(nb->nb_inplace_rshift, tab_irshift);
2590 ADD(nb->nb_inplace_and, tab_iand);
2591 ADD(nb->nb_inplace_xor, tab_ixor);
2592 ADD(nb->nb_inplace_or, tab_ior);
Guido van Rossum874f15a2001-09-25 21:16:33 +00002593 if (type->tp_flags & Py_TPFLAGS_CHECKTYPES) {
2594 ADD(nb->nb_floor_divide, tab_floordiv);
2595 ADD(nb->nb_true_divide, tab_truediv);
2596 ADD(nb->nb_inplace_floor_divide, tab_ifloordiv);
2597 ADD(nb->nb_inplace_true_divide, tab_itruediv);
2598 }
Tim Peters6d6c1a32001-08-02 04:15:00 +00002599 }
2600
2601 ADD(type->tp_getattro, tab_getattr);
2602 ADD(type->tp_setattro, tab_setattr);
2603 ADD(type->tp_compare, tab_cmp);
2604 ADD(type->tp_repr, tab_repr);
2605 ADD(type->tp_hash, tab_hash);
2606 ADD(type->tp_call, tab_call);
2607 ADD(type->tp_str, tab_str);
2608 ADD(type->tp_richcompare, tab_richcmp);
2609 ADD(type->tp_iter, tab_iter);
2610 ADD(type->tp_iternext, tab_next);
2611 ADD(type->tp_descr_get, tab_descr_get);
2612 ADD(type->tp_descr_set, tab_descr_set);
2613 ADD(type->tp_init, tab_init);
2614
Guido van Rossumf040ede2001-08-07 16:40:56 +00002615 if (type->tp_new != NULL) {
2616 if (add_tp_new_wrapper(type) < 0)
2617 return -1;
2618 }
Tim Peters6d6c1a32001-08-02 04:15:00 +00002619
2620 return 0;
2621}
2622
Guido van Rossumf040ede2001-08-07 16:40:56 +00002623/* Slot wrappers that call the corresponding __foo__ slot. See comments
2624 below at override_slots() for more explanation. */
Tim Peters6d6c1a32001-08-02 04:15:00 +00002625
Guido van Rossumdc91b992001-08-08 22:26:22 +00002626#define SLOT0(FUNCNAME, OPSTR) \
Tim Peters6d6c1a32001-08-02 04:15:00 +00002627static PyObject * \
Guido van Rossumdc91b992001-08-08 22:26:22 +00002628FUNCNAME(PyObject *self) \
Tim Peters6d6c1a32001-08-02 04:15:00 +00002629{ \
Guido van Rossum5592e4d2001-08-28 18:28:21 +00002630 static PyObject *cache_str; \
Guido van Rossum717ce002001-09-14 16:58:08 +00002631 return call_method(self, OPSTR, &cache_str, "()"); \
Tim Peters6d6c1a32001-08-02 04:15:00 +00002632}
2633
Guido van Rossumdc91b992001-08-08 22:26:22 +00002634#define SLOT1(FUNCNAME, OPSTR, ARG1TYPE, ARGCODES) \
Tim Peters6d6c1a32001-08-02 04:15:00 +00002635static PyObject * \
Guido van Rossumdc91b992001-08-08 22:26:22 +00002636FUNCNAME(PyObject *self, ARG1TYPE arg1) \
Tim Peters6d6c1a32001-08-02 04:15:00 +00002637{ \
Guido van Rossum2730b132001-08-28 18:22:14 +00002638 static PyObject *cache_str; \
Guido van Rossum717ce002001-09-14 16:58:08 +00002639 return call_method(self, OPSTR, &cache_str, "(" ARGCODES ")", arg1); \
Tim Peters6d6c1a32001-08-02 04:15:00 +00002640}
2641
Guido van Rossumdc91b992001-08-08 22:26:22 +00002642
2643#define SLOT1BINFULL(FUNCNAME, TESTFUNC, SLOTNAME, OPSTR, ROPSTR) \
Tim Peters6d6c1a32001-08-02 04:15:00 +00002644static PyObject * \
Guido van Rossumdc91b992001-08-08 22:26:22 +00002645FUNCNAME(PyObject *self, PyObject *other) \
Tim Peters6d6c1a32001-08-02 04:15:00 +00002646{ \
Guido van Rossum2730b132001-08-28 18:22:14 +00002647 static PyObject *cache_str, *rcache_str; \
Guido van Rossumdc91b992001-08-08 22:26:22 +00002648 if (self->ob_type->tp_as_number != NULL && \
2649 self->ob_type->tp_as_number->SLOTNAME == TESTFUNC) { \
2650 PyObject *r; \
Guido van Rossumf21c6be2001-09-14 17:51:50 +00002651 r = call_maybe( \
Guido van Rossum717ce002001-09-14 16:58:08 +00002652 self, OPSTR, &cache_str, "(O)", other); \
Guido van Rossumdc91b992001-08-08 22:26:22 +00002653 if (r != Py_NotImplemented || \
2654 other->ob_type == self->ob_type) \
2655 return r; \
2656 Py_DECREF(r); \
2657 } \
2658 if (other->ob_type->tp_as_number != NULL && \
2659 other->ob_type->tp_as_number->SLOTNAME == TESTFUNC) { \
Guido van Rossumf21c6be2001-09-14 17:51:50 +00002660 return call_maybe( \
Guido van Rossum717ce002001-09-14 16:58:08 +00002661 other, ROPSTR, &rcache_str, "(O)", self); \
Guido van Rossumdc91b992001-08-08 22:26:22 +00002662 } \
2663 Py_INCREF(Py_NotImplemented); \
2664 return Py_NotImplemented; \
2665}
2666
2667#define SLOT1BIN(FUNCNAME, SLOTNAME, OPSTR, ROPSTR) \
2668 SLOT1BINFULL(FUNCNAME, FUNCNAME, SLOTNAME, OPSTR, ROPSTR)
2669
2670#define SLOT2(FUNCNAME, OPSTR, ARG1TYPE, ARG2TYPE, ARGCODES) \
2671static PyObject * \
2672FUNCNAME(PyObject *self, ARG1TYPE arg1, ARG2TYPE arg2) \
2673{ \
Guido van Rossum2730b132001-08-28 18:22:14 +00002674 static PyObject *cache_str; \
Guido van Rossum717ce002001-09-14 16:58:08 +00002675 return call_method(self, OPSTR, &cache_str, \
2676 "(" ARGCODES ")", arg1, arg2); \
Tim Peters6d6c1a32001-08-02 04:15:00 +00002677}
2678
2679static int
2680slot_sq_length(PyObject *self)
2681{
Guido van Rossum2730b132001-08-28 18:22:14 +00002682 static PyObject *len_str;
Guido van Rossum717ce002001-09-14 16:58:08 +00002683 PyObject *res = call_method(self, "__len__", &len_str, "()");
Tim Peters6d6c1a32001-08-02 04:15:00 +00002684
2685 if (res == NULL)
2686 return -1;
2687 return (int)PyInt_AsLong(res);
2688}
2689
Guido van Rossumdc91b992001-08-08 22:26:22 +00002690SLOT1(slot_sq_concat, "__add__", PyObject *, "O")
2691SLOT1(slot_sq_repeat, "__mul__", int, "i")
2692SLOT1(slot_sq_item, "__getitem__", int, "i")
2693SLOT2(slot_sq_slice, "__getslice__", int, int, "ii")
Tim Peters6d6c1a32001-08-02 04:15:00 +00002694
2695static int
2696slot_sq_ass_item(PyObject *self, int index, PyObject *value)
2697{
2698 PyObject *res;
Guido van Rossum2730b132001-08-28 18:22:14 +00002699 static PyObject *delitem_str, *setitem_str;
Tim Peters6d6c1a32001-08-02 04:15:00 +00002700
2701 if (value == NULL)
Guido van Rossum2730b132001-08-28 18:22:14 +00002702 res = call_method(self, "__delitem__", &delitem_str,
Guido van Rossum717ce002001-09-14 16:58:08 +00002703 "(i)", index);
Tim Peters6d6c1a32001-08-02 04:15:00 +00002704 else
Guido van Rossum2730b132001-08-28 18:22:14 +00002705 res = call_method(self, "__setitem__", &setitem_str,
Guido van Rossum717ce002001-09-14 16:58:08 +00002706 "(iO)", index, value);
Tim Peters6d6c1a32001-08-02 04:15:00 +00002707 if (res == NULL)
2708 return -1;
2709 Py_DECREF(res);
2710 return 0;
2711}
2712
2713static int
2714slot_sq_ass_slice(PyObject *self, int i, int j, PyObject *value)
2715{
2716 PyObject *res;
Guido van Rossum2730b132001-08-28 18:22:14 +00002717 static PyObject *delslice_str, *setslice_str;
Tim Peters6d6c1a32001-08-02 04:15:00 +00002718
2719 if (value == NULL)
Guido van Rossum2730b132001-08-28 18:22:14 +00002720 res = call_method(self, "__delslice__", &delslice_str,
Guido van Rossum717ce002001-09-14 16:58:08 +00002721 "(ii)", i, j);
Tim Peters6d6c1a32001-08-02 04:15:00 +00002722 else
Guido van Rossum2730b132001-08-28 18:22:14 +00002723 res = call_method(self, "__setslice__", &setslice_str,
Guido van Rossum717ce002001-09-14 16:58:08 +00002724 "(iiO)", i, j, value);
Tim Peters6d6c1a32001-08-02 04:15:00 +00002725 if (res == NULL)
2726 return -1;
2727 Py_DECREF(res);
2728 return 0;
2729}
2730
2731static int
2732slot_sq_contains(PyObject *self, PyObject *value)
2733{
Guido van Rossumb8f63662001-08-15 23:57:02 +00002734 PyObject *func, *res, *args;
Guido van Rossum60718732001-08-28 17:47:51 +00002735 static PyObject *contains_str;
Tim Peters6d6c1a32001-08-02 04:15:00 +00002736
Guido van Rossum60718732001-08-28 17:47:51 +00002737 func = lookup_method(self, "__contains__", &contains_str);
Guido van Rossumb8f63662001-08-15 23:57:02 +00002738
2739 if (func != NULL) {
2740 args = Py_BuildValue("(O)", value);
2741 if (args == NULL)
2742 res = NULL;
2743 else {
Guido van Rossum717ce002001-09-14 16:58:08 +00002744 res = PyObject_Call(func, args, NULL);
Guido van Rossumb8f63662001-08-15 23:57:02 +00002745 Py_DECREF(args);
2746 }
2747 Py_DECREF(func);
2748 if (res == NULL)
2749 return -1;
2750 return PyObject_IsTrue(res);
2751 }
2752 else {
2753 PyErr_Clear();
Tim Peters16a77ad2001-09-08 04:00:12 +00002754 return _PySequence_IterSearch(self, value,
2755 PY_ITERSEARCH_CONTAINS);
Guido van Rossumb8f63662001-08-15 23:57:02 +00002756 }
Tim Peters6d6c1a32001-08-02 04:15:00 +00002757}
2758
Guido van Rossumdc91b992001-08-08 22:26:22 +00002759SLOT1(slot_sq_inplace_concat, "__iadd__", PyObject *, "O")
2760SLOT1(slot_sq_inplace_repeat, "__imul__", int, "i")
Tim Peters6d6c1a32001-08-02 04:15:00 +00002761
2762#define slot_mp_length slot_sq_length
2763
Guido van Rossumdc91b992001-08-08 22:26:22 +00002764SLOT1(slot_mp_subscript, "__getitem__", PyObject *, "O")
Tim Peters6d6c1a32001-08-02 04:15:00 +00002765
2766static int
2767slot_mp_ass_subscript(PyObject *self, PyObject *key, PyObject *value)
2768{
2769 PyObject *res;
Guido van Rossum2730b132001-08-28 18:22:14 +00002770 static PyObject *delitem_str, *setitem_str;
Tim Peters6d6c1a32001-08-02 04:15:00 +00002771
2772 if (value == NULL)
Guido van Rossum2730b132001-08-28 18:22:14 +00002773 res = call_method(self, "__delitem__", &delitem_str,
Guido van Rossum717ce002001-09-14 16:58:08 +00002774 "(O)", key);
Tim Peters6d6c1a32001-08-02 04:15:00 +00002775 else
Guido van Rossum2730b132001-08-28 18:22:14 +00002776 res = call_method(self, "__setitem__", &setitem_str,
Guido van Rossum717ce002001-09-14 16:58:08 +00002777 "(OO)", key, value);
Tim Peters6d6c1a32001-08-02 04:15:00 +00002778 if (res == NULL)
2779 return -1;
2780 Py_DECREF(res);
2781 return 0;
2782}
2783
Guido van Rossumdc91b992001-08-08 22:26:22 +00002784SLOT1BIN(slot_nb_add, nb_add, "__add__", "__radd__")
2785SLOT1BIN(slot_nb_subtract, nb_subtract, "__sub__", "__rsub__")
2786SLOT1BIN(slot_nb_multiply, nb_multiply, "__mul__", "__rmul__")
2787SLOT1BIN(slot_nb_divide, nb_divide, "__div__", "__rdiv__")
2788SLOT1BIN(slot_nb_remainder, nb_remainder, "__mod__", "__rmod__")
2789SLOT1BIN(slot_nb_divmod, nb_divmod, "__divmod__", "__rdivmod__")
2790
2791staticforward PyObject *slot_nb_power(PyObject *, PyObject *, PyObject *);
2792
2793SLOT1BINFULL(slot_nb_power_binary, slot_nb_power,
2794 nb_power, "__pow__", "__rpow__")
2795
2796static PyObject *
2797slot_nb_power(PyObject *self, PyObject *other, PyObject *modulus)
2798{
Guido van Rossum2730b132001-08-28 18:22:14 +00002799 static PyObject *pow_str;
2800
Guido van Rossumdc91b992001-08-08 22:26:22 +00002801 if (modulus == Py_None)
2802 return slot_nb_power_binary(self, other);
2803 /* Three-arg power doesn't use __rpow__ */
Guido van Rossum2730b132001-08-28 18:22:14 +00002804 return call_method(self, "__pow__", &pow_str,
Guido van Rossum717ce002001-09-14 16:58:08 +00002805 "(OO)", other, modulus);
Guido van Rossumdc91b992001-08-08 22:26:22 +00002806}
2807
2808SLOT0(slot_nb_negative, "__neg__")
2809SLOT0(slot_nb_positive, "__pos__")
2810SLOT0(slot_nb_absolute, "__abs__")
Tim Peters6d6c1a32001-08-02 04:15:00 +00002811
2812static int
2813slot_nb_nonzero(PyObject *self)
2814{
Guido van Rossumb8f63662001-08-15 23:57:02 +00002815 PyObject *func, *res;
Guido van Rossum60718732001-08-28 17:47:51 +00002816 static PyObject *nonzero_str, *len_str;
Tim Peters6d6c1a32001-08-02 04:15:00 +00002817
Guido van Rossum60718732001-08-28 17:47:51 +00002818 func = lookup_method(self, "__nonzero__", &nonzero_str);
Guido van Rossumb8f63662001-08-15 23:57:02 +00002819 if (func == NULL) {
2820 PyErr_Clear();
Guido van Rossum60718732001-08-28 17:47:51 +00002821 func = lookup_method(self, "__len__", &len_str);
Guido van Rossumb8f63662001-08-15 23:57:02 +00002822 }
2823
2824 if (func != NULL) {
Guido van Rossum717ce002001-09-14 16:58:08 +00002825 res = PyObject_CallObject(func, NULL);
Guido van Rossumb8f63662001-08-15 23:57:02 +00002826 Py_DECREF(func);
2827 if (res == NULL)
2828 return -1;
2829 return PyObject_IsTrue(res);
2830 }
2831 else {
2832 PyErr_Clear();
2833 return 1;
2834 }
Tim Peters6d6c1a32001-08-02 04:15:00 +00002835}
2836
Guido van Rossumdc91b992001-08-08 22:26:22 +00002837SLOT0(slot_nb_invert, "__invert__")
2838SLOT1BIN(slot_nb_lshift, nb_lshift, "__lshift__", "__rlshift__")
2839SLOT1BIN(slot_nb_rshift, nb_rshift, "__rshift__", "__rrshift__")
2840SLOT1BIN(slot_nb_and, nb_and, "__and__", "__rand__")
2841SLOT1BIN(slot_nb_xor, nb_xor, "__xor__", "__rxor__")
2842SLOT1BIN(slot_nb_or, nb_or, "__or__", "__ror__")
Tim Peters6d6c1a32001-08-02 04:15:00 +00002843/* Not coerce() */
Guido van Rossumdc91b992001-08-08 22:26:22 +00002844SLOT0(slot_nb_int, "__int__")
2845SLOT0(slot_nb_long, "__long__")
2846SLOT0(slot_nb_float, "__float__")
2847SLOT0(slot_nb_oct, "__oct__")
2848SLOT0(slot_nb_hex, "__hex__")
2849SLOT1(slot_nb_inplace_add, "__iadd__", PyObject *, "O")
2850SLOT1(slot_nb_inplace_subtract, "__isub__", PyObject *, "O")
2851SLOT1(slot_nb_inplace_multiply, "__imul__", PyObject *, "O")
2852SLOT1(slot_nb_inplace_divide, "__idiv__", PyObject *, "O")
2853SLOT1(slot_nb_inplace_remainder, "__imod__", PyObject *, "O")
2854SLOT2(slot_nb_inplace_power, "__ipow__", PyObject *, PyObject *, "OO")
2855SLOT1(slot_nb_inplace_lshift, "__ilshift__", PyObject *, "O")
2856SLOT1(slot_nb_inplace_rshift, "__irshift__", PyObject *, "O")
2857SLOT1(slot_nb_inplace_and, "__iand__", PyObject *, "O")
2858SLOT1(slot_nb_inplace_xor, "__ixor__", PyObject *, "O")
2859SLOT1(slot_nb_inplace_or, "__ior__", PyObject *, "O")
2860SLOT1BIN(slot_nb_floor_divide, nb_floor_divide,
2861 "__floordiv__", "__rfloordiv__")
2862SLOT1BIN(slot_nb_true_divide, nb_true_divide, "__truediv__", "__rtruediv__")
2863SLOT1(slot_nb_inplace_floor_divide, "__ifloordiv__", PyObject *, "O")
2864SLOT1(slot_nb_inplace_true_divide, "__itruediv__", PyObject *, "O")
Tim Peters6d6c1a32001-08-02 04:15:00 +00002865
2866static int
Guido van Rossumb8f63662001-08-15 23:57:02 +00002867half_compare(PyObject *self, PyObject *other)
Tim Peters6d6c1a32001-08-02 04:15:00 +00002868{
Guido van Rossumb8f63662001-08-15 23:57:02 +00002869 PyObject *func, *args, *res;
Guido van Rossum60718732001-08-28 17:47:51 +00002870 static PyObject *cmp_str;
Guido van Rossumb8f63662001-08-15 23:57:02 +00002871 int c;
Tim Peters6d6c1a32001-08-02 04:15:00 +00002872
Guido van Rossum60718732001-08-28 17:47:51 +00002873 func = lookup_method(self, "__cmp__", &cmp_str);
Guido van Rossumb8f63662001-08-15 23:57:02 +00002874 if (func == NULL) {
2875 PyErr_Clear();
2876 }
2877 else {
2878 args = Py_BuildValue("(O)", other);
2879 if (args == NULL)
2880 res = NULL;
2881 else {
Guido van Rossum717ce002001-09-14 16:58:08 +00002882 res = PyObject_Call(func, args, NULL);
Guido van Rossumb8f63662001-08-15 23:57:02 +00002883 Py_DECREF(args);
2884 }
2885 if (res != Py_NotImplemented) {
2886 if (res == NULL)
2887 return -2;
2888 c = PyInt_AsLong(res);
2889 Py_DECREF(res);
2890 if (c == -1 && PyErr_Occurred())
2891 return -2;
2892 return (c < 0) ? -1 : (c > 0) ? 1 : 0;
2893 }
2894 Py_DECREF(res);
2895 }
2896 return 2;
Tim Peters6d6c1a32001-08-02 04:15:00 +00002897}
2898
Guido van Rossumab3b0342001-09-18 20:38:53 +00002899/* This slot is published for the benefit of try_3way_compare in object.c */
2900int
2901_PyObject_SlotCompare(PyObject *self, PyObject *other)
Guido van Rossumb8f63662001-08-15 23:57:02 +00002902{
2903 int c;
2904
Guido van Rossumab3b0342001-09-18 20:38:53 +00002905 if (self->ob_type->tp_compare == _PyObject_SlotCompare) {
Guido van Rossumb8f63662001-08-15 23:57:02 +00002906 c = half_compare(self, other);
2907 if (c <= 1)
2908 return c;
2909 }
Guido van Rossumab3b0342001-09-18 20:38:53 +00002910 if (other->ob_type->tp_compare == _PyObject_SlotCompare) {
Guido van Rossumb8f63662001-08-15 23:57:02 +00002911 c = half_compare(other, self);
2912 if (c < -1)
2913 return -2;
2914 if (c <= 1)
2915 return -c;
2916 }
2917 return (void *)self < (void *)other ? -1 :
2918 (void *)self > (void *)other ? 1 : 0;
2919}
2920
2921static PyObject *
2922slot_tp_repr(PyObject *self)
2923{
2924 PyObject *func, *res;
Guido van Rossum60718732001-08-28 17:47:51 +00002925 static PyObject *repr_str;
Guido van Rossumb8f63662001-08-15 23:57:02 +00002926
Guido van Rossum60718732001-08-28 17:47:51 +00002927 func = lookup_method(self, "__repr__", &repr_str);
Guido van Rossumb8f63662001-08-15 23:57:02 +00002928 if (func != NULL) {
2929 res = PyEval_CallObject(func, NULL);
2930 Py_DECREF(func);
2931 return res;
2932 }
Barry Warsaw7ce36942001-08-24 18:34:26 +00002933 PyErr_Clear();
2934 return PyString_FromFormat("<%s object at %p>",
2935 self->ob_type->tp_name, self);
Guido van Rossumb8f63662001-08-15 23:57:02 +00002936}
2937
2938static PyObject *
2939slot_tp_str(PyObject *self)
2940{
2941 PyObject *func, *res;
Guido van Rossum60718732001-08-28 17:47:51 +00002942 static PyObject *str_str;
Guido van Rossumb8f63662001-08-15 23:57:02 +00002943
Guido van Rossum60718732001-08-28 17:47:51 +00002944 func = lookup_method(self, "__str__", &str_str);
Guido van Rossumb8f63662001-08-15 23:57:02 +00002945 if (func != NULL) {
2946 res = PyEval_CallObject(func, NULL);
2947 Py_DECREF(func);
2948 return res;
2949 }
2950 else {
2951 PyErr_Clear();
2952 return slot_tp_repr(self);
2953 }
2954}
Tim Peters6d6c1a32001-08-02 04:15:00 +00002955
2956static long
2957slot_tp_hash(PyObject *self)
2958{
Guido van Rossumb8f63662001-08-15 23:57:02 +00002959 PyObject *func, *res;
Guido van Rossum60718732001-08-28 17:47:51 +00002960 static PyObject *hash_str, *eq_str, *cmp_str;
2961
Tim Peters6d6c1a32001-08-02 04:15:00 +00002962 long h;
2963
Guido van Rossum60718732001-08-28 17:47:51 +00002964 func = lookup_method(self, "__hash__", &hash_str);
Guido van Rossumb8f63662001-08-15 23:57:02 +00002965
2966 if (func != NULL) {
2967 res = PyEval_CallObject(func, NULL);
2968 Py_DECREF(func);
2969 if (res == NULL)
2970 return -1;
2971 h = PyInt_AsLong(res);
2972 }
2973 else {
2974 PyErr_Clear();
Guido van Rossum60718732001-08-28 17:47:51 +00002975 func = lookup_method(self, "__eq__", &eq_str);
Guido van Rossumb8f63662001-08-15 23:57:02 +00002976 if (func == NULL) {
2977 PyErr_Clear();
Guido van Rossum60718732001-08-28 17:47:51 +00002978 func = lookup_method(self, "__cmp__", &cmp_str);
Guido van Rossumb8f63662001-08-15 23:57:02 +00002979 }
2980 if (func != NULL) {
2981 Py_DECREF(func);
2982 PyErr_SetString(PyExc_TypeError, "unhashable type");
2983 return -1;
2984 }
2985 PyErr_Clear();
2986 h = _Py_HashPointer((void *)self);
2987 }
Tim Peters6d6c1a32001-08-02 04:15:00 +00002988 if (h == -1 && !PyErr_Occurred())
2989 h = -2;
2990 return h;
2991}
2992
2993static PyObject *
2994slot_tp_call(PyObject *self, PyObject *args, PyObject *kwds)
2995{
Guido van Rossum60718732001-08-28 17:47:51 +00002996 static PyObject *call_str;
2997 PyObject *meth = lookup_method(self, "__call__", &call_str);
Tim Peters6d6c1a32001-08-02 04:15:00 +00002998 PyObject *res;
2999
3000 if (meth == NULL)
3001 return NULL;
3002 res = PyObject_Call(meth, args, kwds);
3003 Py_DECREF(meth);
3004 return res;
3005}
3006
Tim Peters6d6c1a32001-08-02 04:15:00 +00003007static PyObject *
3008slot_tp_getattro(PyObject *self, PyObject *name)
3009{
3010 PyTypeObject *tp = self->ob_type;
Tim Peters6d6c1a32001-08-02 04:15:00 +00003011 PyObject *getattr;
Guido van Rossum8e248182001-08-12 05:17:56 +00003012 static PyObject *getattr_str = NULL;
Tim Peters6d6c1a32001-08-02 04:15:00 +00003013
Guido van Rossum8e248182001-08-12 05:17:56 +00003014 if (getattr_str == NULL) {
Guido van Rossum867a8d22001-09-21 19:29:08 +00003015 getattr_str = PyString_InternFromString("__getattribute__");
Guido van Rossum8e248182001-08-12 05:17:56 +00003016 if (getattr_str == NULL)
3017 return NULL;
Tim Peters6d6c1a32001-08-02 04:15:00 +00003018 }
Guido van Rossum8e248182001-08-12 05:17:56 +00003019 getattr = _PyType_Lookup(tp, getattr_str);
Guido van Rossumc3542212001-08-16 09:18:56 +00003020 if (getattr == NULL) {
3021 /* Avoid further slowdowns */
3022 if (tp->tp_getattro == slot_tp_getattro)
3023 tp->tp_getattro = PyObject_GenericGetAttr;
Guido van Rossum8e248182001-08-12 05:17:56 +00003024 return PyObject_GenericGetAttr(self, name);
Guido van Rossumc3542212001-08-16 09:18:56 +00003025 }
Tim Peters6d6c1a32001-08-02 04:15:00 +00003026 return PyObject_CallFunction(getattr, "OO", self, name);
3027}
3028
Guido van Rossum19c1cd52001-09-21 21:24:49 +00003029static PyObject *
3030slot_tp_getattr_hook(PyObject *self, PyObject *name)
3031{
3032 PyTypeObject *tp = self->ob_type;
3033 PyObject *getattr, *getattribute, *res;
3034 static PyObject *getattribute_str = NULL;
3035 static PyObject *getattr_str = NULL;
3036
3037 if (getattr_str == NULL) {
3038 getattr_str = PyString_InternFromString("__getattr__");
3039 if (getattr_str == NULL)
3040 return NULL;
3041 }
3042 if (getattribute_str == NULL) {
3043 getattribute_str =
3044 PyString_InternFromString("__getattribute__");
3045 if (getattribute_str == NULL)
3046 return NULL;
3047 }
3048 getattr = _PyType_Lookup(tp, getattr_str);
3049 getattribute = _PyType_Lookup(tp, getattribute_str);
3050 if (getattr == NULL && getattribute == NULL) {
3051 /* Avoid further slowdowns */
3052 if (tp->tp_getattro == slot_tp_getattr_hook)
3053 tp->tp_getattro = PyObject_GenericGetAttr;
3054 return PyObject_GenericGetAttr(self, name);
3055 }
3056 if (getattribute == NULL)
3057 res = PyObject_GenericGetAttr(self, name);
3058 else
3059 res = PyObject_CallFunction(getattribute, "OO", self, name);
Guido van Rossum3926a632001-09-25 16:25:58 +00003060 if (getattr != NULL &&
3061 res == NULL && PyErr_ExceptionMatches(PyExc_AttributeError)) {
Guido van Rossum19c1cd52001-09-21 21:24:49 +00003062 PyErr_Clear();
3063 res = PyObject_CallFunction(getattr, "OO", self, name);
3064 }
3065 return res;
3066}
3067
Tim Peters6d6c1a32001-08-02 04:15:00 +00003068static int
3069slot_tp_setattro(PyObject *self, PyObject *name, PyObject *value)
3070{
3071 PyObject *res;
Guido van Rossum2730b132001-08-28 18:22:14 +00003072 static PyObject *delattr_str, *setattr_str;
Tim Peters6d6c1a32001-08-02 04:15:00 +00003073
3074 if (value == NULL)
Guido van Rossum2730b132001-08-28 18:22:14 +00003075 res = call_method(self, "__delattr__", &delattr_str,
Guido van Rossum717ce002001-09-14 16:58:08 +00003076 "(O)", name);
Tim Peters6d6c1a32001-08-02 04:15:00 +00003077 else
Guido van Rossum2730b132001-08-28 18:22:14 +00003078 res = call_method(self, "__setattr__", &setattr_str,
Guido van Rossum717ce002001-09-14 16:58:08 +00003079 "(OO)", name, value);
Tim Peters6d6c1a32001-08-02 04:15:00 +00003080 if (res == NULL)
3081 return -1;
3082 Py_DECREF(res);
3083 return 0;
3084}
3085
3086/* Map rich comparison operators to their __xx__ namesakes */
3087static char *name_op[] = {
3088 "__lt__",
3089 "__le__",
3090 "__eq__",
3091 "__ne__",
3092 "__gt__",
3093 "__ge__",
3094};
3095
3096static PyObject *
Guido van Rossumb8f63662001-08-15 23:57:02 +00003097half_richcompare(PyObject *self, PyObject *other, int op)
Tim Peters6d6c1a32001-08-02 04:15:00 +00003098{
Guido van Rossumb8f63662001-08-15 23:57:02 +00003099 PyObject *func, *args, *res;
Guido van Rossum60718732001-08-28 17:47:51 +00003100 static PyObject *op_str[6];
Tim Peters6d6c1a32001-08-02 04:15:00 +00003101
Guido van Rossum60718732001-08-28 17:47:51 +00003102 func = lookup_method(self, name_op[op], &op_str[op]);
Guido van Rossumb8f63662001-08-15 23:57:02 +00003103 if (func == NULL) {
3104 PyErr_Clear();
3105 Py_INCREF(Py_NotImplemented);
3106 return Py_NotImplemented;
3107 }
3108 args = Py_BuildValue("(O)", other);
3109 if (args == NULL)
3110 res = NULL;
3111 else {
Guido van Rossum717ce002001-09-14 16:58:08 +00003112 res = PyObject_Call(func, args, NULL);
Guido van Rossumb8f63662001-08-15 23:57:02 +00003113 Py_DECREF(args);
3114 }
3115 Py_DECREF(func);
Tim Peters6d6c1a32001-08-02 04:15:00 +00003116 return res;
3117}
3118
Guido van Rossumb8f63662001-08-15 23:57:02 +00003119/* Map rich comparison operators to their swapped version, e.g. LT --> GT */
3120static int swapped_op[] = {Py_GT, Py_GE, Py_EQ, Py_NE, Py_LT, Py_LE};
3121
3122static PyObject *
3123slot_tp_richcompare(PyObject *self, PyObject *other, int op)
3124{
3125 PyObject *res;
3126
3127 if (self->ob_type->tp_richcompare == slot_tp_richcompare) {
3128 res = half_richcompare(self, other, op);
3129 if (res != Py_NotImplemented)
3130 return res;
3131 Py_DECREF(res);
3132 }
3133 if (other->ob_type->tp_richcompare == slot_tp_richcompare) {
3134 res = half_richcompare(other, self, swapped_op[op]);
3135 if (res != Py_NotImplemented) {
3136 return res;
3137 }
3138 Py_DECREF(res);
3139 }
3140 Py_INCREF(Py_NotImplemented);
3141 return Py_NotImplemented;
3142}
3143
3144static PyObject *
3145slot_tp_iter(PyObject *self)
3146{
3147 PyObject *func, *res;
Guido van Rossum60718732001-08-28 17:47:51 +00003148 static PyObject *iter_str, *getitem_str;
Guido van Rossumb8f63662001-08-15 23:57:02 +00003149
Guido van Rossum60718732001-08-28 17:47:51 +00003150 func = lookup_method(self, "__iter__", &iter_str);
Guido van Rossumb8f63662001-08-15 23:57:02 +00003151 if (func != NULL) {
3152 res = PyObject_CallObject(func, NULL);
3153 Py_DECREF(func);
3154 return res;
3155 }
3156 PyErr_Clear();
Guido van Rossum60718732001-08-28 17:47:51 +00003157 func = lookup_method(self, "__getitem__", &getitem_str);
Guido van Rossumb8f63662001-08-15 23:57:02 +00003158 if (func == NULL) {
3159 PyErr_SetString(PyExc_TypeError, "iter() of non-sequence");
3160 return NULL;
3161 }
3162 Py_DECREF(func);
3163 return PySeqIter_New(self);
3164}
Tim Peters6d6c1a32001-08-02 04:15:00 +00003165
3166static PyObject *
3167slot_tp_iternext(PyObject *self)
3168{
Guido van Rossum2730b132001-08-28 18:22:14 +00003169 static PyObject *next_str;
Guido van Rossum717ce002001-09-14 16:58:08 +00003170 return call_method(self, "next", &next_str, "()");
Tim Peters6d6c1a32001-08-02 04:15:00 +00003171}
3172
Guido van Rossum1a493502001-08-17 16:47:50 +00003173static PyObject *
3174slot_tp_descr_get(PyObject *self, PyObject *obj, PyObject *type)
3175{
3176 PyTypeObject *tp = self->ob_type;
3177 PyObject *get;
3178 static PyObject *get_str = NULL;
3179
3180 if (get_str == NULL) {
3181 get_str = PyString_InternFromString("__get__");
3182 if (get_str == NULL)
3183 return NULL;
3184 }
3185 get = _PyType_Lookup(tp, get_str);
3186 if (get == NULL) {
3187 /* Avoid further slowdowns */
3188 if (tp->tp_descr_get == slot_tp_descr_get)
3189 tp->tp_descr_get = NULL;
3190 Py_INCREF(self);
3191 return self;
3192 }
Guido van Rossum2c252392001-08-24 10:13:31 +00003193 if (obj == NULL)
3194 obj = Py_None;
3195 if (type == NULL)
3196 type = Py_None;
Guido van Rossum1a493502001-08-17 16:47:50 +00003197 return PyObject_CallFunction(get, "OOO", self, obj, type);
3198}
Tim Peters6d6c1a32001-08-02 04:15:00 +00003199
3200static int
3201slot_tp_descr_set(PyObject *self, PyObject *target, PyObject *value)
3202{
Guido van Rossum2c252392001-08-24 10:13:31 +00003203 PyObject *res;
Guido van Rossum2730b132001-08-28 18:22:14 +00003204 static PyObject *del_str, *set_str;
Guido van Rossum2c252392001-08-24 10:13:31 +00003205
3206 if (value == NULL)
Guido van Rossum2730b132001-08-28 18:22:14 +00003207 res = call_method(self, "__del__", &del_str,
Guido van Rossum717ce002001-09-14 16:58:08 +00003208 "(O)", target);
Guido van Rossum2c252392001-08-24 10:13:31 +00003209 else
Guido van Rossum2730b132001-08-28 18:22:14 +00003210 res = call_method(self, "__set__", &set_str,
Guido van Rossum717ce002001-09-14 16:58:08 +00003211 "(OO)", target, value);
Tim Peters6d6c1a32001-08-02 04:15:00 +00003212 if (res == NULL)
3213 return -1;
3214 Py_DECREF(res);
3215 return 0;
3216}
3217
3218static int
3219slot_tp_init(PyObject *self, PyObject *args, PyObject *kwds)
3220{
Guido van Rossum60718732001-08-28 17:47:51 +00003221 static PyObject *init_str;
3222 PyObject *meth = lookup_method(self, "__init__", &init_str);
Tim Peters6d6c1a32001-08-02 04:15:00 +00003223 PyObject *res;
3224
3225 if (meth == NULL)
3226 return -1;
3227 res = PyObject_Call(meth, args, kwds);
3228 Py_DECREF(meth);
3229 if (res == NULL)
3230 return -1;
3231 Py_DECREF(res);
3232 return 0;
3233}
3234
3235static PyObject *
3236slot_tp_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
3237{
3238 PyObject *func = PyObject_GetAttrString((PyObject *)type, "__new__");
3239 PyObject *newargs, *x;
3240 int i, n;
3241
3242 if (func == NULL)
3243 return NULL;
3244 assert(PyTuple_Check(args));
3245 n = PyTuple_GET_SIZE(args);
3246 newargs = PyTuple_New(n+1);
3247 if (newargs == NULL)
3248 return NULL;
3249 Py_INCREF(type);
3250 PyTuple_SET_ITEM(newargs, 0, (PyObject *)type);
3251 for (i = 0; i < n; i++) {
3252 x = PyTuple_GET_ITEM(args, i);
3253 Py_INCREF(x);
3254 PyTuple_SET_ITEM(newargs, i+1, x);
3255 }
3256 x = PyObject_Call(func, newargs, kwds);
3257 Py_DECREF(func);
3258 return x;
3259}
3260
Guido van Rossumf040ede2001-08-07 16:40:56 +00003261/* This is called at the very end of type_new() (even after
Guido van Rossum528b7eb2001-08-07 17:24:28 +00003262 PyType_Ready()) to complete the initialization of dynamic types.
Guido van Rossumf040ede2001-08-07 16:40:56 +00003263 The dict argument is the dictionary argument passed to type_new(),
3264 which is the local namespace of the class statement, in other
3265 words, it contains the methods. For each special method (like
3266 __repr__) defined in the dictionary, the corresponding function
3267 slot in the type object (like tp_repr) is set to a special function
3268 whose name is 'slot_' followed by the slot name and whose signature
3269 is whatever is required for that slot. These slot functions look
3270 up the corresponding method in the type's dictionary and call it.
3271 The slot functions have to take care of the various peculiarities
3272 of the mapping between slots and special methods, such as mapping
3273 one slot to multiple methods (tp_richcompare <--> __le__, __lt__
3274 etc.) or mapping multiple slots to a single method (sq_item,
3275 mp_subscript <--> __getitem__). */
3276
Tim Peters6d6c1a32001-08-02 04:15:00 +00003277static void
3278override_slots(PyTypeObject *type, PyObject *dict)
3279{
3280 PySequenceMethods *sq = type->tp_as_sequence;
3281 PyMappingMethods *mp = type->tp_as_mapping;
3282 PyNumberMethods *nb = type->tp_as_number;
3283
Guido van Rossumdc91b992001-08-08 22:26:22 +00003284#define SQSLOT(OPNAME, SLOTNAME, FUNCNAME) \
Guido van Rossum8e248182001-08-12 05:17:56 +00003285 if (dict == NULL || PyDict_GetItemString(dict, OPNAME)) { \
Guido van Rossumdc91b992001-08-08 22:26:22 +00003286 sq->SLOTNAME = FUNCNAME; \
Tim Peters6d6c1a32001-08-02 04:15:00 +00003287 }
3288
Guido van Rossumdc91b992001-08-08 22:26:22 +00003289#define MPSLOT(OPNAME, SLOTNAME, FUNCNAME) \
Guido van Rossum8e248182001-08-12 05:17:56 +00003290 if (dict == NULL || PyDict_GetItemString(dict, OPNAME)) { \
Guido van Rossumdc91b992001-08-08 22:26:22 +00003291 mp->SLOTNAME = FUNCNAME; \
Tim Peters6d6c1a32001-08-02 04:15:00 +00003292 }
3293
Guido van Rossumdc91b992001-08-08 22:26:22 +00003294#define NBSLOT(OPNAME, SLOTNAME, FUNCNAME) \
Guido van Rossum8e248182001-08-12 05:17:56 +00003295 if (dict == NULL || PyDict_GetItemString(dict, OPNAME)) { \
Guido van Rossumdc91b992001-08-08 22:26:22 +00003296 nb->SLOTNAME = FUNCNAME; \
Tim Peters6d6c1a32001-08-02 04:15:00 +00003297 }
3298
Guido van Rossumdc91b992001-08-08 22:26:22 +00003299#define TPSLOT(OPNAME, SLOTNAME, FUNCNAME) \
Guido van Rossum8e248182001-08-12 05:17:56 +00003300 if (dict == NULL || PyDict_GetItemString(dict, OPNAME)) { \
Guido van Rossumdc91b992001-08-08 22:26:22 +00003301 type->SLOTNAME = FUNCNAME; \
Tim Peters6d6c1a32001-08-02 04:15:00 +00003302 }
3303
Guido van Rossumdc91b992001-08-08 22:26:22 +00003304 SQSLOT("__len__", sq_length, slot_sq_length);
3305 SQSLOT("__add__", sq_concat, slot_sq_concat);
3306 SQSLOT("__mul__", sq_repeat, slot_sq_repeat);
3307 SQSLOT("__getitem__", sq_item, slot_sq_item);
3308 SQSLOT("__getslice__", sq_slice, slot_sq_slice);
3309 SQSLOT("__setitem__", sq_ass_item, slot_sq_ass_item);
3310 SQSLOT("__delitem__", sq_ass_item, slot_sq_ass_item);
3311 SQSLOT("__setslice__", sq_ass_slice, slot_sq_ass_slice);
3312 SQSLOT("__delslice__", sq_ass_slice, slot_sq_ass_slice);
3313 SQSLOT("__contains__", sq_contains, slot_sq_contains);
3314 SQSLOT("__iadd__", sq_inplace_concat, slot_sq_inplace_concat);
3315 SQSLOT("__imul__", sq_inplace_repeat, slot_sq_inplace_repeat);
Tim Peters6d6c1a32001-08-02 04:15:00 +00003316
Guido van Rossumdc91b992001-08-08 22:26:22 +00003317 MPSLOT("__len__", mp_length, slot_mp_length);
3318 MPSLOT("__getitem__", mp_subscript, slot_mp_subscript);
3319 MPSLOT("__setitem__", mp_ass_subscript, slot_mp_ass_subscript);
3320 MPSLOT("__delitem__", mp_ass_subscript, slot_mp_ass_subscript);
Tim Peters6d6c1a32001-08-02 04:15:00 +00003321
Guido van Rossumdc91b992001-08-08 22:26:22 +00003322 NBSLOT("__add__", nb_add, slot_nb_add);
3323 NBSLOT("__sub__", nb_subtract, slot_nb_subtract);
3324 NBSLOT("__mul__", nb_multiply, slot_nb_multiply);
3325 NBSLOT("__div__", nb_divide, slot_nb_divide);
3326 NBSLOT("__mod__", nb_remainder, slot_nb_remainder);
3327 NBSLOT("__divmod__", nb_divmod, slot_nb_divmod);
3328 NBSLOT("__pow__", nb_power, slot_nb_power);
3329 NBSLOT("__neg__", nb_negative, slot_nb_negative);
3330 NBSLOT("__pos__", nb_positive, slot_nb_positive);
3331 NBSLOT("__abs__", nb_absolute, slot_nb_absolute);
3332 NBSLOT("__nonzero__", nb_nonzero, slot_nb_nonzero);
3333 NBSLOT("__invert__", nb_invert, slot_nb_invert);
3334 NBSLOT("__lshift__", nb_lshift, slot_nb_lshift);
3335 NBSLOT("__rshift__", nb_rshift, slot_nb_rshift);
3336 NBSLOT("__and__", nb_and, slot_nb_and);
3337 NBSLOT("__xor__", nb_xor, slot_nb_xor);
3338 NBSLOT("__or__", nb_or, slot_nb_or);
Tim Peters6d6c1a32001-08-02 04:15:00 +00003339 /* Not coerce() */
Guido van Rossumdc91b992001-08-08 22:26:22 +00003340 NBSLOT("__int__", nb_int, slot_nb_int);
3341 NBSLOT("__long__", nb_long, slot_nb_long);
3342 NBSLOT("__float__", nb_float, slot_nb_float);
3343 NBSLOT("__oct__", nb_oct, slot_nb_oct);
3344 NBSLOT("__hex__", nb_hex, slot_nb_hex);
3345 NBSLOT("__iadd__", nb_inplace_add, slot_nb_inplace_add);
3346 NBSLOT("__isub__", nb_inplace_subtract, slot_nb_inplace_subtract);
3347 NBSLOT("__imul__", nb_inplace_multiply, slot_nb_inplace_multiply);
3348 NBSLOT("__idiv__", nb_inplace_divide, slot_nb_inplace_divide);
3349 NBSLOT("__imod__", nb_inplace_remainder, slot_nb_inplace_remainder);
3350 NBSLOT("__ipow__", nb_inplace_power, slot_nb_inplace_power);
3351 NBSLOT("__ilshift__", nb_inplace_lshift, slot_nb_inplace_lshift);
3352 NBSLOT("__irshift__", nb_inplace_rshift, slot_nb_inplace_rshift);
3353 NBSLOT("__iand__", nb_inplace_and, slot_nb_inplace_and);
3354 NBSLOT("__ixor__", nb_inplace_xor, slot_nb_inplace_xor);
3355 NBSLOT("__ior__", nb_inplace_or, slot_nb_inplace_or);
3356 NBSLOT("__floordiv__", nb_floor_divide, slot_nb_floor_divide);
3357 NBSLOT("__truediv__", nb_true_divide, slot_nb_true_divide);
3358 NBSLOT("__ifloordiv__", nb_inplace_floor_divide,
3359 slot_nb_inplace_floor_divide);
3360 NBSLOT("__itruediv__", nb_inplace_true_divide,
3361 slot_nb_inplace_true_divide);
Tim Peters6d6c1a32001-08-02 04:15:00 +00003362
Guido van Rossum8e248182001-08-12 05:17:56 +00003363 if (dict == NULL ||
3364 PyDict_GetItemString(dict, "__str__") ||
Tim Peters6d6c1a32001-08-02 04:15:00 +00003365 PyDict_GetItemString(dict, "__repr__"))
3366 type->tp_print = NULL;
3367
Guido van Rossumab3b0342001-09-18 20:38:53 +00003368 TPSLOT("__cmp__", tp_compare, _PyObject_SlotCompare);
Guido van Rossumdc91b992001-08-08 22:26:22 +00003369 TPSLOT("__repr__", tp_repr, slot_tp_repr);
3370 TPSLOT("__hash__", tp_hash, slot_tp_hash);
3371 TPSLOT("__call__", tp_call, slot_tp_call);
3372 TPSLOT("__str__", tp_str, slot_tp_str);
Guido van Rossum867a8d22001-09-21 19:29:08 +00003373 TPSLOT("__getattribute__", tp_getattro, slot_tp_getattro);
Guido van Rossum19c1cd52001-09-21 21:24:49 +00003374 TPSLOT("__getattr__", tp_getattro, slot_tp_getattr_hook);
Guido van Rossumdc91b992001-08-08 22:26:22 +00003375 TPSLOT("__setattr__", tp_setattro, slot_tp_setattro);
3376 TPSLOT("__lt__", tp_richcompare, slot_tp_richcompare);
3377 TPSLOT("__le__", tp_richcompare, slot_tp_richcompare);
3378 TPSLOT("__eq__", tp_richcompare, slot_tp_richcompare);
3379 TPSLOT("__ne__", tp_richcompare, slot_tp_richcompare);
3380 TPSLOT("__gt__", tp_richcompare, slot_tp_richcompare);
3381 TPSLOT("__ge__", tp_richcompare, slot_tp_richcompare);
3382 TPSLOT("__iter__", tp_iter, slot_tp_iter);
3383 TPSLOT("next", tp_iternext, slot_tp_iternext);
3384 TPSLOT("__get__", tp_descr_get, slot_tp_descr_get);
3385 TPSLOT("__set__", tp_descr_set, slot_tp_descr_set);
3386 TPSLOT("__init__", tp_init, slot_tp_init);
3387 TPSLOT("__new__", tp_new, slot_tp_new);
Tim Peters6d6c1a32001-08-02 04:15:00 +00003388}
Guido van Rossum705f0f52001-08-24 16:47:00 +00003389
3390
3391/* Cooperative 'super' */
3392
3393typedef struct {
3394 PyObject_HEAD
Guido van Rossume705ef12001-08-29 15:47:06 +00003395 PyTypeObject *type;
Guido van Rossum705f0f52001-08-24 16:47:00 +00003396 PyObject *obj;
3397} superobject;
3398
Guido van Rossum6f799372001-09-20 20:46:19 +00003399static PyMemberDef super_members[] = {
3400 {"__thisclass__", T_OBJECT, offsetof(superobject, type), READONLY,
3401 "the class invoking super()"},
3402 {"__self__", T_OBJECT, offsetof(superobject, obj), READONLY,
3403 "the instance invoking super(); may be None"},
Guido van Rossum41eb14d2001-08-30 23:13:11 +00003404 {0}
3405};
3406
Guido van Rossum705f0f52001-08-24 16:47:00 +00003407static void
3408super_dealloc(PyObject *self)
3409{
3410 superobject *su = (superobject *)self;
3411
3412 Py_XDECREF(su->obj);
3413 Py_XDECREF(su->type);
3414 self->ob_type->tp_free(self);
3415}
3416
3417static PyObject *
Guido van Rossum41eb14d2001-08-30 23:13:11 +00003418super_repr(PyObject *self)
3419{
3420 superobject *su = (superobject *)self;
3421
3422 if (su->obj)
3423 return PyString_FromFormat(
Guido van Rossuma4cb7882001-09-25 03:56:29 +00003424 "<super: <class '%s'>, <%s object>>",
Guido van Rossum41eb14d2001-08-30 23:13:11 +00003425 su->type ? su->type->tp_name : "NULL",
3426 su->obj->ob_type->tp_name);
3427 else
3428 return PyString_FromFormat(
Guido van Rossuma4cb7882001-09-25 03:56:29 +00003429 "<super: <class '%s'>, NULL>",
Guido van Rossum41eb14d2001-08-30 23:13:11 +00003430 su->type ? su->type->tp_name : "NULL");
3431}
3432
3433static PyObject *
Guido van Rossum705f0f52001-08-24 16:47:00 +00003434super_getattro(PyObject *self, PyObject *name)
3435{
3436 superobject *su = (superobject *)self;
3437
3438 if (su->obj != NULL) {
3439 PyObject *mro, *res, *tmp;
3440 descrgetfunc f;
3441 int i, n;
3442
Guido van Rossume705ef12001-08-29 15:47:06 +00003443 mro = su->obj->ob_type->tp_mro;
Guido van Rossum41eb14d2001-08-30 23:13:11 +00003444 if (mro == NULL)
3445 n = 0;
3446 else {
3447 assert(PyTuple_Check(mro));
3448 n = PyTuple_GET_SIZE(mro);
3449 }
Guido van Rossum705f0f52001-08-24 16:47:00 +00003450 for (i = 0; i < n; i++) {
Guido van Rossume705ef12001-08-29 15:47:06 +00003451 if ((PyObject *)(su->type) == PyTuple_GET_ITEM(mro, i))
Guido van Rossum705f0f52001-08-24 16:47:00 +00003452 break;
3453 }
Guido van Rossume705ef12001-08-29 15:47:06 +00003454 if (i >= n && PyType_Check(su->obj)) {
3455 mro = ((PyTypeObject *)(su->obj))->tp_mro;
Guido van Rossum41eb14d2001-08-30 23:13:11 +00003456 if (mro == NULL)
3457 n = 0;
3458 else {
3459 assert(PyTuple_Check(mro));
3460 n = PyTuple_GET_SIZE(mro);
3461 }
Guido van Rossume705ef12001-08-29 15:47:06 +00003462 for (i = 0; i < n; i++) {
3463 if ((PyObject *)(su->type) ==
3464 PyTuple_GET_ITEM(mro, i))
3465 break;
3466 }
Guido van Rossume705ef12001-08-29 15:47:06 +00003467 }
Guido van Rossum705f0f52001-08-24 16:47:00 +00003468 i++;
3469 res = NULL;
3470 for (; i < n; i++) {
3471 tmp = PyTuple_GET_ITEM(mro, i);
3472 assert(PyType_Check(tmp));
3473 res = PyDict_GetItem(
3474 ((PyTypeObject *)tmp)->tp_defined, name);
3475 if (res != NULL) {
3476 Py_INCREF(res);
3477 f = res->ob_type->tp_descr_get;
3478 if (f != NULL) {
3479 tmp = f(res, su->obj, res);
3480 Py_DECREF(res);
3481 res = tmp;
3482 }
3483 return res;
3484 }
3485 }
3486 }
3487 return PyObject_GenericGetAttr(self, name);
3488}
3489
3490static PyObject *
3491super_descr_get(PyObject *self, PyObject *obj, PyObject *type)
3492{
3493 superobject *su = (superobject *)self;
3494 superobject *new;
3495
3496 if (obj == NULL || obj == Py_None || su->obj != NULL) {
3497 /* Not binding to an object, or already bound */
3498 Py_INCREF(self);
3499 return self;
3500 }
3501 new = (superobject *)PySuper_Type.tp_new(&PySuper_Type, NULL, NULL);
3502 if (new == NULL)
3503 return NULL;
3504 Py_INCREF(su->type);
3505 Py_INCREF(obj);
3506 new->type = su->type;
3507 new->obj = obj;
3508 return (PyObject *)new;
3509}
3510
3511static int
3512super_init(PyObject *self, PyObject *args, PyObject *kwds)
3513{
3514 superobject *su = (superobject *)self;
Guido van Rossume705ef12001-08-29 15:47:06 +00003515 PyTypeObject *type;
3516 PyObject *obj = NULL;
Guido van Rossum705f0f52001-08-24 16:47:00 +00003517
3518 if (!PyArg_ParseTuple(args, "O!|O:super", &PyType_Type, &type, &obj))
3519 return -1;
3520 if (obj == Py_None)
3521 obj = NULL;
Guido van Rossume705ef12001-08-29 15:47:06 +00003522 if (obj != NULL &&
3523 !PyType_IsSubtype(obj->ob_type, type) &&
3524 !(PyType_Check(obj) &&
3525 PyType_IsSubtype((PyTypeObject *)obj, type))) {
Guido van Rossum705f0f52001-08-24 16:47:00 +00003526 PyErr_SetString(PyExc_TypeError,
Guido van Rossume705ef12001-08-29 15:47:06 +00003527 "super(type, obj): "
3528 "obj must be an instance or subtype of type");
Guido van Rossum705f0f52001-08-24 16:47:00 +00003529 return -1;
3530 }
3531 Py_INCREF(type);
3532 Py_XINCREF(obj);
3533 su->type = type;
3534 su->obj = obj;
3535 return 0;
3536}
3537
3538static char super_doc[] =
3539"super(type) -> unbound super object\n"
3540"super(type, obj) -> bound super object; requires isinstance(obj, type)\n"
Guido van Rossume705ef12001-08-29 15:47:06 +00003541"super(type, type2) -> bound super object; requires issubclass(type2, type)\n"
Guido van Rossum705f0f52001-08-24 16:47:00 +00003542"Typical use to call a cooperative superclass method:\n"
3543"class C(B):\n"
3544" def meth(self, arg):\n"
3545" super(C, self).meth(arg)";
3546
3547PyTypeObject PySuper_Type = {
3548 PyObject_HEAD_INIT(&PyType_Type)
3549 0, /* ob_size */
3550 "super", /* tp_name */
3551 sizeof(superobject), /* tp_basicsize */
3552 0, /* tp_itemsize */
3553 /* methods */
3554 super_dealloc, /* tp_dealloc */
3555 0, /* tp_print */
3556 0, /* tp_getattr */
3557 0, /* tp_setattr */
3558 0, /* tp_compare */
Guido van Rossum41eb14d2001-08-30 23:13:11 +00003559 super_repr, /* tp_repr */
Guido van Rossum705f0f52001-08-24 16:47:00 +00003560 0, /* tp_as_number */
3561 0, /* tp_as_sequence */
3562 0, /* tp_as_mapping */
3563 0, /* tp_hash */
3564 0, /* tp_call */
3565 0, /* tp_str */
3566 super_getattro, /* tp_getattro */
3567 0, /* tp_setattro */
3568 0, /* tp_as_buffer */
Guido van Rossum31bcff82001-08-30 04:37:15 +00003569 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
Guido van Rossum705f0f52001-08-24 16:47:00 +00003570 super_doc, /* tp_doc */
3571 0, /* tp_traverse */
3572 0, /* tp_clear */
3573 0, /* tp_richcompare */
3574 0, /* tp_weaklistoffset */
3575 0, /* tp_iter */
3576 0, /* tp_iternext */
3577 0, /* tp_methods */
Guido van Rossum41eb14d2001-08-30 23:13:11 +00003578 super_members, /* tp_members */
Guido van Rossum705f0f52001-08-24 16:47:00 +00003579 0, /* tp_getset */
3580 0, /* tp_base */
3581 0, /* tp_dict */
3582 super_descr_get, /* tp_descr_get */
3583 0, /* tp_descr_set */
3584 0, /* tp_dictoffset */
3585 super_init, /* tp_init */
3586 PyType_GenericAlloc, /* tp_alloc */
3587 PyType_GenericNew, /* tp_new */
3588 _PyObject_Del, /* tp_free */
3589};