blob: c362b800267c411e61cab5fc2a48d75b478eb6ed [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();
Guido van Rossum0add15f1997-05-09 01:07:15 +0000440 if (inst->in_dict == NULL) {
Guido van Rossum4cc6ac72000-07-01 01:00:38 +0000441 inst = (PyInstanceObject *) PyObject_AS_GC(inst);
Guido van Rossumd7823f22000-06-28 23:46:07 +0000442 PyObject_DEL(inst);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000443 return NULL;
444 }
Guido van Rossumd7823f22000-06-28 23:46:07 +0000445 Py_INCREF(class);
446 inst->in_class = (PyClassObject *)class;
Neil Schemenauere3550a62000-10-04 16:20:41 +0000447 PyObject_GC_Init(inst);
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;
Tim Peters6b184912000-09-17 14:40:17 +0000490#ifdef Py_REF_DEBUG
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000491 extern long _Py_RefTotal;
Skip Montanaro4ca150b2000-07-08 12:04:57 +0000492#endif
Tim Peters6b184912000-09-17 14:40:17 +0000493 /* Temporarily resurrect the object. */
Guido van Rossum441e4ab1996-05-23 22:46:51 +0000494#ifdef Py_TRACE_REFS
Tim Peters6b184912000-09-17 14:40:17 +0000495#ifndef Py_REF_DEBUG
496# error "Py_TRACE_REFS defined but Py_REF_DEBUG not."
497#endif
Guido van Rossum441e4ab1996-05-23 22:46:51 +0000498 /* much too complicated if Py_TRACE_REFS defined */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000499 inst->ob_type = &PyInstance_Type;
Guido van Rossumbffd6832000-01-20 22:32:56 +0000500 _Py_NewReference((PyObject *)inst);
Sjoerd Mullender740f3571995-08-28 09:00:43 +0000501#ifdef COUNT_ALLOCS
Tim Peters6b184912000-09-17 14:40:17 +0000502 /* compensate for boost in _Py_NewReference; note that
503 * _Py_RefTotal was also boosted; we'll knock that down later.
504 */
505 inst->ob_type->tp_alloc--;
Sjoerd Mullender740f3571995-08-28 09:00:43 +0000506#endif
Guido van Rossum441e4ab1996-05-23 22:46:51 +0000507#else /* !Py_TRACE_REFS */
Tim Peters6b184912000-09-17 14:40:17 +0000508 /* Py_INCREF boosts _Py_RefTotal if Py_REF_DEBUG is defined */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000509 Py_INCREF(inst);
Guido van Rossum441e4ab1996-05-23 22:46:51 +0000510#endif /* !Py_TRACE_REFS */
Tim Peters6b184912000-09-17 14:40:17 +0000511
512 /* Save the current exception, if any. */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000513 PyErr_Fetch(&error_type, &error_value, &error_traceback);
Tim Peters6b184912000-09-17 14:40:17 +0000514 /* Execute __del__ method, if any. */
Guido van Rossum2878a691996-08-09 20:53:24 +0000515 if (delstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000516 delstr = PyString_InternFromString("__del__");
Jeremy Hylton9e392e22000-04-26 20:39:20 +0000517 if ((del = instance_getattr2(inst, delstr)) != NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000518 PyObject *res = PyEval_CallObject(del, (PyObject *)NULL);
Tim Peters6b184912000-09-17 14:40:17 +0000519 if (res == NULL)
Jeremy Hyltonb709df32000-09-01 02:47:25 +0000520 PyErr_WriteUnraisable(del);
Guido van Rossum0dfcf751996-08-12 22:00:53 +0000521 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000522 Py_DECREF(res);
523 Py_DECREF(del);
Guido van Rossum25831651993-05-19 14:50:45 +0000524 }
Tim Peters6b184912000-09-17 14:40:17 +0000525 /* Restore the saved exception. */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000526 PyErr_Restore(error_type, error_value, error_traceback);
Tim Peters6b184912000-09-17 14:40:17 +0000527 /* Undo the temporary resurrection; can't use DECREF here, it would
528 * cause a recursive call.
529 */
530#ifdef Py_REF_DEBUG
531 /* _Py_RefTotal was boosted either by _Py_NewReference or
532 * Py_INCREF above.
533 */
534 _Py_RefTotal--;
535#endif
Sjoerd Mullender740f3571995-08-28 09:00:43 +0000536 if (--inst->ob_refcnt > 0) {
537#ifdef COUNT_ALLOCS
538 inst->ob_type->tp_free--;
539#endif
Guido van Rossum25831651993-05-19 14:50:45 +0000540 return; /* __del__ added a reference; don't delete now */
Sjoerd Mullender740f3571995-08-28 09:00:43 +0000541 }
Guido van Rossum441e4ab1996-05-23 22:46:51 +0000542#ifdef Py_TRACE_REFS
Guido van Rossumbffd6832000-01-20 22:32:56 +0000543 _Py_ForgetReference((PyObject *)inst);
Tim Peters6b184912000-09-17 14:40:17 +0000544#ifdef COUNT_ALLOCS
545 /* compensate for increment in _Py_ForgetReference */
546 inst->ob_type->tp_free--;
547#endif
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000548#ifndef WITH_CYCLE_GC
Sjoerd Mullender740f3571995-08-28 09:00:43 +0000549 inst->ob_type = NULL;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +0000550#endif
Tim Peters6b184912000-09-17 14:40:17 +0000551#endif
Neil Schemenauerce209672000-09-15 18:57:21 +0000552 PyObject_GC_Fini(inst);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000553 Py_DECREF(inst->in_class);
554 Py_XDECREF(inst->in_dict);
Guido van Rossum4cc6ac72000-07-01 01:00:38 +0000555 inst = (PyInstanceObject *) PyObject_AS_GC(inst);
Guido van Rossumb18618d2000-05-03 23:44:39 +0000556 PyObject_DEL(inst);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000557}
558
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000559static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000560instance_getattr1(register PyInstanceObject *inst, PyObject *name)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000561{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000562 register PyObject *v;
563 register char *sname = PyString_AsString(name);
Guido van Rossum2878a691996-08-09 20:53:24 +0000564 if (sname[0] == '_' && sname[1] == '_') {
565 if (strcmp(sname, "__dict__") == 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000566 if (PyEval_GetRestricted()) {
567 PyErr_SetString(PyExc_RuntimeError,
568 "instance.__dict__ not accessible in restricted mode");
Guido van Rossum10393b11995-01-10 10:39:49 +0000569 return NULL;
570 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000571 Py_INCREF(inst->in_dict);
Sjoerd Mullender3bb8a051993-10-22 12:04:32 +0000572 return inst->in_dict;
573 }
Guido van Rossum2878a691996-08-09 20:53:24 +0000574 if (strcmp(sname, "__class__") == 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000575 Py_INCREF(inst->in_class);
576 return (PyObject *)inst->in_class;
Sjoerd Mullender3bb8a051993-10-22 12:04:32 +0000577 }
Guido van Rossum94308391991-10-20 20:11:48 +0000578 }
Jeremy Hylton9e392e22000-04-26 20:39:20 +0000579 v = instance_getattr2(inst, name);
580 if (v == NULL) {
581 PyErr_Format(PyExc_AttributeError,"'%.50s' instance has no attribute '%.400s'",
582 PyString_AS_STRING(inst->in_class->cl_name), sname);
583 }
584 return v;
585}
586
587static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000588instance_getattr2(register PyInstanceObject *inst, PyObject *name)
Jeremy Hylton9e392e22000-04-26 20:39:20 +0000589{
590 register PyObject *v;
591 PyClassObject *class;
Guido van Rossumeb6b33a1993-05-25 09:38:27 +0000592 class = NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000593 v = PyDict_GetItem(inst->in_dict, name);
Guido van Rossumb3f72581993-05-21 19:56:10 +0000594 if (v == NULL) {
595 v = class_lookup(inst->in_class, name, &class);
Jeremy Hylton9e392e22000-04-26 20:39:20 +0000596 if (v == NULL)
597 return v;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000598 }
Guido van Rossum0add15f1997-05-09 01:07:15 +0000599 Py_INCREF(v);
Guido van Rossum21d335e1993-10-15 13:01:11 +0000600 if (class != NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000601 if (PyFunction_Check(v)) {
602 PyObject *w = PyMethod_New(v, (PyObject *)inst,
603 (PyObject *)class);
604 Py_DECREF(v);
Guido van Rossum21d335e1993-10-15 13:01:11 +0000605 v = w;
606 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000607 else if (PyMethod_Check(v)) {
608 PyObject *im_class = PyMethod_Class(v);
Guido van Rossum21d335e1993-10-15 13:01:11 +0000609 /* Only if classes are compatible */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000610 if (PyClass_IsSubclass((PyObject *)class, im_class)) {
611 PyObject *im_func = PyMethod_Function(v);
612 PyObject *w = PyMethod_New(im_func,
613 (PyObject *)inst, im_class);
614 Py_DECREF(v);
Guido van Rossum21d335e1993-10-15 13:01:11 +0000615 v = w;
616 }
617 }
Guido van Rossumb3f72581993-05-21 19:56:10 +0000618 }
619 return v;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000620}
621
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000622static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000623instance_getattr(register PyInstanceObject *inst, PyObject *name)
Guido van Rossume7737541994-09-05 07:31:41 +0000624{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000625 register PyObject *func, *res;
Guido van Rossum52ca98a1994-09-05 07:32:29 +0000626 res = instance_getattr1(inst, name);
627 if (res == NULL && (func = inst->in_class->cl_getattr) != NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000628 PyObject *args;
629 PyErr_Clear();
630 args = Py_BuildValue("(OO)", inst, name);
Guido van Rossum52ca98a1994-09-05 07:32:29 +0000631 if (args == NULL)
Guido van Rossume7737541994-09-05 07:31:41 +0000632 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000633 res = PyEval_CallObject(func, args);
634 Py_DECREF(args);
Guido van Rossume7737541994-09-05 07:31:41 +0000635 }
636 return res;
637}
638
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000639static int
Fred Drake79912472000-07-09 04:06:11 +0000640instance_setattr1(PyInstanceObject *inst, PyObject *name, PyObject *v)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000641{
Guido van Rossum94472a01992-09-04 09:45:18 +0000642 if (v == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000643 int rv = PyDict_DelItem(inst->in_dict, name);
Guido van Rossum94472a01992-09-04 09:45:18 +0000644 if (rv < 0)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000645 PyErr_SetString(PyExc_AttributeError,
Guido van Rossum94472a01992-09-04 09:45:18 +0000646 "delete non-existing instance attribute");
647 return rv;
648 }
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000649 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000650 return PyDict_SetItem(inst->in_dict, name, v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000651}
652
Guido van Rossume7737541994-09-05 07:31:41 +0000653static int
Fred Drake79912472000-07-09 04:06:11 +0000654instance_setattr(PyInstanceObject *inst, PyObject *name, PyObject *v)
Guido van Rossume7737541994-09-05 07:31:41 +0000655{
Guido van Rossumb2173c31997-08-25 21:23:56 +0000656 PyObject *func, *args, *res, *tmp;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000657 char *sname = PyString_AsString(name);
Guido van Rossumb2173c31997-08-25 21:23:56 +0000658 if (sname[0] == '_' && sname[1] == '_') {
659 int n = PyString_Size(name);
Guido van Rossum2878a691996-08-09 20:53:24 +0000660 if (sname[n-1] == '_' && sname[n-2] == '_') {
Guido van Rossumb2173c31997-08-25 21:23:56 +0000661 if (strcmp(sname, "__dict__") == 0) {
662 if (PyEval_GetRestricted()) {
663 PyErr_SetString(PyExc_RuntimeError,
664 "__dict__ not accessible in restricted mode");
665 return -1;
666 }
667 if (v == NULL || !PyDict_Check(v)) {
668 PyErr_SetString(PyExc_TypeError,
669 "__dict__ must be set to a dictionary");
670 return -1;
671 }
672 tmp = inst->in_dict;
673 Py_INCREF(v);
674 inst->in_dict = v;
675 Py_DECREF(tmp);
676 return 0;
677 }
678 if (strcmp(sname, "__class__") == 0) {
679 if (PyEval_GetRestricted()) {
680 PyErr_SetString(PyExc_RuntimeError,
681 "__class__ not accessible in restricted mode");
682 return -1;
683 }
684 if (v == NULL || !PyClass_Check(v)) {
685 PyErr_SetString(PyExc_TypeError,
686 "__class__ must be set to a class");
687 return -1;
688 }
689 tmp = (PyObject *)(inst->in_class);
690 Py_INCREF(v);
691 inst->in_class = (PyClassObject *)v;
692 Py_DECREF(tmp);
693 return 0;
694 }
Guido van Rossume7737541994-09-05 07:31:41 +0000695 }
Guido van Rossume7737541994-09-05 07:31:41 +0000696 }
Guido van Rossum52ca98a1994-09-05 07:32:29 +0000697 if (v == NULL)
698 func = inst->in_class->cl_delattr;
699 else
700 func = inst->in_class->cl_setattr;
701 if (func == NULL)
702 return instance_setattr1(inst, name, v);
703 if (v == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000704 args = Py_BuildValue("(OO)", inst, name);
Guido van Rossum52ca98a1994-09-05 07:32:29 +0000705 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000706 args = Py_BuildValue("(OOO)", inst, name, v);
Guido van Rossum52ca98a1994-09-05 07:32:29 +0000707 if (args == NULL)
708 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000709 res = PyEval_CallObject(func, args);
710 Py_DECREF(args);
Guido van Rossum52ca98a1994-09-05 07:32:29 +0000711 if (res == NULL)
712 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000713 Py_DECREF(res);
Guido van Rossume7737541994-09-05 07:31:41 +0000714 return 0;
715}
716
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000717static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000718instance_repr(PyInstanceObject *inst)
Guido van Rossum04691fc1992-08-12 15:35:34 +0000719{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000720 PyObject *func;
721 PyObject *res;
722 static PyObject *reprstr;
Guido van Rossum04691fc1992-08-12 15:35:34 +0000723
Guido van Rossum2878a691996-08-09 20:53:24 +0000724 if (reprstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000725 reprstr = PyString_InternFromString("__repr__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000726 func = instance_getattr(inst, reprstr);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000727 if (func == NULL) {
Guido van Rossum25831651993-05-19 14:50:45 +0000728 char buf[140];
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000729 PyObject *classname = inst->in_class->cl_name;
Guido van Rossumb7f1afe1997-12-03 00:06:02 +0000730 PyObject *mod = PyDict_GetItemString(
731 inst->in_class->cl_dict, "__module__");
Guido van Rossum25831651993-05-19 14:50:45 +0000732 char *cname;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000733 if (classname != NULL && PyString_Check(classname))
734 cname = PyString_AsString(classname);
Guido van Rossum25831651993-05-19 14:50:45 +0000735 else
736 cname = "?";
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000737 PyErr_Clear();
Guido van Rossumb7f1afe1997-12-03 00:06:02 +0000738 if (mod == NULL || !PyString_Check(mod))
Fred Drakea44d3532000-06-30 15:01:00 +0000739 sprintf(buf, "<?.%.100s instance at %p>",
740 cname, inst);
Guido van Rossumb7f1afe1997-12-03 00:06:02 +0000741 else
Fred Drakea44d3532000-06-30 15:01:00 +0000742 sprintf(buf, "<%.50s.%.50s instance at %p>",
Guido van Rossumb7f1afe1997-12-03 00:06:02 +0000743 PyString_AsString(mod),
Fred Drakea44d3532000-06-30 15:01:00 +0000744 cname, inst);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000745 return PyString_FromString(buf);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000746 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000747 res = PyEval_CallObject(func, (PyObject *)NULL);
748 Py_DECREF(func);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000749 return res;
750}
751
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000752static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000753instance_compare1(PyObject *inst, PyObject *other)
Guido van Rossume7d444f1995-01-07 12:35:18 +0000754{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000755 return PyInstance_DoBinOp(inst, other, "__cmp__", "__rcmp__",
Guido van Rossume7d444f1995-01-07 12:35:18 +0000756 instance_compare1);
757}
758
Guido van Rossum9bfef441993-03-29 10:43:31 +0000759static int
Fred Drake79912472000-07-09 04:06:11 +0000760instance_compare(PyObject *inst, PyObject *other)
Guido van Rossum04691fc1992-08-12 15:35:34 +0000761{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000762 PyObject *result;
Guido van Rossume7d444f1995-01-07 12:35:18 +0000763 long outcome;
764 result = instance_compare1(inst, other);
Guido van Rossumc8b6df91997-05-23 00:06:51 +0000765 if (result == NULL)
766 return -1;
767 if (!PyInt_Check(result)) {
768 Py_DECREF(result);
769 PyErr_SetString(PyExc_TypeError,
770 "comparison did not return an int");
771 return -1;
Sjoerd Mullenderb9a6d121994-10-19 15:11:52 +0000772 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000773 outcome = PyInt_AsLong(result);
774 Py_DECREF(result);
Guido van Rossum03093a21994-09-28 15:51:32 +0000775 if (outcome < 0)
776 return -1;
777 else if (outcome > 0)
778 return 1;
779 return 0;
Guido van Rossum04691fc1992-08-12 15:35:34 +0000780}
781
Guido van Rossum9bfef441993-03-29 10:43:31 +0000782static long
Fred Drake79912472000-07-09 04:06:11 +0000783instance_hash(PyInstanceObject *inst)
Guido van Rossum9bfef441993-03-29 10:43:31 +0000784{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000785 PyObject *func;
786 PyObject *res;
Guido van Rossum8a0c3451993-04-08 12:56:19 +0000787 long outcome;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000788 static PyObject *hashstr, *cmpstr;
Guido van Rossum9bfef441993-03-29 10:43:31 +0000789
Guido van Rossum2878a691996-08-09 20:53:24 +0000790 if (hashstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000791 hashstr = PyString_InternFromString("__hash__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000792 func = instance_getattr(inst, hashstr);
Guido van Rossum9bfef441993-03-29 10:43:31 +0000793 if (func == NULL) {
794 /* If there is no __cmp__ method, we hash on the address.
795 If a __cmp__ method exists, there must be a __hash__. */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000796 PyErr_Clear();
Guido van Rossum2878a691996-08-09 20:53:24 +0000797 if (cmpstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000798 cmpstr = PyString_InternFromString("__cmp__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000799 func = instance_getattr(inst, cmpstr);
Guido van Rossum8a0c3451993-04-08 12:56:19 +0000800 if (func == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000801 PyErr_Clear();
Fred Drake13634cf2000-06-29 19:17:04 +0000802 return _Py_HashPointer(inst);
Guido van Rossum8a0c3451993-04-08 12:56:19 +0000803 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000804 PyErr_SetString(PyExc_TypeError, "unhashable instance");
Guido van Rossum9bfef441993-03-29 10:43:31 +0000805 return -1;
806 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000807 res = PyEval_CallObject(func, (PyObject *)NULL);
808 Py_DECREF(func);
Guido van Rossum9bfef441993-03-29 10:43:31 +0000809 if (res == NULL)
810 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000811 if (PyInt_Check(res)) {
812 outcome = PyInt_AsLong(res);
Guido van Rossum9bfef441993-03-29 10:43:31 +0000813 if (outcome == -1)
814 outcome = -2;
815 }
816 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000817 PyErr_SetString(PyExc_TypeError,
818 "__hash__() should return an int");
Guido van Rossum9bfef441993-03-29 10:43:31 +0000819 outcome = -1;
820 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000821 Py_DECREF(res);
Guido van Rossum9bfef441993-03-29 10:43:31 +0000822 return outcome;
823}
824
Jeremy Hylton8caad492000-06-23 14:18:11 +0000825static int
826instance_traverse(PyInstanceObject *o, visitproc visit, void *arg)
827{
828 int err;
829 if (o->in_class) {
830 err = visit((PyObject *)(o->in_class), arg);
831 if (err)
832 return err;
833 }
834 if (o->in_dict) {
835 err = visit(o->in_dict, arg);
836 if (err)
837 return err;
838 }
Jeremy Hyltond22162b2000-06-23 17:14:56 +0000839 return 0;
Jeremy Hylton8caad492000-06-23 14:18:11 +0000840}
841
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000842static PyObject *getitemstr, *setitemstr, *delitemstr, *lenstr;
Guido van Rossum2878a691996-08-09 20:53:24 +0000843
Guido van Rossum9bfef441993-03-29 10:43:31 +0000844static int
Fred Drake79912472000-07-09 04:06:11 +0000845instance_length(PyInstanceObject *inst)
Guido van Rossum04691fc1992-08-12 15:35:34 +0000846{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000847 PyObject *func;
848 PyObject *res;
Guido van Rossum04691fc1992-08-12 15:35:34 +0000849 int outcome;
850
Guido van Rossum2878a691996-08-09 20:53:24 +0000851 if (lenstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000852 lenstr = PyString_InternFromString("__len__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000853 func = instance_getattr(inst, lenstr);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000854 if (func == NULL)
855 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000856 res = PyEval_CallObject(func, (PyObject *)NULL);
857 Py_DECREF(func);
Guido van Rossumd014ea61992-11-26 10:30:26 +0000858 if (res == NULL)
859 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000860 if (PyInt_Check(res)) {
861 outcome = PyInt_AsLong(res);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000862 if (outcome < 0)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000863 PyErr_SetString(PyExc_ValueError,
864 "__len__() should return >= 0");
Guido van Rossum04691fc1992-08-12 15:35:34 +0000865 }
866 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000867 PyErr_SetString(PyExc_TypeError,
868 "__len__() should return an int");
Guido van Rossum04691fc1992-08-12 15:35:34 +0000869 outcome = -1;
870 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000871 Py_DECREF(res);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000872 return outcome;
873}
874
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000875static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000876instance_subscript(PyInstanceObject *inst, PyObject *key)
Guido van Rossum04691fc1992-08-12 15:35:34 +0000877{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000878 PyObject *func;
879 PyObject *arg;
880 PyObject *res;
Guido van Rossum04691fc1992-08-12 15:35:34 +0000881
Guido van Rossum2878a691996-08-09 20:53:24 +0000882 if (getitemstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000883 getitemstr = PyString_InternFromString("__getitem__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000884 func = instance_getattr(inst, getitemstr);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000885 if (func == NULL)
886 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000887 arg = Py_BuildValue("(O)", key);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000888 if (arg == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000889 Py_DECREF(func);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000890 return NULL;
891 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000892 res = PyEval_CallObject(func, arg);
893 Py_DECREF(func);
894 Py_DECREF(arg);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000895 return res;
896}
897
Guido van Rossum9bfef441993-03-29 10:43:31 +0000898static int
Fred Drake79912472000-07-09 04:06:11 +0000899instance_ass_subscript(PyInstanceObject *inst, PyObject *key, PyObject *value)
Guido van Rossum04691fc1992-08-12 15:35:34 +0000900{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000901 PyObject *func;
902 PyObject *arg;
903 PyObject *res;
Guido van Rossum04691fc1992-08-12 15:35:34 +0000904
Guido van Rossum2878a691996-08-09 20:53:24 +0000905 if (value == NULL) {
906 if (delitemstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000907 delitemstr = PyString_InternFromString("__delitem__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000908 func = instance_getattr(inst, delitemstr);
909 }
910 else {
911 if (setitemstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000912 setitemstr = PyString_InternFromString("__setitem__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000913 func = instance_getattr(inst, setitemstr);
914 }
Guido van Rossum04691fc1992-08-12 15:35:34 +0000915 if (func == NULL)
916 return -1;
917 if (value == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000918 arg = Py_BuildValue("(O)", key);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000919 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000920 arg = Py_BuildValue("(OO)", key, value);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000921 if (arg == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000922 Py_DECREF(func);
Guido van Rossum6d946f91992-08-14 13:49:30 +0000923 return -1;
Guido van Rossum04691fc1992-08-12 15:35:34 +0000924 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000925 res = PyEval_CallObject(func, arg);
926 Py_DECREF(func);
927 Py_DECREF(arg);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000928 if (res == NULL)
929 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000930 Py_DECREF(res);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000931 return 0;
932}
933
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000934static PyMappingMethods instance_as_mapping = {
Guido van Rossumb6775db1994-08-01 11:34:53 +0000935 (inquiry)instance_length, /*mp_length*/
936 (binaryfunc)instance_subscript, /*mp_subscript*/
937 (objobjargproc)instance_ass_subscript, /*mp_ass_subscript*/
Guido van Rossum04691fc1992-08-12 15:35:34 +0000938};
939
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000940static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000941instance_item(PyInstanceObject *inst, int i)
Guido van Rossum04691fc1992-08-12 15:35:34 +0000942{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000943 PyObject *func, *arg, *res;
Guido van Rossum04691fc1992-08-12 15:35:34 +0000944
Guido van Rossum2878a691996-08-09 20:53:24 +0000945 if (getitemstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000946 getitemstr = PyString_InternFromString("__getitem__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000947 func = instance_getattr(inst, getitemstr);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000948 if (func == NULL)
949 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000950 arg = Py_BuildValue("(i)", i);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000951 if (arg == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000952 Py_DECREF(func);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000953 return NULL;
954 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000955 res = PyEval_CallObject(func, arg);
956 Py_DECREF(func);
957 Py_DECREF(arg);
Guido van Rossum04691fc1992-08-12 15:35:34 +0000958 return res;
959}
960
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000961static PyObject *
Thomas Wouters1d75a792000-08-17 22:37:32 +0000962sliceobj_from_intint(int i, int j)
963{
964 PyObject *start, *end, *res;
965
966 start = PyInt_FromLong((long)i);
967 if (!start)
968 return NULL;
969
970 end = PyInt_FromLong((long)j);
971 if (!end) {
972 Py_DECREF(start);
973 return NULL;
974 }
975 res = PySlice_New(start, end, NULL);
976 Py_DECREF(start);
977 Py_DECREF(end);
978 return res;
979}
980
981
982static PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000983instance_slice(PyInstanceObject *inst, int i, int j)
Guido van Rossum04691fc1992-08-12 15:35:34 +0000984{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000985 PyObject *func, *arg, *res;
986 static PyObject *getslicestr;
Guido van Rossum04691fc1992-08-12 15:35:34 +0000987
Guido van Rossum2878a691996-08-09 20:53:24 +0000988 if (getslicestr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +0000989 getslicestr = PyString_InternFromString("__getslice__");
Guido van Rossum2878a691996-08-09 20:53:24 +0000990 func = instance_getattr(inst, getslicestr);
Thomas Wouters1d75a792000-08-17 22:37:32 +0000991
992 if (func == NULL) {
993 PyErr_Clear();
994
995 if (getitemstr == NULL)
996 getitemstr = PyString_InternFromString("__getitem__");
997 func = instance_getattr(inst, getitemstr);
998 if (func == NULL)
999 return NULL;
1000 arg = Py_BuildValue("(N)", sliceobj_from_intint(i, j));
1001 } else
1002 arg = Py_BuildValue("(ii)", i, j);
1003
Guido van Rossum04691fc1992-08-12 15:35:34 +00001004 if (arg == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001005 Py_DECREF(func);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001006 return NULL;
1007 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001008 res = PyEval_CallObject(func, arg);
1009 Py_DECREF(func);
1010 Py_DECREF(arg);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001011 return res;
1012}
1013
1014static int
Fred Drake79912472000-07-09 04:06:11 +00001015instance_ass_item(PyInstanceObject *inst, int i, PyObject *item)
Guido van Rossum04691fc1992-08-12 15:35:34 +00001016{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001017 PyObject *func, *arg, *res;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001018
Guido van Rossum2878a691996-08-09 20:53:24 +00001019 if (item == NULL) {
1020 if (delitemstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +00001021 delitemstr = PyString_InternFromString("__delitem__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001022 func = instance_getattr(inst, delitemstr);
1023 }
1024 else {
1025 if (setitemstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +00001026 setitemstr = PyString_InternFromString("__setitem__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001027 func = instance_getattr(inst, setitemstr);
1028 }
Guido van Rossum04691fc1992-08-12 15:35:34 +00001029 if (func == NULL)
Guido van Rossum6d946f91992-08-14 13:49:30 +00001030 return -1;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001031 if (item == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001032 arg = Py_BuildValue("i", i);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001033 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001034 arg = Py_BuildValue("(iO)", i, item);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001035 if (arg == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001036 Py_DECREF(func);
Guido van Rossum6d946f91992-08-14 13:49:30 +00001037 return -1;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001038 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001039 res = PyEval_CallObject(func, arg);
1040 Py_DECREF(func);
1041 Py_DECREF(arg);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001042 if (res == NULL)
1043 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001044 Py_DECREF(res);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001045 return 0;
1046}
1047
1048static int
Fred Drake79912472000-07-09 04:06:11 +00001049instance_ass_slice(PyInstanceObject *inst, int i, int j, PyObject *value)
Guido van Rossum04691fc1992-08-12 15:35:34 +00001050{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001051 PyObject *func, *arg, *res;
1052 static PyObject *setslicestr, *delslicestr;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001053
Guido van Rossum2878a691996-08-09 20:53:24 +00001054 if (value == NULL) {
1055 if (delslicestr == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001056 delslicestr =
1057 PyString_InternFromString("__delslice__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001058 func = instance_getattr(inst, delslicestr);
Thomas Wouters1d75a792000-08-17 22:37:32 +00001059 if (func == NULL) {
1060 PyErr_Clear();
1061 if (delitemstr == NULL)
1062 delitemstr =
1063 PyString_InternFromString("__delitem__");
1064 func = instance_getattr(inst, delitemstr);
1065 if (func == NULL)
1066 return -1;
1067
1068 arg = Py_BuildValue("(N)",
1069 sliceobj_from_intint(i, j));
1070 } else
1071 arg = Py_BuildValue("(ii)", i, j);
Guido van Rossum2878a691996-08-09 20:53:24 +00001072 }
1073 else {
1074 if (setslicestr == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001075 setslicestr =
1076 PyString_InternFromString("__setslice__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001077 func = instance_getattr(inst, setslicestr);
Thomas Wouters1d75a792000-08-17 22:37:32 +00001078 if (func == NULL) {
1079 PyErr_Clear();
1080 if (setitemstr == NULL)
1081 setitemstr =
1082 PyString_InternFromString("__setitem__");
1083 func = instance_getattr(inst, setitemstr);
1084 if (func == NULL)
1085 return -1;
1086
1087 arg = Py_BuildValue("(NO)",
1088 sliceobj_from_intint(i, j), value);
1089 } else
1090 arg = Py_BuildValue("(iiO)", i, j, value);
Guido van Rossum2878a691996-08-09 20:53:24 +00001091 }
Guido van Rossum04691fc1992-08-12 15:35:34 +00001092 if (arg == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001093 Py_DECREF(func);
Guido van Rossum6d946f91992-08-14 13:49:30 +00001094 return -1;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001095 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001096 res = PyEval_CallObject(func, arg);
1097 Py_DECREF(func);
1098 Py_DECREF(arg);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001099 if (res == NULL)
1100 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001101 Py_DECREF(res);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001102 return 0;
1103}
1104
Guido van Rossumee28c3a2000-02-28 15:03:15 +00001105static int instance_contains(PyInstanceObject *inst, PyObject *member)
1106{
1107 static PyObject *__contains__;
1108 PyObject *func, *arg, *res;
1109 int ret;
1110
1111 if(__contains__ == NULL) {
1112 __contains__ = PyString_InternFromString("__contains__");
1113 if(__contains__ == NULL)
1114 return -1;
1115 }
1116 func = instance_getattr(inst, __contains__);
1117 if(func == NULL) {
Thomas Wouters7e474022000-07-16 12:04:32 +00001118 /* fall back to previous behavior */
Guido van Rossumee28c3a2000-02-28 15:03:15 +00001119 int i, cmp_res;
1120
1121 if(!PyErr_ExceptionMatches(PyExc_AttributeError))
1122 return -1;
1123 PyErr_Clear();
1124 for(i=0;;i++) {
1125 PyObject *obj = instance_item(inst, i);
1126 int ret = 0;
1127
1128 if(obj == NULL) {
1129 if(!PyErr_ExceptionMatches(PyExc_IndexError))
1130 return -1;
1131 PyErr_Clear();
1132 return 0;
1133 }
1134 if(PyObject_Cmp(obj, member, &cmp_res) == -1)
1135 ret = -1;
1136 if(cmp_res == 0)
1137 ret = 1;
1138 Py_DECREF(obj);
1139 if(ret)
1140 return ret;
1141 }
1142 }
1143 arg = Py_BuildValue("(O)", member);
1144 if(arg == NULL) {
1145 Py_DECREF(func);
1146 return -1;
1147 }
1148 res = PyEval_CallObject(func, arg);
1149 Py_DECREF(func);
1150 Py_DECREF(arg);
1151 if(res == NULL)
1152 return -1;
1153 ret = PyObject_IsTrue(res);
1154 Py_DECREF(res);
1155 return ret;
1156}
1157
Fred Drake79912472000-07-09 04:06:11 +00001158static PySequenceMethods
1159instance_as_sequence = {
Guido van Rossumb6775db1994-08-01 11:34:53 +00001160 (inquiry)instance_length, /*sq_length*/
Guido van Rossum03093a21994-09-28 15:51:32 +00001161 0, /*sq_concat*/
1162 0, /*sq_repeat*/
Guido van Rossumb6775db1994-08-01 11:34:53 +00001163 (intargfunc)instance_item, /*sq_item*/
1164 (intintargfunc)instance_slice, /*sq_slice*/
1165 (intobjargproc)instance_ass_item, /*sq_ass_item*/
1166 (intintobjargproc)instance_ass_slice, /*sq_ass_slice*/
Guido van Rossumee28c3a2000-02-28 15:03:15 +00001167 (objobjproc)instance_contains, /* sq_contains */
Guido van Rossum04691fc1992-08-12 15:35:34 +00001168};
1169
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001170static PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001171generic_unary_op(PyInstanceObject *self, PyObject *methodname)
Guido van Rossum04691fc1992-08-12 15:35:34 +00001172{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001173 PyObject *func, *res;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001174
1175 if ((func = instance_getattr(self, methodname)) == NULL)
1176 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001177 res = PyEval_CallObject(func, (PyObject *)NULL);
1178 Py_DECREF(func);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001179 return res;
1180}
1181
Guido van Rossum03093a21994-09-28 15:51:32 +00001182
Guido van Rossum03093a21994-09-28 15:51:32 +00001183/* Implement a binary operator involving at least one class instance. */
1184
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001185PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001186PyInstance_DoBinOp(PyObject *v, PyObject *w, char *opname, char *ropname,
1187 PyObject * (*thisfunc)(PyObject *, PyObject *))
Guido van Rossum03093a21994-09-28 15:51:32 +00001188{
1189 char buf[256];
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001190 PyObject *result = NULL;
Barry Warsawdc55d712000-08-18 04:57:32 +00001191
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001192 if (PyInstance_HalfBinOp(v, w, opname, &result, thisfunc, 0) <= 0)
Guido van Rossum03093a21994-09-28 15:51:32 +00001193 return result;
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001194 if (PyInstance_HalfBinOp(w, v, ropname, &result, thisfunc, 1) <= 0)
Guido van Rossum03093a21994-09-28 15:51:32 +00001195 return result;
Thomas Wouters7e474022000-07-16 12:04:32 +00001196 /* Sigh -- special case for comparisons */
Guido van Rossumc8b6df91997-05-23 00:06:51 +00001197 if (strcmp(opname, "__cmp__") == 0) {
Barry Warsawdc55d712000-08-18 04:57:32 +00001198 Py_uintptr_t iv = (Py_uintptr_t)v;
1199 Py_uintptr_t iw = (Py_uintptr_t)w;
1200 long c = (iv < iw) ? -1 : (iv > iw) ? 1 : 0;
Guido van Rossumc8b6df91997-05-23 00:06:51 +00001201 return PyInt_FromLong(c);
1202 }
Guido van Rossum03093a21994-09-28 15:51:32 +00001203 sprintf(buf, "%s nor %s defined for these operands", opname, ropname);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001204 PyErr_SetString(PyExc_TypeError, buf);
Guido van Rossum03093a21994-09-28 15:51:32 +00001205 return NULL;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001206}
1207
Guido van Rossum03093a21994-09-28 15:51:32 +00001208
1209/* Try one half of a binary operator involving a class instance.
1210 Return value:
1211 -1 if an exception is to be reported right away
1212 0 if we have a valid result
1213 1 if we could try another operation
1214*/
1215
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001216static PyObject *coerce_obj;
Guido van Rossum2878a691996-08-09 20:53:24 +00001217
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001218int
1219PyInstance_HalfBinOp(PyObject *v, PyObject *w, char *opname, PyObject **r_result,
1220 PyObject * (*thisfunc)(PyObject *, PyObject *), int swapped)
Guido van Rossum03093a21994-09-28 15:51:32 +00001221{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001222 PyObject *func;
1223 PyObject *args;
Guido van Rossum3931df91997-11-18 19:23:07 +00001224 PyObject *coercefunc;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001225 PyObject *coerced = NULL;
1226 PyObject *v1;
Guido van Rossum03093a21994-09-28 15:51:32 +00001227
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001228 if (!PyInstance_Check(v))
Guido van Rossum03093a21994-09-28 15:51:32 +00001229 return 1;
Guido van Rossum2878a691996-08-09 20:53:24 +00001230 if (coerce_obj == NULL) {
Guido van Rossuma412d241997-01-18 07:59:12 +00001231 coerce_obj = PyString_InternFromString("__coerce__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001232 if (coerce_obj == NULL)
1233 return -1;
1234 }
Guido van Rossum3931df91997-11-18 19:23:07 +00001235 coercefunc = PyObject_GetAttr(v, coerce_obj);
1236 if (coercefunc == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001237 PyErr_Clear();
Guido van Rossum03093a21994-09-28 15:51:32 +00001238 }
1239 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001240 args = Py_BuildValue("(O)", w);
Guido van Rossum03093a21994-09-28 15:51:32 +00001241 if (args == NULL) {
Guido van Rossum03093a21994-09-28 15:51:32 +00001242 return -1;
1243 }
Guido van Rossum3931df91997-11-18 19:23:07 +00001244 coerced = PyEval_CallObject(coercefunc, args);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001245 Py_DECREF(args);
Guido van Rossum3931df91997-11-18 19:23:07 +00001246 Py_DECREF(coercefunc);
Guido van Rossum03093a21994-09-28 15:51:32 +00001247 if (coerced == NULL) {
Guido van Rossum03093a21994-09-28 15:51:32 +00001248 return -1;
1249 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001250 if (coerced == Py_None) {
1251 Py_DECREF(coerced);
Guido van Rossum03093a21994-09-28 15:51:32 +00001252 return 1;
1253 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001254 if (!PyTuple_Check(coerced) || PyTuple_Size(coerced) != 2) {
1255 Py_DECREF(coerced);
1256 PyErr_SetString(PyExc_TypeError,
Guido van Rossume7d444f1995-01-07 12:35:18 +00001257 "coercion should return None or 2-tuple");
Guido van Rossum03093a21994-09-28 15:51:32 +00001258 return -1;
1259 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001260 v1 = PyTuple_GetItem(coerced, 0);
1261 w = PyTuple_GetItem(coerced, 1);
Guido van Rossum03093a21994-09-28 15:51:32 +00001262 if (v1 != v) {
1263 v = v1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001264 if (!PyInstance_Check(v) && !PyInstance_Check(w)) {
Guido van Rossume7d444f1995-01-07 12:35:18 +00001265 if (swapped)
1266 *r_result = (*thisfunc)(w, v);
1267 else
1268 *r_result = (*thisfunc)(v, w);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001269 Py_DECREF(coerced);
Guido van Rossume7d444f1995-01-07 12:35:18 +00001270 return *r_result == NULL ? -1 : 0;
Guido van Rossum03093a21994-09-28 15:51:32 +00001271 }
1272 }
Guido van Rossum03093a21994-09-28 15:51:32 +00001273 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001274 func = PyObject_GetAttrString(v, opname);
Guido van Rossume7d444f1995-01-07 12:35:18 +00001275 if (func == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001276 Py_XDECREF(coerced);
Guido van Rossum617c1b01998-05-28 19:50:02 +00001277 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
Guido van Rossume7d444f1995-01-07 12:35:18 +00001278 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001279 PyErr_Clear();
Guido van Rossume7d444f1995-01-07 12:35:18 +00001280 return 1;
1281 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001282 args = Py_BuildValue("(O)", w);
Guido van Rossum03093a21994-09-28 15:51:32 +00001283 if (args == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001284 Py_DECREF(func);
1285 Py_XDECREF(coerced);
Guido van Rossum03093a21994-09-28 15:51:32 +00001286 return -1;
1287 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001288 *r_result = PyEval_CallObject(func, args);
1289 Py_DECREF(args);
1290 Py_DECREF(func);
1291 Py_XDECREF(coerced);
Guido van Rossum03093a21994-09-28 15:51:32 +00001292 return *r_result == NULL ? -1 : 0;
1293}
1294
Guido van Rossum879c5811995-01-10 15:24:06 +00001295static int
Fred Drake79912472000-07-09 04:06:11 +00001296instance_coerce(PyObject **pv, PyObject **pw)
Guido van Rossum879c5811995-01-10 15:24:06 +00001297{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001298 PyObject *v = *pv;
1299 PyObject *w = *pw;
Guido van Rossum3931df91997-11-18 19:23:07 +00001300 PyObject *coercefunc;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001301 PyObject *args;
1302 PyObject *coerced;
Guido van Rossum879c5811995-01-10 15:24:06 +00001303
Guido van Rossum2878a691996-08-09 20:53:24 +00001304 if (coerce_obj == NULL) {
Guido van Rossuma412d241997-01-18 07:59:12 +00001305 coerce_obj = PyString_InternFromString("__coerce__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001306 if (coerce_obj == NULL)
1307 return -1;
1308 }
Guido van Rossum3931df91997-11-18 19:23:07 +00001309 coercefunc = PyObject_GetAttr(v, coerce_obj);
1310 if (coercefunc == NULL) {
Guido van Rossum879c5811995-01-10 15:24:06 +00001311 /* No __coerce__ method: always OK */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001312 PyErr_Clear();
1313 Py_INCREF(v);
1314 Py_INCREF(w);
Guido van Rossum879c5811995-01-10 15:24:06 +00001315 return 0;
1316 }
1317 /* Has __coerce__ method: call it */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001318 args = Py_BuildValue("(O)", w);
Guido van Rossum879c5811995-01-10 15:24:06 +00001319 if (args == NULL) {
1320 return -1;
1321 }
Guido van Rossum3931df91997-11-18 19:23:07 +00001322 coerced = PyEval_CallObject(coercefunc, args);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001323 Py_DECREF(args);
Guido van Rossum3931df91997-11-18 19:23:07 +00001324 Py_DECREF(coercefunc);
Guido van Rossum879c5811995-01-10 15:24:06 +00001325 if (coerced == NULL) {
1326 /* __coerce__ call raised an exception */
1327 return -1;
1328 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001329 if (coerced == Py_None) {
Guido van Rossum879c5811995-01-10 15:24:06 +00001330 /* __coerce__ says "I can't do it" */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001331 Py_DECREF(coerced);
Guido van Rossum879c5811995-01-10 15:24:06 +00001332 return 1;
1333 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001334 if (!PyTuple_Check(coerced) || PyTuple_Size(coerced) != 2) {
Guido van Rossum879c5811995-01-10 15:24:06 +00001335 /* __coerce__ return value is malformed */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001336 Py_DECREF(coerced);
1337 PyErr_SetString(PyExc_TypeError,
Guido van Rossum879c5811995-01-10 15:24:06 +00001338 "coercion should return None or 2-tuple");
1339 return -1;
1340 }
1341 /* __coerce__ returned two new values */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001342 *pv = PyTuple_GetItem(coerced, 0);
1343 *pw = PyTuple_GetItem(coerced, 1);
1344 Py_INCREF(*pv);
1345 Py_INCREF(*pw);
1346 Py_DECREF(coerced);
Guido van Rossum879c5811995-01-10 15:24:06 +00001347 return 0;
1348}
1349
Guido van Rossum03093a21994-09-28 15:51:32 +00001350
Guido van Rossum04691fc1992-08-12 15:35:34 +00001351#define UNARY(funcname, methodname) \
Thomas Woutersc3073522000-07-23 22:09:59 +00001352static PyObject *funcname(PyInstanceObject *self) { \
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001353 static PyObject *o; \
Guido van Rossuma412d241997-01-18 07:59:12 +00001354 if (o == NULL) o = PyString_InternFromString(methodname); \
Guido van Rossum2878a691996-08-09 20:53:24 +00001355 return generic_unary_op(self, o); \
Guido van Rossum04691fc1992-08-12 15:35:34 +00001356}
1357
Guido van Rossum04691fc1992-08-12 15:35:34 +00001358UNARY(instance_neg, "__neg__")
1359UNARY(instance_pos, "__pos__")
1360UNARY(instance_abs, "__abs__")
1361
Guido van Rossum9bfef441993-03-29 10:43:31 +00001362static int
Fred Drake79912472000-07-09 04:06:11 +00001363instance_nonzero(PyInstanceObject *self)
Guido van Rossum04691fc1992-08-12 15:35:34 +00001364{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001365 PyObject *func, *res;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001366 long outcome;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001367 static PyObject *nonzerostr;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001368
Guido van Rossum2878a691996-08-09 20:53:24 +00001369 if (nonzerostr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +00001370 nonzerostr = PyString_InternFromString("__nonzero__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001371 if ((func = instance_getattr(self, nonzerostr)) == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001372 PyErr_Clear();
Guido van Rossum2878a691996-08-09 20:53:24 +00001373 if (lenstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +00001374 lenstr = PyString_InternFromString("__len__");
Guido van Rossum2878a691996-08-09 20:53:24 +00001375 if ((func = instance_getattr(self, lenstr)) == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001376 PyErr_Clear();
Guido van Rossum04691fc1992-08-12 15:35:34 +00001377 /* Fall back to the default behavior:
1378 all instances are nonzero */
1379 return 1;
1380 }
1381 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001382 res = PyEval_CallObject(func, (PyObject *)NULL);
1383 Py_DECREF(func);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001384 if (res == NULL)
1385 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001386 if (!PyInt_Check(res)) {
1387 Py_DECREF(res);
1388 PyErr_SetString(PyExc_TypeError,
1389 "__nonzero__ should return an int");
Guido van Rossum04691fc1992-08-12 15:35:34 +00001390 return -1;
1391 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001392 outcome = PyInt_AsLong(res);
1393 Py_DECREF(res);
Guido van Rossum04691fc1992-08-12 15:35:34 +00001394 if (outcome < 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001395 PyErr_SetString(PyExc_ValueError,
1396 "__nonzero__ should return >= 0");
Guido van Rossum04691fc1992-08-12 15:35:34 +00001397 return -1;
1398 }
1399 return outcome > 0;
1400}
1401
1402UNARY(instance_invert, "__invert__")
Guido van Rossum1899c2e1992-09-12 11:09:23 +00001403UNARY(instance_int, "__int__")
1404UNARY(instance_long, "__long__")
1405UNARY(instance_float, "__float__")
1406UNARY(instance_oct, "__oct__")
1407UNARY(instance_hex, "__hex__")
1408
Guido van Rossum03093a21994-09-28 15:51:32 +00001409/* This version is for ternary calls only (z != None) */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001410static PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001411instance_pow(PyObject *v, PyObject *w, PyObject *z)
Guido van Rossum03093a21994-09-28 15:51:32 +00001412{
1413 /* XXX Doesn't do coercions... */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001414 PyObject *func;
1415 PyObject *args;
1416 PyObject *result;
1417 static PyObject *powstr;
Guido van Rossum2878a691996-08-09 20:53:24 +00001418
1419 if (powstr == NULL)
Guido van Rossuma412d241997-01-18 07:59:12 +00001420 powstr = PyString_InternFromString("__pow__");
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001421 func = PyObject_GetAttr(v, powstr);
Guido van Rossum03093a21994-09-28 15:51:32 +00001422 if (func == NULL)
1423 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001424 args = Py_BuildValue("(OO)", w, z);
Guido van Rossum03093a21994-09-28 15:51:32 +00001425 if (args == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001426 Py_DECREF(func);
Guido van Rossum03093a21994-09-28 15:51:32 +00001427 return NULL;
1428 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001429 result = PyEval_CallObject(func, args);
1430 Py_DECREF(func);
1431 Py_DECREF(args);
Guido van Rossum03093a21994-09-28 15:51:32 +00001432 return result;
1433}
1434
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001435static PyObject *
1436instance_inplace_pow(PyObject *v, PyObject *w, PyObject *z)
1437{
1438 /* XXX Doesn't do coercions... */
1439 PyObject *func;
1440 PyObject *args;
1441 PyObject *result;
1442 static PyObject *ipowstr;
1443
1444 if (ipowstr == NULL)
1445 ipowstr = PyString_InternFromString("__ipow__");
1446 func = PyObject_GetAttr(v, ipowstr);
1447 if (func == NULL) {
1448 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
1449 return NULL;
Thomas Wouters1de2a792000-08-25 10:47:46 +00001450 PyErr_Clear();
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001451 return instance_pow(v, w, z);
1452 }
1453 args = Py_BuildValue("(OO)", w, z);
1454 if (args == NULL) {
1455 Py_DECREF(func);
1456 return NULL;
1457 }
1458 result = PyEval_CallObject(func, args);
1459 Py_DECREF(func);
1460 Py_DECREF(args);
1461 return result;
1462}
1463
1464
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001465static PyNumberMethods instance_as_number = {
Guido van Rossum03093a21994-09-28 15:51:32 +00001466 0, /*nb_add*/
1467 0, /*nb_subtract*/
1468 0, /*nb_multiply*/
1469 0, /*nb_divide*/
1470 0, /*nb_remainder*/
1471 0, /*nb_divmod*/
Guido van Rossum52ca98a1994-09-05 07:32:29 +00001472 (ternaryfunc)instance_pow, /*nb_power*/
Guido van Rossumb6775db1994-08-01 11:34:53 +00001473 (unaryfunc)instance_neg, /*nb_negative*/
1474 (unaryfunc)instance_pos, /*nb_positive*/
1475 (unaryfunc)instance_abs, /*nb_absolute*/
1476 (inquiry)instance_nonzero, /*nb_nonzero*/
1477 (unaryfunc)instance_invert, /*nb_invert*/
Guido van Rossum03093a21994-09-28 15:51:32 +00001478 0, /*nb_lshift*/
1479 0, /*nb_rshift*/
1480 0, /*nb_and*/
1481 0, /*nb_xor*/
1482 0, /*nb_or*/
Guido van Rossum879c5811995-01-10 15:24:06 +00001483 (coercion)instance_coerce, /*nb_coerce*/
Guido van Rossumb6775db1994-08-01 11:34:53 +00001484 (unaryfunc)instance_int, /*nb_int*/
1485 (unaryfunc)instance_long, /*nb_long*/
1486 (unaryfunc)instance_float, /*nb_float*/
1487 (unaryfunc)instance_oct, /*nb_oct*/
1488 (unaryfunc)instance_hex, /*nb_hex*/
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001489 0, /*nb_inplace_add*/
1490 0, /*nb_inplace_subtract*/
1491 0, /*nb_inplace_multiply*/
1492 0, /*nb_inplace_divide*/
1493 0, /*nb_inplace_remainder*/
1494 (ternaryfunc)instance_inplace_pow, /*nb_inplace_power*/
1495 0, /*nb_inplace_lshift*/
1496 0, /*nb_inplace_rshift*/
1497 0, /*nb_inplace_and*/
1498 0, /*nb_inplace_xor*/
1499 0, /*nb_inplace_or*/
Guido van Rossum04691fc1992-08-12 15:35:34 +00001500};
1501
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001502PyTypeObject PyInstance_Type = {
1503 PyObject_HEAD_INIT(&PyType_Type)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001504 0,
Guido van Rossum21ed88c1991-04-04 10:42:10 +00001505 "instance",
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001506 sizeof(PyInstanceObject) + PyGC_HEAD_SIZE,
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001507 0,
Guido van Rossumb6775db1994-08-01 11:34:53 +00001508 (destructor)instance_dealloc, /*tp_dealloc*/
Guido van Rossum7066dd71992-09-17 17:54:56 +00001509 0, /*tp_print*/
Guido van Rossum2878a691996-08-09 20:53:24 +00001510 0, /*tp_getattr*/
1511 0, /*tp_setattr*/
1512 instance_compare, /*tp_compare*/
Guido van Rossumb6775db1994-08-01 11:34:53 +00001513 (reprfunc)instance_repr, /*tp_repr*/
Guido van Rossum04691fc1992-08-12 15:35:34 +00001514 &instance_as_number, /*tp_as_number*/
1515 &instance_as_sequence, /*tp_as_sequence*/
1516 &instance_as_mapping, /*tp_as_mapping*/
Guido van Rossumb6775db1994-08-01 11:34:53 +00001517 (hashfunc)instance_hash, /*tp_hash*/
Guido van Rossum2878a691996-08-09 20:53:24 +00001518 0, /*tp_call*/
1519 0, /*tp_str*/
1520 (getattrofunc)instance_getattr, /*tp_getattro*/
1521 (setattrofunc)instance_setattr, /*tp_setattro*/
Guido van Rossumee28c3a2000-02-28 15:03:15 +00001522 0, /* tp_as_buffer */
Jeremy Hyltond08b4c42000-06-23 19:37:02 +00001523 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_GC, /*tp_flags*/
Jeremy Hylton8caad492000-06-23 14:18:11 +00001524 0, /* tp_doc */
1525 (traverseproc)instance_traverse, /* tp_traverse */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001526};
1527
1528
Guido van Rossum81daa321993-05-20 14:24:46 +00001529/* Instance method objects are used for two purposes:
1530 (a) as bound instance methods (returned by instancename.methodname)
1531 (b) as unbound methods (returned by ClassName.methodname)
1532 In case (b), im_self is NULL
1533*/
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001534
Guido van Rossuma0d349f1997-08-05 02:06:53 +00001535static PyMethodObject *free_list;
1536
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001537PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001538PyMethod_New(PyObject *func, PyObject *self, PyObject *class)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001539{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001540 register PyMethodObject *im;
Guido van Rossum7859f871998-07-08 14:58:16 +00001541 if (!PyCallable_Check(func)) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001542 PyErr_BadInternalCall();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001543 return NULL;
1544 }
Guido van Rossuma0d349f1997-08-05 02:06:53 +00001545 im = free_list;
1546 if (im != NULL) {
1547 free_list = (PyMethodObject *)(im->im_self);
Guido van Rossumb18618d2000-05-03 23:44:39 +00001548 PyObject_INIT(im, &PyMethod_Type);
Guido van Rossuma0d349f1997-08-05 02:06:53 +00001549 }
1550 else {
1551 im = PyObject_NEW(PyMethodObject, &PyMethod_Type);
1552 if (im == NULL)
1553 return NULL;
1554 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001555 Py_INCREF(func);
Guido van Rossume8122f11991-05-05 20:03:07 +00001556 im->im_func = func;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001557 Py_XINCREF(self);
Guido van Rossume8122f11991-05-05 20:03:07 +00001558 im->im_self = self;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001559 Py_INCREF(class);
Guido van Rossum81daa321993-05-20 14:24:46 +00001560 im->im_class = class;
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001561 PyObject_GC_Init(im);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001562 return (PyObject *)im;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001563}
1564
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001565PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001566PyMethod_Function(register PyObject *im)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001567{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001568 if (!PyMethod_Check(im)) {
1569 PyErr_BadInternalCall();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001570 return NULL;
1571 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001572 return ((PyMethodObject *)im)->im_func;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001573}
1574
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001575PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001576PyMethod_Self(register PyObject *im)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001577{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001578 if (!PyMethod_Check(im)) {
1579 PyErr_BadInternalCall();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001580 return NULL;
1581 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001582 return ((PyMethodObject *)im)->im_self;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001583}
1584
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001585PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001586PyMethod_Class(register PyObject *im)
Guido van Rossum81daa321993-05-20 14:24:46 +00001587{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001588 if (!PyMethod_Check(im)) {
1589 PyErr_BadInternalCall();
Guido van Rossum81daa321993-05-20 14:24:46 +00001590 return NULL;
1591 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001592 return ((PyMethodObject *)im)->im_class;
Guido van Rossum81daa321993-05-20 14:24:46 +00001593}
1594
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001595/* Class method methods */
1596
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001597#define OFF(x) offsetof(PyMethodObject, x)
Guido van Rossum3f5da241990-12-20 15:06:42 +00001598
Guido van Rossume8122f11991-05-05 20:03:07 +00001599static struct memberlist instancemethod_memberlist[] = {
1600 {"im_func", T_OBJECT, OFF(im_func)},
1601 {"im_self", T_OBJECT, OFF(im_self)},
Guido van Rossum81daa321993-05-20 14:24:46 +00001602 {"im_class", T_OBJECT, OFF(im_class)},
Guido van Rossum6dabc981996-05-14 21:54:20 +00001603 /* Dummies that are not handled by getattr() except for __members__ */
1604 {"__doc__", T_INT, 0},
1605 {"__name__", T_INT, 0},
Guido van Rossum3f5da241990-12-20 15:06:42 +00001606 {NULL} /* Sentinel */
1607};
1608
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001609static PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001610instancemethod_getattr(register PyMethodObject *im, PyObject *name)
Guido van Rossum3f5da241990-12-20 15:06:42 +00001611{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001612 char *sname = PyString_AsString(name);
Guido van Rossum2878a691996-08-09 20:53:24 +00001613 if (sname[0] == '_') {
Guido van Rossum7859f871998-07-08 14:58:16 +00001614 /* Inherit __name__ and __doc__ from the callable object
1615 implementing the method */
1616 if (strcmp(sname, "__name__") == 0 ||
1617 strcmp(sname, "__doc__") == 0)
1618 return PyObject_GetAttr(im->im_func, name);
Guido van Rossum6dabc981996-05-14 21:54:20 +00001619 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001620 if (PyEval_GetRestricted()) {
1621 PyErr_SetString(PyExc_RuntimeError,
1622 "instance-method attributes not accessible in restricted mode");
Guido van Rossum10393b11995-01-10 10:39:49 +00001623 return NULL;
1624 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001625 return PyMember_Get((char *)im, instancemethod_memberlist, sname);
Guido van Rossum3f5da241990-12-20 15:06:42 +00001626}
1627
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001628static void
Fred Drake79912472000-07-09 04:06:11 +00001629instancemethod_dealloc(register PyMethodObject *im)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001630{
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001631 PyObject_GC_Fini(im);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001632 Py_DECREF(im->im_func);
1633 Py_XDECREF(im->im_self);
1634 Py_DECREF(im->im_class);
Guido van Rossuma0d349f1997-08-05 02:06:53 +00001635 im->im_self = (PyObject *)free_list;
1636 free_list = im;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001637}
1638
Guido van Rossumebc8c511992-09-03 20:39:51 +00001639static int
Fred Drake79912472000-07-09 04:06:11 +00001640instancemethod_compare(PyMethodObject *a, PyMethodObject *b)
Guido van Rossumebc8c511992-09-03 20:39:51 +00001641{
Guido van Rossume9df7271995-04-06 14:46:51 +00001642 if (a->im_self != b->im_self)
1643 return (a->im_self < b->im_self) ? -1 : 1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001644 return PyObject_Compare(a->im_func, b->im_func);
Guido van Rossumebc8c511992-09-03 20:39:51 +00001645}
1646
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001647static PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001648instancemethod_repr(PyMethodObject *a)
Guido van Rossum25831651993-05-19 14:50:45 +00001649{
1650 char buf[240];
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001651 PyInstanceObject *self = (PyInstanceObject *)(a->im_self);
Guido van Rossum7859f871998-07-08 14:58:16 +00001652 PyObject *func = a->im_func;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001653 PyClassObject *class = (PyClassObject *)(a->im_class);
1654 PyObject *fclassname, *iclassname, *funcname;
Guido van Rossum81daa321993-05-20 14:24:46 +00001655 char *fcname, *icname, *fname;
1656 fclassname = class->cl_name;
Guido van Rossum7859f871998-07-08 14:58:16 +00001657 if (PyFunction_Check(func)) {
1658 funcname = ((PyFunctionObject *)func)->func_name;
1659 Py_INCREF(funcname);
1660 }
1661 else {
1662 funcname = PyObject_GetAttrString(func,"__name__");
1663 if (funcname == NULL)
1664 PyErr_Clear();
1665 }
1666 if (funcname != NULL && PyString_Check(funcname))
1667 fname = PyString_AS_STRING(funcname);
1668 else
1669 fname = "?";
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001670 if (fclassname != NULL && PyString_Check(fclassname))
1671 fcname = PyString_AsString(fclassname);
Guido van Rossum25831651993-05-19 14:50:45 +00001672 else
Guido van Rossum81daa321993-05-20 14:24:46 +00001673 fcname = "?";
Guido van Rossum81daa321993-05-20 14:24:46 +00001674 if (self == NULL)
1675 sprintf(buf, "<unbound method %.100s.%.100s>", fcname, fname);
1676 else {
1677 iclassname = self->in_class->cl_name;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001678 if (iclassname != NULL && PyString_Check(iclassname))
1679 icname = PyString_AsString(iclassname);
Guido van Rossum81daa321993-05-20 14:24:46 +00001680 else
1681 icname = "?";
Fred Drakea44d3532000-06-30 15:01:00 +00001682 sprintf(buf, "<method %.60s.%.60s of %.60s instance at %p>",
1683 fcname, fname, icname, self);
Guido van Rossum81daa321993-05-20 14:24:46 +00001684 }
Guido van Rossum42636dc1999-10-11 14:03:12 +00001685 Py_XDECREF(funcname);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001686 return PyString_FromString(buf);
Guido van Rossum25831651993-05-19 14:50:45 +00001687}
1688
Guido van Rossum9bfef441993-03-29 10:43:31 +00001689static long
Fred Drake79912472000-07-09 04:06:11 +00001690instancemethod_hash(PyMethodObject *a)
Guido van Rossum9bfef441993-03-29 10:43:31 +00001691{
1692 long x, y;
Guido van Rossum81daa321993-05-20 14:24:46 +00001693 if (a->im_self == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001694 x = PyObject_Hash(Py_None);
Guido van Rossum81daa321993-05-20 14:24:46 +00001695 else
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001696 x = PyObject_Hash(a->im_self);
Guido van Rossum9bfef441993-03-29 10:43:31 +00001697 if (x == -1)
1698 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001699 y = PyObject_Hash(a->im_func);
Guido van Rossum9bfef441993-03-29 10:43:31 +00001700 if (y == -1)
1701 return -1;
1702 return x ^ y;
1703}
1704
Jeremy Hylton8caad492000-06-23 14:18:11 +00001705static int
1706instancemethod_traverse(PyMethodObject *im, visitproc visit, void *arg)
1707{
1708 int err;
1709 if (im->im_func) {
1710 err = visit(im->im_func, arg);
1711 if (err)
1712 return err;
1713 }
1714 if (im->im_self) {
1715 err = visit(im->im_self, arg);
1716 if (err)
1717 return err;
1718 }
1719 if (im->im_class) {
1720 err = visit(im->im_class, arg);
1721 if (err)
1722 return err;
1723 }
Jeremy Hyltond22162b2000-06-23 17:14:56 +00001724 return 0;
Jeremy Hylton8caad492000-06-23 14:18:11 +00001725}
1726
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001727PyTypeObject PyMethod_Type = {
1728 PyObject_HEAD_INIT(&PyType_Type)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001729 0,
Guido van Rossum569fce71991-04-16 08:38:43 +00001730 "instance method",
Jeremy Hyltonc5007aa2000-06-30 05:02:53 +00001731 sizeof(PyMethodObject) + PyGC_HEAD_SIZE,
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001732 0,
Guido van Rossumb6775db1994-08-01 11:34:53 +00001733 (destructor)instancemethod_dealloc, /*tp_dealloc*/
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001734 0, /*tp_print*/
Guido van Rossum2878a691996-08-09 20:53:24 +00001735 0, /*tp_getattr*/
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001736 0, /*tp_setattr*/
Guido van Rossumb6775db1994-08-01 11:34:53 +00001737 (cmpfunc)instancemethod_compare, /*tp_compare*/
1738 (reprfunc)instancemethod_repr, /*tp_repr*/
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001739 0, /*tp_as_number*/
1740 0, /*tp_as_sequence*/
1741 0, /*tp_as_mapping*/
Guido van Rossumb6775db1994-08-01 11:34:53 +00001742 (hashfunc)instancemethod_hash, /*tp_hash*/
Guido van Rossum2878a691996-08-09 20:53:24 +00001743 0, /*tp_call*/
1744 0, /*tp_str*/
1745 (getattrofunc)instancemethod_getattr, /*tp_getattro*/
1746 0, /*tp_setattro*/
Jeremy Hylton8caad492000-06-23 14:18:11 +00001747 0, /* tp_as_buffer */
Jeremy Hyltond08b4c42000-06-23 19:37:02 +00001748 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_GC, /*tp_flags*/
Jeremy Hylton8caad492000-06-23 14:18:11 +00001749 0, /* tp_doc */
1750 (traverseproc)instancemethod_traverse, /* tp_traverse */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001751};
Guido van Rossuma0d349f1997-08-05 02:06:53 +00001752
1753/* Clear out the free list */
1754
1755void
Fred Drake79912472000-07-09 04:06:11 +00001756PyMethod_Fini(void)
Guido van Rossuma0d349f1997-08-05 02:06:53 +00001757{
1758 while (free_list) {
Guido van Rossumb18618d2000-05-03 23:44:39 +00001759 PyMethodObject *im = free_list;
1760 free_list = (PyMethodObject *)(im->im_self);
Guido van Rossum4cc6ac72000-07-01 01:00:38 +00001761 im = (PyMethodObject *) PyObject_AS_GC(im);
Guido van Rossumb18618d2000-05-03 23:44:39 +00001762 PyObject_DEL(im);
Guido van Rossuma0d349f1997-08-05 02:06:53 +00001763 }
1764}