blob: bb17df174a0756af485dc67874fb83f54383d43d [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 }
456 Py_INCREF(klass);
457 inst->in_class = (PyClassObject *)klass;
458 inst->in_dict = dict;
459 PyObject_GC_Init(inst);
460 return (PyObject *)inst;
461}
462
463PyObject *
464PyInstance_New(PyObject *klass, PyObject *arg, PyObject *kw)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000465{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000466 register PyInstanceObject *inst;
467 PyObject *init;
468 static PyObject *initstr;
Fred Drake5cc2c8c2001-01-28 03:53:08 +0000469
470 inst = (PyInstanceObject *) PyInstance_NewRaw(klass, NULL);
Guido van Rossume8122f11991-05-05 20:03:07 +0000471 if (inst == NULL)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000472 return NULL;
Guido van Rossum2878a691996-08-09 20:53:24 +0000473 if (initstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000474 initstr = PyString_InternFromString("__init__");
Jeremy Hylton9e392e22000-04-26 20:39:20 +0000475 init = instance_getattr2(inst, initstr);
Guido van Rossum25831651993-05-19 14:50:45 +0000476 if (init == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000477 if ((arg != NULL && (!PyTuple_Check(arg) ||
478 PyTuple_Size(arg) != 0))
479 || (kw != NULL && (!PyDict_Check(kw) ||
480 PyDict_Size(kw) != 0))) {
481 PyErr_SetString(PyExc_TypeError,
Guido van Rossuma83f2701995-07-26 18:07:32 +0000482 "this constructor takes no arguments");
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000483 Py_DECREF(inst);
Guido van Rossum25831651993-05-19 14:50:45 +0000484 inst = NULL;
485 }
486 }
487 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000488 PyObject *res = PyEval_CallObjectWithKeywords(init, arg, kw);
489 Py_DECREF(init);
Guido van Rossum25831651993-05-19 14:50:45 +0000490 if (res == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000491 Py_DECREF(inst);
Guido van Rossum25831651993-05-19 14:50:45 +0000492 inst = NULL;
493 }
494 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000495 if (res != Py_None) {
496 PyErr_SetString(PyExc_TypeError,
Guido van Rossum25831651993-05-19 14:50:45 +0000497 "__init__() should return None");
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000498 Py_DECREF(inst);
Guido van Rossum25831651993-05-19 14:50:45 +0000499 inst = NULL;
500 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000501 Py_DECREF(res);
Guido van Rossum25831651993-05-19 14:50:45 +0000502 }
503 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000504 return (PyObject *)inst;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000505}
506
Guido van Rossum21ed88c1991-04-04 10:42:10 +0000507/* Instance methods */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000508
509static void
Fred Drake79912472000-07-09 04:06:11 +0000510instance_dealloc(register PyInstanceObject *inst)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000511{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000512 PyObject *error_type, *error_value, *error_traceback;
513 PyObject *del;
514 static PyObject *delstr;
Tim Peters6b184912000-09-17 14:40:17 +0000515#ifdef Py_REF_DEBUG
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000516 extern long _Py_RefTotal;
Skip Montanaro4ca150b2000-07-08 12:04:57 +0000517#endif
Fred Drake41deb1e2001-02-01 05:27:45 +0000518
519 if (!PyObject_ClearWeakRefs((PyObject *) inst))
520 return;
521
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 Rossum9bfef441993-03-29 10:43:31 +0000784static long
Fred Drake79912472000-07-09 04:06:11 +0000785instance_hash(PyInstanceObject *inst)
Guido van Rossum9bfef441993-03-29 10:43:31 +0000786{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000787 PyObject *func;
788 PyObject *res;
Guido van Rossum8a0c3451993-04-08 12:56:19 +0000789 long outcome;
Guido van Rossum65e8bd72001-01-18 23:46:31 +0000790 static PyObject *hashstr, *eqstr, *cmpstr;
Guido van Rossum9bfef441993-03-29 10:43:31 +0000791
Guido van Rossum2878a691996-08-09 20:53:24 +0000792 if (hashstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000793 hashstr = PyString_InternFromString("__hash__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000794 func = instance_getattr(inst, hashstr);
Guido van Rossum9bfef441993-03-29 10:43:31 +0000795 if (func == NULL) {
Guido van Rossum65e8bd72001-01-18 23:46:31 +0000796 /* If there is no __eq__ and no __cmp__ method, we hash on the
797 address. If an __eq__ or __cmp__ method exists, there must
798 be a __hash__. */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000799 PyErr_Clear();
Guido van Rossum65e8bd72001-01-18 23:46:31 +0000800 if (eqstr == NULL)
801 eqstr = PyString_InternFromString("__eq__");
802 func = instance_getattr(inst, eqstr);
Guido van Rossum8a0c3451993-04-08 12:56:19 +0000803 if (func == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000804 PyErr_Clear();
Guido van Rossum65e8bd72001-01-18 23:46:31 +0000805 if (cmpstr == NULL)
806 cmpstr = PyString_InternFromString("__cmp__");
807 func = instance_getattr(inst, cmpstr);
808 if (func == NULL) {
809 PyErr_Clear();
810 return _Py_HashPointer(inst);
811 }
Guido van Rossum8a0c3451993-04-08 12:56:19 +0000812 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000813 PyErr_SetString(PyExc_TypeError, "unhashable instance");
Guido van Rossum9bfef441993-03-29 10:43:31 +0000814 return -1;
815 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000816 res = PyEval_CallObject(func, (PyObject *)NULL);
817 Py_DECREF(func);
Guido van Rossum9bfef441993-03-29 10:43:31 +0000818 if (res == NULL)
819 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000820 if (PyInt_Check(res)) {
821 outcome = PyInt_AsLong(res);
Guido van Rossum9bfef441993-03-29 10:43:31 +0000822 if (outcome == -1)
823 outcome = -2;
824 }
825 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000826 PyErr_SetString(PyExc_TypeError,
827 "__hash__() should return an int");
Guido van Rossum9bfef441993-03-29 10:43:31 +0000828 outcome = -1;
829 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000830 Py_DECREF(res);
Guido van Rossum9bfef441993-03-29 10:43:31 +0000831 return outcome;
832}
833
Jeremy Hylton8caad492000-06-23 14:18:11 +0000834static int
835instance_traverse(PyInstanceObject *o, visitproc visit, void *arg)
836{
837 int err;
838 if (o->in_class) {
839 err = visit((PyObject *)(o->in_class), arg);
840 if (err)
841 return err;
842 }
843 if (o->in_dict) {
844 err = visit(o->in_dict, arg);
845 if (err)
846 return err;
847 }
Jeremy Hyltond22162b2000-06-23 17:14:56 +0000848 return 0;
Jeremy Hylton8caad492000-06-23 14:18:11 +0000849}
850
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000851static PyObject *getitemstr, *setitemstr, *delitemstr, *lenstr;
Guido van Rossum2878a691996-08-09 20:53:24 +0000852
Guido van Rossum9bfef441993-03-29 10:43:31 +0000853static int
Fred Drake79912472000-07-09 04:06:11 +0000854instance_length(PyInstanceObject *inst)
Guido van Rossum04691fc1992-08-12 15:35:34 +0000855{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000856 PyObject *func;
857 PyObject *res;
Guido van Rossum04691fc1992-08-12 15:35:34 +0000858 int outcome;
859
Guido van Rossum2878a691996-08-09 20:53:24 +0000860 if (lenstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000861 lenstr = PyString_InternFromString("__len__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000862 func = instance_getattr(inst, lenstr);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000863 if (func == NULL)
864 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000865 res = PyEval_CallObject(func, (PyObject *)NULL);
866 Py_DECREF(func);
Guido van Rossumd014ea61992-11-26 10:30:26 +0000867 if (res == NULL)
868 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000869 if (PyInt_Check(res)) {
870 outcome = PyInt_AsLong(res);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000871 if (outcome < 0)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000872 PyErr_SetString(PyExc_ValueError,
873 "__len__() should return >= 0");
Guido van Rossum04691fc1992-08-12 15:35:34 +0000874 }
875 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000876 PyErr_SetString(PyExc_TypeError,
877 "__len__() should return an int");
Guido van Rossum04691fc1992-08-12 15:35:34 +0000878 outcome = -1;
879 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000880 Py_DECREF(res);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000881 return outcome;
882}
883
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000884static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000885instance_subscript(PyInstanceObject *inst, PyObject *key)
Guido van Rossum04691fc1992-08-12 15:35:34 +0000886{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000887 PyObject *func;
888 PyObject *arg;
889 PyObject *res;
Guido van Rossum04691fc1992-08-12 15:35:34 +0000890
Guido van Rossum2878a691996-08-09 20:53:24 +0000891 if (getitemstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000892 getitemstr = PyString_InternFromString("__getitem__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000893 func = instance_getattr(inst, getitemstr);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000894 if (func == NULL)
895 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000896 arg = Py_BuildValue("(O)", key);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000897 if (arg == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000898 Py_DECREF(func);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000899 return NULL;
900 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000901 res = PyEval_CallObject(func, arg);
902 Py_DECREF(func);
903 Py_DECREF(arg);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000904 return res;
905}
906
Guido van Rossum9bfef441993-03-29 10:43:31 +0000907static int
Fred Drake79912472000-07-09 04:06:11 +0000908instance_ass_subscript(PyInstanceObject *inst, PyObject *key, PyObject *value)
Guido van Rossum04691fc1992-08-12 15:35:34 +0000909{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000910 PyObject *func;
911 PyObject *arg;
912 PyObject *res;
Guido van Rossum04691fc1992-08-12 15:35:34 +0000913
Guido van Rossum2878a691996-08-09 20:53:24 +0000914 if (value == NULL) {
915 if (delitemstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000916 delitemstr = PyString_InternFromString("__delitem__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000917 func = instance_getattr(inst, delitemstr);
918 }
919 else {
920 if (setitemstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000921 setitemstr = PyString_InternFromString("__setitem__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000922 func = instance_getattr(inst, setitemstr);
923 }
Guido van Rossum04691fc1992-08-12 15:35:34 +0000924 if (func == NULL)
925 return -1;
926 if (value == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000927 arg = Py_BuildValue("(O)", key);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000928 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000929 arg = Py_BuildValue("(OO)", key, value);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000930 if (arg == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000931 Py_DECREF(func);
Guido van Rossum6d946f91992-08-14 13:49:30 +0000932 return -1;
Guido van Rossum04691fc1992-08-12 15:35:34 +0000933 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000934 res = PyEval_CallObject(func, arg);
935 Py_DECREF(func);
936 Py_DECREF(arg);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000937 if (res == NULL)
938 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000939 Py_DECREF(res);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000940 return 0;
941}
942
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000943static PyMappingMethods instance_as_mapping = {
Guido van Rossum8998b4f2001-01-17 15:28:20 +0000944 (inquiry)instance_length, /* mp_length */
945 (binaryfunc)instance_subscript, /* mp_subscript */
946 (objobjargproc)instance_ass_subscript, /* mp_ass_subscript */
Guido van Rossum04691fc1992-08-12 15:35:34 +0000947};
948
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000949static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000950instance_item(PyInstanceObject *inst, int i)
Guido van Rossum04691fc1992-08-12 15:35:34 +0000951{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000952 PyObject *func, *arg, *res;
Guido van Rossum04691fc1992-08-12 15:35:34 +0000953
Guido van Rossum2878a691996-08-09 20:53:24 +0000954 if (getitemstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000955 getitemstr = PyString_InternFromString("__getitem__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000956 func = instance_getattr(inst, getitemstr);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000957 if (func == NULL)
958 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000959 arg = Py_BuildValue("(i)", i);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000960 if (arg == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000961 Py_DECREF(func);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000962 return NULL;
963 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000964 res = PyEval_CallObject(func, arg);
965 Py_DECREF(func);
966 Py_DECREF(arg);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000967 return res;
968}
969
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000970static PyObject *
Thomas Wouters1d75a792000-08-17 22:37:32 +0000971sliceobj_from_intint(int i, int j)
972{
973 PyObject *start, *end, *res;
974
975 start = PyInt_FromLong((long)i);
976 if (!start)
977 return NULL;
978
979 end = PyInt_FromLong((long)j);
980 if (!end) {
981 Py_DECREF(start);
982 return NULL;
983 }
984 res = PySlice_New(start, end, NULL);
985 Py_DECREF(start);
986 Py_DECREF(end);
987 return res;
988}
989
990
991static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000992instance_slice(PyInstanceObject *inst, int i, int j)
Guido van Rossum04691fc1992-08-12 15:35:34 +0000993{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000994 PyObject *func, *arg, *res;
995 static PyObject *getslicestr;
Guido van Rossum04691fc1992-08-12 15:35:34 +0000996
Guido van Rossum2878a691996-08-09 20:53:24 +0000997 if (getslicestr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000998 getslicestr = PyString_InternFromString("__getslice__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000999 func = instance_getattr(inst, getslicestr);
Thomas Wouters1d75a792000-08-17 22:37:32 +00001000
1001 if (func == NULL) {
1002 PyErr_Clear();
1003
1004 if (getitemstr == NULL)
1005 getitemstr = PyString_InternFromString("__getitem__");
1006 func = instance_getattr(inst, getitemstr);
1007 if (func == NULL)
1008 return NULL;
1009 arg = Py_BuildValue("(N)", sliceobj_from_intint(i, j));
1010 } else
1011 arg = Py_BuildValue("(ii)", i, j);
1012
Guido van Rossum04691fc1992-08-12 15:35:34 +00001013 if (arg == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001014 Py_DECREF(func);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001015 return NULL;
1016 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001017 res = PyEval_CallObject(func, arg);
1018 Py_DECREF(func);
1019 Py_DECREF(arg);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001020 return res;
1021}
1022
1023static int
Fred Drake79912472000-07-09 04:06:11 +00001024instance_ass_item(PyInstanceObject *inst, int i, PyObject *item)
Guido van Rossum04691fc1992-08-12 15:35:34 +00001025{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001026 PyObject *func, *arg, *res;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001027
Guido van Rossum2878a691996-08-09 20:53:24 +00001028 if (item == NULL) {
1029 if (delitemstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +00001030 delitemstr = PyString_InternFromString("__delitem__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001031 func = instance_getattr(inst, delitemstr);
1032 }
1033 else {
1034 if (setitemstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +00001035 setitemstr = PyString_InternFromString("__setitem__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001036 func = instance_getattr(inst, setitemstr);
1037 }
Guido van Rossum04691fc1992-08-12 15:35:34 +00001038 if (func == NULL)
Guido van Rossum6d946f91992-08-14 13:49:30 +00001039 return -1;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001040 if (item == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001041 arg = Py_BuildValue("i", i);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001042 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001043 arg = Py_BuildValue("(iO)", i, item);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001044 if (arg == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001045 Py_DECREF(func);
Guido van Rossum6d946f91992-08-14 13:49:30 +00001046 return -1;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001047 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001048 res = PyEval_CallObject(func, arg);
1049 Py_DECREF(func);
1050 Py_DECREF(arg);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001051 if (res == NULL)
1052 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001053 Py_DECREF(res);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001054 return 0;
1055}
1056
1057static int
Fred Drake79912472000-07-09 04:06:11 +00001058instance_ass_slice(PyInstanceObject *inst, int i, int j, PyObject *value)
Guido van Rossum04691fc1992-08-12 15:35:34 +00001059{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001060 PyObject *func, *arg, *res;
1061 static PyObject *setslicestr, *delslicestr;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001062
Guido van Rossum2878a691996-08-09 20:53:24 +00001063 if (value == NULL) {
1064 if (delslicestr == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001065 delslicestr =
1066 PyString_InternFromString("__delslice__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001067 func = instance_getattr(inst, delslicestr);
Thomas Wouters1d75a792000-08-17 22:37:32 +00001068 if (func == NULL) {
1069 PyErr_Clear();
1070 if (delitemstr == NULL)
1071 delitemstr =
1072 PyString_InternFromString("__delitem__");
1073 func = instance_getattr(inst, delitemstr);
1074 if (func == NULL)
1075 return -1;
1076
1077 arg = Py_BuildValue("(N)",
1078 sliceobj_from_intint(i, j));
1079 } else
1080 arg = Py_BuildValue("(ii)", i, j);
Guido van Rossum2878a691996-08-09 20:53:24 +00001081 }
1082 else {
1083 if (setslicestr == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001084 setslicestr =
1085 PyString_InternFromString("__setslice__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001086 func = instance_getattr(inst, setslicestr);
Thomas Wouters1d75a792000-08-17 22:37:32 +00001087 if (func == NULL) {
1088 PyErr_Clear();
1089 if (setitemstr == NULL)
1090 setitemstr =
1091 PyString_InternFromString("__setitem__");
1092 func = instance_getattr(inst, setitemstr);
1093 if (func == NULL)
1094 return -1;
1095
1096 arg = Py_BuildValue("(NO)",
1097 sliceobj_from_intint(i, j), value);
1098 } else
1099 arg = Py_BuildValue("(iiO)", i, j, value);
Guido van Rossum2878a691996-08-09 20:53:24 +00001100 }
Guido van Rossum04691fc1992-08-12 15:35:34 +00001101 if (arg == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001102 Py_DECREF(func);
Guido van Rossum6d946f91992-08-14 13:49:30 +00001103 return -1;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001104 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001105 res = PyEval_CallObject(func, arg);
1106 Py_DECREF(func);
1107 Py_DECREF(arg);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001108 if (res == NULL)
1109 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001110 Py_DECREF(res);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001111 return 0;
1112}
1113
Guido van Rossumee28c3a2000-02-28 15:03:15 +00001114static int instance_contains(PyInstanceObject *inst, PyObject *member)
1115{
1116 static PyObject *__contains__;
1117 PyObject *func, *arg, *res;
1118 int ret;
1119
1120 if(__contains__ == NULL) {
1121 __contains__ = PyString_InternFromString("__contains__");
1122 if(__contains__ == NULL)
1123 return -1;
1124 }
1125 func = instance_getattr(inst, __contains__);
1126 if(func == NULL) {
Thomas Wouters7e474022000-07-16 12:04:32 +00001127 /* fall back to previous behavior */
Guido van Rossumee28c3a2000-02-28 15:03:15 +00001128 int i, cmp_res;
1129
1130 if(!PyErr_ExceptionMatches(PyExc_AttributeError))
1131 return -1;
1132 PyErr_Clear();
1133 for(i=0;;i++) {
1134 PyObject *obj = instance_item(inst, i);
1135 int ret = 0;
1136
1137 if(obj == NULL) {
1138 if(!PyErr_ExceptionMatches(PyExc_IndexError))
1139 return -1;
1140 PyErr_Clear();
1141 return 0;
1142 }
1143 if(PyObject_Cmp(obj, member, &cmp_res) == -1)
1144 ret = -1;
1145 if(cmp_res == 0)
1146 ret = 1;
1147 Py_DECREF(obj);
1148 if(ret)
1149 return ret;
1150 }
1151 }
1152 arg = Py_BuildValue("(O)", member);
1153 if(arg == NULL) {
1154 Py_DECREF(func);
1155 return -1;
1156 }
1157 res = PyEval_CallObject(func, arg);
1158 Py_DECREF(func);
1159 Py_DECREF(arg);
1160 if(res == NULL)
1161 return -1;
1162 ret = PyObject_IsTrue(res);
1163 Py_DECREF(res);
1164 return ret;
1165}
1166
Fred Drake79912472000-07-09 04:06:11 +00001167static PySequenceMethods
1168instance_as_sequence = {
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001169 (inquiry)instance_length, /* sq_length */
1170 0, /* sq_concat */
1171 0, /* sq_repeat */
1172 (intargfunc)instance_item, /* sq_item */
1173 (intintargfunc)instance_slice, /* sq_slice */
1174 (intobjargproc)instance_ass_item, /* sq_ass_item */
1175 (intintobjargproc)instance_ass_slice, /* sq_ass_slice */
1176 (objobjproc)instance_contains, /* sq_contains */
Guido van Rossum04691fc1992-08-12 15:35:34 +00001177};
1178
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001179static PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001180generic_unary_op(PyInstanceObject *self, PyObject *methodname)
Guido van Rossum04691fc1992-08-12 15:35:34 +00001181{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001182 PyObject *func, *res;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001183
1184 if ((func = instance_getattr(self, methodname)) == NULL)
1185 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001186 res = PyEval_CallObject(func, (PyObject *)NULL);
1187 Py_DECREF(func);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001188 return res;
1189}
1190
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001191static PyObject *
1192generic_binary_op(PyObject *v, PyObject *w, char *opname)
Guido van Rossum03093a21994-09-28 15:51:32 +00001193{
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001194 PyObject *result;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001195 PyObject *args;
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001196 PyObject *func = PyObject_GetAttrString(v, opname);
Guido van Rossume7d444f1995-01-07 12:35:18 +00001197 if (func == NULL) {
Guido van Rossum617c1b01998-05-28 19:50:02 +00001198 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001199 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001200 PyErr_Clear();
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001201 Py_INCREF(Py_NotImplemented);
1202 return Py_NotImplemented;
Guido van Rossume7d444f1995-01-07 12:35:18 +00001203 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001204 args = Py_BuildValue("(O)", w);
Guido van Rossum03093a21994-09-28 15:51:32 +00001205 if (args == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001206 Py_DECREF(func);
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001207 return NULL;
Guido van Rossum03093a21994-09-28 15:51:32 +00001208 }
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001209 result = PyEval_CallObject(func, args);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001210 Py_DECREF(args);
1211 Py_DECREF(func);
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001212 return result;
1213}
1214
1215
1216static PyObject *coerce_obj;
1217
1218/* Try one half of a binary operator involving a class instance. */
1219static PyObject *
1220half_binop(PyObject *v, PyObject *w, char *opname, binaryfunc thisfunc,
1221 int swapped)
1222{
1223 PyObject *args;
1224 PyObject *coercefunc;
1225 PyObject *coerced = NULL;
1226 PyObject *v1;
1227 PyObject *result;
1228
1229 if (!PyInstance_Check(v)) {
1230 Py_INCREF(Py_NotImplemented);
1231 return Py_NotImplemented;
1232 }
1233
1234 if (coerce_obj == NULL) {
1235 coerce_obj = PyString_InternFromString("__coerce__");
1236 if (coerce_obj == NULL)
1237 return NULL;
1238 }
1239 coercefunc = PyObject_GetAttr(v, coerce_obj);
1240 if (coercefunc == NULL) {
1241 PyErr_Clear();
1242 return generic_binary_op(v, w, opname);
1243 }
1244
1245 args = Py_BuildValue("(O)", w);
1246 if (args == NULL) {
1247 return NULL;
1248 }
1249 coerced = PyEval_CallObject(coercefunc, args);
1250 Py_DECREF(args);
1251 Py_DECREF(coercefunc);
1252 if (coerced == NULL) {
1253 return NULL;
1254 }
1255 if (coerced == Py_None || coerced == Py_NotImplemented) {
1256 Py_DECREF(coerced);
1257 return generic_binary_op(v, w, opname);
1258 }
1259 if (!PyTuple_Check(coerced) || PyTuple_Size(coerced) != 2) {
1260 Py_DECREF(coerced);
1261 PyErr_SetString(PyExc_TypeError,
1262 "coercion should return None or 2-tuple");
1263 return NULL;
1264 }
1265 v1 = PyTuple_GetItem(coerced, 0);
1266 w = PyTuple_GetItem(coerced, 1);
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001267 if (v1->ob_type == v->ob_type && PyInstance_Check(v)) {
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001268 /* prevent recursion if __coerce__ returns self as the first
1269 * argument */
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001270 result = generic_binary_op(v1, w, opname);
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001271 } else {
1272 if (swapped)
1273 result = (thisfunc)(w, v1);
1274 else
1275 result = (thisfunc)(v1, w);
1276 }
1277 Py_DECREF(coerced);
1278 return result;
1279}
1280
1281/* Implement a binary operator involving at least one class instance. */
1282static PyObject *
1283do_binop(PyObject *v, PyObject *w, char *opname, char *ropname,
1284 binaryfunc thisfunc)
1285{
1286 PyObject *result = half_binop(v, w, opname, thisfunc, 0);
1287 if (result == Py_NotImplemented) {
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001288 Py_DECREF(result);
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001289 result = half_binop(w, v, ropname, thisfunc, 1);
1290 }
1291 return result;
1292}
1293
1294static PyObject *
1295do_binop_inplace(PyObject *v, PyObject *w, char *iopname, char *opname,
1296 char *ropname, binaryfunc thisfunc)
1297{
1298 PyObject *result = half_binop(v, w, iopname, thisfunc, 0);
1299 if (result == Py_NotImplemented) {
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001300 Py_DECREF(result);
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001301 result = do_binop(v, w, opname, ropname, thisfunc);
1302 }
1303 return result;
Guido van Rossum03093a21994-09-28 15:51:32 +00001304}
1305
Guido van Rossum879c5811995-01-10 15:24:06 +00001306static int
Fred Drake79912472000-07-09 04:06:11 +00001307instance_coerce(PyObject **pv, PyObject **pw)
Guido van Rossum879c5811995-01-10 15:24:06 +00001308{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001309 PyObject *v = *pv;
1310 PyObject *w = *pw;
Guido van Rossum3931df91997-11-18 19:23:07 +00001311 PyObject *coercefunc;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001312 PyObject *args;
1313 PyObject *coerced;
Guido van Rossum879c5811995-01-10 15:24:06 +00001314
Guido van Rossum2878a691996-08-09 20:53:24 +00001315 if (coerce_obj == NULL) {
Guido van Rossuma412d241997-01-18 07:59:12 +00001316 coerce_obj = PyString_InternFromString("__coerce__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001317 if (coerce_obj == NULL)
1318 return -1;
1319 }
Guido van Rossum3931df91997-11-18 19:23:07 +00001320 coercefunc = PyObject_GetAttr(v, coerce_obj);
1321 if (coercefunc == NULL) {
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001322 /* No __coerce__ method */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001323 PyErr_Clear();
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001324 return 1;
Guido van Rossum879c5811995-01-10 15:24:06 +00001325 }
1326 /* Has __coerce__ method: call it */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001327 args = Py_BuildValue("(O)", w);
Guido van Rossum879c5811995-01-10 15:24:06 +00001328 if (args == NULL) {
1329 return -1;
1330 }
Guido van Rossum3931df91997-11-18 19:23:07 +00001331 coerced = PyEval_CallObject(coercefunc, args);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001332 Py_DECREF(args);
Guido van Rossum3931df91997-11-18 19:23:07 +00001333 Py_DECREF(coercefunc);
Guido van Rossum879c5811995-01-10 15:24:06 +00001334 if (coerced == NULL) {
1335 /* __coerce__ call raised an exception */
1336 return -1;
1337 }
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001338 if (coerced == Py_None || coerced == Py_NotImplemented) {
Guido van Rossum879c5811995-01-10 15:24:06 +00001339 /* __coerce__ says "I can't do it" */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001340 Py_DECREF(coerced);
Guido van Rossum879c5811995-01-10 15:24:06 +00001341 return 1;
1342 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001343 if (!PyTuple_Check(coerced) || PyTuple_Size(coerced) != 2) {
Guido van Rossum879c5811995-01-10 15:24:06 +00001344 /* __coerce__ return value is malformed */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001345 Py_DECREF(coerced);
1346 PyErr_SetString(PyExc_TypeError,
Guido van Rossum879c5811995-01-10 15:24:06 +00001347 "coercion should return None or 2-tuple");
1348 return -1;
1349 }
1350 /* __coerce__ returned two new values */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001351 *pv = PyTuple_GetItem(coerced, 0);
1352 *pw = PyTuple_GetItem(coerced, 1);
1353 Py_INCREF(*pv);
1354 Py_INCREF(*pw);
1355 Py_DECREF(coerced);
Guido van Rossum879c5811995-01-10 15:24:06 +00001356 return 0;
1357}
1358
Guido van Rossum04691fc1992-08-12 15:35:34 +00001359#define UNARY(funcname, methodname) \
Thomas Woutersc3073522000-07-23 22:09:59 +00001360static PyObject *funcname(PyInstanceObject *self) { \
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001361 static PyObject *o; \
Guido van Rossuma412d241997-01-18 07:59:12 +00001362 if (o == NULL) o = PyString_InternFromString(methodname); \
Guido van Rossum2878a691996-08-09 20:53:24 +00001363 return generic_unary_op(self, o); \
Guido van Rossum04691fc1992-08-12 15:35:34 +00001364}
1365
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001366#define BINARY(f, m, n) \
1367static PyObject *f(PyObject *v, PyObject *w) { \
1368 return do_binop(v, w, "__" m "__", "__r" m "__", n); \
1369}
1370
1371#define BINARY_INPLACE(f, m, n) \
1372static PyObject *f(PyObject *v, PyObject *w) { \
1373 return do_binop_inplace(v, w, "__i" m "__", "__" m "__", \
1374 "__r" m "__", n); \
1375}
1376
Guido van Rossum04691fc1992-08-12 15:35:34 +00001377UNARY(instance_neg, "__neg__")
1378UNARY(instance_pos, "__pos__")
1379UNARY(instance_abs, "__abs__")
1380
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001381BINARY(instance_or, "or", PyNumber_Or)
1382BINARY(instance_and, "and", PyNumber_And)
1383BINARY(instance_xor, "xor", PyNumber_Xor)
1384BINARY(instance_lshift, "lshift", PyNumber_Lshift)
1385BINARY(instance_rshift, "rshift", PyNumber_Rshift)
1386BINARY(instance_add, "add", PyNumber_Add)
1387BINARY(instance_sub, "sub", PyNumber_Subtract)
1388BINARY(instance_mul, "mul", PyNumber_Multiply)
1389BINARY(instance_div, "div", PyNumber_Divide)
1390BINARY(instance_mod, "mod", PyNumber_Remainder)
1391BINARY(instance_divmod, "divmod", PyNumber_Divmod)
1392
1393BINARY_INPLACE(instance_ior, "or", PyNumber_InPlaceOr)
1394BINARY_INPLACE(instance_ixor, "xor", PyNumber_InPlaceXor)
1395BINARY_INPLACE(instance_iand, "and", PyNumber_InPlaceAnd)
1396BINARY_INPLACE(instance_ilshift, "lshift", PyNumber_InPlaceLshift)
1397BINARY_INPLACE(instance_irshift, "rshift", PyNumber_InPlaceRshift)
1398BINARY_INPLACE(instance_iadd, "add", PyNumber_InPlaceAdd)
1399BINARY_INPLACE(instance_isub, "sub", PyNumber_InPlaceSubtract)
1400BINARY_INPLACE(instance_imul, "mul", PyNumber_InPlaceMultiply)
1401BINARY_INPLACE(instance_idiv, "div", PyNumber_InPlaceDivide)
1402BINARY_INPLACE(instance_imod, "mod", PyNumber_InPlaceRemainder)
1403
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001404/* Try a 3-way comparison, returning an int; v is an instance. Return:
1405 -2 for an exception;
1406 -1 if v < w;
1407 0 if v == w;
1408 1 if v > w;
1409 2 if this particular 3-way comparison is not implemented or undefined.
1410*/
1411static int
1412half_cmp(PyObject *v, PyObject *w)
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001413{
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001414 static PyObject *cmp_obj;
1415 PyObject *args;
Guido van Rossum3202c6f2001-01-29 23:50:25 +00001416 PyObject *cmp_func;
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001417 PyObject *result;
1418 long l;
1419
1420 assert(PyInstance_Check(v));
1421
1422 if (cmp_obj == NULL) {
1423 cmp_obj = PyString_InternFromString("__cmp__");
1424 if (cmp_obj == NULL)
1425 return -2;
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001426 }
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001427
Guido van Rossum3202c6f2001-01-29 23:50:25 +00001428 cmp_func = PyObject_GetAttr(v, cmp_obj);
1429 if (cmp_func == NULL) {
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001430 PyErr_Clear();
1431 return 2;
1432 }
1433
1434 args = Py_BuildValue("(O)", w);
1435 if (args == NULL)
1436 return -2;
1437
Guido van Rossum3202c6f2001-01-29 23:50:25 +00001438 result = PyEval_CallObject(cmp_func, args);
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001439 Py_DECREF(args);
Guido van Rossum3202c6f2001-01-29 23:50:25 +00001440 Py_DECREF(cmp_func);
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001441
1442 if (result == NULL)
1443 return -2;
1444
1445 if (result == Py_NotImplemented) {
1446 Py_DECREF(result);
1447 return 2;
1448 }
1449
1450 l = PyInt_AsLong(result);
1451 Py_DECREF(result);
1452 if (l == -1 && PyErr_Occurred()) {
1453 PyErr_SetString(PyExc_TypeError,
1454 "comparison did not return an int");
1455 return -2;
1456 }
1457
1458 return l < 0 ? -1 : l > 0 ? 1 : 0;
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001459}
1460
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001461/* Try a 3-way comparison, returning an int; either v or w is an instance.
1462 We first try a coercion. Return:
1463 -2 for an exception;
1464 -1 if v < w;
1465 0 if v == w;
1466 1 if v > w;
1467 2 if this particular 3-way comparison is not implemented or undefined.
1468 THIS IS ONLY CALLED FROM object.c!
1469*/
1470static int
1471instance_compare(PyObject *v, PyObject *w)
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001472{
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001473 int c;
1474
1475 c = PyNumber_CoerceEx(&v, &w);
1476 if (c < 0)
1477 return -2;
1478 if (c == 0) {
1479 /* If neither is now an instance, use regular comparison */
1480 if (!PyInstance_Check(v) && !PyInstance_Check(w)) {
1481 c = PyObject_Compare(v, w);
1482 Py_DECREF(v);
1483 Py_DECREF(w);
1484 if (PyErr_Occurred())
1485 return -2;
1486 return c < 0 ? -1 : c > 0 ? 1 : 0;
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001487 }
1488 }
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001489 else {
1490 /* The coercion didn't do anything.
1491 Treat this the same as returning v and w unchanged. */
1492 Py_INCREF(v);
1493 Py_INCREF(w);
1494 }
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001495
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001496 if (PyInstance_Check(v)) {
1497 c = half_cmp(v, w);
1498 if (c <= 1) {
1499 Py_DECREF(v);
1500 Py_DECREF(w);
1501 return c;
1502 }
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001503 }
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001504 if (PyInstance_Check(w)) {
1505 c = half_cmp(w, v);
1506 if (c <= 1) {
1507 Py_DECREF(v);
1508 Py_DECREF(w);
1509 if (c >= -1)
1510 c = -c;
1511 return c;
1512 }
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001513 }
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001514 Py_DECREF(v);
1515 Py_DECREF(w);
1516 return 2;
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001517}
1518
Guido van Rossum9bfef441993-03-29 10:43:31 +00001519static int
Fred Drake79912472000-07-09 04:06:11 +00001520instance_nonzero(PyInstanceObject *self)
Guido van Rossum04691fc1992-08-12 15:35:34 +00001521{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001522 PyObject *func, *res;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001523 long outcome;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001524 static PyObject *nonzerostr;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001525
Guido van Rossum2878a691996-08-09 20:53:24 +00001526 if (nonzerostr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +00001527 nonzerostr = PyString_InternFromString("__nonzero__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001528 if ((func = instance_getattr(self, nonzerostr)) == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001529 PyErr_Clear();
Guido van Rossum2878a691996-08-09 20:53:24 +00001530 if (lenstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +00001531 lenstr = PyString_InternFromString("__len__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001532 if ((func = instance_getattr(self, lenstr)) == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001533 PyErr_Clear();
Guido van Rossum04691fc1992-08-12 15:35:34 +00001534 /* Fall back to the default behavior:
1535 all instances are nonzero */
1536 return 1;
1537 }
1538 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001539 res = PyEval_CallObject(func, (PyObject *)NULL);
1540 Py_DECREF(func);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001541 if (res == NULL)
1542 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001543 if (!PyInt_Check(res)) {
1544 Py_DECREF(res);
1545 PyErr_SetString(PyExc_TypeError,
1546 "__nonzero__ should return an int");
Guido van Rossum04691fc1992-08-12 15:35:34 +00001547 return -1;
1548 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001549 outcome = PyInt_AsLong(res);
1550 Py_DECREF(res);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001551 if (outcome < 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001552 PyErr_SetString(PyExc_ValueError,
1553 "__nonzero__ should return >= 0");
Guido van Rossum04691fc1992-08-12 15:35:34 +00001554 return -1;
1555 }
1556 return outcome > 0;
1557}
1558
1559UNARY(instance_invert, "__invert__")
Guido van Rossum1899c2e1992-09-12 11:09:23 +00001560UNARY(instance_int, "__int__")
1561UNARY(instance_long, "__long__")
1562UNARY(instance_float, "__float__")
1563UNARY(instance_oct, "__oct__")
1564UNARY(instance_hex, "__hex__")
1565
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001566static PyObject *
1567bin_power(PyObject *v, PyObject *w)
1568{
1569 return PyNumber_Power(v, w, Py_None);
1570}
1571
Guido van Rossum03093a21994-09-28 15:51:32 +00001572/* This version is for ternary calls only (z != None) */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001573static PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001574instance_pow(PyObject *v, PyObject *w, PyObject *z)
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001575{
1576 if (z == Py_None) {
1577 return do_binop(v, w, "__pow__", "__rpow__", bin_power);
Guido van Rossum03093a21994-09-28 15:51:32 +00001578 }
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001579 else {
1580 PyObject *func;
1581 PyObject *args;
1582 PyObject *result;
1583
1584 /* XXX Doesn't do coercions... */
1585 func = PyObject_GetAttrString(v, "__pow__");
1586 if (func == NULL)
1587 return NULL;
1588 args = Py_BuildValue("(OO)", w, z);
1589 if (args == NULL) {
1590 Py_DECREF(func);
1591 return NULL;
1592 }
1593 result = PyEval_CallObject(func, args);
1594 Py_DECREF(func);
1595 Py_DECREF(args);
1596 return result;
1597 }
Guido van Rossum03093a21994-09-28 15:51:32 +00001598}
1599
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001600static PyObject *
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001601bin_inplace_power(PyObject *v, PyObject *w)
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001602{
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001603 return PyNumber_InPlacePower(v, w, Py_None);
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001604}
1605
1606
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001607static PyObject *
1608instance_ipow(PyObject *v, PyObject *w, PyObject *z)
1609{
1610 if (z == Py_None) {
1611 return do_binop_inplace(v, w, "__ipow__", "__pow__",
1612 "__rpow__", bin_inplace_power);
1613 }
1614 else {
1615 /* XXX Doesn't do coercions... */
1616 PyObject *func;
1617 PyObject *args;
1618 PyObject *result;
1619
1620 func = PyObject_GetAttrString(v, "__ipow__");
1621 if (func == NULL) {
1622 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
1623 return NULL;
1624 PyErr_Clear();
1625 return instance_pow(v, w, z);
1626 }
1627 args = Py_BuildValue("(OO)", w, z);
1628 if (args == NULL) {
1629 Py_DECREF(func);
1630 return NULL;
1631 }
1632 result = PyEval_CallObject(func, args);
1633 Py_DECREF(func);
1634 Py_DECREF(args);
1635 return result;
1636 }
1637}
1638
1639
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001640/* Map rich comparison operators to their __xx__ namesakes */
1641static char *name_op[] = {
1642 "__lt__",
1643 "__le__",
1644 "__eq__",
1645 "__ne__",
1646 "__gt__",
1647 "__ge__",
1648};
1649
1650static PyObject *
1651half_richcompare(PyObject *v, PyObject *w, int op)
1652{
1653 PyObject *name;
1654 PyObject *method;
1655 PyObject *args;
1656 PyObject *res;
1657
1658 assert(PyInstance_Check(v));
1659
1660 name = PyString_InternFromString(name_op[op]);
1661 if (name == NULL)
1662 return NULL;
1663
1664 method = PyObject_GetAttr(v, name);
1665 Py_DECREF(name);
1666 if (method == NULL) {
1667 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
1668 return NULL;
1669 PyErr_Clear();
1670 res = Py_NotImplemented;
1671 Py_INCREF(res);
1672 return res;
1673 }
1674
1675 args = Py_BuildValue("(O)", w);
1676 if (args == NULL) {
1677 Py_DECREF(method);
1678 return NULL;
1679 }
1680
1681 res = PyEval_CallObject(method, args);
1682 Py_DECREF(args);
1683 Py_DECREF(method);
1684
1685 return res;
1686}
1687
1688/* Map rich comparison operators to their swapped version, e.g. LT --> GT */
1689static int swapped_op[] = {Py_GT, Py_GE, Py_EQ, Py_NE, Py_LT, Py_LE};
1690
1691static PyObject *
1692instance_richcompare(PyObject *v, PyObject *w, int op)
1693{
1694 PyObject *res;
1695
1696 if (PyInstance_Check(v)) {
1697 res = half_richcompare(v, w, op);
1698 if (res != Py_NotImplemented)
1699 return res;
1700 Py_DECREF(res);
1701 }
1702
1703 if (PyInstance_Check(w)) {
1704 res = half_richcompare(w, v, swapped_op[op]);
1705 if (res != Py_NotImplemented)
1706 return res;
1707 Py_DECREF(res);
1708 }
1709
1710 Py_INCREF(Py_NotImplemented);
1711 return Py_NotImplemented;
1712}
1713
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001714
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001715static PyNumberMethods instance_as_number = {
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001716 (binaryfunc)instance_add, /* nb_add */
1717 (binaryfunc)instance_sub, /* nb_subtract */
1718 (binaryfunc)instance_mul, /* nb_multiply */
1719 (binaryfunc)instance_div, /* nb_divide */
1720 (binaryfunc)instance_mod, /* nb_remainder */
1721 (binaryfunc)instance_divmod, /* nb_divmod */
1722 (ternaryfunc)instance_pow, /* nb_power */
1723 (unaryfunc)instance_neg, /* nb_negative */
1724 (unaryfunc)instance_pos, /* nb_positive */
1725 (unaryfunc)instance_abs, /* nb_absolute */
1726 (inquiry)instance_nonzero, /* nb_nonzero */
1727 (unaryfunc)instance_invert, /* nb_invert */
1728 (binaryfunc)instance_lshift, /* nb_lshift */
1729 (binaryfunc)instance_rshift, /* nb_rshift */
1730 (binaryfunc)instance_and, /* nb_and */
1731 (binaryfunc)instance_xor, /* nb_xor */
1732 (binaryfunc)instance_or, /* nb_or */
1733 (coercion)instance_coerce, /* nb_coerce */
1734 (unaryfunc)instance_int, /* nb_int */
1735 (unaryfunc)instance_long, /* nb_long */
1736 (unaryfunc)instance_float, /* nb_float */
1737 (unaryfunc)instance_oct, /* nb_oct */
1738 (unaryfunc)instance_hex, /* nb_hex */
1739 (binaryfunc)instance_iadd, /* nb_inplace_add */
1740 (binaryfunc)instance_isub, /* nb_inplace_subtract */
1741 (binaryfunc)instance_imul, /* nb_inplace_multiply */
1742 (binaryfunc)instance_idiv, /* nb_inplace_divide */
1743 (binaryfunc)instance_imod, /* nb_inplace_remainder */
1744 (ternaryfunc)instance_ipow, /* nb_inplace_power */
1745 (binaryfunc)instance_ilshift, /* nb_inplace_lshift */
1746 (binaryfunc)instance_irshift, /* nb_inplace_rshift */
1747 (binaryfunc)instance_iand, /* nb_inplace_and */
1748 (binaryfunc)instance_ixor, /* nb_inplace_xor */
1749 (binaryfunc)instance_ior, /* nb_inplace_or */
Guido van Rossum04691fc1992-08-12 15:35:34 +00001750};
1751
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001752PyTypeObject PyInstance_Type = {
1753 PyObject_HEAD_INIT(&PyType_Type)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001754 0,
Guido van Rossum21ed88c1991-04-04 10:42:10 +00001755 "instance",
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001756 sizeof(PyInstanceObject) + PyGC_HEAD_SIZE,
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001757 0,
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001758 (destructor)instance_dealloc, /* tp_dealloc */
1759 0, /* tp_print */
1760 0, /* tp_getattr */
1761 0, /* tp_setattr */
1762 instance_compare, /* tp_compare */
1763 (reprfunc)instance_repr, /* tp_repr */
1764 &instance_as_number, /* tp_as_number */
1765 &instance_as_sequence, /* tp_as_sequence */
1766 &instance_as_mapping, /* tp_as_mapping */
1767 (hashfunc)instance_hash, /* tp_hash */
1768 0, /* tp_call */
1769 0, /* tp_str */
1770 (getattrofunc)instance_getattr, /* tp_getattro */
1771 (setattrofunc)instance_setattr, /* tp_setattro */
1772 0, /* tp_as_buffer */
1773 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_GC | Py_TPFLAGS_CHECKTYPES,/*tp_flags*/
1774 0, /* tp_doc */
Jeremy Hylton8caad492000-06-23 14:18:11 +00001775 (traverseproc)instance_traverse, /* tp_traverse */
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001776 0, /* tp_clear */
1777 instance_richcompare, /* tp_richcompare */
Fred Drake41deb1e2001-02-01 05:27:45 +00001778 offsetof(PyInstanceObject, in_weakreflist) /* tp_weaklistoffset */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001779};
1780
1781
Guido van Rossum81daa321993-05-20 14:24:46 +00001782/* Instance method objects are used for two purposes:
1783 (a) as bound instance methods (returned by instancename.methodname)
1784 (b) as unbound methods (returned by ClassName.methodname)
1785 In case (b), im_self is NULL
1786*/
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001787
Guido van Rossuma0d349f1997-08-05 02:06:53 +00001788static PyMethodObject *free_list;
1789
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001790PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001791PyMethod_New(PyObject *func, PyObject *self, PyObject *class)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001792{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001793 register PyMethodObject *im;
Guido van Rossum7859f871998-07-08 14:58:16 +00001794 if (!PyCallable_Check(func)) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001795 PyErr_BadInternalCall();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001796 return NULL;
1797 }
Guido van Rossuma0d349f1997-08-05 02:06:53 +00001798 im = free_list;
1799 if (im != NULL) {
1800 free_list = (PyMethodObject *)(im->im_self);
Guido van Rossumb18618d2000-05-03 23:44:39 +00001801 PyObject_INIT(im, &PyMethod_Type);
Guido van Rossuma0d349f1997-08-05 02:06:53 +00001802 }
1803 else {
1804 im = PyObject_NEW(PyMethodObject, &PyMethod_Type);
1805 if (im == NULL)
1806 return NULL;
1807 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001808 Py_INCREF(func);
Guido van Rossume8122f11991-05-05 20:03:07 +00001809 im->im_func = func;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001810 Py_XINCREF(self);
Guido van Rossume8122f11991-05-05 20:03:07 +00001811 im->im_self = self;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001812 Py_INCREF(class);
Guido van Rossum81daa321993-05-20 14:24:46 +00001813 im->im_class = class;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001814 PyObject_GC_Init(im);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001815 return (PyObject *)im;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001816}
1817
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001818PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001819PyMethod_Function(register PyObject *im)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001820{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001821 if (!PyMethod_Check(im)) {
1822 PyErr_BadInternalCall();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001823 return NULL;
1824 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001825 return ((PyMethodObject *)im)->im_func;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001826}
1827
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001828PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001829PyMethod_Self(register PyObject *im)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001830{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001831 if (!PyMethod_Check(im)) {
1832 PyErr_BadInternalCall();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001833 return NULL;
1834 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001835 return ((PyMethodObject *)im)->im_self;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001836}
1837
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001838PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001839PyMethod_Class(register PyObject *im)
Guido van Rossum81daa321993-05-20 14:24:46 +00001840{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001841 if (!PyMethod_Check(im)) {
1842 PyErr_BadInternalCall();
Guido van Rossum81daa321993-05-20 14:24:46 +00001843 return NULL;
1844 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001845 return ((PyMethodObject *)im)->im_class;
Guido van Rossum81daa321993-05-20 14:24:46 +00001846}
1847
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001848/* Class method methods */
1849
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001850#define OFF(x) offsetof(PyMethodObject, x)
Guido van Rossum3f5da241990-12-20 15:06:42 +00001851
Guido van Rossume8122f11991-05-05 20:03:07 +00001852static struct memberlist instancemethod_memberlist[] = {
1853 {"im_func", T_OBJECT, OFF(im_func)},
1854 {"im_self", T_OBJECT, OFF(im_self)},
Guido van Rossum81daa321993-05-20 14:24:46 +00001855 {"im_class", T_OBJECT, OFF(im_class)},
Guido van Rossum6dabc981996-05-14 21:54:20 +00001856 /* Dummies that are not handled by getattr() except for __members__ */
1857 {"__doc__", T_INT, 0},
1858 {"__name__", T_INT, 0},
Barry Warsawd6a9e842001-01-15 20:40:19 +00001859 {"__dict__", T_OBJECT, 0},
Guido van Rossum3f5da241990-12-20 15:06:42 +00001860 {NULL} /* Sentinel */
1861};
1862
Barry Warsawd6a9e842001-01-15 20:40:19 +00001863static int
1864instancemethod_setattro(register PyMethodObject *im, PyObject *name,
1865 PyObject *v)
Guido van Rossum3f5da241990-12-20 15:06:42 +00001866{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001867 char *sname = PyString_AsString(name);
Barry Warsawd6a9e842001-01-15 20:40:19 +00001868
1869 if (PyEval_GetRestricted() ||
1870 strcmp(sname, "im_func") == 0 ||
1871 strcmp(sname, "im_self") == 0 ||
1872 strcmp(sname, "im_class") == 0)
1873 {
1874 PyErr_Format(PyExc_TypeError, "read-only attribute: %s",
1875 sname);
1876 return -1;
1877 }
1878 if (im->im_self != NULL) {
1879 PyErr_Format(PyExc_TypeError,
1880 "cannot set attributes through bound methods");
1881 return -1;
1882 }
1883 return PyObject_SetAttr(im->im_func, name, v);
1884}
1885
1886
1887static PyObject *
1888instancemethod_getattro(register PyMethodObject *im, PyObject *name)
1889{
1890 PyObject *rtn;
1891 char *sname = PyString_AsString(name);
Guido van Rossum2878a691996-08-09 20:53:24 +00001892 if (sname[0] == '_') {
Guido van Rossum7859f871998-07-08 14:58:16 +00001893 /* Inherit __name__ and __doc__ from the callable object
1894 implementing the method */
1895 if (strcmp(sname, "__name__") == 0 ||
1896 strcmp(sname, "__doc__") == 0)
1897 return PyObject_GetAttr(im->im_func, name);
Guido van Rossum6dabc981996-05-14 21:54:20 +00001898 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001899 if (PyEval_GetRestricted()) {
1900 PyErr_SetString(PyExc_RuntimeError,
1901 "instance-method attributes not accessible in restricted mode");
Guido van Rossum10393b11995-01-10 10:39:49 +00001902 return NULL;
1903 }
Barry Warsawd6a9e842001-01-15 20:40:19 +00001904 if (sname[0] == '_' && strcmp(sname, "__dict__") == 0)
1905 return PyObject_GetAttr(im->im_func, name);
1906
1907 rtn = PyMember_Get((char *)im, instancemethod_memberlist, sname);
1908 if (rtn == NULL && PyErr_ExceptionMatches(PyExc_AttributeError)) {
1909 PyErr_Clear();
1910 rtn = PyObject_GetAttr(im->im_func, name);
1911 }
1912 return rtn;
Guido van Rossum3f5da241990-12-20 15:06:42 +00001913}
1914
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001915static void
Fred Drake79912472000-07-09 04:06:11 +00001916instancemethod_dealloc(register PyMethodObject *im)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001917{
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001918 PyObject_GC_Fini(im);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001919 Py_DECREF(im->im_func);
1920 Py_XDECREF(im->im_self);
1921 Py_DECREF(im->im_class);
Guido van Rossuma0d349f1997-08-05 02:06:53 +00001922 im->im_self = (PyObject *)free_list;
1923 free_list = im;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001924}
1925
Guido van Rossumebc8c511992-09-03 20:39:51 +00001926static int
Fred Drake79912472000-07-09 04:06:11 +00001927instancemethod_compare(PyMethodObject *a, PyMethodObject *b)
Guido van Rossumebc8c511992-09-03 20:39:51 +00001928{
Guido van Rossume9df7271995-04-06 14:46:51 +00001929 if (a->im_self != b->im_self)
1930 return (a->im_self < b->im_self) ? -1 : 1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001931 return PyObject_Compare(a->im_func, b->im_func);
Guido van Rossumebc8c511992-09-03 20:39:51 +00001932}
1933
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001934static PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001935instancemethod_repr(PyMethodObject *a)
Guido van Rossum25831651993-05-19 14:50:45 +00001936{
1937 char buf[240];
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001938 PyInstanceObject *self = (PyInstanceObject *)(a->im_self);
Guido van Rossum7859f871998-07-08 14:58:16 +00001939 PyObject *func = a->im_func;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001940 PyClassObject *class = (PyClassObject *)(a->im_class);
1941 PyObject *fclassname, *iclassname, *funcname;
Guido van Rossum81daa321993-05-20 14:24:46 +00001942 char *fcname, *icname, *fname;
1943 fclassname = class->cl_name;
Guido van Rossum7859f871998-07-08 14:58:16 +00001944 if (PyFunction_Check(func)) {
1945 funcname = ((PyFunctionObject *)func)->func_name;
1946 Py_INCREF(funcname);
1947 }
1948 else {
1949 funcname = PyObject_GetAttrString(func,"__name__");
1950 if (funcname == NULL)
1951 PyErr_Clear();
1952 }
1953 if (funcname != NULL && PyString_Check(funcname))
1954 fname = PyString_AS_STRING(funcname);
1955 else
1956 fname = "?";
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001957 if (fclassname != NULL && PyString_Check(fclassname))
1958 fcname = PyString_AsString(fclassname);
Guido van Rossum25831651993-05-19 14:50:45 +00001959 else
Guido van Rossum81daa321993-05-20 14:24:46 +00001960 fcname = "?";
Guido van Rossum81daa321993-05-20 14:24:46 +00001961 if (self == NULL)
1962 sprintf(buf, "<unbound method %.100s.%.100s>", fcname, fname);
1963 else {
1964 iclassname = self->in_class->cl_name;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001965 if (iclassname != NULL && PyString_Check(iclassname))
1966 icname = PyString_AsString(iclassname);
Guido van Rossum81daa321993-05-20 14:24:46 +00001967 else
1968 icname = "?";
Fred Drakea44d3532000-06-30 15:01:00 +00001969 sprintf(buf, "<method %.60s.%.60s of %.60s instance at %p>",
1970 fcname, fname, icname, self);
Guido van Rossum81daa321993-05-20 14:24:46 +00001971 }
Guido van Rossum42636dc1999-10-11 14:03:12 +00001972 Py_XDECREF(funcname);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001973 return PyString_FromString(buf);
Guido van Rossum25831651993-05-19 14:50:45 +00001974}
1975
Guido van Rossum9bfef441993-03-29 10:43:31 +00001976static long
Fred Drake79912472000-07-09 04:06:11 +00001977instancemethod_hash(PyMethodObject *a)
Guido van Rossum9bfef441993-03-29 10:43:31 +00001978{
1979 long x, y;
Guido van Rossum81daa321993-05-20 14:24:46 +00001980 if (a->im_self == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001981 x = PyObject_Hash(Py_None);
Guido van Rossum81daa321993-05-20 14:24:46 +00001982 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001983 x = PyObject_Hash(a->im_self);
Guido van Rossum9bfef441993-03-29 10:43:31 +00001984 if (x == -1)
1985 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001986 y = PyObject_Hash(a->im_func);
Guido van Rossum9bfef441993-03-29 10:43:31 +00001987 if (y == -1)
1988 return -1;
1989 return x ^ y;
1990}
1991
Jeremy Hylton8caad492000-06-23 14:18:11 +00001992static int
1993instancemethod_traverse(PyMethodObject *im, visitproc visit, void *arg)
1994{
1995 int err;
1996 if (im->im_func) {
1997 err = visit(im->im_func, arg);
1998 if (err)
1999 return err;
2000 }
2001 if (im->im_self) {
2002 err = visit(im->im_self, arg);
2003 if (err)
2004 return err;
2005 }
2006 if (im->im_class) {
2007 err = visit(im->im_class, arg);
2008 if (err)
2009 return err;
2010 }
Jeremy Hyltond22162b2000-06-23 17:14:56 +00002011 return 0;
Jeremy Hylton8caad492000-06-23 14:18:11 +00002012}
2013
Guido van Rossumc0b618a1997-05-02 03:12:38 +00002014PyTypeObject PyMethod_Type = {
2015 PyObject_HEAD_INIT(&PyType_Type)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002016 0,
Guido van Rossum569fce71991-04-16 08:38:43 +00002017 "instance method",
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00002018 sizeof(PyMethodObject) + PyGC_HEAD_SIZE,
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002019 0,
Guido van Rossum8998b4f2001-01-17 15:28:20 +00002020 (destructor)instancemethod_dealloc, /* tp_dealloc */
2021 0, /* tp_print */
2022 0, /* tp_getattr */
2023 0, /* tp_setattr */
2024 (cmpfunc)instancemethod_compare, /* tp_compare */
2025 (reprfunc)instancemethod_repr, /* tp_repr */
2026 0, /* tp_as_number */
2027 0, /* tp_as_sequence */
2028 0, /* tp_as_mapping */
2029 (hashfunc)instancemethod_hash, /* tp_hash */
2030 0, /* tp_call */
2031 0, /* tp_str */
2032 (getattrofunc)instancemethod_getattro, /* tp_getattro */
2033 (setattrofunc)instancemethod_setattro, /* tp_setattro */
2034 0, /* tp_as_buffer */
2035 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_GC, /* tp_flags */
2036 0, /* tp_doc */
Jeremy Hylton8caad492000-06-23 14:18:11 +00002037 (traverseproc)instancemethod_traverse, /* tp_traverse */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002038};
Guido van Rossuma0d349f1997-08-05 02:06:53 +00002039
2040/* Clear out the free list */
2041
2042void
Fred Drake79912472000-07-09 04:06:11 +00002043PyMethod_Fini(void)
Guido van Rossuma0d349f1997-08-05 02:06:53 +00002044{
2045 while (free_list) {
Guido van Rossumb18618d2000-05-03 23:44:39 +00002046 PyMethodObject *im = free_list;
2047 free_list = (PyMethodObject *)(im->im_self);
Guido van Rossum4cc6ac72000-07-01 01:00:38 +00002048 im = (PyMethodObject *) PyObject_AS_GC(im);
Guido van Rossumb18618d2000-05-03 23:44:39 +00002049 PyObject_DEL(im);
Guido van Rossuma0d349f1997-08-05 02:06:53 +00002050 }
2051}