blob: 2babbfbd63fb02a504e3c264fdca3f71b6ad454d [file] [log] [blame]
Guido van Rossumf70e43a1991-02-19 12:39:46 +00001
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002/* Class object implementation */
3
Guido van Rossumc0b618a1997-05-02 03:12:38 +00004#include "Python.h"
Guido van Rossum3f5da241990-12-20 15:06:42 +00005#include "structmember.h"
Guido van Rossum04691fc1992-08-12 15:35:34 +00006
Neil Schemenauer29bfc072001-01-04 01:43:46 +00007
Guido van Rossum52ca98a1994-09-05 07:32:29 +00008/* Forward */
Tim Petersdbd9ba62000-07-09 03:09:57 +00009static PyObject *class_lookup(PyClassObject *, PyObject *,
10 PyClassObject **);
11static PyObject *instance_getattr1(PyInstanceObject *, PyObject *);
12static PyObject *instance_getattr2(PyInstanceObject *, PyObject *);
Guido van Rossum52ca98a1994-09-05 07:32:29 +000013
Guido van Rossuma63eff61998-05-29 21:37:21 +000014static PyObject *getattrstr, *setattrstr, *delattrstr;
15
Fred Drake79912472000-07-09 04:06:11 +000016
Guido van Rossumc0b618a1997-05-02 03:12:38 +000017PyObject *
Fred Drake79912472000-07-09 04:06:11 +000018PyClass_New(PyObject *bases, PyObject *dict, PyObject *name)
19 /* bases is NULL or tuple of classobjects! */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000020{
Guido van Rossumc0b618a1997-05-02 03:12:38 +000021 PyClassObject *op, *dummy;
Guido van Rossum7cc56eb1997-09-12 20:04:46 +000022 static PyObject *docstr, *modstr, *namestr;
Guido van Rossum019f4241996-08-21 14:54:28 +000023 if (docstr == NULL) {
Guido van Rossuma412d241997-01-18 07:59:12 +000024 docstr= PyString_InternFromString("__doc__");
Guido van Rossum019f4241996-08-21 14:54:28 +000025 if (docstr == NULL)
26 return NULL;
27 }
Guido van Rossum7cc56eb1997-09-12 20:04:46 +000028 if (modstr == NULL) {
29 modstr= PyString_InternFromString("__module__");
30 if (modstr == NULL)
31 return NULL;
32 }
33 if (namestr == NULL) {
34 namestr= PyString_InternFromString("__name__");
35 if (namestr == NULL)
36 return NULL;
37 }
Guido van Rossum04d73c41997-10-07 14:54:11 +000038 if (name == NULL || !PyString_Check(name)) {
39 PyErr_SetString(PyExc_SystemError,
40 "PyClass_New: name must be a string");
41 return NULL;
42 }
43 if (dict == NULL || !PyDict_Check(dict)) {
44 PyErr_SetString(PyExc_SystemError,
45 "PyClass_New: dict must be a dictionary");
46 return NULL;
47 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +000048 if (PyDict_GetItem(dict, docstr) == NULL) {
49 if (PyDict_SetItem(dict, docstr, Py_None) < 0)
Guido van Rossume7d444f1995-01-07 12:35:18 +000050 return NULL;
51 }
Guido van Rossum7cc56eb1997-09-12 20:04:46 +000052 if (PyDict_GetItem(dict, modstr) == NULL) {
53 PyObject *globals = PyEval_GetGlobals();
54 if (globals != NULL) {
Guido van Rossum04d73c41997-10-07 14:54:11 +000055 PyObject *modname = PyDict_GetItem(globals, namestr);
56 if (modname != NULL) {
57 if (PyDict_SetItem(dict, modstr, modname) < 0)
Guido van Rossum7cc56eb1997-09-12 20:04:46 +000058 return NULL;
59 }
60 }
61 }
Guido van Rossume2966a61991-12-10 13:53:23 +000062 if (bases == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +000063 bases = PyTuple_New(0);
Guido van Rossume2966a61991-12-10 13:53:23 +000064 if (bases == NULL)
Guido van Rossume5372401993-03-16 12:15:04 +000065 return NULL;
Guido van Rossume2966a61991-12-10 13:53:23 +000066 }
Guido van Rossum04d73c41997-10-07 14:54:11 +000067 else {
68 int i;
69 if (!PyTuple_Check(bases)) {
70 PyErr_SetString(PyExc_SystemError,
71 "PyClass_New: bases must be a tuple");
72 return NULL;
73 }
74 i = PyTuple_Size(bases);
75 while (--i >= 0) {
76 if (!PyClass_Check(PyTuple_GetItem(bases, i))) {
77 PyErr_SetString(PyExc_SystemError,
78 "PyClass_New: base must be a class");
79 return NULL;
80 }
81 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +000082 Py_INCREF(bases);
Guido van Rossum04d73c41997-10-07 14:54:11 +000083 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +000084 op = PyObject_NEW(PyClassObject, &PyClass_Type);
Guido van Rossume2966a61991-12-10 13:53:23 +000085 if (op == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +000086 Py_DECREF(bases);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000087 return NULL;
Guido van Rossume2966a61991-12-10 13:53:23 +000088 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000089 op->cl_bases = bases;
Guido van Rossumc0b618a1997-05-02 03:12:38 +000090 Py_INCREF(dict);
Guido van Rossum81daa321993-05-20 14:24:46 +000091 op->cl_dict = dict;
Guido van Rossumc0b618a1997-05-02 03:12:38 +000092 Py_XINCREF(name);
Guido van Rossum94308391991-10-20 20:11:48 +000093 op->cl_name = name;
Guido van Rossum2878a691996-08-09 20:53:24 +000094 if (getattrstr == NULL) {
Guido van Rossuma412d241997-01-18 07:59:12 +000095 getattrstr = PyString_InternFromString("__getattr__");
96 setattrstr = PyString_InternFromString("__setattr__");
97 delattrstr = PyString_InternFromString("__delattr__");
Guido van Rossum2878a691996-08-09 20:53:24 +000098 }
99 op->cl_getattr = class_lookup(op, getattrstr, &dummy);
100 op->cl_setattr = class_lookup(op, setattrstr, &dummy);
101 op->cl_delattr = class_lookup(op, delattrstr, &dummy);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000102 Py_XINCREF(op->cl_getattr);
103 Py_XINCREF(op->cl_setattr);
104 Py_XINCREF(op->cl_delattr);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000105 PyObject_GC_Init(op);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000106 return (PyObject *) op;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000107}
108
109/* Class methods */
110
111static void
Fred Drake79912472000-07-09 04:06:11 +0000112class_dealloc(PyClassObject *op)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000113{
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000114 PyObject_GC_Fini(op);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000115 Py_DECREF(op->cl_bases);
116 Py_DECREF(op->cl_dict);
117 Py_XDECREF(op->cl_name);
Guido van Rossum152d8171998-08-04 14:59:16 +0000118 Py_XDECREF(op->cl_getattr);
119 Py_XDECREF(op->cl_setattr);
120 Py_XDECREF(op->cl_delattr);
Guido van Rossum4cc6ac72000-07-01 01:00:38 +0000121 op = (PyClassObject *) PyObject_AS_GC(op);
Guido van Rossumb18618d2000-05-03 23:44:39 +0000122 PyObject_DEL(op);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000123}
124
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000125static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000126class_lookup(PyClassObject *cp, PyObject *name, PyClassObject **pclass)
Guido van Rossum81daa321993-05-20 14:24:46 +0000127{
128 int i, n;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000129 PyObject *value = PyDict_GetItem(cp->cl_dict, name);
Guido van Rossum81daa321993-05-20 14:24:46 +0000130 if (value != NULL) {
131 *pclass = cp;
132 return value;
133 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000134 n = PyTuple_Size(cp->cl_bases);
Guido van Rossum81daa321993-05-20 14:24:46 +0000135 for (i = 0; i < n; i++) {
Guido van Rossum7cc56eb1997-09-12 20:04:46 +0000136 /* XXX What if one of the bases is not a class? */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000137 PyObject *v = class_lookup(
138 (PyClassObject *)
139 PyTuple_GetItem(cp->cl_bases, i), name, pclass);
Guido van Rossum81daa321993-05-20 14:24:46 +0000140 if (v != NULL)
141 return v;
142 }
143 return NULL;
144}
145
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000146static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000147class_getattr(register PyClassObject *op, PyObject *name)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000148{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000149 register PyObject *v;
150 register char *sname = PyString_AsString(name);
151 PyClassObject *class;
Guido van Rossum2878a691996-08-09 20:53:24 +0000152 if (sname[0] == '_' && sname[1] == '_') {
153 if (strcmp(sname, "__dict__") == 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000154 if (PyEval_GetRestricted()) {
155 PyErr_SetString(PyExc_RuntimeError,
156 "class.__dict__ not accessible in restricted mode");
Guido van Rossum10393b11995-01-10 10:39:49 +0000157 return NULL;
158 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000159 Py_INCREF(op->cl_dict);
Guido van Rossum10393b11995-01-10 10:39:49 +0000160 return op->cl_dict;
161 }
Guido van Rossum2878a691996-08-09 20:53:24 +0000162 if (strcmp(sname, "__bases__") == 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000163 Py_INCREF(op->cl_bases);
Guido van Rossum10393b11995-01-10 10:39:49 +0000164 return op->cl_bases;
165 }
Guido van Rossum2878a691996-08-09 20:53:24 +0000166 if (strcmp(sname, "__name__") == 0) {
Guido van Rossum10393b11995-01-10 10:39:49 +0000167 if (op->cl_name == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000168 v = Py_None;
Guido van Rossum10393b11995-01-10 10:39:49 +0000169 else
170 v = op->cl_name;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000171 Py_INCREF(v);
Guido van Rossum10393b11995-01-10 10:39:49 +0000172 return v;
173 }
Guido van Rossum94308391991-10-20 20:11:48 +0000174 }
Guido van Rossum81daa321993-05-20 14:24:46 +0000175 v = class_lookup(op, name, &class);
Guido van Rossumb3f72581993-05-21 19:56:10 +0000176 if (v == NULL) {
Fred Drake661ea262000-10-24 19:57:45 +0000177 PyErr_Format(PyExc_AttributeError,
178 "class %.50s has no attribute '%.400s'",
179 PyString_AS_STRING(op->cl_name), sname);
Guido van Rossumb3f72581993-05-21 19:56:10 +0000180 return NULL;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000181 }
Guido van Rossum0add15f1997-05-09 01:07:15 +0000182 Py_INCREF(v);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000183 if (PyFunction_Check(v)) {
184 PyObject *w = PyMethod_New(v, (PyObject *)NULL,
185 (PyObject *)class);
186 Py_DECREF(v);
Guido van Rossumb3f72581993-05-21 19:56:10 +0000187 v = w;
188 }
189 return v;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000190}
191
Guido van Rossuma63eff61998-05-29 21:37:21 +0000192static void
Fred Drake79912472000-07-09 04:06:11 +0000193set_slot(PyObject **slot, PyObject *v)
Guido van Rossuma63eff61998-05-29 21:37:21 +0000194{
195 PyObject *temp = *slot;
196 Py_XINCREF(v);
197 *slot = v;
198 Py_XDECREF(temp);
199}
200
Guido van Rossum7ba30431998-07-08 13:34:48 +0000201static void
Fred Drake79912472000-07-09 04:06:11 +0000202set_attr_slots(PyClassObject *c)
Guido van Rossum7ba30431998-07-08 13:34:48 +0000203{
204 PyClassObject *dummy;
205
206 set_slot(&c->cl_getattr, class_lookup(c, getattrstr, &dummy));
207 set_slot(&c->cl_setattr, class_lookup(c, setattrstr, &dummy));
208 set_slot(&c->cl_delattr, class_lookup(c, delattrstr, &dummy));
209}
210
Guido van Rossuma63eff61998-05-29 21:37:21 +0000211static char *
Fred Drake79912472000-07-09 04:06:11 +0000212set_dict(PyClassObject *c, PyObject *v)
Guido van Rossuma63eff61998-05-29 21:37:21 +0000213{
Guido van Rossuma63eff61998-05-29 21:37:21 +0000214 if (v == NULL || !PyDict_Check(v))
215 return "__dict__ must be a dictionary object";
216 set_slot(&c->cl_dict, v);
Guido van Rossum7ba30431998-07-08 13:34:48 +0000217 set_attr_slots(c);
Guido van Rossuma63eff61998-05-29 21:37:21 +0000218 return "";
219}
220
221static char *
Fred Drake79912472000-07-09 04:06:11 +0000222set_bases(PyClassObject *c, PyObject *v)
Guido van Rossuma63eff61998-05-29 21:37:21 +0000223{
Guido van Rossuma63eff61998-05-29 21:37:21 +0000224 int i, n;
225
226 if (v == NULL || !PyTuple_Check(v))
227 return "__bases__ must be a tuple object";
228 n = PyTuple_Size(v);
229 for (i = 0; i < n; i++) {
230 PyObject *x = PyTuple_GET_ITEM(v, i);
231 if (!PyClass_Check(x))
232 return "__bases__ items must be classes";
233 if (PyClass_IsSubclass(x, (PyObject *)c))
234 return "a __bases__ item causes an inheritance cycle";
235 }
236 set_slot(&c->cl_bases, v);
Guido van Rossum7ba30431998-07-08 13:34:48 +0000237 set_attr_slots(c);
Guido van Rossuma63eff61998-05-29 21:37:21 +0000238 return "";
239}
240
241static char *
Fred Drake79912472000-07-09 04:06:11 +0000242set_name(PyClassObject *c, PyObject *v)
Guido van Rossuma63eff61998-05-29 21:37:21 +0000243{
Guido van Rossuma63eff61998-05-29 21:37:21 +0000244 if (v == NULL || !PyString_Check(v))
245 return "__name__ must be a string object";
Guido van Rossumad89bbc2000-06-28 21:57:18 +0000246 if (strlen(PyString_AS_STRING(v)) != (size_t)PyString_GET_SIZE(v))
Guido van Rossuma63eff61998-05-29 21:37:21 +0000247 return "__name__ must not contain null bytes";
248 set_slot(&c->cl_name, v);
249 return "";
250}
251
Guido van Rossum94308391991-10-20 20:11:48 +0000252static int
Fred Drake79912472000-07-09 04:06:11 +0000253class_setattr(PyClassObject *op, PyObject *name, PyObject *v)
Guido van Rossum94308391991-10-20 20:11:48 +0000254{
Guido van Rossumb2173c31997-08-25 21:23:56 +0000255 char *sname;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000256 if (PyEval_GetRestricted()) {
257 PyErr_SetString(PyExc_RuntimeError,
Guido van Rossumac21f6a1995-08-04 04:05:31 +0000258 "classes are read-only in restricted mode");
259 return -1;
260 }
Guido van Rossumb2173c31997-08-25 21:23:56 +0000261 sname = PyString_AsString(name);
262 if (sname[0] == '_' && sname[1] == '_') {
263 int n = PyString_Size(name);
264 if (sname[n-1] == '_' && sname[n-2] == '_') {
Guido van Rossuma63eff61998-05-29 21:37:21 +0000265 char *err = NULL;
266 if (strcmp(sname, "__dict__") == 0)
267 err = set_dict(op, v);
268 else if (strcmp(sname, "__bases__") == 0)
269 err = set_bases(op, v);
270 else if (strcmp(sname, "__name__") == 0)
271 err = set_name(op, v);
272 else if (strcmp(sname, "__getattr__") == 0)
273 set_slot(&op->cl_getattr, v);
274 else if (strcmp(sname, "__setattr__") == 0)
275 set_slot(&op->cl_setattr, v);
276 else if (strcmp(sname, "__delattr__") == 0)
277 set_slot(&op->cl_delattr, v);
278 /* For the last three, we fall through to update the
279 dictionary as well. */
280 if (err != NULL) {
281 if (*err == '\0')
282 return 0;
283 PyErr_SetString(PyExc_TypeError, err);
Guido van Rossumb2173c31997-08-25 21:23:56 +0000284 return -1;
285 }
286 }
287 }
Guido van Rossum94472a01992-09-04 09:45:18 +0000288 if (v == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000289 int rv = PyDict_DelItem(op->cl_dict, name);
Guido van Rossum94472a01992-09-04 09:45:18 +0000290 if (rv < 0)
Fred Drake661ea262000-10-24 19:57:45 +0000291 PyErr_Format(PyExc_AttributeError,
292 "class %.50s has no attribute '%.400s'",
293 PyString_AS_STRING(op->cl_name), sname);
Guido van Rossum94472a01992-09-04 09:45:18 +0000294 return rv;
295 }
Guido van Rossum94308391991-10-20 20:11:48 +0000296 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000297 return PyDict_SetItem(op->cl_dict, name, v);
Guido van Rossum94308391991-10-20 20:11:48 +0000298}
299
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000300static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000301class_repr(PyClassObject *op)
Guido van Rossum25831651993-05-19 14:50:45 +0000302{
Guido van Rossum4a2a6211997-10-20 23:26:11 +0000303 PyObject *mod = PyDict_GetItemString(op->cl_dict, "__module__");
Guido van Rossum25831651993-05-19 14:50:45 +0000304 char buf[140];
305 char *name;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000306 if (op->cl_name == NULL || !PyString_Check(op->cl_name))
Guido van Rossum25831651993-05-19 14:50:45 +0000307 name = "?";
308 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000309 name = PyString_AsString(op->cl_name);
Guido van Rossum4a2a6211997-10-20 23:26:11 +0000310 if (mod == NULL || !PyString_Check(mod))
Fred Drakea44d3532000-06-30 15:01:00 +0000311 sprintf(buf, "<class ?.%.100s at %p>", name, op);
Guido van Rossum4a2a6211997-10-20 23:26:11 +0000312 else
Fred Drakea44d3532000-06-30 15:01:00 +0000313 sprintf(buf, "<class %.50s.%.50s at %p>",
Guido van Rossum4a2a6211997-10-20 23:26:11 +0000314 PyString_AsString(mod),
Fred Drakea44d3532000-06-30 15:01:00 +0000315 name, op);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000316 return PyString_FromString(buf);
Guido van Rossum25831651993-05-19 14:50:45 +0000317}
318
Guido van Rossum4a2a6211997-10-20 23:26:11 +0000319static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000320class_str(PyClassObject *op)
Guido van Rossum4a2a6211997-10-20 23:26:11 +0000321{
322 PyObject *mod = PyDict_GetItemString(op->cl_dict, "__module__");
323 PyObject *name = op->cl_name;
324 PyObject *res;
325 int m, n;
326
327 if (name == NULL || !PyString_Check(name))
328 return class_repr(op);
329 if (mod == NULL || !PyString_Check(mod)) {
330 Py_INCREF(name);
331 return name;
332 }
333 m = PyString_Size(mod);
334 n = PyString_Size(name);
335 res = PyString_FromStringAndSize((char *)NULL, m+1+n);
336 if (res != NULL) {
337 char *s = PyString_AsString(res);
338 memcpy(s, PyString_AsString(mod), m);
339 s += m;
340 *s++ = '.';
341 memcpy(s, PyString_AsString(name), n);
342 }
343 return res;
344}
345
Jeremy Hylton8caad492000-06-23 14:18:11 +0000346static int
347class_traverse(PyClassObject *o, visitproc visit, void *arg)
348{
349 int err;
350 if (o->cl_bases) {
351 err = visit(o->cl_bases, arg);
352 if (err)
353 return err;
354 }
355 if (o->cl_dict) {
356 err = visit(o->cl_dict, arg);
357 if (err)
358 return err;
359 }
360 if (o->cl_name) {
361 err = visit(o->cl_name, arg);
362 if (err)
363 return err;
364 }
365 if (o->cl_getattr) {
366 err = visit(o->cl_getattr, arg);
367 if (err)
368 return err;
369 }
370 if (o->cl_setattr) {
371 err = visit(o->cl_setattr, arg);
372 if (err)
373 return err;
374 }
375 if (o->cl_delattr) {
376 err = visit(o->cl_delattr, arg);
377 if (err)
378 return err;
379 }
380 return 0;
381}
382
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000383PyTypeObject PyClass_Type = {
384 PyObject_HEAD_INIT(&PyType_Type)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000385 0,
386 "class",
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000387 sizeof(PyClassObject) + PyGC_HEAD_SIZE,
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000388 0,
Guido van Rossum8998b4f2001-01-17 15:28:20 +0000389 (destructor)class_dealloc, /* tp_dealloc */
390 0, /* tp_print */
391 0, /* tp_getattr */
392 0, /* tp_setattr */
393 0, /* tp_compare */
394 (reprfunc)class_repr, /* tp_repr */
395 0, /* tp_as_number */
396 0, /* tp_as_sequence */
397 0, /* tp_as_mapping */
398 0, /* tp_hash */
399 0, /* tp_call */
400 (reprfunc)class_str, /* tp_str */
401 (getattrofunc)class_getattr, /* tp_getattro */
402 (setattrofunc)class_setattr, /* tp_setattro */
403 0, /* tp_as_buffer */
404 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_GC, /* tp_flags */
405 0, /* tp_doc */
406 (traverseproc)class_traverse, /* tp_traverse */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000407};
408
Guido van Rossum81daa321993-05-20 14:24:46 +0000409int
Fred Drake79912472000-07-09 04:06:11 +0000410PyClass_IsSubclass(PyObject *class, PyObject *base)
Guido van Rossum81daa321993-05-20 14:24:46 +0000411{
412 int i, n;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000413 PyClassObject *cp;
Guido van Rossum81daa321993-05-20 14:24:46 +0000414 if (class == base)
415 return 1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000416 if (class == NULL || !PyClass_Check(class))
Guido van Rossumeb6b33a1993-05-25 09:38:27 +0000417 return 0;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000418 cp = (PyClassObject *)class;
419 n = PyTuple_Size(cp->cl_bases);
Guido van Rossum81daa321993-05-20 14:24:46 +0000420 for (i = 0; i < n; i++) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000421 if (PyClass_IsSubclass(PyTuple_GetItem(cp->cl_bases, i), base))
Guido van Rossum81daa321993-05-20 14:24:46 +0000422 return 1;
423 }
424 return 0;
425}
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000426
Guido van Rossum81daa321993-05-20 14:24:46 +0000427
428/* Instance objects */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000429
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000430PyObject *
Fred Drake5cc2c8c2001-01-28 03:53:08 +0000431PyInstance_NewRaw(PyObject *klass, PyObject *dict)
432{
433 PyInstanceObject *inst;
434
435 if (!PyClass_Check(klass)) {
436 PyErr_BadInternalCall();
437 return NULL;
438 }
439 if (dict == NULL) {
440 dict = PyDict_New();
441 if (dict == NULL)
442 return NULL;
443 }
444 else {
445 if (!PyDict_Check(dict)) {
446 PyErr_BadInternalCall();
447 return NULL;
448 }
449 Py_INCREF(dict);
450 }
451 inst = PyObject_NEW(PyInstanceObject, &PyInstance_Type);
452 if (inst == NULL) {
453 Py_DECREF(dict);
454 return NULL;
455 }
Fred Drake4e262a92001-03-22 18:26:47 +0000456 inst->in_weakreflist = NULL;
Fred Drake5cc2c8c2001-01-28 03:53:08 +0000457 Py_INCREF(klass);
458 inst->in_class = (PyClassObject *)klass;
459 inst->in_dict = dict;
460 PyObject_GC_Init(inst);
461 return (PyObject *)inst;
462}
463
464PyObject *
465PyInstance_New(PyObject *klass, PyObject *arg, PyObject *kw)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000466{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000467 register PyInstanceObject *inst;
468 PyObject *init;
469 static PyObject *initstr;
Fred Drake5cc2c8c2001-01-28 03:53:08 +0000470
471 inst = (PyInstanceObject *) PyInstance_NewRaw(klass, NULL);
Guido van Rossume8122f11991-05-05 20:03:07 +0000472 if (inst == NULL)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000473 return NULL;
Guido van Rossum2878a691996-08-09 20:53:24 +0000474 if (initstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000475 initstr = PyString_InternFromString("__init__");
Jeremy Hylton9e392e22000-04-26 20:39:20 +0000476 init = instance_getattr2(inst, initstr);
Guido van Rossum25831651993-05-19 14:50:45 +0000477 if (init == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000478 if ((arg != NULL && (!PyTuple_Check(arg) ||
479 PyTuple_Size(arg) != 0))
480 || (kw != NULL && (!PyDict_Check(kw) ||
481 PyDict_Size(kw) != 0))) {
482 PyErr_SetString(PyExc_TypeError,
Guido van Rossuma83f2701995-07-26 18:07:32 +0000483 "this constructor takes no arguments");
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000484 Py_DECREF(inst);
Guido van Rossum25831651993-05-19 14:50:45 +0000485 inst = NULL;
486 }
487 }
488 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000489 PyObject *res = PyEval_CallObjectWithKeywords(init, arg, kw);
490 Py_DECREF(init);
Guido van Rossum25831651993-05-19 14:50:45 +0000491 if (res == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000492 Py_DECREF(inst);
Guido van Rossum25831651993-05-19 14:50:45 +0000493 inst = NULL;
494 }
495 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000496 if (res != Py_None) {
497 PyErr_SetString(PyExc_TypeError,
Guido van Rossum25831651993-05-19 14:50:45 +0000498 "__init__() should return None");
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000499 Py_DECREF(inst);
Guido van Rossum25831651993-05-19 14:50:45 +0000500 inst = NULL;
501 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000502 Py_DECREF(res);
Guido van Rossum25831651993-05-19 14:50:45 +0000503 }
504 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000505 return (PyObject *)inst;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000506}
507
Guido van Rossum21ed88c1991-04-04 10:42:10 +0000508/* Instance methods */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000509
510static void
Fred Drake79912472000-07-09 04:06:11 +0000511instance_dealloc(register PyInstanceObject *inst)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000512{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000513 PyObject *error_type, *error_value, *error_traceback;
514 PyObject *del;
515 static PyObject *delstr;
Tim Peters6b184912000-09-17 14:40:17 +0000516#ifdef Py_REF_DEBUG
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000517 extern long _Py_RefTotal;
Skip Montanaro4ca150b2000-07-08 12:04:57 +0000518#endif
Fred Drake41deb1e2001-02-01 05:27:45 +0000519
Fred Drakeb60654b2001-02-26 18:56:37 +0000520 PyObject_ClearWeakRefs((PyObject *) inst);
Fred Drake41deb1e2001-02-01 05:27:45 +0000521
Tim Peters6b184912000-09-17 14:40:17 +0000522 /* Temporarily resurrect the object. */
Guido van Rossum441e4ab1996-05-23 22:46:51 +0000523#ifdef Py_TRACE_REFS
Tim Peters6b184912000-09-17 14:40:17 +0000524#ifndef Py_REF_DEBUG
525# error "Py_TRACE_REFS defined but Py_REF_DEBUG not."
526#endif
Guido van Rossum441e4ab1996-05-23 22:46:51 +0000527 /* much too complicated if Py_TRACE_REFS defined */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000528 inst->ob_type = &PyInstance_Type;
Guido van Rossumbffd6832000-01-20 22:32:56 +0000529 _Py_NewReference((PyObject *)inst);
Sjoerd Mullender740f3571995-08-28 09:00:43 +0000530#ifdef COUNT_ALLOCS
Tim Peters6b184912000-09-17 14:40:17 +0000531 /* compensate for boost in _Py_NewReference; note that
532 * _Py_RefTotal was also boosted; we'll knock that down later.
533 */
534 inst->ob_type->tp_alloc--;
Sjoerd Mullender740f3571995-08-28 09:00:43 +0000535#endif
Guido van Rossum441e4ab1996-05-23 22:46:51 +0000536#else /* !Py_TRACE_REFS */
Tim Peters6b184912000-09-17 14:40:17 +0000537 /* Py_INCREF boosts _Py_RefTotal if Py_REF_DEBUG is defined */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000538 Py_INCREF(inst);
Guido van Rossum441e4ab1996-05-23 22:46:51 +0000539#endif /* !Py_TRACE_REFS */
Tim Peters6b184912000-09-17 14:40:17 +0000540
541 /* Save the current exception, if any. */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000542 PyErr_Fetch(&error_type, &error_value, &error_traceback);
Tim Peters6b184912000-09-17 14:40:17 +0000543 /* Execute __del__ method, if any. */
Guido van Rossum2878a691996-08-09 20:53:24 +0000544 if (delstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000545 delstr = PyString_InternFromString("__del__");
Jeremy Hylton9e392e22000-04-26 20:39:20 +0000546 if ((del = instance_getattr2(inst, delstr)) != NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000547 PyObject *res = PyEval_CallObject(del, (PyObject *)NULL);
Tim Peters6b184912000-09-17 14:40:17 +0000548 if (res == NULL)
Jeremy Hyltonb709df32000-09-01 02:47:25 +0000549 PyErr_WriteUnraisable(del);
Guido van Rossum0dfcf751996-08-12 22:00:53 +0000550 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000551 Py_DECREF(res);
552 Py_DECREF(del);
Guido van Rossum25831651993-05-19 14:50:45 +0000553 }
Tim Peters6b184912000-09-17 14:40:17 +0000554 /* Restore the saved exception. */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000555 PyErr_Restore(error_type, error_value, error_traceback);
Tim Peters6b184912000-09-17 14:40:17 +0000556 /* Undo the temporary resurrection; can't use DECREF here, it would
557 * cause a recursive call.
558 */
559#ifdef Py_REF_DEBUG
560 /* _Py_RefTotal was boosted either by _Py_NewReference or
561 * Py_INCREF above.
562 */
563 _Py_RefTotal--;
564#endif
Sjoerd Mullender740f3571995-08-28 09:00:43 +0000565 if (--inst->ob_refcnt > 0) {
566#ifdef COUNT_ALLOCS
567 inst->ob_type->tp_free--;
568#endif
Guido van Rossum25831651993-05-19 14:50:45 +0000569 return; /* __del__ added a reference; don't delete now */
Sjoerd Mullender740f3571995-08-28 09:00:43 +0000570 }
Guido van Rossum441e4ab1996-05-23 22:46:51 +0000571#ifdef Py_TRACE_REFS
Guido van Rossumbffd6832000-01-20 22:32:56 +0000572 _Py_ForgetReference((PyObject *)inst);
Tim Peters6b184912000-09-17 14:40:17 +0000573#ifdef COUNT_ALLOCS
574 /* compensate for increment in _Py_ForgetReference */
575 inst->ob_type->tp_free--;
576#endif
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000577#ifndef WITH_CYCLE_GC
Sjoerd Mullender740f3571995-08-28 09:00:43 +0000578 inst->ob_type = NULL;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000579#endif
Tim Peters6b184912000-09-17 14:40:17 +0000580#endif
Neil Schemenauerce209672000-09-15 18:57:21 +0000581 PyObject_GC_Fini(inst);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000582 Py_DECREF(inst->in_class);
583 Py_XDECREF(inst->in_dict);
Guido van Rossum4cc6ac72000-07-01 01:00:38 +0000584 inst = (PyInstanceObject *) PyObject_AS_GC(inst);
Guido van Rossumb18618d2000-05-03 23:44:39 +0000585 PyObject_DEL(inst);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000586}
587
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000588static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000589instance_getattr1(register PyInstanceObject *inst, PyObject *name)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000590{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000591 register PyObject *v;
592 register char *sname = PyString_AsString(name);
Guido van Rossum2878a691996-08-09 20:53:24 +0000593 if (sname[0] == '_' && sname[1] == '_') {
594 if (strcmp(sname, "__dict__") == 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000595 if (PyEval_GetRestricted()) {
596 PyErr_SetString(PyExc_RuntimeError,
597 "instance.__dict__ not accessible in restricted mode");
Guido van Rossum10393b11995-01-10 10:39:49 +0000598 return NULL;
599 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000600 Py_INCREF(inst->in_dict);
Sjoerd Mullender3bb8a051993-10-22 12:04:32 +0000601 return inst->in_dict;
602 }
Guido van Rossum2878a691996-08-09 20:53:24 +0000603 if (strcmp(sname, "__class__") == 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000604 Py_INCREF(inst->in_class);
605 return (PyObject *)inst->in_class;
Sjoerd Mullender3bb8a051993-10-22 12:04:32 +0000606 }
Guido van Rossum94308391991-10-20 20:11:48 +0000607 }
Jeremy Hylton9e392e22000-04-26 20:39:20 +0000608 v = instance_getattr2(inst, name);
609 if (v == NULL) {
Fred Drake661ea262000-10-24 19:57:45 +0000610 PyErr_Format(PyExc_AttributeError,
611 "%.50s instance has no attribute '%.400s'",
Jeremy Hylton9e392e22000-04-26 20:39:20 +0000612 PyString_AS_STRING(inst->in_class->cl_name), sname);
613 }
614 return v;
615}
616
617static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000618instance_getattr2(register PyInstanceObject *inst, PyObject *name)
Jeremy Hylton9e392e22000-04-26 20:39:20 +0000619{
620 register PyObject *v;
621 PyClassObject *class;
Guido van Rossumeb6b33a1993-05-25 09:38:27 +0000622 class = NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000623 v = PyDict_GetItem(inst->in_dict, name);
Guido van Rossumb3f72581993-05-21 19:56:10 +0000624 if (v == NULL) {
625 v = class_lookup(inst->in_class, name, &class);
Jeremy Hylton9e392e22000-04-26 20:39:20 +0000626 if (v == NULL)
627 return v;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000628 }
Guido van Rossum0add15f1997-05-09 01:07:15 +0000629 Py_INCREF(v);
Guido van Rossum21d335e1993-10-15 13:01:11 +0000630 if (class != NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000631 if (PyFunction_Check(v)) {
632 PyObject *w = PyMethod_New(v, (PyObject *)inst,
Jeremy Hylton09ac89a2001-01-29 22:38:32 +0000633 (PyObject *)class);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000634 Py_DECREF(v);
Guido van Rossum21d335e1993-10-15 13:01:11 +0000635 v = w;
636 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000637 else if (PyMethod_Check(v)) {
638 PyObject *im_class = PyMethod_Class(v);
Guido van Rossum21d335e1993-10-15 13:01:11 +0000639 /* Only if classes are compatible */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000640 if (PyClass_IsSubclass((PyObject *)class, im_class)) {
641 PyObject *im_func = PyMethod_Function(v);
642 PyObject *w = PyMethod_New(im_func,
643 (PyObject *)inst, im_class);
644 Py_DECREF(v);
Guido van Rossum21d335e1993-10-15 13:01:11 +0000645 v = w;
646 }
647 }
Guido van Rossumb3f72581993-05-21 19:56:10 +0000648 }
649 return v;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000650}
651
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000652static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000653instance_getattr(register PyInstanceObject *inst, PyObject *name)
Guido van Rossume7737541994-09-05 07:31:41 +0000654{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000655 register PyObject *func, *res;
Guido van Rossum52ca98a1994-09-05 07:32:29 +0000656 res = instance_getattr1(inst, name);
657 if (res == NULL && (func = inst->in_class->cl_getattr) != NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000658 PyObject *args;
659 PyErr_Clear();
660 args = Py_BuildValue("(OO)", inst, name);
Guido van Rossum52ca98a1994-09-05 07:32:29 +0000661 if (args == NULL)
Guido van Rossume7737541994-09-05 07:31:41 +0000662 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000663 res = PyEval_CallObject(func, args);
664 Py_DECREF(args);
Guido van Rossume7737541994-09-05 07:31:41 +0000665 }
666 return res;
667}
668
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000669static int
Fred Drake79912472000-07-09 04:06:11 +0000670instance_setattr1(PyInstanceObject *inst, PyObject *name, PyObject *v)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000671{
Guido van Rossum94472a01992-09-04 09:45:18 +0000672 if (v == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000673 int rv = PyDict_DelItem(inst->in_dict, name);
Guido van Rossum94472a01992-09-04 09:45:18 +0000674 if (rv < 0)
Fred Drake661ea262000-10-24 19:57:45 +0000675 PyErr_Format(PyExc_AttributeError,
676 "%.50s instance has no attribute '%.400s'",
677 PyString_AS_STRING(inst->in_class->cl_name),
678 PyString_AS_STRING(name));
Guido van Rossum94472a01992-09-04 09:45:18 +0000679 return rv;
680 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000681 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000682 return PyDict_SetItem(inst->in_dict, name, v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000683}
684
Guido van Rossume7737541994-09-05 07:31:41 +0000685static int
Fred Drake79912472000-07-09 04:06:11 +0000686instance_setattr(PyInstanceObject *inst, PyObject *name, PyObject *v)
Guido van Rossume7737541994-09-05 07:31:41 +0000687{
Guido van Rossumb2173c31997-08-25 21:23:56 +0000688 PyObject *func, *args, *res, *tmp;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000689 char *sname = PyString_AsString(name);
Guido van Rossumb2173c31997-08-25 21:23:56 +0000690 if (sname[0] == '_' && sname[1] == '_') {
691 int n = PyString_Size(name);
Guido van Rossum2878a691996-08-09 20:53:24 +0000692 if (sname[n-1] == '_' && sname[n-2] == '_') {
Guido van Rossumb2173c31997-08-25 21:23:56 +0000693 if (strcmp(sname, "__dict__") == 0) {
694 if (PyEval_GetRestricted()) {
695 PyErr_SetString(PyExc_RuntimeError,
696 "__dict__ not accessible in restricted mode");
697 return -1;
698 }
699 if (v == NULL || !PyDict_Check(v)) {
700 PyErr_SetString(PyExc_TypeError,
701 "__dict__ must be set to a dictionary");
702 return -1;
703 }
704 tmp = inst->in_dict;
705 Py_INCREF(v);
706 inst->in_dict = v;
707 Py_DECREF(tmp);
708 return 0;
709 }
710 if (strcmp(sname, "__class__") == 0) {
711 if (PyEval_GetRestricted()) {
712 PyErr_SetString(PyExc_RuntimeError,
713 "__class__ not accessible in restricted mode");
714 return -1;
715 }
716 if (v == NULL || !PyClass_Check(v)) {
717 PyErr_SetString(PyExc_TypeError,
718 "__class__ must be set to a class");
719 return -1;
720 }
721 tmp = (PyObject *)(inst->in_class);
722 Py_INCREF(v);
723 inst->in_class = (PyClassObject *)v;
724 Py_DECREF(tmp);
725 return 0;
726 }
Guido van Rossume7737541994-09-05 07:31:41 +0000727 }
Guido van Rossume7737541994-09-05 07:31:41 +0000728 }
Guido van Rossum52ca98a1994-09-05 07:32:29 +0000729 if (v == NULL)
730 func = inst->in_class->cl_delattr;
731 else
732 func = inst->in_class->cl_setattr;
733 if (func == NULL)
734 return instance_setattr1(inst, name, v);
735 if (v == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000736 args = Py_BuildValue("(OO)", inst, name);
Guido van Rossum52ca98a1994-09-05 07:32:29 +0000737 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000738 args = Py_BuildValue("(OOO)", inst, name, v);
Guido van Rossum52ca98a1994-09-05 07:32:29 +0000739 if (args == NULL)
740 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000741 res = PyEval_CallObject(func, args);
742 Py_DECREF(args);
Guido van Rossum52ca98a1994-09-05 07:32:29 +0000743 if (res == NULL)
744 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000745 Py_DECREF(res);
Guido van Rossume7737541994-09-05 07:31:41 +0000746 return 0;
747}
748
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000749static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000750instance_repr(PyInstanceObject *inst)
Guido van Rossum04691fc1992-08-12 15:35:34 +0000751{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000752 PyObject *func;
753 PyObject *res;
754 static PyObject *reprstr;
Guido van Rossum04691fc1992-08-12 15:35:34 +0000755
Guido van Rossum2878a691996-08-09 20:53:24 +0000756 if (reprstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000757 reprstr = PyString_InternFromString("__repr__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000758 func = instance_getattr(inst, reprstr);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000759 if (func == NULL) {
Guido van Rossum25831651993-05-19 14:50:45 +0000760 char buf[140];
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000761 PyObject *classname = inst->in_class->cl_name;
Guido van Rossumb7f1afe1997-12-03 00:06:02 +0000762 PyObject *mod = PyDict_GetItemString(
763 inst->in_class->cl_dict, "__module__");
Guido van Rossum25831651993-05-19 14:50:45 +0000764 char *cname;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000765 if (classname != NULL && PyString_Check(classname))
766 cname = PyString_AsString(classname);
Guido van Rossum25831651993-05-19 14:50:45 +0000767 else
768 cname = "?";
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000769 PyErr_Clear();
Guido van Rossumb7f1afe1997-12-03 00:06:02 +0000770 if (mod == NULL || !PyString_Check(mod))
Fred Drakea44d3532000-06-30 15:01:00 +0000771 sprintf(buf, "<?.%.100s instance at %p>",
772 cname, inst);
Guido van Rossumb7f1afe1997-12-03 00:06:02 +0000773 else
Fred Drakea44d3532000-06-30 15:01:00 +0000774 sprintf(buf, "<%.50s.%.50s instance at %p>",
Guido van Rossumb7f1afe1997-12-03 00:06:02 +0000775 PyString_AsString(mod),
Fred Drakea44d3532000-06-30 15:01:00 +0000776 cname, inst);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000777 return PyString_FromString(buf);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000778 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000779 res = PyEval_CallObject(func, (PyObject *)NULL);
780 Py_DECREF(func);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000781 return res;
782}
783
Guido van Rossum82c690f2001-04-30 14:39:18 +0000784static PyObject *
785instance_str(PyInstanceObject *inst)
786{
787 PyObject *func;
788 PyObject *res;
789 static PyObject *strstr;
790
791 if (strstr == NULL)
792 strstr = PyString_InternFromString("__str__");
793 func = instance_getattr(inst, strstr);
794 if (func == NULL) {
795 PyErr_Clear();
796 return instance_repr(inst);
797 }
798 res = PyEval_CallObject(func, (PyObject *)NULL);
799 Py_DECREF(func);
800 return res;
801}
802
Guido van Rossum9bfef441993-03-29 10:43:31 +0000803static long
Fred Drake79912472000-07-09 04:06:11 +0000804instance_hash(PyInstanceObject *inst)
Guido van Rossum9bfef441993-03-29 10:43:31 +0000805{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000806 PyObject *func;
807 PyObject *res;
Guido van Rossum8a0c3451993-04-08 12:56:19 +0000808 long outcome;
Guido van Rossum65e8bd72001-01-18 23:46:31 +0000809 static PyObject *hashstr, *eqstr, *cmpstr;
Guido van Rossum9bfef441993-03-29 10:43:31 +0000810
Guido van Rossum2878a691996-08-09 20:53:24 +0000811 if (hashstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000812 hashstr = PyString_InternFromString("__hash__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000813 func = instance_getattr(inst, hashstr);
Guido van Rossum9bfef441993-03-29 10:43:31 +0000814 if (func == NULL) {
Guido van Rossum65e8bd72001-01-18 23:46:31 +0000815 /* If there is no __eq__ and no __cmp__ method, we hash on the
816 address. If an __eq__ or __cmp__ method exists, there must
817 be a __hash__. */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000818 PyErr_Clear();
Guido van Rossum65e8bd72001-01-18 23:46:31 +0000819 if (eqstr == NULL)
820 eqstr = PyString_InternFromString("__eq__");
821 func = instance_getattr(inst, eqstr);
Guido van Rossum8a0c3451993-04-08 12:56:19 +0000822 if (func == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000823 PyErr_Clear();
Guido van Rossum65e8bd72001-01-18 23:46:31 +0000824 if (cmpstr == NULL)
825 cmpstr = PyString_InternFromString("__cmp__");
826 func = instance_getattr(inst, cmpstr);
827 if (func == NULL) {
828 PyErr_Clear();
829 return _Py_HashPointer(inst);
830 }
Guido van Rossum8a0c3451993-04-08 12:56:19 +0000831 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000832 PyErr_SetString(PyExc_TypeError, "unhashable instance");
Guido van Rossum9bfef441993-03-29 10:43:31 +0000833 return -1;
834 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000835 res = PyEval_CallObject(func, (PyObject *)NULL);
836 Py_DECREF(func);
Guido van Rossum9bfef441993-03-29 10:43:31 +0000837 if (res == NULL)
838 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000839 if (PyInt_Check(res)) {
840 outcome = PyInt_AsLong(res);
Guido van Rossum9bfef441993-03-29 10:43:31 +0000841 if (outcome == -1)
842 outcome = -2;
843 }
844 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000845 PyErr_SetString(PyExc_TypeError,
846 "__hash__() should return an int");
Guido van Rossum9bfef441993-03-29 10:43:31 +0000847 outcome = -1;
848 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000849 Py_DECREF(res);
Guido van Rossum9bfef441993-03-29 10:43:31 +0000850 return outcome;
851}
852
Jeremy Hylton8caad492000-06-23 14:18:11 +0000853static int
854instance_traverse(PyInstanceObject *o, visitproc visit, void *arg)
855{
856 int err;
857 if (o->in_class) {
858 err = visit((PyObject *)(o->in_class), arg);
859 if (err)
860 return err;
861 }
862 if (o->in_dict) {
863 err = visit(o->in_dict, arg);
864 if (err)
865 return err;
866 }
Jeremy Hyltond22162b2000-06-23 17:14:56 +0000867 return 0;
Jeremy Hylton8caad492000-06-23 14:18:11 +0000868}
869
Guido van Rossum213c7a62001-04-23 14:08:49 +0000870static PyObject *getitemstr, *setitemstr, *delitemstr, *lenstr;
871static PyObject *iterstr, *nextstr;
Guido van Rossum2878a691996-08-09 20:53:24 +0000872
Guido van Rossum9bfef441993-03-29 10:43:31 +0000873static int
Fred Drake79912472000-07-09 04:06:11 +0000874instance_length(PyInstanceObject *inst)
Guido van Rossum04691fc1992-08-12 15:35:34 +0000875{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000876 PyObject *func;
877 PyObject *res;
Guido van Rossum04691fc1992-08-12 15:35:34 +0000878 int outcome;
879
Guido van Rossum2878a691996-08-09 20:53:24 +0000880 if (lenstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000881 lenstr = PyString_InternFromString("__len__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000882 func = instance_getattr(inst, lenstr);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000883 if (func == NULL)
884 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000885 res = PyEval_CallObject(func, (PyObject *)NULL);
886 Py_DECREF(func);
Guido van Rossumd014ea61992-11-26 10:30:26 +0000887 if (res == NULL)
888 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000889 if (PyInt_Check(res)) {
890 outcome = PyInt_AsLong(res);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000891 if (outcome < 0)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000892 PyErr_SetString(PyExc_ValueError,
893 "__len__() should return >= 0");
Guido van Rossum04691fc1992-08-12 15:35:34 +0000894 }
895 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000896 PyErr_SetString(PyExc_TypeError,
897 "__len__() should return an int");
Guido van Rossum04691fc1992-08-12 15:35:34 +0000898 outcome = -1;
899 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000900 Py_DECREF(res);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000901 return outcome;
902}
903
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000904static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000905instance_subscript(PyInstanceObject *inst, PyObject *key)
Guido van Rossum04691fc1992-08-12 15:35:34 +0000906{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000907 PyObject *func;
908 PyObject *arg;
909 PyObject *res;
Guido van Rossum04691fc1992-08-12 15:35:34 +0000910
Guido van Rossum2878a691996-08-09 20:53:24 +0000911 if (getitemstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000912 getitemstr = PyString_InternFromString("__getitem__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000913 func = instance_getattr(inst, getitemstr);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000914 if (func == NULL)
915 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000916 arg = Py_BuildValue("(O)", key);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000917 if (arg == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000918 Py_DECREF(func);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000919 return NULL;
920 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000921 res = PyEval_CallObject(func, arg);
922 Py_DECREF(func);
923 Py_DECREF(arg);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000924 return res;
925}
926
Guido van Rossum9bfef441993-03-29 10:43:31 +0000927static int
Fred Drake79912472000-07-09 04:06:11 +0000928instance_ass_subscript(PyInstanceObject *inst, PyObject *key, PyObject *value)
Guido van Rossum04691fc1992-08-12 15:35:34 +0000929{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000930 PyObject *func;
931 PyObject *arg;
932 PyObject *res;
Guido van Rossum04691fc1992-08-12 15:35:34 +0000933
Guido van Rossum2878a691996-08-09 20:53:24 +0000934 if (value == NULL) {
935 if (delitemstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000936 delitemstr = PyString_InternFromString("__delitem__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000937 func = instance_getattr(inst, delitemstr);
938 }
939 else {
940 if (setitemstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000941 setitemstr = PyString_InternFromString("__setitem__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000942 func = instance_getattr(inst, setitemstr);
943 }
Guido van Rossum04691fc1992-08-12 15:35:34 +0000944 if (func == NULL)
945 return -1;
946 if (value == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000947 arg = Py_BuildValue("(O)", key);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000948 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000949 arg = Py_BuildValue("(OO)", key, value);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000950 if (arg == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000951 Py_DECREF(func);
Guido van Rossum6d946f91992-08-14 13:49:30 +0000952 return -1;
Guido van Rossum04691fc1992-08-12 15:35:34 +0000953 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000954 res = PyEval_CallObject(func, arg);
955 Py_DECREF(func);
956 Py_DECREF(arg);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000957 if (res == NULL)
958 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000959 Py_DECREF(res);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000960 return 0;
961}
962
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000963static PyMappingMethods instance_as_mapping = {
Guido van Rossum8998b4f2001-01-17 15:28:20 +0000964 (inquiry)instance_length, /* mp_length */
965 (binaryfunc)instance_subscript, /* mp_subscript */
966 (objobjargproc)instance_ass_subscript, /* mp_ass_subscript */
Guido van Rossum04691fc1992-08-12 15:35:34 +0000967};
968
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000969static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000970instance_item(PyInstanceObject *inst, int i)
Guido van Rossum04691fc1992-08-12 15:35:34 +0000971{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000972 PyObject *func, *arg, *res;
Guido van Rossum04691fc1992-08-12 15:35:34 +0000973
Guido van Rossum2878a691996-08-09 20:53:24 +0000974 if (getitemstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000975 getitemstr = PyString_InternFromString("__getitem__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000976 func = instance_getattr(inst, getitemstr);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000977 if (func == NULL)
978 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000979 arg = Py_BuildValue("(i)", i);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000980 if (arg == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000981 Py_DECREF(func);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000982 return NULL;
983 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000984 res = PyEval_CallObject(func, arg);
985 Py_DECREF(func);
986 Py_DECREF(arg);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000987 return res;
988}
989
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000990static PyObject *
Thomas Wouters1d75a792000-08-17 22:37:32 +0000991sliceobj_from_intint(int i, int j)
992{
993 PyObject *start, *end, *res;
994
995 start = PyInt_FromLong((long)i);
996 if (!start)
997 return NULL;
998
999 end = PyInt_FromLong((long)j);
1000 if (!end) {
1001 Py_DECREF(start);
1002 return NULL;
1003 }
1004 res = PySlice_New(start, end, NULL);
1005 Py_DECREF(start);
1006 Py_DECREF(end);
1007 return res;
1008}
1009
1010
1011static PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001012instance_slice(PyInstanceObject *inst, int i, int j)
Guido van Rossum04691fc1992-08-12 15:35:34 +00001013{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001014 PyObject *func, *arg, *res;
1015 static PyObject *getslicestr;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001016
Guido van Rossum2878a691996-08-09 20:53:24 +00001017 if (getslicestr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +00001018 getslicestr = PyString_InternFromString("__getslice__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001019 func = instance_getattr(inst, getslicestr);
Thomas Wouters1d75a792000-08-17 22:37:32 +00001020
1021 if (func == NULL) {
1022 PyErr_Clear();
1023
1024 if (getitemstr == NULL)
1025 getitemstr = PyString_InternFromString("__getitem__");
1026 func = instance_getattr(inst, getitemstr);
1027 if (func == NULL)
1028 return NULL;
1029 arg = Py_BuildValue("(N)", sliceobj_from_intint(i, j));
1030 } else
1031 arg = Py_BuildValue("(ii)", i, j);
1032
Guido van Rossum04691fc1992-08-12 15:35:34 +00001033 if (arg == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001034 Py_DECREF(func);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001035 return NULL;
1036 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001037 res = PyEval_CallObject(func, arg);
1038 Py_DECREF(func);
1039 Py_DECREF(arg);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001040 return res;
1041}
1042
1043static int
Fred Drake79912472000-07-09 04:06:11 +00001044instance_ass_item(PyInstanceObject *inst, int i, PyObject *item)
Guido van Rossum04691fc1992-08-12 15:35:34 +00001045{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001046 PyObject *func, *arg, *res;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001047
Guido van Rossum2878a691996-08-09 20:53:24 +00001048 if (item == NULL) {
1049 if (delitemstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +00001050 delitemstr = PyString_InternFromString("__delitem__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001051 func = instance_getattr(inst, delitemstr);
1052 }
1053 else {
1054 if (setitemstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +00001055 setitemstr = PyString_InternFromString("__setitem__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001056 func = instance_getattr(inst, setitemstr);
1057 }
Guido van Rossum04691fc1992-08-12 15:35:34 +00001058 if (func == NULL)
Guido van Rossum6d946f91992-08-14 13:49:30 +00001059 return -1;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001060 if (item == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001061 arg = Py_BuildValue("i", i);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001062 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001063 arg = Py_BuildValue("(iO)", i, item);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001064 if (arg == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001065 Py_DECREF(func);
Guido van Rossum6d946f91992-08-14 13:49:30 +00001066 return -1;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001067 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001068 res = PyEval_CallObject(func, arg);
1069 Py_DECREF(func);
1070 Py_DECREF(arg);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001071 if (res == NULL)
1072 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001073 Py_DECREF(res);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001074 return 0;
1075}
1076
1077static int
Fred Drake79912472000-07-09 04:06:11 +00001078instance_ass_slice(PyInstanceObject *inst, int i, int j, PyObject *value)
Guido van Rossum04691fc1992-08-12 15:35:34 +00001079{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001080 PyObject *func, *arg, *res;
1081 static PyObject *setslicestr, *delslicestr;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001082
Guido van Rossum2878a691996-08-09 20:53:24 +00001083 if (value == NULL) {
1084 if (delslicestr == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001085 delslicestr =
1086 PyString_InternFromString("__delslice__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001087 func = instance_getattr(inst, delslicestr);
Thomas Wouters1d75a792000-08-17 22:37:32 +00001088 if (func == NULL) {
1089 PyErr_Clear();
1090 if (delitemstr == NULL)
1091 delitemstr =
1092 PyString_InternFromString("__delitem__");
1093 func = instance_getattr(inst, delitemstr);
1094 if (func == NULL)
1095 return -1;
1096
1097 arg = Py_BuildValue("(N)",
1098 sliceobj_from_intint(i, j));
1099 } else
1100 arg = Py_BuildValue("(ii)", i, j);
Guido van Rossum2878a691996-08-09 20:53:24 +00001101 }
1102 else {
1103 if (setslicestr == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001104 setslicestr =
1105 PyString_InternFromString("__setslice__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001106 func = instance_getattr(inst, setslicestr);
Thomas Wouters1d75a792000-08-17 22:37:32 +00001107 if (func == NULL) {
1108 PyErr_Clear();
1109 if (setitemstr == NULL)
1110 setitemstr =
1111 PyString_InternFromString("__setitem__");
1112 func = instance_getattr(inst, setitemstr);
1113 if (func == NULL)
1114 return -1;
1115
1116 arg = Py_BuildValue("(NO)",
1117 sliceobj_from_intint(i, j), value);
1118 } else
1119 arg = Py_BuildValue("(iiO)", i, j, value);
Guido van Rossum2878a691996-08-09 20:53:24 +00001120 }
Guido van Rossum04691fc1992-08-12 15:35:34 +00001121 if (arg == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001122 Py_DECREF(func);
Guido van Rossum6d946f91992-08-14 13:49:30 +00001123 return -1;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001124 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001125 res = PyEval_CallObject(func, arg);
1126 Py_DECREF(func);
1127 Py_DECREF(arg);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001128 if (res == NULL)
1129 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001130 Py_DECREF(res);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001131 return 0;
1132}
1133
Guido van Rossumee28c3a2000-02-28 15:03:15 +00001134static int instance_contains(PyInstanceObject *inst, PyObject *member)
1135{
1136 static PyObject *__contains__;
1137 PyObject *func, *arg, *res;
1138 int ret;
1139
1140 if(__contains__ == NULL) {
1141 __contains__ = PyString_InternFromString("__contains__");
1142 if(__contains__ == NULL)
1143 return -1;
1144 }
1145 func = instance_getattr(inst, __contains__);
1146 if(func == NULL) {
Thomas Wouters7e474022000-07-16 12:04:32 +00001147 /* fall back to previous behavior */
Guido van Rossumee28c3a2000-02-28 15:03:15 +00001148 int i, cmp_res;
1149
1150 if(!PyErr_ExceptionMatches(PyExc_AttributeError))
1151 return -1;
1152 PyErr_Clear();
1153 for(i=0;;i++) {
1154 PyObject *obj = instance_item(inst, i);
1155 int ret = 0;
1156
1157 if(obj == NULL) {
1158 if(!PyErr_ExceptionMatches(PyExc_IndexError))
1159 return -1;
1160 PyErr_Clear();
1161 return 0;
1162 }
1163 if(PyObject_Cmp(obj, member, &cmp_res) == -1)
1164 ret = -1;
1165 if(cmp_res == 0)
1166 ret = 1;
1167 Py_DECREF(obj);
1168 if(ret)
1169 return ret;
1170 }
1171 }
1172 arg = Py_BuildValue("(O)", member);
1173 if(arg == NULL) {
1174 Py_DECREF(func);
1175 return -1;
1176 }
1177 res = PyEval_CallObject(func, arg);
1178 Py_DECREF(func);
1179 Py_DECREF(arg);
1180 if(res == NULL)
1181 return -1;
1182 ret = PyObject_IsTrue(res);
1183 Py_DECREF(res);
1184 return ret;
1185}
1186
Fred Drake79912472000-07-09 04:06:11 +00001187static PySequenceMethods
1188instance_as_sequence = {
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001189 (inquiry)instance_length, /* sq_length */
1190 0, /* sq_concat */
1191 0, /* sq_repeat */
1192 (intargfunc)instance_item, /* sq_item */
1193 (intintargfunc)instance_slice, /* sq_slice */
1194 (intobjargproc)instance_ass_item, /* sq_ass_item */
1195 (intintobjargproc)instance_ass_slice, /* sq_ass_slice */
1196 (objobjproc)instance_contains, /* sq_contains */
Guido van Rossum04691fc1992-08-12 15:35:34 +00001197};
1198
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001199static PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001200generic_unary_op(PyInstanceObject *self, PyObject *methodname)
Guido van Rossum04691fc1992-08-12 15:35:34 +00001201{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001202 PyObject *func, *res;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001203
1204 if ((func = instance_getattr(self, methodname)) == NULL)
1205 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001206 res = PyEval_CallObject(func, (PyObject *)NULL);
1207 Py_DECREF(func);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001208 return res;
1209}
1210
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001211static PyObject *
1212generic_binary_op(PyObject *v, PyObject *w, char *opname)
Guido van Rossum03093a21994-09-28 15:51:32 +00001213{
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001214 PyObject *result;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001215 PyObject *args;
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001216 PyObject *func = PyObject_GetAttrString(v, opname);
Guido van Rossume7d444f1995-01-07 12:35:18 +00001217 if (func == NULL) {
Guido van Rossum617c1b01998-05-28 19:50:02 +00001218 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001219 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001220 PyErr_Clear();
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001221 Py_INCREF(Py_NotImplemented);
1222 return Py_NotImplemented;
Guido van Rossume7d444f1995-01-07 12:35:18 +00001223 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001224 args = Py_BuildValue("(O)", w);
Guido van Rossum03093a21994-09-28 15:51:32 +00001225 if (args == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001226 Py_DECREF(func);
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001227 return NULL;
Guido van Rossum03093a21994-09-28 15:51:32 +00001228 }
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001229 result = PyEval_CallObject(func, args);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001230 Py_DECREF(args);
1231 Py_DECREF(func);
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001232 return result;
1233}
1234
1235
1236static PyObject *coerce_obj;
1237
1238/* Try one half of a binary operator involving a class instance. */
1239static PyObject *
1240half_binop(PyObject *v, PyObject *w, char *opname, binaryfunc thisfunc,
1241 int swapped)
1242{
1243 PyObject *args;
1244 PyObject *coercefunc;
1245 PyObject *coerced = NULL;
1246 PyObject *v1;
1247 PyObject *result;
1248
1249 if (!PyInstance_Check(v)) {
1250 Py_INCREF(Py_NotImplemented);
1251 return Py_NotImplemented;
1252 }
1253
1254 if (coerce_obj == NULL) {
1255 coerce_obj = PyString_InternFromString("__coerce__");
1256 if (coerce_obj == NULL)
1257 return NULL;
1258 }
1259 coercefunc = PyObject_GetAttr(v, coerce_obj);
1260 if (coercefunc == NULL) {
1261 PyErr_Clear();
1262 return generic_binary_op(v, w, opname);
1263 }
1264
1265 args = Py_BuildValue("(O)", w);
1266 if (args == NULL) {
1267 return NULL;
1268 }
1269 coerced = PyEval_CallObject(coercefunc, args);
1270 Py_DECREF(args);
1271 Py_DECREF(coercefunc);
1272 if (coerced == NULL) {
1273 return NULL;
1274 }
1275 if (coerced == Py_None || coerced == Py_NotImplemented) {
1276 Py_DECREF(coerced);
1277 return generic_binary_op(v, w, opname);
1278 }
1279 if (!PyTuple_Check(coerced) || PyTuple_Size(coerced) != 2) {
1280 Py_DECREF(coerced);
1281 PyErr_SetString(PyExc_TypeError,
1282 "coercion should return None or 2-tuple");
1283 return NULL;
1284 }
1285 v1 = PyTuple_GetItem(coerced, 0);
1286 w = PyTuple_GetItem(coerced, 1);
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001287 if (v1->ob_type == v->ob_type && PyInstance_Check(v)) {
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001288 /* prevent recursion if __coerce__ returns self as the first
1289 * argument */
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001290 result = generic_binary_op(v1, w, opname);
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001291 } else {
1292 if (swapped)
1293 result = (thisfunc)(w, v1);
1294 else
1295 result = (thisfunc)(v1, w);
1296 }
1297 Py_DECREF(coerced);
1298 return result;
1299}
1300
1301/* Implement a binary operator involving at least one class instance. */
1302static PyObject *
1303do_binop(PyObject *v, PyObject *w, char *opname, char *ropname,
1304 binaryfunc thisfunc)
1305{
1306 PyObject *result = half_binop(v, w, opname, thisfunc, 0);
1307 if (result == Py_NotImplemented) {
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001308 Py_DECREF(result);
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001309 result = half_binop(w, v, ropname, thisfunc, 1);
1310 }
1311 return result;
1312}
1313
1314static PyObject *
1315do_binop_inplace(PyObject *v, PyObject *w, char *iopname, char *opname,
1316 char *ropname, binaryfunc thisfunc)
1317{
1318 PyObject *result = half_binop(v, w, iopname, thisfunc, 0);
1319 if (result == Py_NotImplemented) {
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001320 Py_DECREF(result);
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001321 result = do_binop(v, w, opname, ropname, thisfunc);
1322 }
1323 return result;
Guido van Rossum03093a21994-09-28 15:51:32 +00001324}
1325
Guido van Rossum879c5811995-01-10 15:24:06 +00001326static int
Fred Drake79912472000-07-09 04:06:11 +00001327instance_coerce(PyObject **pv, PyObject **pw)
Guido van Rossum879c5811995-01-10 15:24:06 +00001328{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001329 PyObject *v = *pv;
1330 PyObject *w = *pw;
Guido van Rossum3931df91997-11-18 19:23:07 +00001331 PyObject *coercefunc;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001332 PyObject *args;
1333 PyObject *coerced;
Guido van Rossum879c5811995-01-10 15:24:06 +00001334
Guido van Rossum2878a691996-08-09 20:53:24 +00001335 if (coerce_obj == NULL) {
Guido van Rossuma412d241997-01-18 07:59:12 +00001336 coerce_obj = PyString_InternFromString("__coerce__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001337 if (coerce_obj == NULL)
1338 return -1;
1339 }
Guido van Rossum3931df91997-11-18 19:23:07 +00001340 coercefunc = PyObject_GetAttr(v, coerce_obj);
1341 if (coercefunc == NULL) {
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001342 /* No __coerce__ method */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001343 PyErr_Clear();
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001344 return 1;
Guido van Rossum879c5811995-01-10 15:24:06 +00001345 }
1346 /* Has __coerce__ method: call it */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001347 args = Py_BuildValue("(O)", w);
Guido van Rossum879c5811995-01-10 15:24:06 +00001348 if (args == NULL) {
1349 return -1;
1350 }
Guido van Rossum3931df91997-11-18 19:23:07 +00001351 coerced = PyEval_CallObject(coercefunc, args);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001352 Py_DECREF(args);
Guido van Rossum3931df91997-11-18 19:23:07 +00001353 Py_DECREF(coercefunc);
Guido van Rossum879c5811995-01-10 15:24:06 +00001354 if (coerced == NULL) {
1355 /* __coerce__ call raised an exception */
1356 return -1;
1357 }
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001358 if (coerced == Py_None || coerced == Py_NotImplemented) {
Guido van Rossum879c5811995-01-10 15:24:06 +00001359 /* __coerce__ says "I can't do it" */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001360 Py_DECREF(coerced);
Guido van Rossum879c5811995-01-10 15:24:06 +00001361 return 1;
1362 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001363 if (!PyTuple_Check(coerced) || PyTuple_Size(coerced) != 2) {
Guido van Rossum879c5811995-01-10 15:24:06 +00001364 /* __coerce__ return value is malformed */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001365 Py_DECREF(coerced);
1366 PyErr_SetString(PyExc_TypeError,
Guido van Rossum879c5811995-01-10 15:24:06 +00001367 "coercion should return None or 2-tuple");
1368 return -1;
1369 }
1370 /* __coerce__ returned two new values */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001371 *pv = PyTuple_GetItem(coerced, 0);
1372 *pw = PyTuple_GetItem(coerced, 1);
1373 Py_INCREF(*pv);
1374 Py_INCREF(*pw);
1375 Py_DECREF(coerced);
Guido van Rossum879c5811995-01-10 15:24:06 +00001376 return 0;
1377}
1378
Guido van Rossum04691fc1992-08-12 15:35:34 +00001379#define UNARY(funcname, methodname) \
Thomas Woutersc3073522000-07-23 22:09:59 +00001380static PyObject *funcname(PyInstanceObject *self) { \
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001381 static PyObject *o; \
Guido van Rossuma412d241997-01-18 07:59:12 +00001382 if (o == NULL) o = PyString_InternFromString(methodname); \
Guido van Rossum2878a691996-08-09 20:53:24 +00001383 return generic_unary_op(self, o); \
Guido van Rossum04691fc1992-08-12 15:35:34 +00001384}
1385
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001386#define BINARY(f, m, n) \
1387static PyObject *f(PyObject *v, PyObject *w) { \
1388 return do_binop(v, w, "__" m "__", "__r" m "__", n); \
1389}
1390
1391#define BINARY_INPLACE(f, m, n) \
1392static PyObject *f(PyObject *v, PyObject *w) { \
1393 return do_binop_inplace(v, w, "__i" m "__", "__" m "__", \
1394 "__r" m "__", n); \
1395}
1396
Guido van Rossum04691fc1992-08-12 15:35:34 +00001397UNARY(instance_neg, "__neg__")
1398UNARY(instance_pos, "__pos__")
1399UNARY(instance_abs, "__abs__")
1400
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001401BINARY(instance_or, "or", PyNumber_Or)
1402BINARY(instance_and, "and", PyNumber_And)
1403BINARY(instance_xor, "xor", PyNumber_Xor)
1404BINARY(instance_lshift, "lshift", PyNumber_Lshift)
1405BINARY(instance_rshift, "rshift", PyNumber_Rshift)
1406BINARY(instance_add, "add", PyNumber_Add)
1407BINARY(instance_sub, "sub", PyNumber_Subtract)
1408BINARY(instance_mul, "mul", PyNumber_Multiply)
1409BINARY(instance_div, "div", PyNumber_Divide)
1410BINARY(instance_mod, "mod", PyNumber_Remainder)
1411BINARY(instance_divmod, "divmod", PyNumber_Divmod)
1412
1413BINARY_INPLACE(instance_ior, "or", PyNumber_InPlaceOr)
1414BINARY_INPLACE(instance_ixor, "xor", PyNumber_InPlaceXor)
1415BINARY_INPLACE(instance_iand, "and", PyNumber_InPlaceAnd)
1416BINARY_INPLACE(instance_ilshift, "lshift", PyNumber_InPlaceLshift)
1417BINARY_INPLACE(instance_irshift, "rshift", PyNumber_InPlaceRshift)
1418BINARY_INPLACE(instance_iadd, "add", PyNumber_InPlaceAdd)
1419BINARY_INPLACE(instance_isub, "sub", PyNumber_InPlaceSubtract)
1420BINARY_INPLACE(instance_imul, "mul", PyNumber_InPlaceMultiply)
1421BINARY_INPLACE(instance_idiv, "div", PyNumber_InPlaceDivide)
1422BINARY_INPLACE(instance_imod, "mod", PyNumber_InPlaceRemainder)
1423
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001424/* Try a 3-way comparison, returning an int; v is an instance. Return:
1425 -2 for an exception;
1426 -1 if v < w;
1427 0 if v == w;
1428 1 if v > w;
1429 2 if this particular 3-way comparison is not implemented or undefined.
1430*/
1431static int
1432half_cmp(PyObject *v, PyObject *w)
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001433{
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001434 static PyObject *cmp_obj;
1435 PyObject *args;
Guido van Rossum3202c6f2001-01-29 23:50:25 +00001436 PyObject *cmp_func;
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001437 PyObject *result;
1438 long l;
1439
1440 assert(PyInstance_Check(v));
1441
1442 if (cmp_obj == NULL) {
1443 cmp_obj = PyString_InternFromString("__cmp__");
1444 if (cmp_obj == NULL)
1445 return -2;
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001446 }
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001447
Guido van Rossum3202c6f2001-01-29 23:50:25 +00001448 cmp_func = PyObject_GetAttr(v, cmp_obj);
1449 if (cmp_func == NULL) {
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001450 PyErr_Clear();
1451 return 2;
1452 }
1453
1454 args = Py_BuildValue("(O)", w);
1455 if (args == NULL)
1456 return -2;
1457
Guido van Rossum3202c6f2001-01-29 23:50:25 +00001458 result = PyEval_CallObject(cmp_func, args);
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001459 Py_DECREF(args);
Guido van Rossum3202c6f2001-01-29 23:50:25 +00001460 Py_DECREF(cmp_func);
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001461
1462 if (result == NULL)
1463 return -2;
1464
1465 if (result == Py_NotImplemented) {
1466 Py_DECREF(result);
1467 return 2;
1468 }
1469
1470 l = PyInt_AsLong(result);
1471 Py_DECREF(result);
1472 if (l == -1 && PyErr_Occurred()) {
1473 PyErr_SetString(PyExc_TypeError,
1474 "comparison did not return an int");
1475 return -2;
1476 }
1477
1478 return l < 0 ? -1 : l > 0 ? 1 : 0;
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001479}
1480
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001481/* Try a 3-way comparison, returning an int; either v or w is an instance.
1482 We first try a coercion. Return:
1483 -2 for an exception;
1484 -1 if v < w;
1485 0 if v == w;
1486 1 if v > w;
1487 2 if this particular 3-way comparison is not implemented or undefined.
1488 THIS IS ONLY CALLED FROM object.c!
1489*/
1490static int
1491instance_compare(PyObject *v, PyObject *w)
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001492{
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001493 int c;
1494
1495 c = PyNumber_CoerceEx(&v, &w);
1496 if (c < 0)
1497 return -2;
1498 if (c == 0) {
1499 /* If neither is now an instance, use regular comparison */
1500 if (!PyInstance_Check(v) && !PyInstance_Check(w)) {
1501 c = PyObject_Compare(v, w);
1502 Py_DECREF(v);
1503 Py_DECREF(w);
1504 if (PyErr_Occurred())
1505 return -2;
1506 return c < 0 ? -1 : c > 0 ? 1 : 0;
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001507 }
1508 }
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001509 else {
1510 /* The coercion didn't do anything.
1511 Treat this the same as returning v and w unchanged. */
1512 Py_INCREF(v);
1513 Py_INCREF(w);
1514 }
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001515
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001516 if (PyInstance_Check(v)) {
1517 c = half_cmp(v, w);
1518 if (c <= 1) {
1519 Py_DECREF(v);
1520 Py_DECREF(w);
1521 return c;
1522 }
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001523 }
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001524 if (PyInstance_Check(w)) {
1525 c = half_cmp(w, v);
1526 if (c <= 1) {
1527 Py_DECREF(v);
1528 Py_DECREF(w);
1529 if (c >= -1)
1530 c = -c;
1531 return c;
1532 }
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001533 }
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001534 Py_DECREF(v);
1535 Py_DECREF(w);
1536 return 2;
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001537}
1538
Guido van Rossum9bfef441993-03-29 10:43:31 +00001539static int
Fred Drake79912472000-07-09 04:06:11 +00001540instance_nonzero(PyInstanceObject *self)
Guido van Rossum04691fc1992-08-12 15:35:34 +00001541{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001542 PyObject *func, *res;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001543 long outcome;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001544 static PyObject *nonzerostr;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001545
Guido van Rossum2878a691996-08-09 20:53:24 +00001546 if (nonzerostr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +00001547 nonzerostr = PyString_InternFromString("__nonzero__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001548 if ((func = instance_getattr(self, nonzerostr)) == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001549 PyErr_Clear();
Guido van Rossum2878a691996-08-09 20:53:24 +00001550 if (lenstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +00001551 lenstr = PyString_InternFromString("__len__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001552 if ((func = instance_getattr(self, lenstr)) == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001553 PyErr_Clear();
Guido van Rossum04691fc1992-08-12 15:35:34 +00001554 /* Fall back to the default behavior:
1555 all instances are nonzero */
1556 return 1;
1557 }
1558 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001559 res = PyEval_CallObject(func, (PyObject *)NULL);
1560 Py_DECREF(func);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001561 if (res == NULL)
1562 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001563 if (!PyInt_Check(res)) {
1564 Py_DECREF(res);
1565 PyErr_SetString(PyExc_TypeError,
1566 "__nonzero__ should return an int");
Guido van Rossum04691fc1992-08-12 15:35:34 +00001567 return -1;
1568 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001569 outcome = PyInt_AsLong(res);
1570 Py_DECREF(res);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001571 if (outcome < 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001572 PyErr_SetString(PyExc_ValueError,
1573 "__nonzero__ should return >= 0");
Guido van Rossum04691fc1992-08-12 15:35:34 +00001574 return -1;
1575 }
1576 return outcome > 0;
1577}
1578
1579UNARY(instance_invert, "__invert__")
Guido van Rossum1899c2e1992-09-12 11:09:23 +00001580UNARY(instance_int, "__int__")
1581UNARY(instance_long, "__long__")
1582UNARY(instance_float, "__float__")
1583UNARY(instance_oct, "__oct__")
1584UNARY(instance_hex, "__hex__")
1585
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001586static PyObject *
1587bin_power(PyObject *v, PyObject *w)
1588{
1589 return PyNumber_Power(v, w, Py_None);
1590}
1591
Guido van Rossum03093a21994-09-28 15:51:32 +00001592/* This version is for ternary calls only (z != None) */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001593static PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001594instance_pow(PyObject *v, PyObject *w, PyObject *z)
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001595{
1596 if (z == Py_None) {
1597 return do_binop(v, w, "__pow__", "__rpow__", bin_power);
Guido van Rossum03093a21994-09-28 15:51:32 +00001598 }
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001599 else {
1600 PyObject *func;
1601 PyObject *args;
1602 PyObject *result;
1603
1604 /* XXX Doesn't do coercions... */
1605 func = PyObject_GetAttrString(v, "__pow__");
1606 if (func == NULL)
1607 return NULL;
1608 args = Py_BuildValue("(OO)", w, z);
1609 if (args == NULL) {
1610 Py_DECREF(func);
1611 return NULL;
1612 }
1613 result = PyEval_CallObject(func, args);
1614 Py_DECREF(func);
1615 Py_DECREF(args);
1616 return result;
1617 }
Guido van Rossum03093a21994-09-28 15:51:32 +00001618}
1619
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001620static PyObject *
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001621bin_inplace_power(PyObject *v, PyObject *w)
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001622{
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001623 return PyNumber_InPlacePower(v, w, Py_None);
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001624}
1625
1626
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001627static PyObject *
1628instance_ipow(PyObject *v, PyObject *w, PyObject *z)
1629{
1630 if (z == Py_None) {
1631 return do_binop_inplace(v, w, "__ipow__", "__pow__",
1632 "__rpow__", bin_inplace_power);
1633 }
1634 else {
1635 /* XXX Doesn't do coercions... */
1636 PyObject *func;
1637 PyObject *args;
1638 PyObject *result;
1639
1640 func = PyObject_GetAttrString(v, "__ipow__");
1641 if (func == NULL) {
1642 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
1643 return NULL;
1644 PyErr_Clear();
1645 return instance_pow(v, w, z);
1646 }
1647 args = Py_BuildValue("(OO)", w, z);
1648 if (args == NULL) {
1649 Py_DECREF(func);
1650 return NULL;
1651 }
1652 result = PyEval_CallObject(func, args);
1653 Py_DECREF(func);
1654 Py_DECREF(args);
1655 return result;
1656 }
1657}
1658
1659
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001660/* Map rich comparison operators to their __xx__ namesakes */
1661static char *name_op[] = {
1662 "__lt__",
1663 "__le__",
1664 "__eq__",
1665 "__ne__",
1666 "__gt__",
1667 "__ge__",
1668};
1669
1670static PyObject *
1671half_richcompare(PyObject *v, PyObject *w, int op)
1672{
1673 PyObject *name;
1674 PyObject *method;
1675 PyObject *args;
1676 PyObject *res;
1677
1678 assert(PyInstance_Check(v));
1679
1680 name = PyString_InternFromString(name_op[op]);
1681 if (name == NULL)
1682 return NULL;
1683
1684 method = PyObject_GetAttr(v, name);
1685 Py_DECREF(name);
1686 if (method == NULL) {
1687 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
1688 return NULL;
1689 PyErr_Clear();
1690 res = Py_NotImplemented;
1691 Py_INCREF(res);
1692 return res;
1693 }
1694
1695 args = Py_BuildValue("(O)", w);
1696 if (args == NULL) {
1697 Py_DECREF(method);
1698 return NULL;
1699 }
1700
1701 res = PyEval_CallObject(method, args);
1702 Py_DECREF(args);
1703 Py_DECREF(method);
1704
1705 return res;
1706}
1707
1708/* Map rich comparison operators to their swapped version, e.g. LT --> GT */
1709static int swapped_op[] = {Py_GT, Py_GE, Py_EQ, Py_NE, Py_LT, Py_LE};
1710
1711static PyObject *
1712instance_richcompare(PyObject *v, PyObject *w, int op)
1713{
1714 PyObject *res;
1715
1716 if (PyInstance_Check(v)) {
1717 res = half_richcompare(v, w, op);
1718 if (res != Py_NotImplemented)
1719 return res;
1720 Py_DECREF(res);
1721 }
1722
1723 if (PyInstance_Check(w)) {
1724 res = half_richcompare(w, v, swapped_op[op]);
1725 if (res != Py_NotImplemented)
1726 return res;
1727 Py_DECREF(res);
1728 }
1729
1730 Py_INCREF(Py_NotImplemented);
1731 return Py_NotImplemented;
1732}
1733
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001734
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00001735/* Get the iterator */
1736static PyObject *
1737instance_getiter(PyInstanceObject *self)
1738{
1739 PyObject *func;
1740
1741 if (iterstr == NULL)
1742 iterstr = PyString_InternFromString("__iter__");
1743 if (getitemstr == NULL)
1744 getitemstr = PyString_InternFromString("__getitem__");
1745
1746 if ((func = instance_getattr(self, iterstr)) != NULL) {
1747 PyObject *res = PyEval_CallObject(func, (PyObject *)NULL);
1748 Py_DECREF(func);
Guido van Rossum213c7a62001-04-23 14:08:49 +00001749 if (res != NULL && !PyIter_Check(res)) {
1750 PyErr_Format(PyExc_TypeError,
1751 "__iter__ returned non-iterator "
1752 "of type '%.100s'",
1753 res->ob_type->tp_name);
1754 Py_DECREF(res);
1755 res = NULL;
1756 }
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00001757 return res;
1758 }
1759 PyErr_Clear();
1760 if ((func = instance_getattr(self, getitemstr)) == NULL) {
1761 PyErr_SetString(PyExc_TypeError, "iter() of non-sequence");
1762 return NULL;
1763 }
1764 Py_DECREF(func);
Guido van Rossum213c7a62001-04-23 14:08:49 +00001765 return PySeqIter_New((PyObject *)self);
1766}
1767
1768
1769/* Call the iterator's next */
1770static PyObject *
1771instance_iternext(PyInstanceObject *self)
1772{
1773 PyObject *func;
1774
1775 if (nextstr == NULL)
1776 nextstr = PyString_InternFromString("next");
1777
1778 if ((func = instance_getattr(self, nextstr)) != NULL) {
1779 PyObject *res = PyEval_CallObject(func, (PyObject *)NULL);
1780 Py_DECREF(func);
1781 if (res != NULL) {
1782 return res;
1783 }
1784 if (PyErr_ExceptionMatches(PyExc_StopIteration)) {
1785 PyErr_Clear();
1786 return NULL;
1787 }
1788 return NULL;
1789 }
1790 PyErr_SetString(PyExc_TypeError, "instance has no next() method");
1791 return NULL;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00001792}
1793
1794
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001795static PyNumberMethods instance_as_number = {
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001796 (binaryfunc)instance_add, /* nb_add */
1797 (binaryfunc)instance_sub, /* nb_subtract */
1798 (binaryfunc)instance_mul, /* nb_multiply */
1799 (binaryfunc)instance_div, /* nb_divide */
1800 (binaryfunc)instance_mod, /* nb_remainder */
1801 (binaryfunc)instance_divmod, /* nb_divmod */
1802 (ternaryfunc)instance_pow, /* nb_power */
1803 (unaryfunc)instance_neg, /* nb_negative */
1804 (unaryfunc)instance_pos, /* nb_positive */
1805 (unaryfunc)instance_abs, /* nb_absolute */
1806 (inquiry)instance_nonzero, /* nb_nonzero */
1807 (unaryfunc)instance_invert, /* nb_invert */
1808 (binaryfunc)instance_lshift, /* nb_lshift */
1809 (binaryfunc)instance_rshift, /* nb_rshift */
1810 (binaryfunc)instance_and, /* nb_and */
1811 (binaryfunc)instance_xor, /* nb_xor */
1812 (binaryfunc)instance_or, /* nb_or */
1813 (coercion)instance_coerce, /* nb_coerce */
1814 (unaryfunc)instance_int, /* nb_int */
1815 (unaryfunc)instance_long, /* nb_long */
1816 (unaryfunc)instance_float, /* nb_float */
1817 (unaryfunc)instance_oct, /* nb_oct */
1818 (unaryfunc)instance_hex, /* nb_hex */
1819 (binaryfunc)instance_iadd, /* nb_inplace_add */
1820 (binaryfunc)instance_isub, /* nb_inplace_subtract */
1821 (binaryfunc)instance_imul, /* nb_inplace_multiply */
1822 (binaryfunc)instance_idiv, /* nb_inplace_divide */
1823 (binaryfunc)instance_imod, /* nb_inplace_remainder */
1824 (ternaryfunc)instance_ipow, /* nb_inplace_power */
1825 (binaryfunc)instance_ilshift, /* nb_inplace_lshift */
1826 (binaryfunc)instance_irshift, /* nb_inplace_rshift */
1827 (binaryfunc)instance_iand, /* nb_inplace_and */
1828 (binaryfunc)instance_ixor, /* nb_inplace_xor */
1829 (binaryfunc)instance_ior, /* nb_inplace_or */
Guido van Rossum04691fc1992-08-12 15:35:34 +00001830};
1831
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001832PyTypeObject PyInstance_Type = {
1833 PyObject_HEAD_INIT(&PyType_Type)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001834 0,
Guido van Rossum21ed88c1991-04-04 10:42:10 +00001835 "instance",
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001836 sizeof(PyInstanceObject) + PyGC_HEAD_SIZE,
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001837 0,
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001838 (destructor)instance_dealloc, /* tp_dealloc */
1839 0, /* tp_print */
1840 0, /* tp_getattr */
1841 0, /* tp_setattr */
1842 instance_compare, /* tp_compare */
1843 (reprfunc)instance_repr, /* tp_repr */
1844 &instance_as_number, /* tp_as_number */
1845 &instance_as_sequence, /* tp_as_sequence */
1846 &instance_as_mapping, /* tp_as_mapping */
1847 (hashfunc)instance_hash, /* tp_hash */
1848 0, /* tp_call */
Guido van Rossum82c690f2001-04-30 14:39:18 +00001849 (reprfunc)instance_str, /* tp_str */
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001850 (getattrofunc)instance_getattr, /* tp_getattro */
1851 (setattrofunc)instance_setattr, /* tp_setattro */
1852 0, /* tp_as_buffer */
1853 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_GC | Py_TPFLAGS_CHECKTYPES,/*tp_flags*/
1854 0, /* tp_doc */
Jeremy Hylton8caad492000-06-23 14:18:11 +00001855 (traverseproc)instance_traverse, /* tp_traverse */
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001856 0, /* tp_clear */
1857 instance_richcompare, /* tp_richcompare */
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00001858 offsetof(PyInstanceObject, in_weakreflist), /* tp_weaklistoffset */
1859 (getiterfunc)instance_getiter, /* tp_iter */
Guido van Rossum213c7a62001-04-23 14:08:49 +00001860 (iternextfunc)instance_iternext, /* tp_iternext */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001861};
1862
1863
Guido van Rossum81daa321993-05-20 14:24:46 +00001864/* Instance method objects are used for two purposes:
1865 (a) as bound instance methods (returned by instancename.methodname)
1866 (b) as unbound methods (returned by ClassName.methodname)
1867 In case (b), im_self is NULL
1868*/
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001869
Guido van Rossuma0d349f1997-08-05 02:06:53 +00001870static PyMethodObject *free_list;
1871
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001872PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001873PyMethod_New(PyObject *func, PyObject *self, PyObject *class)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001874{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001875 register PyMethodObject *im;
Guido van Rossum7859f871998-07-08 14:58:16 +00001876 if (!PyCallable_Check(func)) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001877 PyErr_BadInternalCall();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001878 return NULL;
1879 }
Guido van Rossuma0d349f1997-08-05 02:06:53 +00001880 im = free_list;
1881 if (im != NULL) {
1882 free_list = (PyMethodObject *)(im->im_self);
Guido van Rossumb18618d2000-05-03 23:44:39 +00001883 PyObject_INIT(im, &PyMethod_Type);
Guido van Rossuma0d349f1997-08-05 02:06:53 +00001884 }
1885 else {
1886 im = PyObject_NEW(PyMethodObject, &PyMethod_Type);
1887 if (im == NULL)
1888 return NULL;
1889 }
Fred Drakedb81e8d2001-03-23 04:19:27 +00001890 im->im_weakreflist = NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001891 Py_INCREF(func);
Guido van Rossume8122f11991-05-05 20:03:07 +00001892 im->im_func = func;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001893 Py_XINCREF(self);
Guido van Rossume8122f11991-05-05 20:03:07 +00001894 im->im_self = self;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001895 Py_INCREF(class);
Guido van Rossum81daa321993-05-20 14:24:46 +00001896 im->im_class = class;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001897 PyObject_GC_Init(im);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001898 return (PyObject *)im;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001899}
1900
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001901PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001902PyMethod_Function(register PyObject *im)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001903{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001904 if (!PyMethod_Check(im)) {
1905 PyErr_BadInternalCall();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001906 return NULL;
1907 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001908 return ((PyMethodObject *)im)->im_func;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001909}
1910
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001911PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001912PyMethod_Self(register PyObject *im)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001913{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001914 if (!PyMethod_Check(im)) {
1915 PyErr_BadInternalCall();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001916 return NULL;
1917 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001918 return ((PyMethodObject *)im)->im_self;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001919}
1920
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001921PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001922PyMethod_Class(register PyObject *im)
Guido van Rossum81daa321993-05-20 14:24:46 +00001923{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001924 if (!PyMethod_Check(im)) {
1925 PyErr_BadInternalCall();
Guido van Rossum81daa321993-05-20 14:24:46 +00001926 return NULL;
1927 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001928 return ((PyMethodObject *)im)->im_class;
Guido van Rossum81daa321993-05-20 14:24:46 +00001929}
1930
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001931/* Class method methods */
1932
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001933#define OFF(x) offsetof(PyMethodObject, x)
Guido van Rossum3f5da241990-12-20 15:06:42 +00001934
Guido van Rossume8122f11991-05-05 20:03:07 +00001935static struct memberlist instancemethod_memberlist[] = {
1936 {"im_func", T_OBJECT, OFF(im_func)},
1937 {"im_self", T_OBJECT, OFF(im_self)},
Guido van Rossum81daa321993-05-20 14:24:46 +00001938 {"im_class", T_OBJECT, OFF(im_class)},
Guido van Rossum6dabc981996-05-14 21:54:20 +00001939 /* Dummies that are not handled by getattr() except for __members__ */
1940 {"__doc__", T_INT, 0},
1941 {"__name__", T_INT, 0},
Barry Warsawd6a9e842001-01-15 20:40:19 +00001942 {"__dict__", T_OBJECT, 0},
Guido van Rossum3f5da241990-12-20 15:06:42 +00001943 {NULL} /* Sentinel */
1944};
1945
Barry Warsawd6a9e842001-01-15 20:40:19 +00001946static int
1947instancemethod_setattro(register PyMethodObject *im, PyObject *name,
1948 PyObject *v)
Guido van Rossum3f5da241990-12-20 15:06:42 +00001949{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001950 char *sname = PyString_AsString(name);
Barry Warsawd6a9e842001-01-15 20:40:19 +00001951
Barry Warsaw4f9b13b2001-02-26 18:09:15 +00001952 PyErr_Format(PyExc_TypeError, "read-only attribute: %s", sname);
1953 return -1;
Barry Warsawd6a9e842001-01-15 20:40:19 +00001954}
1955
1956
1957static PyObject *
1958instancemethod_getattro(register PyMethodObject *im, PyObject *name)
1959{
1960 PyObject *rtn;
1961 char *sname = PyString_AsString(name);
Guido van Rossum2878a691996-08-09 20:53:24 +00001962 if (sname[0] == '_') {
Guido van Rossum7859f871998-07-08 14:58:16 +00001963 /* Inherit __name__ and __doc__ from the callable object
1964 implementing the method */
1965 if (strcmp(sname, "__name__") == 0 ||
1966 strcmp(sname, "__doc__") == 0)
1967 return PyObject_GetAttr(im->im_func, name);
Guido van Rossum6dabc981996-05-14 21:54:20 +00001968 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001969 if (PyEval_GetRestricted()) {
1970 PyErr_SetString(PyExc_RuntimeError,
1971 "instance-method attributes not accessible in restricted mode");
Guido van Rossum10393b11995-01-10 10:39:49 +00001972 return NULL;
1973 }
Barry Warsawd6a9e842001-01-15 20:40:19 +00001974 if (sname[0] == '_' && strcmp(sname, "__dict__") == 0)
1975 return PyObject_GetAttr(im->im_func, name);
1976
1977 rtn = PyMember_Get((char *)im, instancemethod_memberlist, sname);
1978 if (rtn == NULL && PyErr_ExceptionMatches(PyExc_AttributeError)) {
1979 PyErr_Clear();
1980 rtn = PyObject_GetAttr(im->im_func, name);
1981 }
1982 return rtn;
Guido van Rossum3f5da241990-12-20 15:06:42 +00001983}
1984
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001985static void
Fred Drake79912472000-07-09 04:06:11 +00001986instancemethod_dealloc(register PyMethodObject *im)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001987{
Fred Drakedb81e8d2001-03-23 04:19:27 +00001988 PyObject_ClearWeakRefs((PyObject *)im);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001989 PyObject_GC_Fini(im);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001990 Py_DECREF(im->im_func);
1991 Py_XDECREF(im->im_self);
1992 Py_DECREF(im->im_class);
Guido van Rossuma0d349f1997-08-05 02:06:53 +00001993 im->im_self = (PyObject *)free_list;
1994 free_list = im;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001995}
1996
Guido van Rossumebc8c511992-09-03 20:39:51 +00001997static int
Fred Drake79912472000-07-09 04:06:11 +00001998instancemethod_compare(PyMethodObject *a, PyMethodObject *b)
Guido van Rossumebc8c511992-09-03 20:39:51 +00001999{
Guido van Rossume9df7271995-04-06 14:46:51 +00002000 if (a->im_self != b->im_self)
2001 return (a->im_self < b->im_self) ? -1 : 1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00002002 return PyObject_Compare(a->im_func, b->im_func);
Guido van Rossumebc8c511992-09-03 20:39:51 +00002003}
2004
Guido van Rossumc0b618a1997-05-02 03:12:38 +00002005static PyObject *
Fred Drake79912472000-07-09 04:06:11 +00002006instancemethod_repr(PyMethodObject *a)
Guido van Rossum25831651993-05-19 14:50:45 +00002007{
2008 char buf[240];
Guido van Rossumc0b618a1997-05-02 03:12:38 +00002009 PyInstanceObject *self = (PyInstanceObject *)(a->im_self);
Guido van Rossum7859f871998-07-08 14:58:16 +00002010 PyObject *func = a->im_func;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00002011 PyClassObject *class = (PyClassObject *)(a->im_class);
2012 PyObject *fclassname, *iclassname, *funcname;
Guido van Rossum81daa321993-05-20 14:24:46 +00002013 char *fcname, *icname, *fname;
2014 fclassname = class->cl_name;
Guido van Rossum7859f871998-07-08 14:58:16 +00002015 if (PyFunction_Check(func)) {
2016 funcname = ((PyFunctionObject *)func)->func_name;
2017 Py_INCREF(funcname);
2018 }
2019 else {
2020 funcname = PyObject_GetAttrString(func,"__name__");
2021 if (funcname == NULL)
2022 PyErr_Clear();
2023 }
2024 if (funcname != NULL && PyString_Check(funcname))
2025 fname = PyString_AS_STRING(funcname);
2026 else
2027 fname = "?";
Guido van Rossumc0b618a1997-05-02 03:12:38 +00002028 if (fclassname != NULL && PyString_Check(fclassname))
2029 fcname = PyString_AsString(fclassname);
Guido van Rossum25831651993-05-19 14:50:45 +00002030 else
Guido van Rossum81daa321993-05-20 14:24:46 +00002031 fcname = "?";
Guido van Rossum81daa321993-05-20 14:24:46 +00002032 if (self == NULL)
2033 sprintf(buf, "<unbound method %.100s.%.100s>", fcname, fname);
2034 else {
2035 iclassname = self->in_class->cl_name;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00002036 if (iclassname != NULL && PyString_Check(iclassname))
2037 icname = PyString_AsString(iclassname);
Guido van Rossum81daa321993-05-20 14:24:46 +00002038 else
2039 icname = "?";
Fred Drakea44d3532000-06-30 15:01:00 +00002040 sprintf(buf, "<method %.60s.%.60s of %.60s instance at %p>",
2041 fcname, fname, icname, self);
Guido van Rossum81daa321993-05-20 14:24:46 +00002042 }
Guido van Rossum42636dc1999-10-11 14:03:12 +00002043 Py_XDECREF(funcname);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00002044 return PyString_FromString(buf);
Guido van Rossum25831651993-05-19 14:50:45 +00002045}
2046
Guido van Rossum9bfef441993-03-29 10:43:31 +00002047static long
Fred Drake79912472000-07-09 04:06:11 +00002048instancemethod_hash(PyMethodObject *a)
Guido van Rossum9bfef441993-03-29 10:43:31 +00002049{
2050 long x, y;
Guido van Rossum81daa321993-05-20 14:24:46 +00002051 if (a->im_self == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00002052 x = PyObject_Hash(Py_None);
Guido van Rossum81daa321993-05-20 14:24:46 +00002053 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +00002054 x = PyObject_Hash(a->im_self);
Guido van Rossum9bfef441993-03-29 10:43:31 +00002055 if (x == -1)
2056 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00002057 y = PyObject_Hash(a->im_func);
Guido van Rossum9bfef441993-03-29 10:43:31 +00002058 if (y == -1)
2059 return -1;
2060 return x ^ y;
2061}
2062
Jeremy Hylton8caad492000-06-23 14:18:11 +00002063static int
2064instancemethod_traverse(PyMethodObject *im, visitproc visit, void *arg)
2065{
2066 int err;
2067 if (im->im_func) {
2068 err = visit(im->im_func, arg);
2069 if (err)
2070 return err;
2071 }
2072 if (im->im_self) {
2073 err = visit(im->im_self, arg);
2074 if (err)
2075 return err;
2076 }
2077 if (im->im_class) {
2078 err = visit(im->im_class, arg);
2079 if (err)
2080 return err;
2081 }
Jeremy Hyltond22162b2000-06-23 17:14:56 +00002082 return 0;
Jeremy Hylton8caad492000-06-23 14:18:11 +00002083}
2084
Guido van Rossumc0b618a1997-05-02 03:12:38 +00002085PyTypeObject PyMethod_Type = {
2086 PyObject_HEAD_INIT(&PyType_Type)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002087 0,
Guido van Rossum569fce71991-04-16 08:38:43 +00002088 "instance method",
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00002089 sizeof(PyMethodObject) + PyGC_HEAD_SIZE,
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002090 0,
Guido van Rossum8998b4f2001-01-17 15:28:20 +00002091 (destructor)instancemethod_dealloc, /* tp_dealloc */
2092 0, /* tp_print */
2093 0, /* tp_getattr */
2094 0, /* tp_setattr */
2095 (cmpfunc)instancemethod_compare, /* tp_compare */
2096 (reprfunc)instancemethod_repr, /* tp_repr */
2097 0, /* tp_as_number */
2098 0, /* tp_as_sequence */
2099 0, /* tp_as_mapping */
2100 (hashfunc)instancemethod_hash, /* tp_hash */
2101 0, /* tp_call */
2102 0, /* tp_str */
2103 (getattrofunc)instancemethod_getattro, /* tp_getattro */
2104 (setattrofunc)instancemethod_setattro, /* tp_setattro */
2105 0, /* tp_as_buffer */
Fred Drake4dcb85b2001-05-03 16:04:13 +00002106 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_GC, /* tp_flags */
Guido van Rossum8998b4f2001-01-17 15:28:20 +00002107 0, /* tp_doc */
Jeremy Hylton8caad492000-06-23 14:18:11 +00002108 (traverseproc)instancemethod_traverse, /* tp_traverse */
Fred Drakedb81e8d2001-03-23 04:19:27 +00002109 0, /* tp_clear */
2110 0, /* tp_richcompare */
2111 offsetof(PyMethodObject, im_weakreflist) /* tp_weaklistoffset */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002112};
Guido van Rossuma0d349f1997-08-05 02:06:53 +00002113
2114/* Clear out the free list */
2115
2116void
Fred Drake79912472000-07-09 04:06:11 +00002117PyMethod_Fini(void)
Guido van Rossuma0d349f1997-08-05 02:06:53 +00002118{
2119 while (free_list) {
Guido van Rossumb18618d2000-05-03 23:44:39 +00002120 PyMethodObject *im = free_list;
2121 free_list = (PyMethodObject *)(im->im_self);
Guido van Rossum4cc6ac72000-07-01 01:00:38 +00002122 im = (PyMethodObject *) PyObject_AS_GC(im);
Guido van Rossumb18618d2000-05-03 23:44:39 +00002123 PyObject_DEL(im);
Guido van Rossuma0d349f1997-08-05 02:06:53 +00002124 }
2125}