blob: 00cfdde6272572f22bdc66dc7b66962df0abf0c7 [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
Fred Drakeb60654b2001-02-26 18:56:37 +0000519 PyObject_ClearWeakRefs((PyObject *) inst);
Fred Drake41deb1e2001-02-01 05:27:45 +0000520
Tim Peters6b184912000-09-17 14:40:17 +0000521 /* Temporarily resurrect the object. */
Guido van Rossum441e4ab1996-05-23 22:46:51 +0000522#ifdef Py_TRACE_REFS
Tim Peters6b184912000-09-17 14:40:17 +0000523#ifndef Py_REF_DEBUG
524# error "Py_TRACE_REFS defined but Py_REF_DEBUG not."
525#endif
Guido van Rossum441e4ab1996-05-23 22:46:51 +0000526 /* much too complicated if Py_TRACE_REFS defined */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000527 inst->ob_type = &PyInstance_Type;
Guido van Rossumbffd6832000-01-20 22:32:56 +0000528 _Py_NewReference((PyObject *)inst);
Sjoerd Mullender740f3571995-08-28 09:00:43 +0000529#ifdef COUNT_ALLOCS
Tim Peters6b184912000-09-17 14:40:17 +0000530 /* compensate for boost in _Py_NewReference; note that
531 * _Py_RefTotal was also boosted; we'll knock that down later.
532 */
533 inst->ob_type->tp_alloc--;
Sjoerd Mullender740f3571995-08-28 09:00:43 +0000534#endif
Guido van Rossum441e4ab1996-05-23 22:46:51 +0000535#else /* !Py_TRACE_REFS */
Tim Peters6b184912000-09-17 14:40:17 +0000536 /* Py_INCREF boosts _Py_RefTotal if Py_REF_DEBUG is defined */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000537 Py_INCREF(inst);
Guido van Rossum441e4ab1996-05-23 22:46:51 +0000538#endif /* !Py_TRACE_REFS */
Tim Peters6b184912000-09-17 14:40:17 +0000539
540 /* Save the current exception, if any. */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000541 PyErr_Fetch(&error_type, &error_value, &error_traceback);
Tim Peters6b184912000-09-17 14:40:17 +0000542 /* Execute __del__ method, if any. */
Guido van Rossum2878a691996-08-09 20:53:24 +0000543 if (delstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000544 delstr = PyString_InternFromString("__del__");
Jeremy Hylton9e392e22000-04-26 20:39:20 +0000545 if ((del = instance_getattr2(inst, delstr)) != NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000546 PyObject *res = PyEval_CallObject(del, (PyObject *)NULL);
Tim Peters6b184912000-09-17 14:40:17 +0000547 if (res == NULL)
Jeremy Hyltonb709df32000-09-01 02:47:25 +0000548 PyErr_WriteUnraisable(del);
Guido van Rossum0dfcf751996-08-12 22:00:53 +0000549 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000550 Py_DECREF(res);
551 Py_DECREF(del);
Guido van Rossum25831651993-05-19 14:50:45 +0000552 }
Tim Peters6b184912000-09-17 14:40:17 +0000553 /* Restore the saved exception. */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000554 PyErr_Restore(error_type, error_value, error_traceback);
Tim Peters6b184912000-09-17 14:40:17 +0000555 /* Undo the temporary resurrection; can't use DECREF here, it would
556 * cause a recursive call.
557 */
558#ifdef Py_REF_DEBUG
559 /* _Py_RefTotal was boosted either by _Py_NewReference or
560 * Py_INCREF above.
561 */
562 _Py_RefTotal--;
563#endif
Sjoerd Mullender740f3571995-08-28 09:00:43 +0000564 if (--inst->ob_refcnt > 0) {
565#ifdef COUNT_ALLOCS
566 inst->ob_type->tp_free--;
567#endif
Guido van Rossum25831651993-05-19 14:50:45 +0000568 return; /* __del__ added a reference; don't delete now */
Sjoerd Mullender740f3571995-08-28 09:00:43 +0000569 }
Guido van Rossum441e4ab1996-05-23 22:46:51 +0000570#ifdef Py_TRACE_REFS
Guido van Rossumbffd6832000-01-20 22:32:56 +0000571 _Py_ForgetReference((PyObject *)inst);
Tim Peters6b184912000-09-17 14:40:17 +0000572#ifdef COUNT_ALLOCS
573 /* compensate for increment in _Py_ForgetReference */
574 inst->ob_type->tp_free--;
575#endif
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000576#ifndef WITH_CYCLE_GC
Sjoerd Mullender740f3571995-08-28 09:00:43 +0000577 inst->ob_type = NULL;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000578#endif
Tim Peters6b184912000-09-17 14:40:17 +0000579#endif
Neil Schemenauerce209672000-09-15 18:57:21 +0000580 PyObject_GC_Fini(inst);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000581 Py_DECREF(inst->in_class);
582 Py_XDECREF(inst->in_dict);
Guido van Rossum4cc6ac72000-07-01 01:00:38 +0000583 inst = (PyInstanceObject *) PyObject_AS_GC(inst);
Guido van Rossumb18618d2000-05-03 23:44:39 +0000584 PyObject_DEL(inst);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000585}
586
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000587static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000588instance_getattr1(register PyInstanceObject *inst, PyObject *name)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000589{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000590 register PyObject *v;
591 register char *sname = PyString_AsString(name);
Guido van Rossum2878a691996-08-09 20:53:24 +0000592 if (sname[0] == '_' && sname[1] == '_') {
593 if (strcmp(sname, "__dict__") == 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000594 if (PyEval_GetRestricted()) {
595 PyErr_SetString(PyExc_RuntimeError,
596 "instance.__dict__ not accessible in restricted mode");
Guido van Rossum10393b11995-01-10 10:39:49 +0000597 return NULL;
598 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000599 Py_INCREF(inst->in_dict);
Sjoerd Mullender3bb8a051993-10-22 12:04:32 +0000600 return inst->in_dict;
601 }
Guido van Rossum2878a691996-08-09 20:53:24 +0000602 if (strcmp(sname, "__class__") == 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000603 Py_INCREF(inst->in_class);
604 return (PyObject *)inst->in_class;
Sjoerd Mullender3bb8a051993-10-22 12:04:32 +0000605 }
Guido van Rossum94308391991-10-20 20:11:48 +0000606 }
Jeremy Hylton9e392e22000-04-26 20:39:20 +0000607 v = instance_getattr2(inst, name);
608 if (v == NULL) {
Fred Drake661ea262000-10-24 19:57:45 +0000609 PyErr_Format(PyExc_AttributeError,
610 "%.50s instance has no attribute '%.400s'",
Jeremy Hylton9e392e22000-04-26 20:39:20 +0000611 PyString_AS_STRING(inst->in_class->cl_name), sname);
612 }
613 return v;
614}
615
616static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000617instance_getattr2(register PyInstanceObject *inst, PyObject *name)
Jeremy Hylton9e392e22000-04-26 20:39:20 +0000618{
619 register PyObject *v;
620 PyClassObject *class;
Guido van Rossumeb6b33a1993-05-25 09:38:27 +0000621 class = NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000622 v = PyDict_GetItem(inst->in_dict, name);
Guido van Rossumb3f72581993-05-21 19:56:10 +0000623 if (v == NULL) {
624 v = class_lookup(inst->in_class, name, &class);
Jeremy Hylton9e392e22000-04-26 20:39:20 +0000625 if (v == NULL)
626 return v;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000627 }
Guido van Rossum0add15f1997-05-09 01:07:15 +0000628 Py_INCREF(v);
Guido van Rossum21d335e1993-10-15 13:01:11 +0000629 if (class != NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000630 if (PyFunction_Check(v)) {
631 PyObject *w = PyMethod_New(v, (PyObject *)inst,
Jeremy Hylton09ac89a2001-01-29 22:38:32 +0000632 (PyObject *)class);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000633 Py_DECREF(v);
Guido van Rossum21d335e1993-10-15 13:01:11 +0000634 v = w;
635 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000636 else if (PyMethod_Check(v)) {
637 PyObject *im_class = PyMethod_Class(v);
Guido van Rossum21d335e1993-10-15 13:01:11 +0000638 /* Only if classes are compatible */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000639 if (PyClass_IsSubclass((PyObject *)class, im_class)) {
640 PyObject *im_func = PyMethod_Function(v);
641 PyObject *w = PyMethod_New(im_func,
642 (PyObject *)inst, im_class);
643 Py_DECREF(v);
Guido van Rossum21d335e1993-10-15 13:01:11 +0000644 v = w;
645 }
646 }
Guido van Rossumb3f72581993-05-21 19:56:10 +0000647 }
648 return v;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000649}
650
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000651static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000652instance_getattr(register PyInstanceObject *inst, PyObject *name)
Guido van Rossume7737541994-09-05 07:31:41 +0000653{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000654 register PyObject *func, *res;
Guido van Rossum52ca98a1994-09-05 07:32:29 +0000655 res = instance_getattr1(inst, name);
656 if (res == NULL && (func = inst->in_class->cl_getattr) != NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000657 PyObject *args;
658 PyErr_Clear();
659 args = Py_BuildValue("(OO)", inst, name);
Guido van Rossum52ca98a1994-09-05 07:32:29 +0000660 if (args == NULL)
Guido van Rossume7737541994-09-05 07:31:41 +0000661 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000662 res = PyEval_CallObject(func, args);
663 Py_DECREF(args);
Guido van Rossume7737541994-09-05 07:31:41 +0000664 }
665 return res;
666}
667
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000668static int
Fred Drake79912472000-07-09 04:06:11 +0000669instance_setattr1(PyInstanceObject *inst, PyObject *name, PyObject *v)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000670{
Guido van Rossum94472a01992-09-04 09:45:18 +0000671 if (v == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000672 int rv = PyDict_DelItem(inst->in_dict, name);
Guido van Rossum94472a01992-09-04 09:45:18 +0000673 if (rv < 0)
Fred Drake661ea262000-10-24 19:57:45 +0000674 PyErr_Format(PyExc_AttributeError,
675 "%.50s instance has no attribute '%.400s'",
676 PyString_AS_STRING(inst->in_class->cl_name),
677 PyString_AS_STRING(name));
Guido van Rossum94472a01992-09-04 09:45:18 +0000678 return rv;
679 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000680 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000681 return PyDict_SetItem(inst->in_dict, name, v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000682}
683
Guido van Rossume7737541994-09-05 07:31:41 +0000684static int
Fred Drake79912472000-07-09 04:06:11 +0000685instance_setattr(PyInstanceObject *inst, PyObject *name, PyObject *v)
Guido van Rossume7737541994-09-05 07:31:41 +0000686{
Guido van Rossumb2173c31997-08-25 21:23:56 +0000687 PyObject *func, *args, *res, *tmp;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000688 char *sname = PyString_AsString(name);
Guido van Rossumb2173c31997-08-25 21:23:56 +0000689 if (sname[0] == '_' && sname[1] == '_') {
690 int n = PyString_Size(name);
Guido van Rossum2878a691996-08-09 20:53:24 +0000691 if (sname[n-1] == '_' && sname[n-2] == '_') {
Guido van Rossumb2173c31997-08-25 21:23:56 +0000692 if (strcmp(sname, "__dict__") == 0) {
693 if (PyEval_GetRestricted()) {
694 PyErr_SetString(PyExc_RuntimeError,
695 "__dict__ not accessible in restricted mode");
696 return -1;
697 }
698 if (v == NULL || !PyDict_Check(v)) {
699 PyErr_SetString(PyExc_TypeError,
700 "__dict__ must be set to a dictionary");
701 return -1;
702 }
703 tmp = inst->in_dict;
704 Py_INCREF(v);
705 inst->in_dict = v;
706 Py_DECREF(tmp);
707 return 0;
708 }
709 if (strcmp(sname, "__class__") == 0) {
710 if (PyEval_GetRestricted()) {
711 PyErr_SetString(PyExc_RuntimeError,
712 "__class__ not accessible in restricted mode");
713 return -1;
714 }
715 if (v == NULL || !PyClass_Check(v)) {
716 PyErr_SetString(PyExc_TypeError,
717 "__class__ must be set to a class");
718 return -1;
719 }
720 tmp = (PyObject *)(inst->in_class);
721 Py_INCREF(v);
722 inst->in_class = (PyClassObject *)v;
723 Py_DECREF(tmp);
724 return 0;
725 }
Guido van Rossume7737541994-09-05 07:31:41 +0000726 }
Guido van Rossume7737541994-09-05 07:31:41 +0000727 }
Guido van Rossum52ca98a1994-09-05 07:32:29 +0000728 if (v == NULL)
729 func = inst->in_class->cl_delattr;
730 else
731 func = inst->in_class->cl_setattr;
732 if (func == NULL)
733 return instance_setattr1(inst, name, v);
734 if (v == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000735 args = Py_BuildValue("(OO)", inst, name);
Guido van Rossum52ca98a1994-09-05 07:32:29 +0000736 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000737 args = Py_BuildValue("(OOO)", inst, name, v);
Guido van Rossum52ca98a1994-09-05 07:32:29 +0000738 if (args == NULL)
739 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000740 res = PyEval_CallObject(func, args);
741 Py_DECREF(args);
Guido van Rossum52ca98a1994-09-05 07:32:29 +0000742 if (res == NULL)
743 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000744 Py_DECREF(res);
Guido van Rossume7737541994-09-05 07:31:41 +0000745 return 0;
746}
747
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000748static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000749instance_repr(PyInstanceObject *inst)
Guido van Rossum04691fc1992-08-12 15:35:34 +0000750{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000751 PyObject *func;
752 PyObject *res;
753 static PyObject *reprstr;
Guido van Rossum04691fc1992-08-12 15:35:34 +0000754
Guido van Rossum2878a691996-08-09 20:53:24 +0000755 if (reprstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000756 reprstr = PyString_InternFromString("__repr__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000757 func = instance_getattr(inst, reprstr);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000758 if (func == NULL) {
Guido van Rossum25831651993-05-19 14:50:45 +0000759 char buf[140];
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000760 PyObject *classname = inst->in_class->cl_name;
Guido van Rossumb7f1afe1997-12-03 00:06:02 +0000761 PyObject *mod = PyDict_GetItemString(
762 inst->in_class->cl_dict, "__module__");
Guido van Rossum25831651993-05-19 14:50:45 +0000763 char *cname;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000764 if (classname != NULL && PyString_Check(classname))
765 cname = PyString_AsString(classname);
Guido van Rossum25831651993-05-19 14:50:45 +0000766 else
767 cname = "?";
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000768 PyErr_Clear();
Guido van Rossumb7f1afe1997-12-03 00:06:02 +0000769 if (mod == NULL || !PyString_Check(mod))
Fred Drakea44d3532000-06-30 15:01:00 +0000770 sprintf(buf, "<?.%.100s instance at %p>",
771 cname, inst);
Guido van Rossumb7f1afe1997-12-03 00:06:02 +0000772 else
Fred Drakea44d3532000-06-30 15:01:00 +0000773 sprintf(buf, "<%.50s.%.50s instance at %p>",
Guido van Rossumb7f1afe1997-12-03 00:06:02 +0000774 PyString_AsString(mod),
Fred Drakea44d3532000-06-30 15:01:00 +0000775 cname, inst);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000776 return PyString_FromString(buf);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000777 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000778 res = PyEval_CallObject(func, (PyObject *)NULL);
779 Py_DECREF(func);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000780 return res;
781}
782
Guido van Rossum9bfef441993-03-29 10:43:31 +0000783static long
Fred Drake79912472000-07-09 04:06:11 +0000784instance_hash(PyInstanceObject *inst)
Guido van Rossum9bfef441993-03-29 10:43:31 +0000785{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000786 PyObject *func;
787 PyObject *res;
Guido van Rossum8a0c3451993-04-08 12:56:19 +0000788 long outcome;
Guido van Rossum65e8bd72001-01-18 23:46:31 +0000789 static PyObject *hashstr, *eqstr, *cmpstr;
Guido van Rossum9bfef441993-03-29 10:43:31 +0000790
Guido van Rossum2878a691996-08-09 20:53:24 +0000791 if (hashstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000792 hashstr = PyString_InternFromString("__hash__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000793 func = instance_getattr(inst, hashstr);
Guido van Rossum9bfef441993-03-29 10:43:31 +0000794 if (func == NULL) {
Guido van Rossum65e8bd72001-01-18 23:46:31 +0000795 /* If there is no __eq__ and no __cmp__ method, we hash on the
796 address. If an __eq__ or __cmp__ method exists, there must
797 be a __hash__. */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000798 PyErr_Clear();
Guido van Rossum65e8bd72001-01-18 23:46:31 +0000799 if (eqstr == NULL)
800 eqstr = PyString_InternFromString("__eq__");
801 func = instance_getattr(inst, eqstr);
Guido van Rossum8a0c3451993-04-08 12:56:19 +0000802 if (func == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000803 PyErr_Clear();
Guido van Rossum65e8bd72001-01-18 23:46:31 +0000804 if (cmpstr == NULL)
805 cmpstr = PyString_InternFromString("__cmp__");
806 func = instance_getattr(inst, cmpstr);
807 if (func == NULL) {
808 PyErr_Clear();
809 return _Py_HashPointer(inst);
810 }
Guido van Rossum8a0c3451993-04-08 12:56:19 +0000811 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000812 PyErr_SetString(PyExc_TypeError, "unhashable instance");
Guido van Rossum9bfef441993-03-29 10:43:31 +0000813 return -1;
814 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000815 res = PyEval_CallObject(func, (PyObject *)NULL);
816 Py_DECREF(func);
Guido van Rossum9bfef441993-03-29 10:43:31 +0000817 if (res == NULL)
818 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000819 if (PyInt_Check(res)) {
820 outcome = PyInt_AsLong(res);
Guido van Rossum9bfef441993-03-29 10:43:31 +0000821 if (outcome == -1)
822 outcome = -2;
823 }
824 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000825 PyErr_SetString(PyExc_TypeError,
826 "__hash__() should return an int");
Guido van Rossum9bfef441993-03-29 10:43:31 +0000827 outcome = -1;
828 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000829 Py_DECREF(res);
Guido van Rossum9bfef441993-03-29 10:43:31 +0000830 return outcome;
831}
832
Jeremy Hylton8caad492000-06-23 14:18:11 +0000833static int
834instance_traverse(PyInstanceObject *o, visitproc visit, void *arg)
835{
836 int err;
837 if (o->in_class) {
838 err = visit((PyObject *)(o->in_class), arg);
839 if (err)
840 return err;
841 }
842 if (o->in_dict) {
843 err = visit(o->in_dict, arg);
844 if (err)
845 return err;
846 }
Jeremy Hyltond22162b2000-06-23 17:14:56 +0000847 return 0;
Jeremy Hylton8caad492000-06-23 14:18:11 +0000848}
849
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000850static PyObject *getitemstr, *setitemstr, *delitemstr, *lenstr;
Guido van Rossum2878a691996-08-09 20:53:24 +0000851
Guido van Rossum9bfef441993-03-29 10:43:31 +0000852static int
Fred Drake79912472000-07-09 04:06:11 +0000853instance_length(PyInstanceObject *inst)
Guido van Rossum04691fc1992-08-12 15:35:34 +0000854{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000855 PyObject *func;
856 PyObject *res;
Guido van Rossum04691fc1992-08-12 15:35:34 +0000857 int outcome;
858
Guido van Rossum2878a691996-08-09 20:53:24 +0000859 if (lenstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000860 lenstr = PyString_InternFromString("__len__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000861 func = instance_getattr(inst, lenstr);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000862 if (func == NULL)
863 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000864 res = PyEval_CallObject(func, (PyObject *)NULL);
865 Py_DECREF(func);
Guido van Rossumd014ea61992-11-26 10:30:26 +0000866 if (res == NULL)
867 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000868 if (PyInt_Check(res)) {
869 outcome = PyInt_AsLong(res);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000870 if (outcome < 0)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000871 PyErr_SetString(PyExc_ValueError,
872 "__len__() should return >= 0");
Guido van Rossum04691fc1992-08-12 15:35:34 +0000873 }
874 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000875 PyErr_SetString(PyExc_TypeError,
876 "__len__() should return an int");
Guido van Rossum04691fc1992-08-12 15:35:34 +0000877 outcome = -1;
878 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000879 Py_DECREF(res);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000880 return outcome;
881}
882
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000883static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000884instance_subscript(PyInstanceObject *inst, PyObject *key)
Guido van Rossum04691fc1992-08-12 15:35:34 +0000885{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000886 PyObject *func;
887 PyObject *arg;
888 PyObject *res;
Guido van Rossum04691fc1992-08-12 15:35:34 +0000889
Guido van Rossum2878a691996-08-09 20:53:24 +0000890 if (getitemstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000891 getitemstr = PyString_InternFromString("__getitem__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000892 func = instance_getattr(inst, getitemstr);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000893 if (func == NULL)
894 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000895 arg = Py_BuildValue("(O)", key);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000896 if (arg == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000897 Py_DECREF(func);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000898 return NULL;
899 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000900 res = PyEval_CallObject(func, arg);
901 Py_DECREF(func);
902 Py_DECREF(arg);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000903 return res;
904}
905
Guido van Rossum9bfef441993-03-29 10:43:31 +0000906static int
Fred Drake79912472000-07-09 04:06:11 +0000907instance_ass_subscript(PyInstanceObject *inst, PyObject *key, PyObject *value)
Guido van Rossum04691fc1992-08-12 15:35:34 +0000908{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000909 PyObject *func;
910 PyObject *arg;
911 PyObject *res;
Guido van Rossum04691fc1992-08-12 15:35:34 +0000912
Guido van Rossum2878a691996-08-09 20:53:24 +0000913 if (value == NULL) {
914 if (delitemstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000915 delitemstr = PyString_InternFromString("__delitem__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000916 func = instance_getattr(inst, delitemstr);
917 }
918 else {
919 if (setitemstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000920 setitemstr = PyString_InternFromString("__setitem__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000921 func = instance_getattr(inst, setitemstr);
922 }
Guido van Rossum04691fc1992-08-12 15:35:34 +0000923 if (func == NULL)
924 return -1;
925 if (value == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000926 arg = Py_BuildValue("(O)", key);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000927 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000928 arg = Py_BuildValue("(OO)", key, value);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000929 if (arg == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000930 Py_DECREF(func);
Guido van Rossum6d946f91992-08-14 13:49:30 +0000931 return -1;
Guido van Rossum04691fc1992-08-12 15:35:34 +0000932 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000933 res = PyEval_CallObject(func, arg);
934 Py_DECREF(func);
935 Py_DECREF(arg);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000936 if (res == NULL)
937 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000938 Py_DECREF(res);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000939 return 0;
940}
941
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000942static PyMappingMethods instance_as_mapping = {
Guido van Rossum8998b4f2001-01-17 15:28:20 +0000943 (inquiry)instance_length, /* mp_length */
944 (binaryfunc)instance_subscript, /* mp_subscript */
945 (objobjargproc)instance_ass_subscript, /* mp_ass_subscript */
Guido van Rossum04691fc1992-08-12 15:35:34 +0000946};
947
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000948static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000949instance_item(PyInstanceObject *inst, int i)
Guido van Rossum04691fc1992-08-12 15:35:34 +0000950{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000951 PyObject *func, *arg, *res;
Guido van Rossum04691fc1992-08-12 15:35:34 +0000952
Guido van Rossum2878a691996-08-09 20:53:24 +0000953 if (getitemstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000954 getitemstr = PyString_InternFromString("__getitem__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000955 func = instance_getattr(inst, getitemstr);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000956 if (func == NULL)
957 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000958 arg = Py_BuildValue("(i)", i);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000959 if (arg == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000960 Py_DECREF(func);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000961 return NULL;
962 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000963 res = PyEval_CallObject(func, arg);
964 Py_DECREF(func);
965 Py_DECREF(arg);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000966 return res;
967}
968
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000969static PyObject *
Thomas Wouters1d75a792000-08-17 22:37:32 +0000970sliceobj_from_intint(int i, int j)
971{
972 PyObject *start, *end, *res;
973
974 start = PyInt_FromLong((long)i);
975 if (!start)
976 return NULL;
977
978 end = PyInt_FromLong((long)j);
979 if (!end) {
980 Py_DECREF(start);
981 return NULL;
982 }
983 res = PySlice_New(start, end, NULL);
984 Py_DECREF(start);
985 Py_DECREF(end);
986 return res;
987}
988
989
990static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000991instance_slice(PyInstanceObject *inst, int i, int j)
Guido van Rossum04691fc1992-08-12 15:35:34 +0000992{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000993 PyObject *func, *arg, *res;
994 static PyObject *getslicestr;
Guido van Rossum04691fc1992-08-12 15:35:34 +0000995
Guido van Rossum2878a691996-08-09 20:53:24 +0000996 if (getslicestr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000997 getslicestr = PyString_InternFromString("__getslice__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000998 func = instance_getattr(inst, getslicestr);
Thomas Wouters1d75a792000-08-17 22:37:32 +0000999
1000 if (func == NULL) {
1001 PyErr_Clear();
1002
1003 if (getitemstr == NULL)
1004 getitemstr = PyString_InternFromString("__getitem__");
1005 func = instance_getattr(inst, getitemstr);
1006 if (func == NULL)
1007 return NULL;
1008 arg = Py_BuildValue("(N)", sliceobj_from_intint(i, j));
1009 } else
1010 arg = Py_BuildValue("(ii)", i, j);
1011
Guido van Rossum04691fc1992-08-12 15:35:34 +00001012 if (arg == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001013 Py_DECREF(func);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001014 return NULL;
1015 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001016 res = PyEval_CallObject(func, arg);
1017 Py_DECREF(func);
1018 Py_DECREF(arg);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001019 return res;
1020}
1021
1022static int
Fred Drake79912472000-07-09 04:06:11 +00001023instance_ass_item(PyInstanceObject *inst, int i, PyObject *item)
Guido van Rossum04691fc1992-08-12 15:35:34 +00001024{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001025 PyObject *func, *arg, *res;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001026
Guido van Rossum2878a691996-08-09 20:53:24 +00001027 if (item == NULL) {
1028 if (delitemstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +00001029 delitemstr = PyString_InternFromString("__delitem__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001030 func = instance_getattr(inst, delitemstr);
1031 }
1032 else {
1033 if (setitemstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +00001034 setitemstr = PyString_InternFromString("__setitem__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001035 func = instance_getattr(inst, setitemstr);
1036 }
Guido van Rossum04691fc1992-08-12 15:35:34 +00001037 if (func == NULL)
Guido van Rossum6d946f91992-08-14 13:49:30 +00001038 return -1;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001039 if (item == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001040 arg = Py_BuildValue("i", i);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001041 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001042 arg = Py_BuildValue("(iO)", i, item);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001043 if (arg == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001044 Py_DECREF(func);
Guido van Rossum6d946f91992-08-14 13:49:30 +00001045 return -1;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001046 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001047 res = PyEval_CallObject(func, arg);
1048 Py_DECREF(func);
1049 Py_DECREF(arg);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001050 if (res == NULL)
1051 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001052 Py_DECREF(res);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001053 return 0;
1054}
1055
1056static int
Fred Drake79912472000-07-09 04:06:11 +00001057instance_ass_slice(PyInstanceObject *inst, int i, int j, PyObject *value)
Guido van Rossum04691fc1992-08-12 15:35:34 +00001058{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001059 PyObject *func, *arg, *res;
1060 static PyObject *setslicestr, *delslicestr;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001061
Guido van Rossum2878a691996-08-09 20:53:24 +00001062 if (value == NULL) {
1063 if (delslicestr == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001064 delslicestr =
1065 PyString_InternFromString("__delslice__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001066 func = instance_getattr(inst, delslicestr);
Thomas Wouters1d75a792000-08-17 22:37:32 +00001067 if (func == NULL) {
1068 PyErr_Clear();
1069 if (delitemstr == NULL)
1070 delitemstr =
1071 PyString_InternFromString("__delitem__");
1072 func = instance_getattr(inst, delitemstr);
1073 if (func == NULL)
1074 return -1;
1075
1076 arg = Py_BuildValue("(N)",
1077 sliceobj_from_intint(i, j));
1078 } else
1079 arg = Py_BuildValue("(ii)", i, j);
Guido van Rossum2878a691996-08-09 20:53:24 +00001080 }
1081 else {
1082 if (setslicestr == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001083 setslicestr =
1084 PyString_InternFromString("__setslice__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001085 func = instance_getattr(inst, setslicestr);
Thomas Wouters1d75a792000-08-17 22:37:32 +00001086 if (func == NULL) {
1087 PyErr_Clear();
1088 if (setitemstr == NULL)
1089 setitemstr =
1090 PyString_InternFromString("__setitem__");
1091 func = instance_getattr(inst, setitemstr);
1092 if (func == NULL)
1093 return -1;
1094
1095 arg = Py_BuildValue("(NO)",
1096 sliceobj_from_intint(i, j), value);
1097 } else
1098 arg = Py_BuildValue("(iiO)", i, j, value);
Guido van Rossum2878a691996-08-09 20:53:24 +00001099 }
Guido van Rossum04691fc1992-08-12 15:35:34 +00001100 if (arg == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001101 Py_DECREF(func);
Guido van Rossum6d946f91992-08-14 13:49:30 +00001102 return -1;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001103 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001104 res = PyEval_CallObject(func, arg);
1105 Py_DECREF(func);
1106 Py_DECREF(arg);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001107 if (res == NULL)
1108 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001109 Py_DECREF(res);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001110 return 0;
1111}
1112
Guido van Rossumee28c3a2000-02-28 15:03:15 +00001113static int instance_contains(PyInstanceObject *inst, PyObject *member)
1114{
1115 static PyObject *__contains__;
1116 PyObject *func, *arg, *res;
1117 int ret;
1118
1119 if(__contains__ == NULL) {
1120 __contains__ = PyString_InternFromString("__contains__");
1121 if(__contains__ == NULL)
1122 return -1;
1123 }
1124 func = instance_getattr(inst, __contains__);
1125 if(func == NULL) {
Thomas Wouters7e474022000-07-16 12:04:32 +00001126 /* fall back to previous behavior */
Guido van Rossumee28c3a2000-02-28 15:03:15 +00001127 int i, cmp_res;
1128
1129 if(!PyErr_ExceptionMatches(PyExc_AttributeError))
1130 return -1;
1131 PyErr_Clear();
1132 for(i=0;;i++) {
1133 PyObject *obj = instance_item(inst, i);
1134 int ret = 0;
1135
1136 if(obj == NULL) {
1137 if(!PyErr_ExceptionMatches(PyExc_IndexError))
1138 return -1;
1139 PyErr_Clear();
1140 return 0;
1141 }
1142 if(PyObject_Cmp(obj, member, &cmp_res) == -1)
1143 ret = -1;
1144 if(cmp_res == 0)
1145 ret = 1;
1146 Py_DECREF(obj);
1147 if(ret)
1148 return ret;
1149 }
1150 }
1151 arg = Py_BuildValue("(O)", member);
1152 if(arg == NULL) {
1153 Py_DECREF(func);
1154 return -1;
1155 }
1156 res = PyEval_CallObject(func, arg);
1157 Py_DECREF(func);
1158 Py_DECREF(arg);
1159 if(res == NULL)
1160 return -1;
1161 ret = PyObject_IsTrue(res);
1162 Py_DECREF(res);
1163 return ret;
1164}
1165
Fred Drake79912472000-07-09 04:06:11 +00001166static PySequenceMethods
1167instance_as_sequence = {
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001168 (inquiry)instance_length, /* sq_length */
1169 0, /* sq_concat */
1170 0, /* sq_repeat */
1171 (intargfunc)instance_item, /* sq_item */
1172 (intintargfunc)instance_slice, /* sq_slice */
1173 (intobjargproc)instance_ass_item, /* sq_ass_item */
1174 (intintobjargproc)instance_ass_slice, /* sq_ass_slice */
1175 (objobjproc)instance_contains, /* sq_contains */
Guido van Rossum04691fc1992-08-12 15:35:34 +00001176};
1177
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001178static PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001179generic_unary_op(PyInstanceObject *self, PyObject *methodname)
Guido van Rossum04691fc1992-08-12 15:35:34 +00001180{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001181 PyObject *func, *res;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001182
1183 if ((func = instance_getattr(self, methodname)) == NULL)
1184 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001185 res = PyEval_CallObject(func, (PyObject *)NULL);
1186 Py_DECREF(func);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001187 return res;
1188}
1189
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001190static PyObject *
1191generic_binary_op(PyObject *v, PyObject *w, char *opname)
Guido van Rossum03093a21994-09-28 15:51:32 +00001192{
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001193 PyObject *result;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001194 PyObject *args;
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001195 PyObject *func = PyObject_GetAttrString(v, opname);
Guido van Rossume7d444f1995-01-07 12:35:18 +00001196 if (func == NULL) {
Guido van Rossum617c1b01998-05-28 19:50:02 +00001197 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001198 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001199 PyErr_Clear();
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001200 Py_INCREF(Py_NotImplemented);
1201 return Py_NotImplemented;
Guido van Rossume7d444f1995-01-07 12:35:18 +00001202 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001203 args = Py_BuildValue("(O)", w);
Guido van Rossum03093a21994-09-28 15:51:32 +00001204 if (args == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001205 Py_DECREF(func);
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001206 return NULL;
Guido van Rossum03093a21994-09-28 15:51:32 +00001207 }
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001208 result = PyEval_CallObject(func, args);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001209 Py_DECREF(args);
1210 Py_DECREF(func);
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001211 return result;
1212}
1213
1214
1215static PyObject *coerce_obj;
1216
1217/* Try one half of a binary operator involving a class instance. */
1218static PyObject *
1219half_binop(PyObject *v, PyObject *w, char *opname, binaryfunc thisfunc,
1220 int swapped)
1221{
1222 PyObject *args;
1223 PyObject *coercefunc;
1224 PyObject *coerced = NULL;
1225 PyObject *v1;
1226 PyObject *result;
1227
1228 if (!PyInstance_Check(v)) {
1229 Py_INCREF(Py_NotImplemented);
1230 return Py_NotImplemented;
1231 }
1232
1233 if (coerce_obj == NULL) {
1234 coerce_obj = PyString_InternFromString("__coerce__");
1235 if (coerce_obj == NULL)
1236 return NULL;
1237 }
1238 coercefunc = PyObject_GetAttr(v, coerce_obj);
1239 if (coercefunc == NULL) {
1240 PyErr_Clear();
1241 return generic_binary_op(v, w, opname);
1242 }
1243
1244 args = Py_BuildValue("(O)", w);
1245 if (args == NULL) {
1246 return NULL;
1247 }
1248 coerced = PyEval_CallObject(coercefunc, args);
1249 Py_DECREF(args);
1250 Py_DECREF(coercefunc);
1251 if (coerced == NULL) {
1252 return NULL;
1253 }
1254 if (coerced == Py_None || coerced == Py_NotImplemented) {
1255 Py_DECREF(coerced);
1256 return generic_binary_op(v, w, opname);
1257 }
1258 if (!PyTuple_Check(coerced) || PyTuple_Size(coerced) != 2) {
1259 Py_DECREF(coerced);
1260 PyErr_SetString(PyExc_TypeError,
1261 "coercion should return None or 2-tuple");
1262 return NULL;
1263 }
1264 v1 = PyTuple_GetItem(coerced, 0);
1265 w = PyTuple_GetItem(coerced, 1);
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001266 if (v1->ob_type == v->ob_type && PyInstance_Check(v)) {
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001267 /* prevent recursion if __coerce__ returns self as the first
1268 * argument */
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001269 result = generic_binary_op(v1, w, opname);
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001270 } else {
1271 if (swapped)
1272 result = (thisfunc)(w, v1);
1273 else
1274 result = (thisfunc)(v1, w);
1275 }
1276 Py_DECREF(coerced);
1277 return result;
1278}
1279
1280/* Implement a binary operator involving at least one class instance. */
1281static PyObject *
1282do_binop(PyObject *v, PyObject *w, char *opname, char *ropname,
1283 binaryfunc thisfunc)
1284{
1285 PyObject *result = half_binop(v, w, opname, thisfunc, 0);
1286 if (result == Py_NotImplemented) {
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001287 Py_DECREF(result);
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001288 result = half_binop(w, v, ropname, thisfunc, 1);
1289 }
1290 return result;
1291}
1292
1293static PyObject *
1294do_binop_inplace(PyObject *v, PyObject *w, char *iopname, char *opname,
1295 char *ropname, binaryfunc thisfunc)
1296{
1297 PyObject *result = half_binop(v, w, iopname, thisfunc, 0);
1298 if (result == Py_NotImplemented) {
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001299 Py_DECREF(result);
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001300 result = do_binop(v, w, opname, ropname, thisfunc);
1301 }
1302 return result;
Guido van Rossum03093a21994-09-28 15:51:32 +00001303}
1304
Guido van Rossum879c5811995-01-10 15:24:06 +00001305static int
Fred Drake79912472000-07-09 04:06:11 +00001306instance_coerce(PyObject **pv, PyObject **pw)
Guido van Rossum879c5811995-01-10 15:24:06 +00001307{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001308 PyObject *v = *pv;
1309 PyObject *w = *pw;
Guido van Rossum3931df91997-11-18 19:23:07 +00001310 PyObject *coercefunc;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001311 PyObject *args;
1312 PyObject *coerced;
Guido van Rossum879c5811995-01-10 15:24:06 +00001313
Guido van Rossum2878a691996-08-09 20:53:24 +00001314 if (coerce_obj == NULL) {
Guido van Rossuma412d241997-01-18 07:59:12 +00001315 coerce_obj = PyString_InternFromString("__coerce__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001316 if (coerce_obj == NULL)
1317 return -1;
1318 }
Guido van Rossum3931df91997-11-18 19:23:07 +00001319 coercefunc = PyObject_GetAttr(v, coerce_obj);
1320 if (coercefunc == NULL) {
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001321 /* No __coerce__ method */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001322 PyErr_Clear();
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001323 return 1;
Guido van Rossum879c5811995-01-10 15:24:06 +00001324 }
1325 /* Has __coerce__ method: call it */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001326 args = Py_BuildValue("(O)", w);
Guido van Rossum879c5811995-01-10 15:24:06 +00001327 if (args == NULL) {
1328 return -1;
1329 }
Guido van Rossum3931df91997-11-18 19:23:07 +00001330 coerced = PyEval_CallObject(coercefunc, args);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001331 Py_DECREF(args);
Guido van Rossum3931df91997-11-18 19:23:07 +00001332 Py_DECREF(coercefunc);
Guido van Rossum879c5811995-01-10 15:24:06 +00001333 if (coerced == NULL) {
1334 /* __coerce__ call raised an exception */
1335 return -1;
1336 }
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001337 if (coerced == Py_None || coerced == Py_NotImplemented) {
Guido van Rossum879c5811995-01-10 15:24:06 +00001338 /* __coerce__ says "I can't do it" */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001339 Py_DECREF(coerced);
Guido van Rossum879c5811995-01-10 15:24:06 +00001340 return 1;
1341 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001342 if (!PyTuple_Check(coerced) || PyTuple_Size(coerced) != 2) {
Guido van Rossum879c5811995-01-10 15:24:06 +00001343 /* __coerce__ return value is malformed */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001344 Py_DECREF(coerced);
1345 PyErr_SetString(PyExc_TypeError,
Guido van Rossum879c5811995-01-10 15:24:06 +00001346 "coercion should return None or 2-tuple");
1347 return -1;
1348 }
1349 /* __coerce__ returned two new values */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001350 *pv = PyTuple_GetItem(coerced, 0);
1351 *pw = PyTuple_GetItem(coerced, 1);
1352 Py_INCREF(*pv);
1353 Py_INCREF(*pw);
1354 Py_DECREF(coerced);
Guido van Rossum879c5811995-01-10 15:24:06 +00001355 return 0;
1356}
1357
Guido van Rossum04691fc1992-08-12 15:35:34 +00001358#define UNARY(funcname, methodname) \
Thomas Woutersc3073522000-07-23 22:09:59 +00001359static PyObject *funcname(PyInstanceObject *self) { \
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001360 static PyObject *o; \
Guido van Rossuma412d241997-01-18 07:59:12 +00001361 if (o == NULL) o = PyString_InternFromString(methodname); \
Guido van Rossum2878a691996-08-09 20:53:24 +00001362 return generic_unary_op(self, o); \
Guido van Rossum04691fc1992-08-12 15:35:34 +00001363}
1364
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001365#define BINARY(f, m, n) \
1366static PyObject *f(PyObject *v, PyObject *w) { \
1367 return do_binop(v, w, "__" m "__", "__r" m "__", n); \
1368}
1369
1370#define BINARY_INPLACE(f, m, n) \
1371static PyObject *f(PyObject *v, PyObject *w) { \
1372 return do_binop_inplace(v, w, "__i" m "__", "__" m "__", \
1373 "__r" m "__", n); \
1374}
1375
Guido van Rossum04691fc1992-08-12 15:35:34 +00001376UNARY(instance_neg, "__neg__")
1377UNARY(instance_pos, "__pos__")
1378UNARY(instance_abs, "__abs__")
1379
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001380BINARY(instance_or, "or", PyNumber_Or)
1381BINARY(instance_and, "and", PyNumber_And)
1382BINARY(instance_xor, "xor", PyNumber_Xor)
1383BINARY(instance_lshift, "lshift", PyNumber_Lshift)
1384BINARY(instance_rshift, "rshift", PyNumber_Rshift)
1385BINARY(instance_add, "add", PyNumber_Add)
1386BINARY(instance_sub, "sub", PyNumber_Subtract)
1387BINARY(instance_mul, "mul", PyNumber_Multiply)
1388BINARY(instance_div, "div", PyNumber_Divide)
1389BINARY(instance_mod, "mod", PyNumber_Remainder)
1390BINARY(instance_divmod, "divmod", PyNumber_Divmod)
1391
1392BINARY_INPLACE(instance_ior, "or", PyNumber_InPlaceOr)
1393BINARY_INPLACE(instance_ixor, "xor", PyNumber_InPlaceXor)
1394BINARY_INPLACE(instance_iand, "and", PyNumber_InPlaceAnd)
1395BINARY_INPLACE(instance_ilshift, "lshift", PyNumber_InPlaceLshift)
1396BINARY_INPLACE(instance_irshift, "rshift", PyNumber_InPlaceRshift)
1397BINARY_INPLACE(instance_iadd, "add", PyNumber_InPlaceAdd)
1398BINARY_INPLACE(instance_isub, "sub", PyNumber_InPlaceSubtract)
1399BINARY_INPLACE(instance_imul, "mul", PyNumber_InPlaceMultiply)
1400BINARY_INPLACE(instance_idiv, "div", PyNumber_InPlaceDivide)
1401BINARY_INPLACE(instance_imod, "mod", PyNumber_InPlaceRemainder)
1402
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001403/* Try a 3-way comparison, returning an int; v is an instance. Return:
1404 -2 for an exception;
1405 -1 if v < w;
1406 0 if v == w;
1407 1 if v > w;
1408 2 if this particular 3-way comparison is not implemented or undefined.
1409*/
1410static int
1411half_cmp(PyObject *v, PyObject *w)
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001412{
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001413 static PyObject *cmp_obj;
1414 PyObject *args;
Guido van Rossum3202c6f2001-01-29 23:50:25 +00001415 PyObject *cmp_func;
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001416 PyObject *result;
1417 long l;
1418
1419 assert(PyInstance_Check(v));
1420
1421 if (cmp_obj == NULL) {
1422 cmp_obj = PyString_InternFromString("__cmp__");
1423 if (cmp_obj == NULL)
1424 return -2;
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001425 }
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001426
Guido van Rossum3202c6f2001-01-29 23:50:25 +00001427 cmp_func = PyObject_GetAttr(v, cmp_obj);
1428 if (cmp_func == NULL) {
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001429 PyErr_Clear();
1430 return 2;
1431 }
1432
1433 args = Py_BuildValue("(O)", w);
1434 if (args == NULL)
1435 return -2;
1436
Guido van Rossum3202c6f2001-01-29 23:50:25 +00001437 result = PyEval_CallObject(cmp_func, args);
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001438 Py_DECREF(args);
Guido van Rossum3202c6f2001-01-29 23:50:25 +00001439 Py_DECREF(cmp_func);
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001440
1441 if (result == NULL)
1442 return -2;
1443
1444 if (result == Py_NotImplemented) {
1445 Py_DECREF(result);
1446 return 2;
1447 }
1448
1449 l = PyInt_AsLong(result);
1450 Py_DECREF(result);
1451 if (l == -1 && PyErr_Occurred()) {
1452 PyErr_SetString(PyExc_TypeError,
1453 "comparison did not return an int");
1454 return -2;
1455 }
1456
1457 return l < 0 ? -1 : l > 0 ? 1 : 0;
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001458}
1459
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001460/* Try a 3-way comparison, returning an int; either v or w is an instance.
1461 We first try a coercion. Return:
1462 -2 for an exception;
1463 -1 if v < w;
1464 0 if v == w;
1465 1 if v > w;
1466 2 if this particular 3-way comparison is not implemented or undefined.
1467 THIS IS ONLY CALLED FROM object.c!
1468*/
1469static int
1470instance_compare(PyObject *v, PyObject *w)
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001471{
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001472 int c;
1473
1474 c = PyNumber_CoerceEx(&v, &w);
1475 if (c < 0)
1476 return -2;
1477 if (c == 0) {
1478 /* If neither is now an instance, use regular comparison */
1479 if (!PyInstance_Check(v) && !PyInstance_Check(w)) {
1480 c = PyObject_Compare(v, w);
1481 Py_DECREF(v);
1482 Py_DECREF(w);
1483 if (PyErr_Occurred())
1484 return -2;
1485 return c < 0 ? -1 : c > 0 ? 1 : 0;
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001486 }
1487 }
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001488 else {
1489 /* The coercion didn't do anything.
1490 Treat this the same as returning v and w unchanged. */
1491 Py_INCREF(v);
1492 Py_INCREF(w);
1493 }
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001494
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001495 if (PyInstance_Check(v)) {
1496 c = half_cmp(v, w);
1497 if (c <= 1) {
1498 Py_DECREF(v);
1499 Py_DECREF(w);
1500 return c;
1501 }
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001502 }
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001503 if (PyInstance_Check(w)) {
1504 c = half_cmp(w, v);
1505 if (c <= 1) {
1506 Py_DECREF(v);
1507 Py_DECREF(w);
1508 if (c >= -1)
1509 c = -c;
1510 return c;
1511 }
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001512 }
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001513 Py_DECREF(v);
1514 Py_DECREF(w);
1515 return 2;
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001516}
1517
Guido van Rossum9bfef441993-03-29 10:43:31 +00001518static int
Fred Drake79912472000-07-09 04:06:11 +00001519instance_nonzero(PyInstanceObject *self)
Guido van Rossum04691fc1992-08-12 15:35:34 +00001520{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001521 PyObject *func, *res;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001522 long outcome;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001523 static PyObject *nonzerostr;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001524
Guido van Rossum2878a691996-08-09 20:53:24 +00001525 if (nonzerostr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +00001526 nonzerostr = PyString_InternFromString("__nonzero__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001527 if ((func = instance_getattr(self, nonzerostr)) == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001528 PyErr_Clear();
Guido van Rossum2878a691996-08-09 20:53:24 +00001529 if (lenstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +00001530 lenstr = PyString_InternFromString("__len__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001531 if ((func = instance_getattr(self, lenstr)) == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001532 PyErr_Clear();
Guido van Rossum04691fc1992-08-12 15:35:34 +00001533 /* Fall back to the default behavior:
1534 all instances are nonzero */
1535 return 1;
1536 }
1537 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001538 res = PyEval_CallObject(func, (PyObject *)NULL);
1539 Py_DECREF(func);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001540 if (res == NULL)
1541 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001542 if (!PyInt_Check(res)) {
1543 Py_DECREF(res);
1544 PyErr_SetString(PyExc_TypeError,
1545 "__nonzero__ should return an int");
Guido van Rossum04691fc1992-08-12 15:35:34 +00001546 return -1;
1547 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001548 outcome = PyInt_AsLong(res);
1549 Py_DECREF(res);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001550 if (outcome < 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001551 PyErr_SetString(PyExc_ValueError,
1552 "__nonzero__ should return >= 0");
Guido van Rossum04691fc1992-08-12 15:35:34 +00001553 return -1;
1554 }
1555 return outcome > 0;
1556}
1557
1558UNARY(instance_invert, "__invert__")
Guido van Rossum1899c2e1992-09-12 11:09:23 +00001559UNARY(instance_int, "__int__")
1560UNARY(instance_long, "__long__")
1561UNARY(instance_float, "__float__")
1562UNARY(instance_oct, "__oct__")
1563UNARY(instance_hex, "__hex__")
1564
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001565static PyObject *
1566bin_power(PyObject *v, PyObject *w)
1567{
1568 return PyNumber_Power(v, w, Py_None);
1569}
1570
Guido van Rossum03093a21994-09-28 15:51:32 +00001571/* This version is for ternary calls only (z != None) */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001572static PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001573instance_pow(PyObject *v, PyObject *w, PyObject *z)
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001574{
1575 if (z == Py_None) {
1576 return do_binop(v, w, "__pow__", "__rpow__", bin_power);
Guido van Rossum03093a21994-09-28 15:51:32 +00001577 }
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001578 else {
1579 PyObject *func;
1580 PyObject *args;
1581 PyObject *result;
1582
1583 /* XXX Doesn't do coercions... */
1584 func = PyObject_GetAttrString(v, "__pow__");
1585 if (func == NULL)
1586 return NULL;
1587 args = Py_BuildValue("(OO)", w, z);
1588 if (args == NULL) {
1589 Py_DECREF(func);
1590 return NULL;
1591 }
1592 result = PyEval_CallObject(func, args);
1593 Py_DECREF(func);
1594 Py_DECREF(args);
1595 return result;
1596 }
Guido van Rossum03093a21994-09-28 15:51:32 +00001597}
1598
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001599static PyObject *
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001600bin_inplace_power(PyObject *v, PyObject *w)
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001601{
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001602 return PyNumber_InPlacePower(v, w, Py_None);
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001603}
1604
1605
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001606static PyObject *
1607instance_ipow(PyObject *v, PyObject *w, PyObject *z)
1608{
1609 if (z == Py_None) {
1610 return do_binop_inplace(v, w, "__ipow__", "__pow__",
1611 "__rpow__", bin_inplace_power);
1612 }
1613 else {
1614 /* XXX Doesn't do coercions... */
1615 PyObject *func;
1616 PyObject *args;
1617 PyObject *result;
1618
1619 func = PyObject_GetAttrString(v, "__ipow__");
1620 if (func == NULL) {
1621 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
1622 return NULL;
1623 PyErr_Clear();
1624 return instance_pow(v, w, z);
1625 }
1626 args = Py_BuildValue("(OO)", w, z);
1627 if (args == NULL) {
1628 Py_DECREF(func);
1629 return NULL;
1630 }
1631 result = PyEval_CallObject(func, args);
1632 Py_DECREF(func);
1633 Py_DECREF(args);
1634 return result;
1635 }
1636}
1637
1638
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001639/* Map rich comparison operators to their __xx__ namesakes */
1640static char *name_op[] = {
1641 "__lt__",
1642 "__le__",
1643 "__eq__",
1644 "__ne__",
1645 "__gt__",
1646 "__ge__",
1647};
1648
1649static PyObject *
1650half_richcompare(PyObject *v, PyObject *w, int op)
1651{
1652 PyObject *name;
1653 PyObject *method;
1654 PyObject *args;
1655 PyObject *res;
1656
1657 assert(PyInstance_Check(v));
1658
1659 name = PyString_InternFromString(name_op[op]);
1660 if (name == NULL)
1661 return NULL;
1662
1663 method = PyObject_GetAttr(v, name);
1664 Py_DECREF(name);
1665 if (method == NULL) {
1666 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
1667 return NULL;
1668 PyErr_Clear();
1669 res = Py_NotImplemented;
1670 Py_INCREF(res);
1671 return res;
1672 }
1673
1674 args = Py_BuildValue("(O)", w);
1675 if (args == NULL) {
1676 Py_DECREF(method);
1677 return NULL;
1678 }
1679
1680 res = PyEval_CallObject(method, args);
1681 Py_DECREF(args);
1682 Py_DECREF(method);
1683
1684 return res;
1685}
1686
1687/* Map rich comparison operators to their swapped version, e.g. LT --> GT */
1688static int swapped_op[] = {Py_GT, Py_GE, Py_EQ, Py_NE, Py_LT, Py_LE};
1689
1690static PyObject *
1691instance_richcompare(PyObject *v, PyObject *w, int op)
1692{
1693 PyObject *res;
1694
1695 if (PyInstance_Check(v)) {
1696 res = half_richcompare(v, w, op);
1697 if (res != Py_NotImplemented)
1698 return res;
1699 Py_DECREF(res);
1700 }
1701
1702 if (PyInstance_Check(w)) {
1703 res = half_richcompare(w, v, swapped_op[op]);
1704 if (res != Py_NotImplemented)
1705 return res;
1706 Py_DECREF(res);
1707 }
1708
1709 Py_INCREF(Py_NotImplemented);
1710 return Py_NotImplemented;
1711}
1712
Neil Schemenauer29bfc072001-01-04 01:43:46 +00001713
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001714static PyNumberMethods instance_as_number = {
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001715 (binaryfunc)instance_add, /* nb_add */
1716 (binaryfunc)instance_sub, /* nb_subtract */
1717 (binaryfunc)instance_mul, /* nb_multiply */
1718 (binaryfunc)instance_div, /* nb_divide */
1719 (binaryfunc)instance_mod, /* nb_remainder */
1720 (binaryfunc)instance_divmod, /* nb_divmod */
1721 (ternaryfunc)instance_pow, /* nb_power */
1722 (unaryfunc)instance_neg, /* nb_negative */
1723 (unaryfunc)instance_pos, /* nb_positive */
1724 (unaryfunc)instance_abs, /* nb_absolute */
1725 (inquiry)instance_nonzero, /* nb_nonzero */
1726 (unaryfunc)instance_invert, /* nb_invert */
1727 (binaryfunc)instance_lshift, /* nb_lshift */
1728 (binaryfunc)instance_rshift, /* nb_rshift */
1729 (binaryfunc)instance_and, /* nb_and */
1730 (binaryfunc)instance_xor, /* nb_xor */
1731 (binaryfunc)instance_or, /* nb_or */
1732 (coercion)instance_coerce, /* nb_coerce */
1733 (unaryfunc)instance_int, /* nb_int */
1734 (unaryfunc)instance_long, /* nb_long */
1735 (unaryfunc)instance_float, /* nb_float */
1736 (unaryfunc)instance_oct, /* nb_oct */
1737 (unaryfunc)instance_hex, /* nb_hex */
1738 (binaryfunc)instance_iadd, /* nb_inplace_add */
1739 (binaryfunc)instance_isub, /* nb_inplace_subtract */
1740 (binaryfunc)instance_imul, /* nb_inplace_multiply */
1741 (binaryfunc)instance_idiv, /* nb_inplace_divide */
1742 (binaryfunc)instance_imod, /* nb_inplace_remainder */
1743 (ternaryfunc)instance_ipow, /* nb_inplace_power */
1744 (binaryfunc)instance_ilshift, /* nb_inplace_lshift */
1745 (binaryfunc)instance_irshift, /* nb_inplace_rshift */
1746 (binaryfunc)instance_iand, /* nb_inplace_and */
1747 (binaryfunc)instance_ixor, /* nb_inplace_xor */
1748 (binaryfunc)instance_ior, /* nb_inplace_or */
Guido van Rossum04691fc1992-08-12 15:35:34 +00001749};
1750
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001751PyTypeObject PyInstance_Type = {
1752 PyObject_HEAD_INIT(&PyType_Type)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001753 0,
Guido van Rossum21ed88c1991-04-04 10:42:10 +00001754 "instance",
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001755 sizeof(PyInstanceObject) + PyGC_HEAD_SIZE,
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001756 0,
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001757 (destructor)instance_dealloc, /* tp_dealloc */
1758 0, /* tp_print */
1759 0, /* tp_getattr */
1760 0, /* tp_setattr */
1761 instance_compare, /* tp_compare */
1762 (reprfunc)instance_repr, /* tp_repr */
1763 &instance_as_number, /* tp_as_number */
1764 &instance_as_sequence, /* tp_as_sequence */
1765 &instance_as_mapping, /* tp_as_mapping */
1766 (hashfunc)instance_hash, /* tp_hash */
1767 0, /* tp_call */
1768 0, /* tp_str */
1769 (getattrofunc)instance_getattr, /* tp_getattro */
1770 (setattrofunc)instance_setattr, /* tp_setattro */
1771 0, /* tp_as_buffer */
1772 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_GC | Py_TPFLAGS_CHECKTYPES,/*tp_flags*/
1773 0, /* tp_doc */
Jeremy Hylton8caad492000-06-23 14:18:11 +00001774 (traverseproc)instance_traverse, /* tp_traverse */
Guido van Rossum8998b4f2001-01-17 15:28:20 +00001775 0, /* tp_clear */
1776 instance_richcompare, /* tp_richcompare */
Fred Drake41deb1e2001-02-01 05:27:45 +00001777 offsetof(PyInstanceObject, in_weakreflist) /* tp_weaklistoffset */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001778};
1779
1780
Guido van Rossum81daa321993-05-20 14:24:46 +00001781/* Instance method objects are used for two purposes:
1782 (a) as bound instance methods (returned by instancename.methodname)
1783 (b) as unbound methods (returned by ClassName.methodname)
1784 In case (b), im_self is NULL
1785*/
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001786
Guido van Rossuma0d349f1997-08-05 02:06:53 +00001787static PyMethodObject *free_list;
1788
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001789PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001790PyMethod_New(PyObject *func, PyObject *self, PyObject *class)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001791{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001792 register PyMethodObject *im;
Guido van Rossum7859f871998-07-08 14:58:16 +00001793 if (!PyCallable_Check(func)) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001794 PyErr_BadInternalCall();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001795 return NULL;
1796 }
Guido van Rossuma0d349f1997-08-05 02:06:53 +00001797 im = free_list;
1798 if (im != NULL) {
1799 free_list = (PyMethodObject *)(im->im_self);
Guido van Rossumb18618d2000-05-03 23:44:39 +00001800 PyObject_INIT(im, &PyMethod_Type);
Guido van Rossuma0d349f1997-08-05 02:06:53 +00001801 }
1802 else {
1803 im = PyObject_NEW(PyMethodObject, &PyMethod_Type);
1804 if (im == NULL)
1805 return NULL;
1806 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001807 Py_INCREF(func);
Guido van Rossume8122f11991-05-05 20:03:07 +00001808 im->im_func = func;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001809 Py_XINCREF(self);
Guido van Rossume8122f11991-05-05 20:03:07 +00001810 im->im_self = self;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001811 Py_INCREF(class);
Guido van Rossum81daa321993-05-20 14:24:46 +00001812 im->im_class = class;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001813 PyObject_GC_Init(im);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001814 return (PyObject *)im;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001815}
1816
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001817PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001818PyMethod_Function(register PyObject *im)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001819{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001820 if (!PyMethod_Check(im)) {
1821 PyErr_BadInternalCall();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001822 return NULL;
1823 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001824 return ((PyMethodObject *)im)->im_func;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001825}
1826
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001827PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001828PyMethod_Self(register PyObject *im)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001829{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001830 if (!PyMethod_Check(im)) {
1831 PyErr_BadInternalCall();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001832 return NULL;
1833 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001834 return ((PyMethodObject *)im)->im_self;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001835}
1836
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001837PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001838PyMethod_Class(register PyObject *im)
Guido van Rossum81daa321993-05-20 14:24:46 +00001839{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001840 if (!PyMethod_Check(im)) {
1841 PyErr_BadInternalCall();
Guido van Rossum81daa321993-05-20 14:24:46 +00001842 return NULL;
1843 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001844 return ((PyMethodObject *)im)->im_class;
Guido van Rossum81daa321993-05-20 14:24:46 +00001845}
1846
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001847/* Class method methods */
1848
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001849#define OFF(x) offsetof(PyMethodObject, x)
Guido van Rossum3f5da241990-12-20 15:06:42 +00001850
Guido van Rossume8122f11991-05-05 20:03:07 +00001851static struct memberlist instancemethod_memberlist[] = {
1852 {"im_func", T_OBJECT, OFF(im_func)},
1853 {"im_self", T_OBJECT, OFF(im_self)},
Guido van Rossum81daa321993-05-20 14:24:46 +00001854 {"im_class", T_OBJECT, OFF(im_class)},
Guido van Rossum6dabc981996-05-14 21:54:20 +00001855 /* Dummies that are not handled by getattr() except for __members__ */
1856 {"__doc__", T_INT, 0},
1857 {"__name__", T_INT, 0},
Barry Warsawd6a9e842001-01-15 20:40:19 +00001858 {"__dict__", T_OBJECT, 0},
Guido van Rossum3f5da241990-12-20 15:06:42 +00001859 {NULL} /* Sentinel */
1860};
1861
Barry Warsawd6a9e842001-01-15 20:40:19 +00001862static int
1863instancemethod_setattro(register PyMethodObject *im, PyObject *name,
1864 PyObject *v)
Guido van Rossum3f5da241990-12-20 15:06:42 +00001865{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001866 char *sname = PyString_AsString(name);
Barry Warsawd6a9e842001-01-15 20:40:19 +00001867
Barry Warsaw4f9b13b2001-02-26 18:09:15 +00001868 PyErr_Format(PyExc_TypeError, "read-only attribute: %s", sname);
1869 return -1;
Barry Warsawd6a9e842001-01-15 20:40:19 +00001870}
1871
1872
1873static PyObject *
1874instancemethod_getattro(register PyMethodObject *im, PyObject *name)
1875{
1876 PyObject *rtn;
1877 char *sname = PyString_AsString(name);
Guido van Rossum2878a691996-08-09 20:53:24 +00001878 if (sname[0] == '_') {
Guido van Rossum7859f871998-07-08 14:58:16 +00001879 /* Inherit __name__ and __doc__ from the callable object
1880 implementing the method */
1881 if (strcmp(sname, "__name__") == 0 ||
1882 strcmp(sname, "__doc__") == 0)
1883 return PyObject_GetAttr(im->im_func, name);
Guido van Rossum6dabc981996-05-14 21:54:20 +00001884 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001885 if (PyEval_GetRestricted()) {
1886 PyErr_SetString(PyExc_RuntimeError,
1887 "instance-method attributes not accessible in restricted mode");
Guido van Rossum10393b11995-01-10 10:39:49 +00001888 return NULL;
1889 }
Barry Warsawd6a9e842001-01-15 20:40:19 +00001890 if (sname[0] == '_' && strcmp(sname, "__dict__") == 0)
1891 return PyObject_GetAttr(im->im_func, name);
1892
1893 rtn = PyMember_Get((char *)im, instancemethod_memberlist, sname);
1894 if (rtn == NULL && PyErr_ExceptionMatches(PyExc_AttributeError)) {
1895 PyErr_Clear();
1896 rtn = PyObject_GetAttr(im->im_func, name);
1897 }
1898 return rtn;
Guido van Rossum3f5da241990-12-20 15:06:42 +00001899}
1900
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001901static void
Fred Drake79912472000-07-09 04:06:11 +00001902instancemethod_dealloc(register PyMethodObject *im)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001903{
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001904 PyObject_GC_Fini(im);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001905 Py_DECREF(im->im_func);
1906 Py_XDECREF(im->im_self);
1907 Py_DECREF(im->im_class);
Guido van Rossuma0d349f1997-08-05 02:06:53 +00001908 im->im_self = (PyObject *)free_list;
1909 free_list = im;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001910}
1911
Guido van Rossumebc8c511992-09-03 20:39:51 +00001912static int
Fred Drake79912472000-07-09 04:06:11 +00001913instancemethod_compare(PyMethodObject *a, PyMethodObject *b)
Guido van Rossumebc8c511992-09-03 20:39:51 +00001914{
Guido van Rossume9df7271995-04-06 14:46:51 +00001915 if (a->im_self != b->im_self)
1916 return (a->im_self < b->im_self) ? -1 : 1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001917 return PyObject_Compare(a->im_func, b->im_func);
Guido van Rossumebc8c511992-09-03 20:39:51 +00001918}
1919
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001920static PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001921instancemethod_repr(PyMethodObject *a)
Guido van Rossum25831651993-05-19 14:50:45 +00001922{
1923 char buf[240];
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001924 PyInstanceObject *self = (PyInstanceObject *)(a->im_self);
Guido van Rossum7859f871998-07-08 14:58:16 +00001925 PyObject *func = a->im_func;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001926 PyClassObject *class = (PyClassObject *)(a->im_class);
1927 PyObject *fclassname, *iclassname, *funcname;
Guido van Rossum81daa321993-05-20 14:24:46 +00001928 char *fcname, *icname, *fname;
1929 fclassname = class->cl_name;
Guido van Rossum7859f871998-07-08 14:58:16 +00001930 if (PyFunction_Check(func)) {
1931 funcname = ((PyFunctionObject *)func)->func_name;
1932 Py_INCREF(funcname);
1933 }
1934 else {
1935 funcname = PyObject_GetAttrString(func,"__name__");
1936 if (funcname == NULL)
1937 PyErr_Clear();
1938 }
1939 if (funcname != NULL && PyString_Check(funcname))
1940 fname = PyString_AS_STRING(funcname);
1941 else
1942 fname = "?";
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001943 if (fclassname != NULL && PyString_Check(fclassname))
1944 fcname = PyString_AsString(fclassname);
Guido van Rossum25831651993-05-19 14:50:45 +00001945 else
Guido van Rossum81daa321993-05-20 14:24:46 +00001946 fcname = "?";
Guido van Rossum81daa321993-05-20 14:24:46 +00001947 if (self == NULL)
1948 sprintf(buf, "<unbound method %.100s.%.100s>", fcname, fname);
1949 else {
1950 iclassname = self->in_class->cl_name;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001951 if (iclassname != NULL && PyString_Check(iclassname))
1952 icname = PyString_AsString(iclassname);
Guido van Rossum81daa321993-05-20 14:24:46 +00001953 else
1954 icname = "?";
Fred Drakea44d3532000-06-30 15:01:00 +00001955 sprintf(buf, "<method %.60s.%.60s of %.60s instance at %p>",
1956 fcname, fname, icname, self);
Guido van Rossum81daa321993-05-20 14:24:46 +00001957 }
Guido van Rossum42636dc1999-10-11 14:03:12 +00001958 Py_XDECREF(funcname);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001959 return PyString_FromString(buf);
Guido van Rossum25831651993-05-19 14:50:45 +00001960}
1961
Guido van Rossum9bfef441993-03-29 10:43:31 +00001962static long
Fred Drake79912472000-07-09 04:06:11 +00001963instancemethod_hash(PyMethodObject *a)
Guido van Rossum9bfef441993-03-29 10:43:31 +00001964{
1965 long x, y;
Guido van Rossum81daa321993-05-20 14:24:46 +00001966 if (a->im_self == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001967 x = PyObject_Hash(Py_None);
Guido van Rossum81daa321993-05-20 14:24:46 +00001968 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001969 x = PyObject_Hash(a->im_self);
Guido van Rossum9bfef441993-03-29 10:43:31 +00001970 if (x == -1)
1971 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001972 y = PyObject_Hash(a->im_func);
Guido van Rossum9bfef441993-03-29 10:43:31 +00001973 if (y == -1)
1974 return -1;
1975 return x ^ y;
1976}
1977
Jeremy Hylton8caad492000-06-23 14:18:11 +00001978static int
1979instancemethod_traverse(PyMethodObject *im, visitproc visit, void *arg)
1980{
1981 int err;
1982 if (im->im_func) {
1983 err = visit(im->im_func, arg);
1984 if (err)
1985 return err;
1986 }
1987 if (im->im_self) {
1988 err = visit(im->im_self, arg);
1989 if (err)
1990 return err;
1991 }
1992 if (im->im_class) {
1993 err = visit(im->im_class, arg);
1994 if (err)
1995 return err;
1996 }
Jeremy Hyltond22162b2000-06-23 17:14:56 +00001997 return 0;
Jeremy Hylton8caad492000-06-23 14:18:11 +00001998}
1999
Guido van Rossumc0b618a1997-05-02 03:12:38 +00002000PyTypeObject PyMethod_Type = {
2001 PyObject_HEAD_INIT(&PyType_Type)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002002 0,
Guido van Rossum569fce71991-04-16 08:38:43 +00002003 "instance method",
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00002004 sizeof(PyMethodObject) + PyGC_HEAD_SIZE,
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002005 0,
Guido van Rossum8998b4f2001-01-17 15:28:20 +00002006 (destructor)instancemethod_dealloc, /* tp_dealloc */
2007 0, /* tp_print */
2008 0, /* tp_getattr */
2009 0, /* tp_setattr */
2010 (cmpfunc)instancemethod_compare, /* tp_compare */
2011 (reprfunc)instancemethod_repr, /* tp_repr */
2012 0, /* tp_as_number */
2013 0, /* tp_as_sequence */
2014 0, /* tp_as_mapping */
2015 (hashfunc)instancemethod_hash, /* tp_hash */
2016 0, /* tp_call */
2017 0, /* tp_str */
2018 (getattrofunc)instancemethod_getattro, /* tp_getattro */
2019 (setattrofunc)instancemethod_setattro, /* tp_setattro */
2020 0, /* tp_as_buffer */
2021 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_GC, /* tp_flags */
2022 0, /* tp_doc */
Jeremy Hylton8caad492000-06-23 14:18:11 +00002023 (traverseproc)instancemethod_traverse, /* tp_traverse */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002024};
Guido van Rossuma0d349f1997-08-05 02:06:53 +00002025
2026/* Clear out the free list */
2027
2028void
Fred Drake79912472000-07-09 04:06:11 +00002029PyMethod_Fini(void)
Guido van Rossuma0d349f1997-08-05 02:06:53 +00002030{
2031 while (free_list) {
Guido van Rossumb18618d2000-05-03 23:44:39 +00002032 PyMethodObject *im = free_list;
2033 free_list = (PyMethodObject *)(im->im_self);
Guido van Rossum4cc6ac72000-07-01 01:00:38 +00002034 im = (PyMethodObject *) PyObject_AS_GC(im);
Guido van Rossumb18618d2000-05-03 23:44:39 +00002035 PyObject_DEL(im);
Guido van Rossuma0d349f1997-08-05 02:06:53 +00002036 }
2037}