blob: 8bbd834cdbda1386bcb309b989b0bbf911ddc54d [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
Guido van Rossum52ca98a1994-09-05 07:32:29 +00007/* Forward */
Tim Petersdbd9ba62000-07-09 03:09:57 +00008static PyObject *class_lookup(PyClassObject *, PyObject *,
9 PyClassObject **);
10static PyObject *instance_getattr1(PyInstanceObject *, PyObject *);
11static PyObject *instance_getattr2(PyInstanceObject *, PyObject *);
Guido van Rossum52ca98a1994-09-05 07:32:29 +000012
Guido van Rossuma63eff61998-05-29 21:37:21 +000013static PyObject *getattrstr, *setattrstr, *delattrstr;
14
Fred Drake79912472000-07-09 04:06:11 +000015
Guido van Rossumc0b618a1997-05-02 03:12:38 +000016PyObject *
Fred Drake79912472000-07-09 04:06:11 +000017PyClass_New(PyObject *bases, PyObject *dict, PyObject *name)
18 /* bases is NULL or tuple of classobjects! */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000019{
Guido van Rossumc0b618a1997-05-02 03:12:38 +000020 PyClassObject *op, *dummy;
Guido van Rossum7cc56eb1997-09-12 20:04:46 +000021 static PyObject *docstr, *modstr, *namestr;
Guido van Rossum019f4241996-08-21 14:54:28 +000022 if (docstr == NULL) {
Guido van Rossuma412d241997-01-18 07:59:12 +000023 docstr= PyString_InternFromString("__doc__");
Guido van Rossum019f4241996-08-21 14:54:28 +000024 if (docstr == NULL)
25 return NULL;
26 }
Guido van Rossum7cc56eb1997-09-12 20:04:46 +000027 if (modstr == NULL) {
28 modstr= PyString_InternFromString("__module__");
29 if (modstr == NULL)
30 return NULL;
31 }
32 if (namestr == NULL) {
33 namestr= PyString_InternFromString("__name__");
34 if (namestr == NULL)
35 return NULL;
36 }
Guido van Rossum04d73c41997-10-07 14:54:11 +000037 if (name == NULL || !PyString_Check(name)) {
38 PyErr_SetString(PyExc_SystemError,
39 "PyClass_New: name must be a string");
40 return NULL;
41 }
42 if (dict == NULL || !PyDict_Check(dict)) {
43 PyErr_SetString(PyExc_SystemError,
44 "PyClass_New: dict must be a dictionary");
45 return NULL;
46 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +000047 if (PyDict_GetItem(dict, docstr) == NULL) {
48 if (PyDict_SetItem(dict, docstr, Py_None) < 0)
Guido van Rossume7d444f1995-01-07 12:35:18 +000049 return NULL;
50 }
Guido van Rossum7cc56eb1997-09-12 20:04:46 +000051 if (PyDict_GetItem(dict, modstr) == NULL) {
52 PyObject *globals = PyEval_GetGlobals();
53 if (globals != NULL) {
Guido van Rossum04d73c41997-10-07 14:54:11 +000054 PyObject *modname = PyDict_GetItem(globals, namestr);
55 if (modname != NULL) {
56 if (PyDict_SetItem(dict, modstr, modname) < 0)
Guido van Rossum7cc56eb1997-09-12 20:04:46 +000057 return NULL;
58 }
59 }
60 }
Guido van Rossume2966a61991-12-10 13:53:23 +000061 if (bases == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +000062 bases = PyTuple_New(0);
Guido van Rossume2966a61991-12-10 13:53:23 +000063 if (bases == NULL)
Guido van Rossume5372401993-03-16 12:15:04 +000064 return NULL;
Guido van Rossume2966a61991-12-10 13:53:23 +000065 }
Guido van Rossum04d73c41997-10-07 14:54:11 +000066 else {
67 int i;
68 if (!PyTuple_Check(bases)) {
69 PyErr_SetString(PyExc_SystemError,
70 "PyClass_New: bases must be a tuple");
71 return NULL;
72 }
73 i = PyTuple_Size(bases);
74 while (--i >= 0) {
75 if (!PyClass_Check(PyTuple_GetItem(bases, i))) {
76 PyErr_SetString(PyExc_SystemError,
77 "PyClass_New: base must be a class");
78 return NULL;
79 }
80 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +000081 Py_INCREF(bases);
Guido van Rossum04d73c41997-10-07 14:54:11 +000082 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +000083 op = PyObject_NEW(PyClassObject, &PyClass_Type);
Guido van Rossume2966a61991-12-10 13:53:23 +000084 if (op == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +000085 Py_DECREF(bases);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000086 return NULL;
Guido van Rossume2966a61991-12-10 13:53:23 +000087 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000088 op->cl_bases = bases;
Guido van Rossumc0b618a1997-05-02 03:12:38 +000089 Py_INCREF(dict);
Guido van Rossum81daa321993-05-20 14:24:46 +000090 op->cl_dict = dict;
Guido van Rossumc0b618a1997-05-02 03:12:38 +000091 Py_XINCREF(name);
Guido van Rossum94308391991-10-20 20:11:48 +000092 op->cl_name = name;
Guido van Rossum2878a691996-08-09 20:53:24 +000093 if (getattrstr == NULL) {
Guido van Rossuma412d241997-01-18 07:59:12 +000094 getattrstr = PyString_InternFromString("__getattr__");
95 setattrstr = PyString_InternFromString("__setattr__");
96 delattrstr = PyString_InternFromString("__delattr__");
Guido van Rossum2878a691996-08-09 20:53:24 +000097 }
98 op->cl_getattr = class_lookup(op, getattrstr, &dummy);
99 op->cl_setattr = class_lookup(op, setattrstr, &dummy);
100 op->cl_delattr = class_lookup(op, delattrstr, &dummy);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000101 Py_XINCREF(op->cl_getattr);
102 Py_XINCREF(op->cl_setattr);
103 Py_XINCREF(op->cl_delattr);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000104 PyObject_GC_Init(op);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000105 return (PyObject *) op;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000106}
107
108/* Class methods */
109
110static void
Fred Drake79912472000-07-09 04:06:11 +0000111class_dealloc(PyClassObject *op)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000112{
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000113 PyObject_GC_Fini(op);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000114 Py_DECREF(op->cl_bases);
115 Py_DECREF(op->cl_dict);
116 Py_XDECREF(op->cl_name);
Guido van Rossum152d8171998-08-04 14:59:16 +0000117 Py_XDECREF(op->cl_getattr);
118 Py_XDECREF(op->cl_setattr);
119 Py_XDECREF(op->cl_delattr);
Guido van Rossum4cc6ac72000-07-01 01:00:38 +0000120 op = (PyClassObject *) PyObject_AS_GC(op);
Guido van Rossumb18618d2000-05-03 23:44:39 +0000121 PyObject_DEL(op);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000122}
123
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000124static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000125class_lookup(PyClassObject *cp, PyObject *name, PyClassObject **pclass)
Guido van Rossum81daa321993-05-20 14:24:46 +0000126{
127 int i, n;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000128 PyObject *value = PyDict_GetItem(cp->cl_dict, name);
Guido van Rossum81daa321993-05-20 14:24:46 +0000129 if (value != NULL) {
130 *pclass = cp;
131 return value;
132 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000133 n = PyTuple_Size(cp->cl_bases);
Guido van Rossum81daa321993-05-20 14:24:46 +0000134 for (i = 0; i < n; i++) {
Guido van Rossum7cc56eb1997-09-12 20:04:46 +0000135 /* XXX What if one of the bases is not a class? */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000136 PyObject *v = class_lookup(
137 (PyClassObject *)
138 PyTuple_GetItem(cp->cl_bases, i), name, pclass);
Guido van Rossum81daa321993-05-20 14:24:46 +0000139 if (v != NULL)
140 return v;
141 }
142 return NULL;
143}
144
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000145static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000146class_getattr(register PyClassObject *op, PyObject *name)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000147{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000148 register PyObject *v;
149 register char *sname = PyString_AsString(name);
150 PyClassObject *class;
Guido van Rossum2878a691996-08-09 20:53:24 +0000151 if (sname[0] == '_' && sname[1] == '_') {
152 if (strcmp(sname, "__dict__") == 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000153 if (PyEval_GetRestricted()) {
154 PyErr_SetString(PyExc_RuntimeError,
155 "class.__dict__ not accessible in restricted mode");
Guido van Rossum10393b11995-01-10 10:39:49 +0000156 return NULL;
157 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000158 Py_INCREF(op->cl_dict);
Guido van Rossum10393b11995-01-10 10:39:49 +0000159 return op->cl_dict;
160 }
Guido van Rossum2878a691996-08-09 20:53:24 +0000161 if (strcmp(sname, "__bases__") == 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000162 Py_INCREF(op->cl_bases);
Guido van Rossum10393b11995-01-10 10:39:49 +0000163 return op->cl_bases;
164 }
Guido van Rossum2878a691996-08-09 20:53:24 +0000165 if (strcmp(sname, "__name__") == 0) {
Guido van Rossum10393b11995-01-10 10:39:49 +0000166 if (op->cl_name == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000167 v = Py_None;
Guido van Rossum10393b11995-01-10 10:39:49 +0000168 else
169 v = op->cl_name;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000170 Py_INCREF(v);
Guido van Rossum10393b11995-01-10 10:39:49 +0000171 return v;
172 }
Guido van Rossum94308391991-10-20 20:11:48 +0000173 }
Guido van Rossum81daa321993-05-20 14:24:46 +0000174 v = class_lookup(op, name, &class);
Guido van Rossumb3f72581993-05-21 19:56:10 +0000175 if (v == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000176 PyErr_SetObject(PyExc_AttributeError, name);
Guido van Rossumb3f72581993-05-21 19:56:10 +0000177 return NULL;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000178 }
Guido van Rossum0add15f1997-05-09 01:07:15 +0000179 Py_INCREF(v);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000180 if (PyFunction_Check(v)) {
181 PyObject *w = PyMethod_New(v, (PyObject *)NULL,
182 (PyObject *)class);
183 Py_DECREF(v);
Guido van Rossumb3f72581993-05-21 19:56:10 +0000184 v = w;
185 }
186 return v;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000187}
188
Guido van Rossuma63eff61998-05-29 21:37:21 +0000189static void
Fred Drake79912472000-07-09 04:06:11 +0000190set_slot(PyObject **slot, PyObject *v)
Guido van Rossuma63eff61998-05-29 21:37:21 +0000191{
192 PyObject *temp = *slot;
193 Py_XINCREF(v);
194 *slot = v;
195 Py_XDECREF(temp);
196}
197
Guido van Rossum7ba30431998-07-08 13:34:48 +0000198static void
Fred Drake79912472000-07-09 04:06:11 +0000199set_attr_slots(PyClassObject *c)
Guido van Rossum7ba30431998-07-08 13:34:48 +0000200{
201 PyClassObject *dummy;
202
203 set_slot(&c->cl_getattr, class_lookup(c, getattrstr, &dummy));
204 set_slot(&c->cl_setattr, class_lookup(c, setattrstr, &dummy));
205 set_slot(&c->cl_delattr, class_lookup(c, delattrstr, &dummy));
206}
207
Guido van Rossuma63eff61998-05-29 21:37:21 +0000208static char *
Fred Drake79912472000-07-09 04:06:11 +0000209set_dict(PyClassObject *c, PyObject *v)
Guido van Rossuma63eff61998-05-29 21:37:21 +0000210{
Guido van Rossuma63eff61998-05-29 21:37:21 +0000211 if (v == NULL || !PyDict_Check(v))
212 return "__dict__ must be a dictionary object";
213 set_slot(&c->cl_dict, v);
Guido van Rossum7ba30431998-07-08 13:34:48 +0000214 set_attr_slots(c);
Guido van Rossuma63eff61998-05-29 21:37:21 +0000215 return "";
216}
217
218static char *
Fred Drake79912472000-07-09 04:06:11 +0000219set_bases(PyClassObject *c, PyObject *v)
Guido van Rossuma63eff61998-05-29 21:37:21 +0000220{
Guido van Rossuma63eff61998-05-29 21:37:21 +0000221 int i, n;
222
223 if (v == NULL || !PyTuple_Check(v))
224 return "__bases__ must be a tuple object";
225 n = PyTuple_Size(v);
226 for (i = 0; i < n; i++) {
227 PyObject *x = PyTuple_GET_ITEM(v, i);
228 if (!PyClass_Check(x))
229 return "__bases__ items must be classes";
230 if (PyClass_IsSubclass(x, (PyObject *)c))
231 return "a __bases__ item causes an inheritance cycle";
232 }
233 set_slot(&c->cl_bases, v);
Guido van Rossum7ba30431998-07-08 13:34:48 +0000234 set_attr_slots(c);
Guido van Rossuma63eff61998-05-29 21:37:21 +0000235 return "";
236}
237
238static char *
Fred Drake79912472000-07-09 04:06:11 +0000239set_name(PyClassObject *c, PyObject *v)
Guido van Rossuma63eff61998-05-29 21:37:21 +0000240{
Guido van Rossuma63eff61998-05-29 21:37:21 +0000241 if (v == NULL || !PyString_Check(v))
242 return "__name__ must be a string object";
Guido van Rossumad89bbc2000-06-28 21:57:18 +0000243 if (strlen(PyString_AS_STRING(v)) != (size_t)PyString_GET_SIZE(v))
Guido van Rossuma63eff61998-05-29 21:37:21 +0000244 return "__name__ must not contain null bytes";
245 set_slot(&c->cl_name, v);
246 return "";
247}
248
Guido van Rossum94308391991-10-20 20:11:48 +0000249static int
Fred Drake79912472000-07-09 04:06:11 +0000250class_setattr(PyClassObject *op, PyObject *name, PyObject *v)
Guido van Rossum94308391991-10-20 20:11:48 +0000251{
Guido van Rossumb2173c31997-08-25 21:23:56 +0000252 char *sname;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000253 if (PyEval_GetRestricted()) {
254 PyErr_SetString(PyExc_RuntimeError,
Guido van Rossumac21f6a1995-08-04 04:05:31 +0000255 "classes are read-only in restricted mode");
256 return -1;
257 }
Guido van Rossumb2173c31997-08-25 21:23:56 +0000258 sname = PyString_AsString(name);
259 if (sname[0] == '_' && sname[1] == '_') {
260 int n = PyString_Size(name);
261 if (sname[n-1] == '_' && sname[n-2] == '_') {
Guido van Rossuma63eff61998-05-29 21:37:21 +0000262 char *err = NULL;
263 if (strcmp(sname, "__dict__") == 0)
264 err = set_dict(op, v);
265 else if (strcmp(sname, "__bases__") == 0)
266 err = set_bases(op, v);
267 else if (strcmp(sname, "__name__") == 0)
268 err = set_name(op, v);
269 else if (strcmp(sname, "__getattr__") == 0)
270 set_slot(&op->cl_getattr, v);
271 else if (strcmp(sname, "__setattr__") == 0)
272 set_slot(&op->cl_setattr, v);
273 else if (strcmp(sname, "__delattr__") == 0)
274 set_slot(&op->cl_delattr, v);
275 /* For the last three, we fall through to update the
276 dictionary as well. */
277 if (err != NULL) {
278 if (*err == '\0')
279 return 0;
280 PyErr_SetString(PyExc_TypeError, err);
Guido van Rossumb2173c31997-08-25 21:23:56 +0000281 return -1;
282 }
283 }
284 }
Guido van Rossum94472a01992-09-04 09:45:18 +0000285 if (v == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000286 int rv = PyDict_DelItem(op->cl_dict, name);
Guido van Rossum94472a01992-09-04 09:45:18 +0000287 if (rv < 0)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000288 PyErr_SetString(PyExc_AttributeError,
Guido van Rossum94472a01992-09-04 09:45:18 +0000289 "delete non-existing class attribute");
290 return rv;
291 }
Guido van Rossum94308391991-10-20 20:11:48 +0000292 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000293 return PyDict_SetItem(op->cl_dict, name, v);
Guido van Rossum94308391991-10-20 20:11:48 +0000294}
295
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000296static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000297class_repr(PyClassObject *op)
Guido van Rossum25831651993-05-19 14:50:45 +0000298{
Guido van Rossum4a2a6211997-10-20 23:26:11 +0000299 PyObject *mod = PyDict_GetItemString(op->cl_dict, "__module__");
Guido van Rossum25831651993-05-19 14:50:45 +0000300 char buf[140];
301 char *name;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000302 if (op->cl_name == NULL || !PyString_Check(op->cl_name))
Guido van Rossum25831651993-05-19 14:50:45 +0000303 name = "?";
304 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000305 name = PyString_AsString(op->cl_name);
Guido van Rossum4a2a6211997-10-20 23:26:11 +0000306 if (mod == NULL || !PyString_Check(mod))
Fred Drakea44d3532000-06-30 15:01:00 +0000307 sprintf(buf, "<class ?.%.100s at %p>", name, op);
Guido van Rossum4a2a6211997-10-20 23:26:11 +0000308 else
Fred Drakea44d3532000-06-30 15:01:00 +0000309 sprintf(buf, "<class %.50s.%.50s at %p>",
Guido van Rossum4a2a6211997-10-20 23:26:11 +0000310 PyString_AsString(mod),
Fred Drakea44d3532000-06-30 15:01:00 +0000311 name, op);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000312 return PyString_FromString(buf);
Guido van Rossum25831651993-05-19 14:50:45 +0000313}
314
Guido van Rossum4a2a6211997-10-20 23:26:11 +0000315static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000316class_str(PyClassObject *op)
Guido van Rossum4a2a6211997-10-20 23:26:11 +0000317{
318 PyObject *mod = PyDict_GetItemString(op->cl_dict, "__module__");
319 PyObject *name = op->cl_name;
320 PyObject *res;
321 int m, n;
322
323 if (name == NULL || !PyString_Check(name))
324 return class_repr(op);
325 if (mod == NULL || !PyString_Check(mod)) {
326 Py_INCREF(name);
327 return name;
328 }
329 m = PyString_Size(mod);
330 n = PyString_Size(name);
331 res = PyString_FromStringAndSize((char *)NULL, m+1+n);
332 if (res != NULL) {
333 char *s = PyString_AsString(res);
334 memcpy(s, PyString_AsString(mod), m);
335 s += m;
336 *s++ = '.';
337 memcpy(s, PyString_AsString(name), n);
338 }
339 return res;
340}
341
Jeremy Hylton8caad492000-06-23 14:18:11 +0000342static int
343class_traverse(PyClassObject *o, visitproc visit, void *arg)
344{
345 int err;
346 if (o->cl_bases) {
347 err = visit(o->cl_bases, arg);
348 if (err)
349 return err;
350 }
351 if (o->cl_dict) {
352 err = visit(o->cl_dict, arg);
353 if (err)
354 return err;
355 }
356 if (o->cl_name) {
357 err = visit(o->cl_name, arg);
358 if (err)
359 return err;
360 }
361 if (o->cl_getattr) {
362 err = visit(o->cl_getattr, arg);
363 if (err)
364 return err;
365 }
366 if (o->cl_setattr) {
367 err = visit(o->cl_setattr, arg);
368 if (err)
369 return err;
370 }
371 if (o->cl_delattr) {
372 err = visit(o->cl_delattr, arg);
373 if (err)
374 return err;
375 }
376 return 0;
377}
378
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000379PyTypeObject PyClass_Type = {
380 PyObject_HEAD_INIT(&PyType_Type)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000381 0,
382 "class",
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000383 sizeof(PyClassObject) + PyGC_HEAD_SIZE,
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000384 0,
Guido van Rossumb6775db1994-08-01 11:34:53 +0000385 (destructor)class_dealloc, /*tp_dealloc*/
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000386 0, /*tp_print*/
Guido van Rossum2878a691996-08-09 20:53:24 +0000387 0, /*tp_getattr*/
388 0, /*tp_setattr*/
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000389 0, /*tp_compare*/
Guido van Rossumb6775db1994-08-01 11:34:53 +0000390 (reprfunc)class_repr, /*tp_repr*/
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000391 0, /*tp_as_number*/
392 0, /*tp_as_sequence*/
393 0, /*tp_as_mapping*/
Guido van Rossum2878a691996-08-09 20:53:24 +0000394 0, /*tp_hash*/
395 0, /*tp_call*/
Guido van Rossum4a2a6211997-10-20 23:26:11 +0000396 (reprfunc)class_str, /*tp_str*/
Guido van Rossum2878a691996-08-09 20:53:24 +0000397 (getattrofunc)class_getattr, /*tp_getattro*/
398 (setattrofunc)class_setattr, /*tp_setattro*/
Jeremy Hylton8caad492000-06-23 14:18:11 +0000399 0, /* tp_as_buffer */
Jeremy Hyltond08b4c42000-06-23 19:37:02 +0000400 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_GC, /*tp_flags*/
Jeremy Hylton8caad492000-06-23 14:18:11 +0000401 0, /* tp_doc */
402 (traverseproc)class_traverse, /* tp_traverse */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000403};
404
Guido van Rossum81daa321993-05-20 14:24:46 +0000405int
Fred Drake79912472000-07-09 04:06:11 +0000406PyClass_IsSubclass(PyObject *class, PyObject *base)
Guido van Rossum81daa321993-05-20 14:24:46 +0000407{
408 int i, n;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000409 PyClassObject *cp;
Guido van Rossum81daa321993-05-20 14:24:46 +0000410 if (class == base)
411 return 1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000412 if (class == NULL || !PyClass_Check(class))
Guido van Rossumeb6b33a1993-05-25 09:38:27 +0000413 return 0;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000414 cp = (PyClassObject *)class;
415 n = PyTuple_Size(cp->cl_bases);
Guido van Rossum81daa321993-05-20 14:24:46 +0000416 for (i = 0; i < n; i++) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000417 if (PyClass_IsSubclass(PyTuple_GetItem(cp->cl_bases, i), base))
Guido van Rossum81daa321993-05-20 14:24:46 +0000418 return 1;
419 }
420 return 0;
421}
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000422
Guido van Rossum81daa321993-05-20 14:24:46 +0000423
424/* Instance objects */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000425
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000426PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000427PyInstance_New(PyObject *class, PyObject *arg, PyObject *kw)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000428{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000429 register PyInstanceObject *inst;
430 PyObject *init;
431 static PyObject *initstr;
432 if (!PyClass_Check(class)) {
433 PyErr_BadInternalCall();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000434 return NULL;
435 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000436 inst = PyObject_NEW(PyInstanceObject, &PyInstance_Type);
Guido van Rossume8122f11991-05-05 20:03:07 +0000437 if (inst == NULL)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000438 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000439 inst->in_dict = PyDict_New();
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000440 PyObject_GC_Init(inst);
Guido van Rossum0add15f1997-05-09 01:07:15 +0000441 if (inst->in_dict == NULL) {
Guido van Rossum4cc6ac72000-07-01 01:00:38 +0000442 inst = (PyInstanceObject *) PyObject_AS_GC(inst);
Guido van Rossumd7823f22000-06-28 23:46:07 +0000443 PyObject_DEL(inst);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000444 return NULL;
445 }
Guido van Rossumd7823f22000-06-28 23:46:07 +0000446 Py_INCREF(class);
447 inst->in_class = (PyClassObject *)class;
Guido van Rossum2878a691996-08-09 20:53:24 +0000448 if (initstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000449 initstr = PyString_InternFromString("__init__");
Jeremy Hylton9e392e22000-04-26 20:39:20 +0000450 init = instance_getattr2(inst, initstr);
Guido van Rossum25831651993-05-19 14:50:45 +0000451 if (init == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000452 if ((arg != NULL && (!PyTuple_Check(arg) ||
453 PyTuple_Size(arg) != 0))
454 || (kw != NULL && (!PyDict_Check(kw) ||
455 PyDict_Size(kw) != 0))) {
456 PyErr_SetString(PyExc_TypeError,
Guido van Rossuma83f2701995-07-26 18:07:32 +0000457 "this constructor takes no arguments");
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000458 Py_DECREF(inst);
Guido van Rossum25831651993-05-19 14:50:45 +0000459 inst = NULL;
460 }
461 }
462 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000463 PyObject *res = PyEval_CallObjectWithKeywords(init, arg, kw);
464 Py_DECREF(init);
Guido van Rossum25831651993-05-19 14:50:45 +0000465 if (res == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000466 Py_DECREF(inst);
Guido van Rossum25831651993-05-19 14:50:45 +0000467 inst = NULL;
468 }
469 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000470 if (res != Py_None) {
471 PyErr_SetString(PyExc_TypeError,
Guido van Rossum25831651993-05-19 14:50:45 +0000472 "__init__() should return None");
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000473 Py_DECREF(inst);
Guido van Rossum25831651993-05-19 14:50:45 +0000474 inst = NULL;
475 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000476 Py_DECREF(res);
Guido van Rossum25831651993-05-19 14:50:45 +0000477 }
478 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000479 return (PyObject *)inst;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000480}
481
Guido van Rossum21ed88c1991-04-04 10:42:10 +0000482/* Instance methods */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000483
484static void
Fred Drake79912472000-07-09 04:06:11 +0000485instance_dealloc(register PyInstanceObject *inst)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000486{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000487 PyObject *error_type, *error_value, *error_traceback;
488 PyObject *del;
489 static PyObject *delstr;
Skip Montanaro4ca150b2000-07-08 12:04:57 +0000490#ifdef Py_TRACE_REFS
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000491 extern long _Py_RefTotal;
Skip Montanaro4ca150b2000-07-08 12:04:57 +0000492#endif
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000493 PyObject_GC_Fini(inst);
Guido van Rossum25831651993-05-19 14:50:45 +0000494 /* Call the __del__ method if it exists. First temporarily
495 revive the object and save the current exception, if any. */
Guido van Rossum441e4ab1996-05-23 22:46:51 +0000496#ifdef Py_TRACE_REFS
497 /* much too complicated if Py_TRACE_REFS defined */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000498 inst->ob_type = &PyInstance_Type;
Guido van Rossumbffd6832000-01-20 22:32:56 +0000499 _Py_NewReference((PyObject *)inst);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000500 _Py_RefTotal--; /* compensate for increment in NEWREF */
Sjoerd Mullender740f3571995-08-28 09:00:43 +0000501#ifdef COUNT_ALLOCS
502 inst->ob_type->tp_alloc--; /* ditto */
503#endif
Guido van Rossum441e4ab1996-05-23 22:46:51 +0000504#else /* !Py_TRACE_REFS */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000505 Py_INCREF(inst);
Guido van Rossum441e4ab1996-05-23 22:46:51 +0000506#endif /* !Py_TRACE_REFS */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000507 PyErr_Fetch(&error_type, &error_value, &error_traceback);
Guido van Rossum2878a691996-08-09 20:53:24 +0000508 if (delstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000509 delstr = PyString_InternFromString("__del__");
Jeremy Hylton9e392e22000-04-26 20:39:20 +0000510 if ((del = instance_getattr2(inst, delstr)) != NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000511 PyObject *res = PyEval_CallObject(del, (PyObject *)NULL);
Guido van Rossum0dfcf751996-08-12 22:00:53 +0000512 if (res == NULL) {
Jeremy Hyltonb709df32000-09-01 02:47:25 +0000513 PyErr_WriteUnraisable(del);
Guido van Rossum0dfcf751996-08-12 22:00:53 +0000514 }
515 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000516 Py_DECREF(res);
517 Py_DECREF(del);
Guido van Rossum25831651993-05-19 14:50:45 +0000518 }
519 /* Restore the saved exception and undo the temporary revival */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000520 PyErr_Restore(error_type, error_value, error_traceback);
Guido van Rossum25831651993-05-19 14:50:45 +0000521 /* Can't use DECREF here, it would cause a recursive call */
Sjoerd Mullender740f3571995-08-28 09:00:43 +0000522 if (--inst->ob_refcnt > 0) {
523#ifdef COUNT_ALLOCS
524 inst->ob_type->tp_free--;
525#endif
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000526 PyObject_GC_Init((PyObject *)inst);
Guido van Rossum25831651993-05-19 14:50:45 +0000527 return; /* __del__ added a reference; don't delete now */
Sjoerd Mullender740f3571995-08-28 09:00:43 +0000528 }
Guido van Rossum441e4ab1996-05-23 22:46:51 +0000529#ifdef Py_TRACE_REFS
Sjoerd Mullender740f3571995-08-28 09:00:43 +0000530#ifdef COUNT_ALLOCS
531 inst->ob_type->tp_free--; /* compensate for increment in UNREF */
532#endif
Guido van Rossumbffd6832000-01-20 22:32:56 +0000533 _Py_ForgetReference((PyObject *)inst);
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000534#ifndef WITH_CYCLE_GC
Sjoerd Mullender740f3571995-08-28 09:00:43 +0000535 inst->ob_type = NULL;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000536#endif
Guido van Rossum441e4ab1996-05-23 22:46:51 +0000537#endif /* Py_TRACE_REFS */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000538 Py_DECREF(inst->in_class);
539 Py_XDECREF(inst->in_dict);
Guido van Rossum4cc6ac72000-07-01 01:00:38 +0000540 inst = (PyInstanceObject *) PyObject_AS_GC(inst);
Guido van Rossumb18618d2000-05-03 23:44:39 +0000541 PyObject_DEL(inst);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000542}
543
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000544static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000545instance_getattr1(register PyInstanceObject *inst, PyObject *name)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000546{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000547 register PyObject *v;
548 register char *sname = PyString_AsString(name);
Guido van Rossum2878a691996-08-09 20:53:24 +0000549 if (sname[0] == '_' && sname[1] == '_') {
550 if (strcmp(sname, "__dict__") == 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000551 if (PyEval_GetRestricted()) {
552 PyErr_SetString(PyExc_RuntimeError,
553 "instance.__dict__ not accessible in restricted mode");
Guido van Rossum10393b11995-01-10 10:39:49 +0000554 return NULL;
555 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000556 Py_INCREF(inst->in_dict);
Sjoerd Mullender3bb8a051993-10-22 12:04:32 +0000557 return inst->in_dict;
558 }
Guido van Rossum2878a691996-08-09 20:53:24 +0000559 if (strcmp(sname, "__class__") == 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000560 Py_INCREF(inst->in_class);
561 return (PyObject *)inst->in_class;
Sjoerd Mullender3bb8a051993-10-22 12:04:32 +0000562 }
Guido van Rossum94308391991-10-20 20:11:48 +0000563 }
Jeremy Hylton9e392e22000-04-26 20:39:20 +0000564 v = instance_getattr2(inst, name);
565 if (v == NULL) {
566 PyErr_Format(PyExc_AttributeError,"'%.50s' instance has no attribute '%.400s'",
567 PyString_AS_STRING(inst->in_class->cl_name), sname);
568 }
569 return v;
570}
571
572static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000573instance_getattr2(register PyInstanceObject *inst, PyObject *name)
Jeremy Hylton9e392e22000-04-26 20:39:20 +0000574{
575 register PyObject *v;
576 PyClassObject *class;
Guido van Rossumeb6b33a1993-05-25 09:38:27 +0000577 class = NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000578 v = PyDict_GetItem(inst->in_dict, name);
Guido van Rossumb3f72581993-05-21 19:56:10 +0000579 if (v == NULL) {
580 v = class_lookup(inst->in_class, name, &class);
Jeremy Hylton9e392e22000-04-26 20:39:20 +0000581 if (v == NULL)
582 return v;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000583 }
Guido van Rossum0add15f1997-05-09 01:07:15 +0000584 Py_INCREF(v);
Guido van Rossum21d335e1993-10-15 13:01:11 +0000585 if (class != NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000586 if (PyFunction_Check(v)) {
587 PyObject *w = PyMethod_New(v, (PyObject *)inst,
588 (PyObject *)class);
589 Py_DECREF(v);
Guido van Rossum21d335e1993-10-15 13:01:11 +0000590 v = w;
591 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000592 else if (PyMethod_Check(v)) {
593 PyObject *im_class = PyMethod_Class(v);
Guido van Rossum21d335e1993-10-15 13:01:11 +0000594 /* Only if classes are compatible */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000595 if (PyClass_IsSubclass((PyObject *)class, im_class)) {
596 PyObject *im_func = PyMethod_Function(v);
597 PyObject *w = PyMethod_New(im_func,
598 (PyObject *)inst, im_class);
599 Py_DECREF(v);
Guido van Rossum21d335e1993-10-15 13:01:11 +0000600 v = w;
601 }
602 }
Guido van Rossumb3f72581993-05-21 19:56:10 +0000603 }
604 return v;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000605}
606
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000607static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000608instance_getattr(register PyInstanceObject *inst, PyObject *name)
Guido van Rossume7737541994-09-05 07:31:41 +0000609{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000610 register PyObject *func, *res;
Guido van Rossum52ca98a1994-09-05 07:32:29 +0000611 res = instance_getattr1(inst, name);
612 if (res == NULL && (func = inst->in_class->cl_getattr) != NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000613 PyObject *args;
614 PyErr_Clear();
615 args = Py_BuildValue("(OO)", inst, name);
Guido van Rossum52ca98a1994-09-05 07:32:29 +0000616 if (args == NULL)
Guido van Rossume7737541994-09-05 07:31:41 +0000617 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000618 res = PyEval_CallObject(func, args);
619 Py_DECREF(args);
Guido van Rossume7737541994-09-05 07:31:41 +0000620 }
621 return res;
622}
623
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000624static int
Fred Drake79912472000-07-09 04:06:11 +0000625instance_setattr1(PyInstanceObject *inst, PyObject *name, PyObject *v)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000626{
Guido van Rossum94472a01992-09-04 09:45:18 +0000627 if (v == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000628 int rv = PyDict_DelItem(inst->in_dict, name);
Guido van Rossum94472a01992-09-04 09:45:18 +0000629 if (rv < 0)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000630 PyErr_SetString(PyExc_AttributeError,
Guido van Rossum94472a01992-09-04 09:45:18 +0000631 "delete non-existing instance attribute");
632 return rv;
633 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000634 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000635 return PyDict_SetItem(inst->in_dict, name, v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000636}
637
Guido van Rossume7737541994-09-05 07:31:41 +0000638static int
Fred Drake79912472000-07-09 04:06:11 +0000639instance_setattr(PyInstanceObject *inst, PyObject *name, PyObject *v)
Guido van Rossume7737541994-09-05 07:31:41 +0000640{
Guido van Rossumb2173c31997-08-25 21:23:56 +0000641 PyObject *func, *args, *res, *tmp;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000642 char *sname = PyString_AsString(name);
Guido van Rossumb2173c31997-08-25 21:23:56 +0000643 if (sname[0] == '_' && sname[1] == '_') {
644 int n = PyString_Size(name);
Guido van Rossum2878a691996-08-09 20:53:24 +0000645 if (sname[n-1] == '_' && sname[n-2] == '_') {
Guido van Rossumb2173c31997-08-25 21:23:56 +0000646 if (strcmp(sname, "__dict__") == 0) {
647 if (PyEval_GetRestricted()) {
648 PyErr_SetString(PyExc_RuntimeError,
649 "__dict__ not accessible in restricted mode");
650 return -1;
651 }
652 if (v == NULL || !PyDict_Check(v)) {
653 PyErr_SetString(PyExc_TypeError,
654 "__dict__ must be set to a dictionary");
655 return -1;
656 }
657 tmp = inst->in_dict;
658 Py_INCREF(v);
659 inst->in_dict = v;
660 Py_DECREF(tmp);
661 return 0;
662 }
663 if (strcmp(sname, "__class__") == 0) {
664 if (PyEval_GetRestricted()) {
665 PyErr_SetString(PyExc_RuntimeError,
666 "__class__ not accessible in restricted mode");
667 return -1;
668 }
669 if (v == NULL || !PyClass_Check(v)) {
670 PyErr_SetString(PyExc_TypeError,
671 "__class__ must be set to a class");
672 return -1;
673 }
674 tmp = (PyObject *)(inst->in_class);
675 Py_INCREF(v);
676 inst->in_class = (PyClassObject *)v;
677 Py_DECREF(tmp);
678 return 0;
679 }
Guido van Rossume7737541994-09-05 07:31:41 +0000680 }
Guido van Rossume7737541994-09-05 07:31:41 +0000681 }
Guido van Rossum52ca98a1994-09-05 07:32:29 +0000682 if (v == NULL)
683 func = inst->in_class->cl_delattr;
684 else
685 func = inst->in_class->cl_setattr;
686 if (func == NULL)
687 return instance_setattr1(inst, name, v);
688 if (v == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000689 args = Py_BuildValue("(OO)", inst, name);
Guido van Rossum52ca98a1994-09-05 07:32:29 +0000690 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000691 args = Py_BuildValue("(OOO)", inst, name, v);
Guido van Rossum52ca98a1994-09-05 07:32:29 +0000692 if (args == NULL)
693 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000694 res = PyEval_CallObject(func, args);
695 Py_DECREF(args);
Guido van Rossum52ca98a1994-09-05 07:32:29 +0000696 if (res == NULL)
697 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000698 Py_DECREF(res);
Guido van Rossume7737541994-09-05 07:31:41 +0000699 return 0;
700}
701
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000702static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000703instance_repr(PyInstanceObject *inst)
Guido van Rossum04691fc1992-08-12 15:35:34 +0000704{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000705 PyObject *func;
706 PyObject *res;
707 static PyObject *reprstr;
Guido van Rossum04691fc1992-08-12 15:35:34 +0000708
Guido van Rossum2878a691996-08-09 20:53:24 +0000709 if (reprstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000710 reprstr = PyString_InternFromString("__repr__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000711 func = instance_getattr(inst, reprstr);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000712 if (func == NULL) {
Guido van Rossum25831651993-05-19 14:50:45 +0000713 char buf[140];
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000714 PyObject *classname = inst->in_class->cl_name;
Guido van Rossumb7f1afe1997-12-03 00:06:02 +0000715 PyObject *mod = PyDict_GetItemString(
716 inst->in_class->cl_dict, "__module__");
Guido van Rossum25831651993-05-19 14:50:45 +0000717 char *cname;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000718 if (classname != NULL && PyString_Check(classname))
719 cname = PyString_AsString(classname);
Guido van Rossum25831651993-05-19 14:50:45 +0000720 else
721 cname = "?";
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000722 PyErr_Clear();
Guido van Rossumb7f1afe1997-12-03 00:06:02 +0000723 if (mod == NULL || !PyString_Check(mod))
Fred Drakea44d3532000-06-30 15:01:00 +0000724 sprintf(buf, "<?.%.100s instance at %p>",
725 cname, inst);
Guido van Rossumb7f1afe1997-12-03 00:06:02 +0000726 else
Fred Drakea44d3532000-06-30 15:01:00 +0000727 sprintf(buf, "<%.50s.%.50s instance at %p>",
Guido van Rossumb7f1afe1997-12-03 00:06:02 +0000728 PyString_AsString(mod),
Fred Drakea44d3532000-06-30 15:01:00 +0000729 cname, inst);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000730 return PyString_FromString(buf);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000731 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000732 res = PyEval_CallObject(func, (PyObject *)NULL);
733 Py_DECREF(func);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000734 return res;
735}
736
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000737static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000738instance_compare1(PyObject *inst, PyObject *other)
Guido van Rossume7d444f1995-01-07 12:35:18 +0000739{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000740 return PyInstance_DoBinOp(inst, other, "__cmp__", "__rcmp__",
Guido van Rossume7d444f1995-01-07 12:35:18 +0000741 instance_compare1);
742}
743
Guido van Rossum9bfef441993-03-29 10:43:31 +0000744static int
Fred Drake79912472000-07-09 04:06:11 +0000745instance_compare(PyObject *inst, PyObject *other)
Guido van Rossum04691fc1992-08-12 15:35:34 +0000746{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000747 PyObject *result;
Guido van Rossume7d444f1995-01-07 12:35:18 +0000748 long outcome;
749 result = instance_compare1(inst, other);
Guido van Rossumc8b6df91997-05-23 00:06:51 +0000750 if (result == NULL)
751 return -1;
752 if (!PyInt_Check(result)) {
753 Py_DECREF(result);
754 PyErr_SetString(PyExc_TypeError,
755 "comparison did not return an int");
756 return -1;
Sjoerd Mullenderb9a6d121994-10-19 15:11:52 +0000757 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000758 outcome = PyInt_AsLong(result);
759 Py_DECREF(result);
Guido van Rossum03093a21994-09-28 15:51:32 +0000760 if (outcome < 0)
761 return -1;
762 else if (outcome > 0)
763 return 1;
764 return 0;
Guido van Rossum04691fc1992-08-12 15:35:34 +0000765}
766
Guido van Rossum9bfef441993-03-29 10:43:31 +0000767static long
Fred Drake79912472000-07-09 04:06:11 +0000768instance_hash(PyInstanceObject *inst)
Guido van Rossum9bfef441993-03-29 10:43:31 +0000769{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000770 PyObject *func;
771 PyObject *res;
Guido van Rossum8a0c3451993-04-08 12:56:19 +0000772 long outcome;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000773 static PyObject *hashstr, *cmpstr;
Guido van Rossum9bfef441993-03-29 10:43:31 +0000774
Guido van Rossum2878a691996-08-09 20:53:24 +0000775 if (hashstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000776 hashstr = PyString_InternFromString("__hash__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000777 func = instance_getattr(inst, hashstr);
Guido van Rossum9bfef441993-03-29 10:43:31 +0000778 if (func == NULL) {
779 /* If there is no __cmp__ method, we hash on the address.
780 If a __cmp__ method exists, there must be a __hash__. */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000781 PyErr_Clear();
Guido van Rossum2878a691996-08-09 20:53:24 +0000782 if (cmpstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000783 cmpstr = PyString_InternFromString("__cmp__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000784 func = instance_getattr(inst, cmpstr);
Guido van Rossum8a0c3451993-04-08 12:56:19 +0000785 if (func == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000786 PyErr_Clear();
Fred Drake13634cf2000-06-29 19:17:04 +0000787 return _Py_HashPointer(inst);
Guido van Rossum8a0c3451993-04-08 12:56:19 +0000788 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000789 PyErr_SetString(PyExc_TypeError, "unhashable instance");
Guido van Rossum9bfef441993-03-29 10:43:31 +0000790 return -1;
791 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000792 res = PyEval_CallObject(func, (PyObject *)NULL);
793 Py_DECREF(func);
Guido van Rossum9bfef441993-03-29 10:43:31 +0000794 if (res == NULL)
795 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000796 if (PyInt_Check(res)) {
797 outcome = PyInt_AsLong(res);
Guido van Rossum9bfef441993-03-29 10:43:31 +0000798 if (outcome == -1)
799 outcome = -2;
800 }
801 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000802 PyErr_SetString(PyExc_TypeError,
803 "__hash__() should return an int");
Guido van Rossum9bfef441993-03-29 10:43:31 +0000804 outcome = -1;
805 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000806 Py_DECREF(res);
Guido van Rossum9bfef441993-03-29 10:43:31 +0000807 return outcome;
808}
809
Jeremy Hylton8caad492000-06-23 14:18:11 +0000810static int
811instance_traverse(PyInstanceObject *o, visitproc visit, void *arg)
812{
813 int err;
814 if (o->in_class) {
815 err = visit((PyObject *)(o->in_class), arg);
816 if (err)
817 return err;
818 }
819 if (o->in_dict) {
820 err = visit(o->in_dict, arg);
821 if (err)
822 return err;
823 }
Jeremy Hyltond22162b2000-06-23 17:14:56 +0000824 return 0;
Jeremy Hylton8caad492000-06-23 14:18:11 +0000825}
826
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000827static PyObject *getitemstr, *setitemstr, *delitemstr, *lenstr;
Guido van Rossum2878a691996-08-09 20:53:24 +0000828
Guido van Rossum9bfef441993-03-29 10:43:31 +0000829static int
Fred Drake79912472000-07-09 04:06:11 +0000830instance_length(PyInstanceObject *inst)
Guido van Rossum04691fc1992-08-12 15:35:34 +0000831{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000832 PyObject *func;
833 PyObject *res;
Guido van Rossum04691fc1992-08-12 15:35:34 +0000834 int outcome;
835
Guido van Rossum2878a691996-08-09 20:53:24 +0000836 if (lenstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000837 lenstr = PyString_InternFromString("__len__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000838 func = instance_getattr(inst, lenstr);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000839 if (func == NULL)
840 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000841 res = PyEval_CallObject(func, (PyObject *)NULL);
842 Py_DECREF(func);
Guido van Rossumd014ea61992-11-26 10:30:26 +0000843 if (res == NULL)
844 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000845 if (PyInt_Check(res)) {
846 outcome = PyInt_AsLong(res);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000847 if (outcome < 0)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000848 PyErr_SetString(PyExc_ValueError,
849 "__len__() should return >= 0");
Guido van Rossum04691fc1992-08-12 15:35:34 +0000850 }
851 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000852 PyErr_SetString(PyExc_TypeError,
853 "__len__() should return an int");
Guido van Rossum04691fc1992-08-12 15:35:34 +0000854 outcome = -1;
855 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000856 Py_DECREF(res);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000857 return outcome;
858}
859
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000860static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000861instance_subscript(PyInstanceObject *inst, PyObject *key)
Guido van Rossum04691fc1992-08-12 15:35:34 +0000862{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000863 PyObject *func;
864 PyObject *arg;
865 PyObject *res;
Guido van Rossum04691fc1992-08-12 15:35:34 +0000866
Guido van Rossum2878a691996-08-09 20:53:24 +0000867 if (getitemstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000868 getitemstr = PyString_InternFromString("__getitem__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000869 func = instance_getattr(inst, getitemstr);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000870 if (func == NULL)
871 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000872 arg = Py_BuildValue("(O)", key);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000873 if (arg == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000874 Py_DECREF(func);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000875 return NULL;
876 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000877 res = PyEval_CallObject(func, arg);
878 Py_DECREF(func);
879 Py_DECREF(arg);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000880 return res;
881}
882
Guido van Rossum9bfef441993-03-29 10:43:31 +0000883static int
Fred Drake79912472000-07-09 04:06:11 +0000884instance_ass_subscript(PyInstanceObject *inst, PyObject *key, PyObject *value)
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 (value == NULL) {
891 if (delitemstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000892 delitemstr = PyString_InternFromString("__delitem__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000893 func = instance_getattr(inst, delitemstr);
894 }
895 else {
896 if (setitemstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000897 setitemstr = PyString_InternFromString("__setitem__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000898 func = instance_getattr(inst, setitemstr);
899 }
Guido van Rossum04691fc1992-08-12 15:35:34 +0000900 if (func == NULL)
901 return -1;
902 if (value == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000903 arg = Py_BuildValue("(O)", key);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000904 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000905 arg = Py_BuildValue("(OO)", key, value);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000906 if (arg == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000907 Py_DECREF(func);
Guido van Rossum6d946f91992-08-14 13:49:30 +0000908 return -1;
Guido van Rossum04691fc1992-08-12 15:35:34 +0000909 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000910 res = PyEval_CallObject(func, arg);
911 Py_DECREF(func);
912 Py_DECREF(arg);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000913 if (res == NULL)
914 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000915 Py_DECREF(res);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000916 return 0;
917}
918
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000919static PyMappingMethods instance_as_mapping = {
Guido van Rossumb6775db1994-08-01 11:34:53 +0000920 (inquiry)instance_length, /*mp_length*/
921 (binaryfunc)instance_subscript, /*mp_subscript*/
922 (objobjargproc)instance_ass_subscript, /*mp_ass_subscript*/
Guido van Rossum04691fc1992-08-12 15:35:34 +0000923};
924
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000925static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000926instance_item(PyInstanceObject *inst, int i)
Guido van Rossum04691fc1992-08-12 15:35:34 +0000927{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000928 PyObject *func, *arg, *res;
Guido van Rossum04691fc1992-08-12 15:35:34 +0000929
Guido van Rossum2878a691996-08-09 20:53:24 +0000930 if (getitemstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000931 getitemstr = PyString_InternFromString("__getitem__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000932 func = instance_getattr(inst, getitemstr);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000933 if (func == NULL)
934 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000935 arg = Py_BuildValue("(i)", i);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000936 if (arg == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000937 Py_DECREF(func);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000938 return NULL;
939 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000940 res = PyEval_CallObject(func, arg);
941 Py_DECREF(func);
942 Py_DECREF(arg);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000943 return res;
944}
945
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000946static PyObject *
Thomas Wouters1d75a792000-08-17 22:37:32 +0000947sliceobj_from_intint(int i, int j)
948{
949 PyObject *start, *end, *res;
950
951 start = PyInt_FromLong((long)i);
952 if (!start)
953 return NULL;
954
955 end = PyInt_FromLong((long)j);
956 if (!end) {
957 Py_DECREF(start);
958 return NULL;
959 }
960 res = PySlice_New(start, end, NULL);
961 Py_DECREF(start);
962 Py_DECREF(end);
963 return res;
964}
965
966
967static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000968instance_slice(PyInstanceObject *inst, int i, int j)
Guido van Rossum04691fc1992-08-12 15:35:34 +0000969{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000970 PyObject *func, *arg, *res;
971 static PyObject *getslicestr;
Guido van Rossum04691fc1992-08-12 15:35:34 +0000972
Guido van Rossum2878a691996-08-09 20:53:24 +0000973 if (getslicestr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000974 getslicestr = PyString_InternFromString("__getslice__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000975 func = instance_getattr(inst, getslicestr);
Thomas Wouters1d75a792000-08-17 22:37:32 +0000976
977 if (func == NULL) {
978 PyErr_Clear();
979
980 if (getitemstr == NULL)
981 getitemstr = PyString_InternFromString("__getitem__");
982 func = instance_getattr(inst, getitemstr);
983 if (func == NULL)
984 return NULL;
985 arg = Py_BuildValue("(N)", sliceobj_from_intint(i, j));
986 } else
987 arg = Py_BuildValue("(ii)", i, j);
988
Guido van Rossum04691fc1992-08-12 15:35:34 +0000989 if (arg == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000990 Py_DECREF(func);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000991 return NULL;
992 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000993 res = PyEval_CallObject(func, arg);
994 Py_DECREF(func);
995 Py_DECREF(arg);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000996 return res;
997}
998
999static int
Fred Drake79912472000-07-09 04:06:11 +00001000instance_ass_item(PyInstanceObject *inst, int i, PyObject *item)
Guido van Rossum04691fc1992-08-12 15:35:34 +00001001{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001002 PyObject *func, *arg, *res;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001003
Guido van Rossum2878a691996-08-09 20:53:24 +00001004 if (item == NULL) {
1005 if (delitemstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +00001006 delitemstr = PyString_InternFromString("__delitem__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001007 func = instance_getattr(inst, delitemstr);
1008 }
1009 else {
1010 if (setitemstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +00001011 setitemstr = PyString_InternFromString("__setitem__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001012 func = instance_getattr(inst, setitemstr);
1013 }
Guido van Rossum04691fc1992-08-12 15:35:34 +00001014 if (func == NULL)
Guido van Rossum6d946f91992-08-14 13:49:30 +00001015 return -1;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001016 if (item == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001017 arg = Py_BuildValue("i", i);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001018 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001019 arg = Py_BuildValue("(iO)", i, item);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001020 if (arg == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001021 Py_DECREF(func);
Guido van Rossum6d946f91992-08-14 13:49:30 +00001022 return -1;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001023 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001024 res = PyEval_CallObject(func, arg);
1025 Py_DECREF(func);
1026 Py_DECREF(arg);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001027 if (res == NULL)
1028 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001029 Py_DECREF(res);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001030 return 0;
1031}
1032
1033static int
Fred Drake79912472000-07-09 04:06:11 +00001034instance_ass_slice(PyInstanceObject *inst, int i, int j, PyObject *value)
Guido van Rossum04691fc1992-08-12 15:35:34 +00001035{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001036 PyObject *func, *arg, *res;
1037 static PyObject *setslicestr, *delslicestr;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001038
Guido van Rossum2878a691996-08-09 20:53:24 +00001039 if (value == NULL) {
1040 if (delslicestr == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001041 delslicestr =
1042 PyString_InternFromString("__delslice__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001043 func = instance_getattr(inst, delslicestr);
Thomas Wouters1d75a792000-08-17 22:37:32 +00001044 if (func == NULL) {
1045 PyErr_Clear();
1046 if (delitemstr == NULL)
1047 delitemstr =
1048 PyString_InternFromString("__delitem__");
1049 func = instance_getattr(inst, delitemstr);
1050 if (func == NULL)
1051 return -1;
1052
1053 arg = Py_BuildValue("(N)",
1054 sliceobj_from_intint(i, j));
1055 } else
1056 arg = Py_BuildValue("(ii)", i, j);
Guido van Rossum2878a691996-08-09 20:53:24 +00001057 }
1058 else {
1059 if (setslicestr == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001060 setslicestr =
1061 PyString_InternFromString("__setslice__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001062 func = instance_getattr(inst, setslicestr);
Thomas Wouters1d75a792000-08-17 22:37:32 +00001063 if (func == NULL) {
1064 PyErr_Clear();
1065 if (setitemstr == NULL)
1066 setitemstr =
1067 PyString_InternFromString("__setitem__");
1068 func = instance_getattr(inst, setitemstr);
1069 if (func == NULL)
1070 return -1;
1071
1072 arg = Py_BuildValue("(NO)",
1073 sliceobj_from_intint(i, j), value);
1074 } else
1075 arg = Py_BuildValue("(iiO)", i, j, value);
Guido van Rossum2878a691996-08-09 20:53:24 +00001076 }
Guido van Rossum04691fc1992-08-12 15:35:34 +00001077 if (arg == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001078 Py_DECREF(func);
Guido van Rossum6d946f91992-08-14 13:49:30 +00001079 return -1;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001080 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001081 res = PyEval_CallObject(func, arg);
1082 Py_DECREF(func);
1083 Py_DECREF(arg);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001084 if (res == NULL)
1085 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001086 Py_DECREF(res);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001087 return 0;
1088}
1089
Guido van Rossumee28c3a2000-02-28 15:03:15 +00001090static int instance_contains(PyInstanceObject *inst, PyObject *member)
1091{
1092 static PyObject *__contains__;
1093 PyObject *func, *arg, *res;
1094 int ret;
1095
1096 if(__contains__ == NULL) {
1097 __contains__ = PyString_InternFromString("__contains__");
1098 if(__contains__ == NULL)
1099 return -1;
1100 }
1101 func = instance_getattr(inst, __contains__);
1102 if(func == NULL) {
Thomas Wouters7e474022000-07-16 12:04:32 +00001103 /* fall back to previous behavior */
Guido van Rossumee28c3a2000-02-28 15:03:15 +00001104 int i, cmp_res;
1105
1106 if(!PyErr_ExceptionMatches(PyExc_AttributeError))
1107 return -1;
1108 PyErr_Clear();
1109 for(i=0;;i++) {
1110 PyObject *obj = instance_item(inst, i);
1111 int ret = 0;
1112
1113 if(obj == NULL) {
1114 if(!PyErr_ExceptionMatches(PyExc_IndexError))
1115 return -1;
1116 PyErr_Clear();
1117 return 0;
1118 }
1119 if(PyObject_Cmp(obj, member, &cmp_res) == -1)
1120 ret = -1;
1121 if(cmp_res == 0)
1122 ret = 1;
1123 Py_DECREF(obj);
1124 if(ret)
1125 return ret;
1126 }
1127 }
1128 arg = Py_BuildValue("(O)", member);
1129 if(arg == NULL) {
1130 Py_DECREF(func);
1131 return -1;
1132 }
1133 res = PyEval_CallObject(func, arg);
1134 Py_DECREF(func);
1135 Py_DECREF(arg);
1136 if(res == NULL)
1137 return -1;
1138 ret = PyObject_IsTrue(res);
1139 Py_DECREF(res);
1140 return ret;
1141}
1142
Fred Drake79912472000-07-09 04:06:11 +00001143static PySequenceMethods
1144instance_as_sequence = {
Guido van Rossumb6775db1994-08-01 11:34:53 +00001145 (inquiry)instance_length, /*sq_length*/
Guido van Rossum03093a21994-09-28 15:51:32 +00001146 0, /*sq_concat*/
1147 0, /*sq_repeat*/
Guido van Rossumb6775db1994-08-01 11:34:53 +00001148 (intargfunc)instance_item, /*sq_item*/
1149 (intintargfunc)instance_slice, /*sq_slice*/
1150 (intobjargproc)instance_ass_item, /*sq_ass_item*/
1151 (intintobjargproc)instance_ass_slice, /*sq_ass_slice*/
Guido van Rossumee28c3a2000-02-28 15:03:15 +00001152 (objobjproc)instance_contains, /* sq_contains */
Guido van Rossum04691fc1992-08-12 15:35:34 +00001153};
1154
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001155static PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001156generic_unary_op(PyInstanceObject *self, PyObject *methodname)
Guido van Rossum04691fc1992-08-12 15:35:34 +00001157{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001158 PyObject *func, *res;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001159
1160 if ((func = instance_getattr(self, methodname)) == NULL)
1161 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001162 res = PyEval_CallObject(func, (PyObject *)NULL);
1163 Py_DECREF(func);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001164 return res;
1165}
1166
Guido van Rossum03093a21994-09-28 15:51:32 +00001167
Guido van Rossum03093a21994-09-28 15:51:32 +00001168/* Implement a binary operator involving at least one class instance. */
1169
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001170PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001171PyInstance_DoBinOp(PyObject *v, PyObject *w, char *opname, char *ropname,
1172 PyObject * (*thisfunc)(PyObject *, PyObject *))
Guido van Rossum03093a21994-09-28 15:51:32 +00001173{
1174 char buf[256];
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001175 PyObject *result = NULL;
Barry Warsawdc55d712000-08-18 04:57:32 +00001176
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001177 if (PyInstance_HalfBinOp(v, w, opname, &result, thisfunc, 0) <= 0)
Guido van Rossum03093a21994-09-28 15:51:32 +00001178 return result;
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001179 if (PyInstance_HalfBinOp(w, v, ropname, &result, thisfunc, 1) <= 0)
Guido van Rossum03093a21994-09-28 15:51:32 +00001180 return result;
Thomas Wouters7e474022000-07-16 12:04:32 +00001181 /* Sigh -- special case for comparisons */
Guido van Rossumc8b6df91997-05-23 00:06:51 +00001182 if (strcmp(opname, "__cmp__") == 0) {
Barry Warsawdc55d712000-08-18 04:57:32 +00001183 Py_uintptr_t iv = (Py_uintptr_t)v;
1184 Py_uintptr_t iw = (Py_uintptr_t)w;
1185 long c = (iv < iw) ? -1 : (iv > iw) ? 1 : 0;
Guido van Rossumc8b6df91997-05-23 00:06:51 +00001186 return PyInt_FromLong(c);
1187 }
Guido van Rossum03093a21994-09-28 15:51:32 +00001188 sprintf(buf, "%s nor %s defined for these operands", opname, ropname);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001189 PyErr_SetString(PyExc_TypeError, buf);
Guido van Rossum03093a21994-09-28 15:51:32 +00001190 return NULL;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001191}
1192
Guido van Rossum03093a21994-09-28 15:51:32 +00001193
1194/* Try one half of a binary operator involving a class instance.
1195 Return value:
1196 -1 if an exception is to be reported right away
1197 0 if we have a valid result
1198 1 if we could try another operation
1199*/
1200
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001201static PyObject *coerce_obj;
Guido van Rossum2878a691996-08-09 20:53:24 +00001202
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001203int
1204PyInstance_HalfBinOp(PyObject *v, PyObject *w, char *opname, PyObject **r_result,
1205 PyObject * (*thisfunc)(PyObject *, PyObject *), int swapped)
Guido van Rossum03093a21994-09-28 15:51:32 +00001206{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001207 PyObject *func;
1208 PyObject *args;
Guido van Rossum3931df91997-11-18 19:23:07 +00001209 PyObject *coercefunc;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001210 PyObject *coerced = NULL;
1211 PyObject *v1;
Guido van Rossum03093a21994-09-28 15:51:32 +00001212
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001213 if (!PyInstance_Check(v))
Guido van Rossum03093a21994-09-28 15:51:32 +00001214 return 1;
Guido van Rossum2878a691996-08-09 20:53:24 +00001215 if (coerce_obj == NULL) {
Guido van Rossuma412d241997-01-18 07:59:12 +00001216 coerce_obj = PyString_InternFromString("__coerce__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001217 if (coerce_obj == NULL)
1218 return -1;
1219 }
Guido van Rossum3931df91997-11-18 19:23:07 +00001220 coercefunc = PyObject_GetAttr(v, coerce_obj);
1221 if (coercefunc == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001222 PyErr_Clear();
Guido van Rossum03093a21994-09-28 15:51:32 +00001223 }
1224 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001225 args = Py_BuildValue("(O)", w);
Guido van Rossum03093a21994-09-28 15:51:32 +00001226 if (args == NULL) {
Guido van Rossum03093a21994-09-28 15:51:32 +00001227 return -1;
1228 }
Guido van Rossum3931df91997-11-18 19:23:07 +00001229 coerced = PyEval_CallObject(coercefunc, args);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001230 Py_DECREF(args);
Guido van Rossum3931df91997-11-18 19:23:07 +00001231 Py_DECREF(coercefunc);
Guido van Rossum03093a21994-09-28 15:51:32 +00001232 if (coerced == NULL) {
Guido van Rossum03093a21994-09-28 15:51:32 +00001233 return -1;
1234 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001235 if (coerced == Py_None) {
1236 Py_DECREF(coerced);
Guido van Rossum03093a21994-09-28 15:51:32 +00001237 return 1;
1238 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001239 if (!PyTuple_Check(coerced) || PyTuple_Size(coerced) != 2) {
1240 Py_DECREF(coerced);
1241 PyErr_SetString(PyExc_TypeError,
Guido van Rossume7d444f1995-01-07 12:35:18 +00001242 "coercion should return None or 2-tuple");
Guido van Rossum03093a21994-09-28 15:51:32 +00001243 return -1;
1244 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001245 v1 = PyTuple_GetItem(coerced, 0);
1246 w = PyTuple_GetItem(coerced, 1);
Guido van Rossum03093a21994-09-28 15:51:32 +00001247 if (v1 != v) {
1248 v = v1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001249 if (!PyInstance_Check(v) && !PyInstance_Check(w)) {
Guido van Rossume7d444f1995-01-07 12:35:18 +00001250 if (swapped)
1251 *r_result = (*thisfunc)(w, v);
1252 else
1253 *r_result = (*thisfunc)(v, w);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001254 Py_DECREF(coerced);
Guido van Rossume7d444f1995-01-07 12:35:18 +00001255 return *r_result == NULL ? -1 : 0;
Guido van Rossum03093a21994-09-28 15:51:32 +00001256 }
1257 }
Guido van Rossum03093a21994-09-28 15:51:32 +00001258 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001259 func = PyObject_GetAttrString(v, opname);
Guido van Rossume7d444f1995-01-07 12:35:18 +00001260 if (func == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001261 Py_XDECREF(coerced);
Guido van Rossum617c1b01998-05-28 19:50:02 +00001262 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
Guido van Rossume7d444f1995-01-07 12:35:18 +00001263 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001264 PyErr_Clear();
Guido van Rossume7d444f1995-01-07 12:35:18 +00001265 return 1;
1266 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001267 args = Py_BuildValue("(O)", w);
Guido van Rossum03093a21994-09-28 15:51:32 +00001268 if (args == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001269 Py_DECREF(func);
1270 Py_XDECREF(coerced);
Guido van Rossum03093a21994-09-28 15:51:32 +00001271 return -1;
1272 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001273 *r_result = PyEval_CallObject(func, args);
1274 Py_DECREF(args);
1275 Py_DECREF(func);
1276 Py_XDECREF(coerced);
Guido van Rossum03093a21994-09-28 15:51:32 +00001277 return *r_result == NULL ? -1 : 0;
1278}
1279
Guido van Rossum879c5811995-01-10 15:24:06 +00001280static int
Fred Drake79912472000-07-09 04:06:11 +00001281instance_coerce(PyObject **pv, PyObject **pw)
Guido van Rossum879c5811995-01-10 15:24:06 +00001282{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001283 PyObject *v = *pv;
1284 PyObject *w = *pw;
Guido van Rossum3931df91997-11-18 19:23:07 +00001285 PyObject *coercefunc;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001286 PyObject *args;
1287 PyObject *coerced;
Guido van Rossum879c5811995-01-10 15:24:06 +00001288
Guido van Rossum2878a691996-08-09 20:53:24 +00001289 if (coerce_obj == NULL) {
Guido van Rossuma412d241997-01-18 07:59:12 +00001290 coerce_obj = PyString_InternFromString("__coerce__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001291 if (coerce_obj == NULL)
1292 return -1;
1293 }
Guido van Rossum3931df91997-11-18 19:23:07 +00001294 coercefunc = PyObject_GetAttr(v, coerce_obj);
1295 if (coercefunc == NULL) {
Guido van Rossum879c5811995-01-10 15:24:06 +00001296 /* No __coerce__ method: always OK */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001297 PyErr_Clear();
1298 Py_INCREF(v);
1299 Py_INCREF(w);
Guido van Rossum879c5811995-01-10 15:24:06 +00001300 return 0;
1301 }
1302 /* Has __coerce__ method: call it */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001303 args = Py_BuildValue("(O)", w);
Guido van Rossum879c5811995-01-10 15:24:06 +00001304 if (args == NULL) {
1305 return -1;
1306 }
Guido van Rossum3931df91997-11-18 19:23:07 +00001307 coerced = PyEval_CallObject(coercefunc, args);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001308 Py_DECREF(args);
Guido van Rossum3931df91997-11-18 19:23:07 +00001309 Py_DECREF(coercefunc);
Guido van Rossum879c5811995-01-10 15:24:06 +00001310 if (coerced == NULL) {
1311 /* __coerce__ call raised an exception */
1312 return -1;
1313 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001314 if (coerced == Py_None) {
Guido van Rossum879c5811995-01-10 15:24:06 +00001315 /* __coerce__ says "I can't do it" */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001316 Py_DECREF(coerced);
Guido van Rossum879c5811995-01-10 15:24:06 +00001317 return 1;
1318 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001319 if (!PyTuple_Check(coerced) || PyTuple_Size(coerced) != 2) {
Guido van Rossum879c5811995-01-10 15:24:06 +00001320 /* __coerce__ return value is malformed */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001321 Py_DECREF(coerced);
1322 PyErr_SetString(PyExc_TypeError,
Guido van Rossum879c5811995-01-10 15:24:06 +00001323 "coercion should return None or 2-tuple");
1324 return -1;
1325 }
1326 /* __coerce__ returned two new values */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001327 *pv = PyTuple_GetItem(coerced, 0);
1328 *pw = PyTuple_GetItem(coerced, 1);
1329 Py_INCREF(*pv);
1330 Py_INCREF(*pw);
1331 Py_DECREF(coerced);
Guido van Rossum879c5811995-01-10 15:24:06 +00001332 return 0;
1333}
1334
Guido van Rossum03093a21994-09-28 15:51:32 +00001335
Guido van Rossum04691fc1992-08-12 15:35:34 +00001336#define UNARY(funcname, methodname) \
Thomas Woutersc3073522000-07-23 22:09:59 +00001337static PyObject *funcname(PyInstanceObject *self) { \
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001338 static PyObject *o; \
Guido van Rossuma412d241997-01-18 07:59:12 +00001339 if (o == NULL) o = PyString_InternFromString(methodname); \
Guido van Rossum2878a691996-08-09 20:53:24 +00001340 return generic_unary_op(self, o); \
Guido van Rossum04691fc1992-08-12 15:35:34 +00001341}
1342
Guido van Rossum04691fc1992-08-12 15:35:34 +00001343UNARY(instance_neg, "__neg__")
1344UNARY(instance_pos, "__pos__")
1345UNARY(instance_abs, "__abs__")
1346
Guido van Rossum9bfef441993-03-29 10:43:31 +00001347static int
Fred Drake79912472000-07-09 04:06:11 +00001348instance_nonzero(PyInstanceObject *self)
Guido van Rossum04691fc1992-08-12 15:35:34 +00001349{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001350 PyObject *func, *res;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001351 long outcome;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001352 static PyObject *nonzerostr;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001353
Guido van Rossum2878a691996-08-09 20:53:24 +00001354 if (nonzerostr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +00001355 nonzerostr = PyString_InternFromString("__nonzero__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001356 if ((func = instance_getattr(self, nonzerostr)) == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001357 PyErr_Clear();
Guido van Rossum2878a691996-08-09 20:53:24 +00001358 if (lenstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +00001359 lenstr = PyString_InternFromString("__len__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001360 if ((func = instance_getattr(self, lenstr)) == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001361 PyErr_Clear();
Guido van Rossum04691fc1992-08-12 15:35:34 +00001362 /* Fall back to the default behavior:
1363 all instances are nonzero */
1364 return 1;
1365 }
1366 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001367 res = PyEval_CallObject(func, (PyObject *)NULL);
1368 Py_DECREF(func);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001369 if (res == NULL)
1370 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001371 if (!PyInt_Check(res)) {
1372 Py_DECREF(res);
1373 PyErr_SetString(PyExc_TypeError,
1374 "__nonzero__ should return an int");
Guido van Rossum04691fc1992-08-12 15:35:34 +00001375 return -1;
1376 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001377 outcome = PyInt_AsLong(res);
1378 Py_DECREF(res);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001379 if (outcome < 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001380 PyErr_SetString(PyExc_ValueError,
1381 "__nonzero__ should return >= 0");
Guido van Rossum04691fc1992-08-12 15:35:34 +00001382 return -1;
1383 }
1384 return outcome > 0;
1385}
1386
1387UNARY(instance_invert, "__invert__")
Guido van Rossum1899c2e1992-09-12 11:09:23 +00001388UNARY(instance_int, "__int__")
1389UNARY(instance_long, "__long__")
1390UNARY(instance_float, "__float__")
1391UNARY(instance_oct, "__oct__")
1392UNARY(instance_hex, "__hex__")
1393
Guido van Rossum03093a21994-09-28 15:51:32 +00001394/* This version is for ternary calls only (z != None) */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001395static PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001396instance_pow(PyObject *v, PyObject *w, PyObject *z)
Guido van Rossum03093a21994-09-28 15:51:32 +00001397{
1398 /* XXX Doesn't do coercions... */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001399 PyObject *func;
1400 PyObject *args;
1401 PyObject *result;
1402 static PyObject *powstr;
Guido van Rossum2878a691996-08-09 20:53:24 +00001403
1404 if (powstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +00001405 powstr = PyString_InternFromString("__pow__");
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001406 func = PyObject_GetAttr(v, powstr);
Guido van Rossum03093a21994-09-28 15:51:32 +00001407 if (func == NULL)
1408 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001409 args = Py_BuildValue("(OO)", w, z);
Guido van Rossum03093a21994-09-28 15:51:32 +00001410 if (args == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001411 Py_DECREF(func);
Guido van Rossum03093a21994-09-28 15:51:32 +00001412 return NULL;
1413 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001414 result = PyEval_CallObject(func, args);
1415 Py_DECREF(func);
1416 Py_DECREF(args);
Guido van Rossum03093a21994-09-28 15:51:32 +00001417 return result;
1418}
1419
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001420static PyObject *
1421instance_inplace_pow(PyObject *v, PyObject *w, PyObject *z)
1422{
1423 /* XXX Doesn't do coercions... */
1424 PyObject *func;
1425 PyObject *args;
1426 PyObject *result;
1427 static PyObject *ipowstr;
1428
1429 if (ipowstr == NULL)
1430 ipowstr = PyString_InternFromString("__ipow__");
1431 func = PyObject_GetAttr(v, ipowstr);
1432 if (func == NULL) {
1433 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
1434 return NULL;
Thomas Wouters1de2a792000-08-25 10:47:46 +00001435 PyErr_Clear();
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001436 return instance_pow(v, w, z);
1437 }
1438 args = Py_BuildValue("(OO)", w, z);
1439 if (args == NULL) {
1440 Py_DECREF(func);
1441 return NULL;
1442 }
1443 result = PyEval_CallObject(func, args);
1444 Py_DECREF(func);
1445 Py_DECREF(args);
1446 return result;
1447}
1448
1449
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001450static PyNumberMethods instance_as_number = {
Guido van Rossum03093a21994-09-28 15:51:32 +00001451 0, /*nb_add*/
1452 0, /*nb_subtract*/
1453 0, /*nb_multiply*/
1454 0, /*nb_divide*/
1455 0, /*nb_remainder*/
1456 0, /*nb_divmod*/
Guido van Rossum52ca98a1994-09-05 07:32:29 +00001457 (ternaryfunc)instance_pow, /*nb_power*/
Guido van Rossumb6775db1994-08-01 11:34:53 +00001458 (unaryfunc)instance_neg, /*nb_negative*/
1459 (unaryfunc)instance_pos, /*nb_positive*/
1460 (unaryfunc)instance_abs, /*nb_absolute*/
1461 (inquiry)instance_nonzero, /*nb_nonzero*/
1462 (unaryfunc)instance_invert, /*nb_invert*/
Guido van Rossum03093a21994-09-28 15:51:32 +00001463 0, /*nb_lshift*/
1464 0, /*nb_rshift*/
1465 0, /*nb_and*/
1466 0, /*nb_xor*/
1467 0, /*nb_or*/
Guido van Rossum879c5811995-01-10 15:24:06 +00001468 (coercion)instance_coerce, /*nb_coerce*/
Guido van Rossumb6775db1994-08-01 11:34:53 +00001469 (unaryfunc)instance_int, /*nb_int*/
1470 (unaryfunc)instance_long, /*nb_long*/
1471 (unaryfunc)instance_float, /*nb_float*/
1472 (unaryfunc)instance_oct, /*nb_oct*/
1473 (unaryfunc)instance_hex, /*nb_hex*/
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001474 0, /*nb_inplace_add*/
1475 0, /*nb_inplace_subtract*/
1476 0, /*nb_inplace_multiply*/
1477 0, /*nb_inplace_divide*/
1478 0, /*nb_inplace_remainder*/
1479 (ternaryfunc)instance_inplace_pow, /*nb_inplace_power*/
1480 0, /*nb_inplace_lshift*/
1481 0, /*nb_inplace_rshift*/
1482 0, /*nb_inplace_and*/
1483 0, /*nb_inplace_xor*/
1484 0, /*nb_inplace_or*/
Guido van Rossum04691fc1992-08-12 15:35:34 +00001485};
1486
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001487PyTypeObject PyInstance_Type = {
1488 PyObject_HEAD_INIT(&PyType_Type)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001489 0,
Guido van Rossum21ed88c1991-04-04 10:42:10 +00001490 "instance",
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001491 sizeof(PyInstanceObject) + PyGC_HEAD_SIZE,
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001492 0,
Guido van Rossumb6775db1994-08-01 11:34:53 +00001493 (destructor)instance_dealloc, /*tp_dealloc*/
Guido van Rossum7066dd71992-09-17 17:54:56 +00001494 0, /*tp_print*/
Guido van Rossum2878a691996-08-09 20:53:24 +00001495 0, /*tp_getattr*/
1496 0, /*tp_setattr*/
1497 instance_compare, /*tp_compare*/
Guido van Rossumb6775db1994-08-01 11:34:53 +00001498 (reprfunc)instance_repr, /*tp_repr*/
Guido van Rossum04691fc1992-08-12 15:35:34 +00001499 &instance_as_number, /*tp_as_number*/
1500 &instance_as_sequence, /*tp_as_sequence*/
1501 &instance_as_mapping, /*tp_as_mapping*/
Guido van Rossumb6775db1994-08-01 11:34:53 +00001502 (hashfunc)instance_hash, /*tp_hash*/
Guido van Rossum2878a691996-08-09 20:53:24 +00001503 0, /*tp_call*/
1504 0, /*tp_str*/
1505 (getattrofunc)instance_getattr, /*tp_getattro*/
1506 (setattrofunc)instance_setattr, /*tp_setattro*/
Guido van Rossumee28c3a2000-02-28 15:03:15 +00001507 0, /* tp_as_buffer */
Jeremy Hyltond08b4c42000-06-23 19:37:02 +00001508 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_GC, /*tp_flags*/
Jeremy Hylton8caad492000-06-23 14:18:11 +00001509 0, /* tp_doc */
1510 (traverseproc)instance_traverse, /* tp_traverse */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001511};
1512
1513
Guido van Rossum81daa321993-05-20 14:24:46 +00001514/* Instance method objects are used for two purposes:
1515 (a) as bound instance methods (returned by instancename.methodname)
1516 (b) as unbound methods (returned by ClassName.methodname)
1517 In case (b), im_self is NULL
1518*/
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001519
Guido van Rossuma0d349f1997-08-05 02:06:53 +00001520static PyMethodObject *free_list;
1521
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001522PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001523PyMethod_New(PyObject *func, PyObject *self, PyObject *class)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001524{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001525 register PyMethodObject *im;
Guido van Rossum7859f871998-07-08 14:58:16 +00001526 if (!PyCallable_Check(func)) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001527 PyErr_BadInternalCall();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001528 return NULL;
1529 }
Guido van Rossuma0d349f1997-08-05 02:06:53 +00001530 im = free_list;
1531 if (im != NULL) {
1532 free_list = (PyMethodObject *)(im->im_self);
Guido van Rossumb18618d2000-05-03 23:44:39 +00001533 PyObject_INIT(im, &PyMethod_Type);
Guido van Rossuma0d349f1997-08-05 02:06:53 +00001534 }
1535 else {
1536 im = PyObject_NEW(PyMethodObject, &PyMethod_Type);
1537 if (im == NULL)
1538 return NULL;
1539 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001540 Py_INCREF(func);
Guido van Rossume8122f11991-05-05 20:03:07 +00001541 im->im_func = func;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001542 Py_XINCREF(self);
Guido van Rossume8122f11991-05-05 20:03:07 +00001543 im->im_self = self;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001544 Py_INCREF(class);
Guido van Rossum81daa321993-05-20 14:24:46 +00001545 im->im_class = class;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001546 PyObject_GC_Init(im);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001547 return (PyObject *)im;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001548}
1549
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001550PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001551PyMethod_Function(register PyObject *im)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001552{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001553 if (!PyMethod_Check(im)) {
1554 PyErr_BadInternalCall();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001555 return NULL;
1556 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001557 return ((PyMethodObject *)im)->im_func;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001558}
1559
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001560PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001561PyMethod_Self(register PyObject *im)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001562{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001563 if (!PyMethod_Check(im)) {
1564 PyErr_BadInternalCall();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001565 return NULL;
1566 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001567 return ((PyMethodObject *)im)->im_self;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001568}
1569
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001570PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001571PyMethod_Class(register PyObject *im)
Guido van Rossum81daa321993-05-20 14:24:46 +00001572{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001573 if (!PyMethod_Check(im)) {
1574 PyErr_BadInternalCall();
Guido van Rossum81daa321993-05-20 14:24:46 +00001575 return NULL;
1576 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001577 return ((PyMethodObject *)im)->im_class;
Guido van Rossum81daa321993-05-20 14:24:46 +00001578}
1579
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001580/* Class method methods */
1581
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001582#define OFF(x) offsetof(PyMethodObject, x)
Guido van Rossum3f5da241990-12-20 15:06:42 +00001583
Guido van Rossume8122f11991-05-05 20:03:07 +00001584static struct memberlist instancemethod_memberlist[] = {
1585 {"im_func", T_OBJECT, OFF(im_func)},
1586 {"im_self", T_OBJECT, OFF(im_self)},
Guido van Rossum81daa321993-05-20 14:24:46 +00001587 {"im_class", T_OBJECT, OFF(im_class)},
Guido van Rossum6dabc981996-05-14 21:54:20 +00001588 /* Dummies that are not handled by getattr() except for __members__ */
1589 {"__doc__", T_INT, 0},
1590 {"__name__", T_INT, 0},
Guido van Rossum3f5da241990-12-20 15:06:42 +00001591 {NULL} /* Sentinel */
1592};
1593
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001594static PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001595instancemethod_getattr(register PyMethodObject *im, PyObject *name)
Guido van Rossum3f5da241990-12-20 15:06:42 +00001596{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001597 char *sname = PyString_AsString(name);
Guido van Rossum2878a691996-08-09 20:53:24 +00001598 if (sname[0] == '_') {
Guido van Rossum7859f871998-07-08 14:58:16 +00001599 /* Inherit __name__ and __doc__ from the callable object
1600 implementing the method */
1601 if (strcmp(sname, "__name__") == 0 ||
1602 strcmp(sname, "__doc__") == 0)
1603 return PyObject_GetAttr(im->im_func, name);
Guido van Rossum6dabc981996-05-14 21:54:20 +00001604 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001605 if (PyEval_GetRestricted()) {
1606 PyErr_SetString(PyExc_RuntimeError,
1607 "instance-method attributes not accessible in restricted mode");
Guido van Rossum10393b11995-01-10 10:39:49 +00001608 return NULL;
1609 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001610 return PyMember_Get((char *)im, instancemethod_memberlist, sname);
Guido van Rossum3f5da241990-12-20 15:06:42 +00001611}
1612
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001613static void
Fred Drake79912472000-07-09 04:06:11 +00001614instancemethod_dealloc(register PyMethodObject *im)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001615{
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001616 PyObject_GC_Fini(im);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001617 Py_DECREF(im->im_func);
1618 Py_XDECREF(im->im_self);
1619 Py_DECREF(im->im_class);
Guido van Rossuma0d349f1997-08-05 02:06:53 +00001620 im->im_self = (PyObject *)free_list;
1621 free_list = im;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001622}
1623
Guido van Rossumebc8c511992-09-03 20:39:51 +00001624static int
Fred Drake79912472000-07-09 04:06:11 +00001625instancemethod_compare(PyMethodObject *a, PyMethodObject *b)
Guido van Rossumebc8c511992-09-03 20:39:51 +00001626{
Guido van Rossume9df7271995-04-06 14:46:51 +00001627 if (a->im_self != b->im_self)
1628 return (a->im_self < b->im_self) ? -1 : 1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001629 return PyObject_Compare(a->im_func, b->im_func);
Guido van Rossumebc8c511992-09-03 20:39:51 +00001630}
1631
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001632static PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001633instancemethod_repr(PyMethodObject *a)
Guido van Rossum25831651993-05-19 14:50:45 +00001634{
1635 char buf[240];
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001636 PyInstanceObject *self = (PyInstanceObject *)(a->im_self);
Guido van Rossum7859f871998-07-08 14:58:16 +00001637 PyObject *func = a->im_func;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001638 PyClassObject *class = (PyClassObject *)(a->im_class);
1639 PyObject *fclassname, *iclassname, *funcname;
Guido van Rossum81daa321993-05-20 14:24:46 +00001640 char *fcname, *icname, *fname;
1641 fclassname = class->cl_name;
Guido van Rossum7859f871998-07-08 14:58:16 +00001642 if (PyFunction_Check(func)) {
1643 funcname = ((PyFunctionObject *)func)->func_name;
1644 Py_INCREF(funcname);
1645 }
1646 else {
1647 funcname = PyObject_GetAttrString(func,"__name__");
1648 if (funcname == NULL)
1649 PyErr_Clear();
1650 }
1651 if (funcname != NULL && PyString_Check(funcname))
1652 fname = PyString_AS_STRING(funcname);
1653 else
1654 fname = "?";
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001655 if (fclassname != NULL && PyString_Check(fclassname))
1656 fcname = PyString_AsString(fclassname);
Guido van Rossum25831651993-05-19 14:50:45 +00001657 else
Guido van Rossum81daa321993-05-20 14:24:46 +00001658 fcname = "?";
Guido van Rossum81daa321993-05-20 14:24:46 +00001659 if (self == NULL)
1660 sprintf(buf, "<unbound method %.100s.%.100s>", fcname, fname);
1661 else {
1662 iclassname = self->in_class->cl_name;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001663 if (iclassname != NULL && PyString_Check(iclassname))
1664 icname = PyString_AsString(iclassname);
Guido van Rossum81daa321993-05-20 14:24:46 +00001665 else
1666 icname = "?";
Fred Drakea44d3532000-06-30 15:01:00 +00001667 sprintf(buf, "<method %.60s.%.60s of %.60s instance at %p>",
1668 fcname, fname, icname, self);
Guido van Rossum81daa321993-05-20 14:24:46 +00001669 }
Guido van Rossum42636dc1999-10-11 14:03:12 +00001670 Py_XDECREF(funcname);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001671 return PyString_FromString(buf);
Guido van Rossum25831651993-05-19 14:50:45 +00001672}
1673
Guido van Rossum9bfef441993-03-29 10:43:31 +00001674static long
Fred Drake79912472000-07-09 04:06:11 +00001675instancemethod_hash(PyMethodObject *a)
Guido van Rossum9bfef441993-03-29 10:43:31 +00001676{
1677 long x, y;
Guido van Rossum81daa321993-05-20 14:24:46 +00001678 if (a->im_self == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001679 x = PyObject_Hash(Py_None);
Guido van Rossum81daa321993-05-20 14:24:46 +00001680 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001681 x = PyObject_Hash(a->im_self);
Guido van Rossum9bfef441993-03-29 10:43:31 +00001682 if (x == -1)
1683 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001684 y = PyObject_Hash(a->im_func);
Guido van Rossum9bfef441993-03-29 10:43:31 +00001685 if (y == -1)
1686 return -1;
1687 return x ^ y;
1688}
1689
Jeremy Hylton8caad492000-06-23 14:18:11 +00001690static int
1691instancemethod_traverse(PyMethodObject *im, visitproc visit, void *arg)
1692{
1693 int err;
1694 if (im->im_func) {
1695 err = visit(im->im_func, arg);
1696 if (err)
1697 return err;
1698 }
1699 if (im->im_self) {
1700 err = visit(im->im_self, arg);
1701 if (err)
1702 return err;
1703 }
1704 if (im->im_class) {
1705 err = visit(im->im_class, arg);
1706 if (err)
1707 return err;
1708 }
Jeremy Hyltond22162b2000-06-23 17:14:56 +00001709 return 0;
Jeremy Hylton8caad492000-06-23 14:18:11 +00001710}
1711
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001712PyTypeObject PyMethod_Type = {
1713 PyObject_HEAD_INIT(&PyType_Type)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001714 0,
Guido van Rossum569fce71991-04-16 08:38:43 +00001715 "instance method",
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001716 sizeof(PyMethodObject) + PyGC_HEAD_SIZE,
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001717 0,
Guido van Rossumb6775db1994-08-01 11:34:53 +00001718 (destructor)instancemethod_dealloc, /*tp_dealloc*/
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001719 0, /*tp_print*/
Guido van Rossum2878a691996-08-09 20:53:24 +00001720 0, /*tp_getattr*/
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001721 0, /*tp_setattr*/
Guido van Rossumb6775db1994-08-01 11:34:53 +00001722 (cmpfunc)instancemethod_compare, /*tp_compare*/
1723 (reprfunc)instancemethod_repr, /*tp_repr*/
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001724 0, /*tp_as_number*/
1725 0, /*tp_as_sequence*/
1726 0, /*tp_as_mapping*/
Guido van Rossumb6775db1994-08-01 11:34:53 +00001727 (hashfunc)instancemethod_hash, /*tp_hash*/
Guido van Rossum2878a691996-08-09 20:53:24 +00001728 0, /*tp_call*/
1729 0, /*tp_str*/
1730 (getattrofunc)instancemethod_getattr, /*tp_getattro*/
1731 0, /*tp_setattro*/
Jeremy Hylton8caad492000-06-23 14:18:11 +00001732 0, /* tp_as_buffer */
Jeremy Hyltond08b4c42000-06-23 19:37:02 +00001733 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_GC, /*tp_flags*/
Jeremy Hylton8caad492000-06-23 14:18:11 +00001734 0, /* tp_doc */
1735 (traverseproc)instancemethod_traverse, /* tp_traverse */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001736};
Guido van Rossuma0d349f1997-08-05 02:06:53 +00001737
1738/* Clear out the free list */
1739
1740void
Fred Drake79912472000-07-09 04:06:11 +00001741PyMethod_Fini(void)
Guido van Rossuma0d349f1997-08-05 02:06:53 +00001742{
1743 while (free_list) {
Guido van Rossumb18618d2000-05-03 23:44:39 +00001744 PyMethodObject *im = free_list;
1745 free_list = (PyMethodObject *)(im->im_self);
Guido van Rossum4cc6ac72000-07-01 01:00:38 +00001746 im = (PyMethodObject *) PyObject_AS_GC(im);
Guido van Rossumb18618d2000-05-03 23:44:39 +00001747 PyObject_DEL(im);
Guido van Rossuma0d349f1997-08-05 02:06:53 +00001748 }
1749}