objtype.c 19.9 KB
Newer Older
1
2
#include <string.h>
#include <assert.h>
3
4
5
6

#include "nlr.h"
#include "misc.h"
#include "mpconfig.h"
7
#include "qstr.h"
8
#include "obj.h"
9
#include "map.h"
10
#include "runtime0.h"
11
12
13
14
15
16
17
18
19
20
21
#include "runtime.h"

/******************************************************************************/
// class object
// creating an instance of a class makes one of these objects

typedef struct _mp_obj_class_t {
    mp_obj_base_t base;
    mp_map_t members;
} mp_obj_class_t;

22
STATIC mp_obj_t mp_obj_new_class(mp_obj_t class) {
23
24
25
26
27
28
    mp_obj_class_t *o = m_new_obj(mp_obj_class_t);
    o->base.type = class;
    mp_map_init(&o->members, 0);
    return o;
}

29
// will return MP_OBJ_NULL if not found
30
STATIC mp_obj_t mp_obj_class_lookup(const mp_obj_type_t *type, qstr attr) {
31
    for (;;) {
32
33
34
35
        if (type->locals_dict != NULL) {
            // search locals_dict (the dynamically created set of methods/attributes)

            assert(MP_OBJ_IS_TYPE(type->locals_dict, &dict_type)); // Micro Python restriction, for now
36
            mp_map_t *locals_map = mp_obj_dict_get_map(type->locals_dict);
37
38
39
40
            mp_map_elem_t *elem = mp_map_lookup(locals_map, MP_OBJ_NEW_QSTR(attr), MP_MAP_LOOKUP);
            if (elem != NULL) {
                return elem->value;
            }
41
            /*
42
43
44
        } else if (type->methods != NULL) {
            // search methods (the const set of methods)

45
46
            for (const mp_method_t *meth = type->methods; meth->name != MP_QSTR_NULL; meth++) {
                if (meth->name == attr) {
47
48
49
                    return (mp_obj_t)meth->fun;
                }
            }
50
            */
51
52
53
54
55
        }

        // attribute not found, keep searching base classes

        // for a const struct, this entry might be NULL
56
        if (type->bases_tuple == MP_OBJ_NULL) {
57
58
59
60
61
            return NULL;
        }

        uint len;
        mp_obj_t *items;
62
        mp_obj_tuple_get(type->bases_tuple, &len, &items);
63
64
65
66
        if (len == 0) {
            return NULL;
        }
        for (uint i = 0; i < len - 1; i++) {
67
            assert(MP_OBJ_IS_TYPE(items[i], &mp_type_type));
68
69
70
            mp_obj_t obj = mp_obj_class_lookup((mp_obj_type_t*)items[i], attr);
            if (obj != MP_OBJ_NULL) {
                return obj;
71
72
73
74
            }
        }

        // search last base (simple tail recursion elimination)
75
        assert(MP_OBJ_IS_TYPE(items[len - 1], &mp_type_type));
76
        type = (mp_obj_type_t*)items[len - 1];
77
78
79
    }
}

80
STATIC void class_print(void (*print)(void *env, const char *fmt, ...), void *env, mp_obj_t self_in, mp_print_kind_t kind) {
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
    mp_obj_class_t *self = self_in;
    qstr meth = (kind == PRINT_STR) ? MP_QSTR___str__ : MP_QSTR___repr__;
    mp_obj_t member = mp_obj_class_lookup(self->base.type, meth);
    if (member == MP_OBJ_NULL && kind == PRINT_STR) {
        // If there's no __str__, fall back to __repr__
        member = mp_obj_class_lookup(self->base.type, MP_QSTR___repr__);
    }

    if (member != MP_OBJ_NULL) {
        mp_obj_t r = rt_call_function_1(member, self_in);
        mp_obj_print_helper(print, env, r, PRINT_STR);
        return;
    }

    // TODO: CPython prints fully-qualified type name
96
97
98
    print(env, "<%s object at %p>", mp_obj_get_type_str(self_in), self_in);
}

99
STATIC mp_obj_t class_make_new(mp_obj_t self_in, uint n_args, uint n_kw, const mp_obj_t *args) {
100
    assert(MP_OBJ_IS_TYPE(self_in, &mp_type_type));
101
    mp_obj_type_t *self = self_in;
102
103
104
105

    mp_obj_t o = mp_obj_new_class(self_in);

    // look for __init__ function
106
    mp_obj_t init_fn = mp_obj_class_lookup(self, MP_QSTR___init__);
107

108
    if (init_fn != MP_OBJ_NULL) {
109
110
        // call __init__ function
        mp_obj_t init_ret;
111
        if (n_args == 0 && n_kw == 0) {
112
            init_ret = rt_call_function_n_kw(init_fn, 1, 0, (mp_obj_t*)&o);
113
        } else {
114
115
116
            mp_obj_t *args2 = m_new(mp_obj_t, 1 + n_args + 2 * n_kw);
            args2[0] = o;
            memcpy(args2 + 1, args, (n_args + 2 * n_kw) * sizeof(mp_obj_t));
117
            init_ret = rt_call_function_n_kw(init_fn, n_args + 1, n_kw, args2);
118
            m_del(mp_obj_t, args2, 1 + n_args + 2 * n_kw);
119
120
        }
        if (init_ret != mp_const_none) {
121
            nlr_jump(mp_obj_new_exception_msg_varg(&mp_type_TypeError, "__init__() should return None, not '%s'", mp_obj_get_type_str(init_ret)));
122
123
124
125
126
        }

    } else {
        // TODO
        if (n_args != 0) {
127
            nlr_jump(mp_obj_new_exception_msg_varg(&mp_type_TypeError, "function takes 0 positional arguments but %d were given", n_args));
128
129
130
131
132
133
        }
    }

    return o;
}

134
STATIC const qstr unary_op_method_name[] = {
135
136
137
138
139
    [RT_UNARY_OP_BOOL] = MP_QSTR___bool__,
    [RT_UNARY_OP_LEN] = MP_QSTR___len__,
    //[RT_UNARY_OP_POSITIVE,
    //[RT_UNARY_OP_NEGATIVE,
    //[RT_UNARY_OP_INVERT,
140
    [RT_UNARY_OP_NOT] = MP_QSTR_, // don't need to implement this, used to make sure array has full size
141
142
};

143
STATIC mp_obj_t class_unary_op(int op, mp_obj_t self_in) {
144
145
146
147
148
149
150
151
152
153
154
155
156
    mp_obj_class_t *self = self_in;
    qstr op_name = unary_op_method_name[op];
    if (op_name == 0) {
        return MP_OBJ_NULL;
    }
    mp_obj_t member = mp_obj_class_lookup(self->base.type, op_name);
    if (member != MP_OBJ_NULL) {
        return rt_call_function_1(member, self_in);
    } else {
        return MP_OBJ_NULL;
    }
}

157
STATIC const qstr binary_op_method_name[] = {
158
    [RT_BINARY_OP_SUBSCR] = MP_QSTR___getitem__,
159
160
161
162
163
164
165
    /*
    RT_BINARY_OP_OR,
    RT_BINARY_OP_XOR,
    RT_BINARY_OP_AND,
    RT_BINARY_OP_LSHIFT,
    RT_BINARY_OP_RSHIFT,
    */
166
167
    [RT_BINARY_OP_ADD] = MP_QSTR___add__,
    [RT_BINARY_OP_SUBTRACT] = MP_QSTR___sub__,
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
    /*
    RT_BINARY_OP_MULTIPLY,
    RT_BINARY_OP_FLOOR_DIVIDE,
    RT_BINARY_OP_TRUE_DIVIDE,
    RT_BINARY_OP_MODULO,
    RT_BINARY_OP_POWER,
    RT_BINARY_OP_INPLACE_OR,
    RT_BINARY_OP_INPLACE_XOR,
    RT_BINARY_OP_INPLACE_AND,
    RT_BINARY_OP_INPLACE_LSHIFT,
    RT_BINARY_OP_INPLACE_RSHIFT,
    RT_BINARY_OP_INPLACE_ADD,
    RT_BINARY_OP_INPLACE_SUBTRACT,
    RT_BINARY_OP_INPLACE_MULTIPLY,
    RT_BINARY_OP_INPLACE_FLOOR_DIVIDE,
    RT_BINARY_OP_INPLACE_TRUE_DIVIDE,
    RT_BINARY_OP_INPLACE_MODULO,
    RT_BINARY_OP_INPLACE_POWER,
186
187
188
189
190
191
192
193
    RT_BINARY_OP_LESS,
    RT_BINARY_OP_MORE,
    RT_BINARY_OP_EQUAL,
    RT_BINARY_OP_LESS_EQUAL,
    RT_BINARY_OP_MORE_EQUAL,
    RT_BINARY_OP_NOT_EQUAL,
    RT_BINARY_OP_IN,
    RT_BINARY_OP_IS,
194
    */
195
    [RT_BINARY_OP_EXCEPTION_MATCH] = MP_QSTR_, // not implemented, used to make sure array has full size
196
197
};

198
STATIC mp_obj_t class_binary_op(int op, mp_obj_t lhs_in, mp_obj_t rhs_in) {
199
    mp_obj_class_t *lhs = lhs_in;
200
201
    qstr op_name = binary_op_method_name[op];
    if (op_name == 0) {
202
203
        return MP_OBJ_NULL;
    }
204
    mp_obj_t member = mp_obj_class_lookup(lhs->base.type, op_name);
205
206
    if (member != MP_OBJ_NULL) {
        return rt_call_function_2(member, lhs_in, rhs_in);
207
208
209
210
211
    } else {
        return MP_OBJ_NULL;
    }
}

212
STATIC void class_load_attr(mp_obj_t self_in, qstr attr, mp_obj_t *dest) {
213
214
215
216
217
    // logic: look in obj members then class locals (TODO check this against CPython)
    mp_obj_class_t *self = self_in;
    mp_map_elem_t *elem = mp_map_lookup(&self->members, MP_OBJ_NEW_QSTR(attr), MP_MAP_LOOKUP);
    if (elem != NULL) {
        // object member, always treated as a value
218
        dest[0] = elem->value;
219
220
        return;
    }
221
222
223
    mp_obj_t member = mp_obj_class_lookup(self->base.type, attr);
    if (member != MP_OBJ_NULL) {
        if (mp_obj_is_callable(member)) {
224
            // class member is callable so build a bound method
225
226
227
228
229
            // check if the methods are functions, static or class methods
            // see http://docs.python.org/3.3/howto/descriptor.html
            // TODO check that this is the correct place to have this logic
            if (MP_OBJ_IS_TYPE(member, &mp_type_staticmethod)) {
                // return just the function
230
                dest[0] = ((mp_obj_static_class_method_t*)member)->fun;
231
232
            } else if (MP_OBJ_IS_TYPE(member, &mp_type_classmethod)) {
                // return a bound method, with self being the type of this object
233
                dest[0] = ((mp_obj_static_class_method_t*)member)->fun;
234
235
236
237
238
239
                dest[1] = mp_obj_get_type(self_in);
            } else {
                // return a bound method, with self being this object
                dest[0] = member;
                dest[1] = self_in;
            }
240
241
242
            return;
        } else {
            // class member is a value, so just return that value
243
            dest[0] = member;
244
245
246
247
248
            return;
        }
    }
}

249
STATIC bool class_store_attr(mp_obj_t self_in, qstr attr, mp_obj_t value) {
250
    mp_obj_class_t *self = self_in;
251
    mp_map_lookup(&self->members, MP_OBJ_NEW_QSTR(attr), MP_MAP_LOOKUP_ADD_IF_NOT_FOUND)->value = value;
252
253
254
    return true;
}

255
256
bool class_store_item(mp_obj_t self_in, mp_obj_t index, mp_obj_t value) {
    mp_obj_class_t *self = self_in;
257
    mp_obj_t member = mp_obj_class_lookup(self->base.type, MP_QSTR___setitem__);
258
    if (member != MP_OBJ_NULL) {
259
        mp_obj_t args[3] = {self_in, index, value};
260
261
        rt_call_function_n_kw(member, 3, 0, args);
        return true;
262
    } else {
263
        return false;
264
265
266
    }
}

267
268
269
/******************************************************************************/
// type object
//  - the struct is mp_obj_type_t and is defined in obj.h so const types can be made
270
//  - there is a constant mp_obj_type_t (called mp_type_type) for the 'type' object
271
//  - creating a new class (a new type) creates a new mp_obj_type_t
272

273
STATIC void type_print(void (*print)(void *env, const char *fmt, ...), void *env, mp_obj_t self_in, mp_print_kind_t kind) {
274
    mp_obj_type_t *self = self_in;
275
    print(env, "<class '%s'>", qstr_str(self->name));
276
277
}

278
STATIC mp_obj_t type_make_new(mp_obj_t type_in, uint n_args, uint n_kw, const mp_obj_t *args) {
279
280
    // TODO check n_kw == 0

281
282
283
284
285
    switch (n_args) {
        case 1:
            return mp_obj_get_type(args[0]);

        case 3:
286
            // args[0] = name
287
            // args[1] = bases tuple
288
            // args[2] = locals dict
289
            return mp_obj_new_type(mp_obj_str_get_qstr(args[0]), args[1], args[2]);
290
291

        default:
292
            nlr_jump(mp_obj_new_exception_msg(&mp_type_TypeError, "type takes 1 or 3 arguments"));
293
294
295
    }
}

296
STATIC mp_obj_t type_call(mp_obj_t self_in, uint n_args, uint n_kw, const mp_obj_t *args) {
297
298
    // instantiate an instance of a class

299
    mp_obj_type_t *self = self_in;
300
301

    if (self->make_new == NULL) {
302
        nlr_jump(mp_obj_new_exception_msg_varg(&mp_type_TypeError, "cannot create '%s' instances", qstr_str(self->name)));
303
    }
304
305

    // make new instance
306
    mp_obj_t o = self->make_new(self, n_args, n_kw, args);
307
308
309
310
311

    // return new instance
    return o;
}

312
// for fail, do nothing; for attr, dest[0] = value; for method, dest[0] = method, dest[1] = self
313
STATIC void type_load_attr(mp_obj_t self_in, qstr attr, mp_obj_t *dest) {
314
    assert(MP_OBJ_IS_TYPE(self_in, &mp_type_type));
315
    mp_obj_type_t *self = self_in;
316
317
318
319
320
321
    mp_obj_t member = mp_obj_class_lookup(self, attr);
    if (member != MP_OBJ_NULL) {
        // check if the methods are functions, static or class methods
        // see http://docs.python.org/3.3/howto/descriptor.html
        if (MP_OBJ_IS_TYPE(member, &mp_type_staticmethod)) {
            // return just the function
322
            dest[0] = ((mp_obj_static_class_method_t*)member)->fun;
323
324
        } else if (MP_OBJ_IS_TYPE(member, &mp_type_classmethod)) {
            // return a bound method, with self being this class
325
            dest[0] = ((mp_obj_static_class_method_t*)member)->fun;
326
327
328
329
330
            dest[1] = self_in;
        } else {
            // return just the function
            // TODO need to wrap in a type check for the first argument; eg list.append(1,1) needs to throw an exception
            dest[0] = (mp_obj_t)member;
331
332
        }
    }
333
334
}

335
STATIC bool type_store_attr(mp_obj_t self_in, qstr attr, mp_obj_t value) {
336
    assert(MP_OBJ_IS_TYPE(self_in, &mp_type_type));
337
338
    mp_obj_type_t *self = self_in;

339
340
    // TODO CPython allows STORE_ATTR to a class, but is this the correct implementation?

341
342
343
344
    if (self->locals_dict != NULL) {
        assert(MP_OBJ_IS_TYPE(self->locals_dict, &dict_type)); // Micro Python restriction, for now
        mp_map_t *locals_map = ((void*)self->locals_dict + sizeof(mp_obj_base_t)); // XXX hack to get map object from dict object
        mp_map_lookup(locals_map, MP_OBJ_NEW_QSTR(attr), MP_MAP_LOOKUP_ADD_IF_NOT_FOUND)->value = value;
345
346
347
348
        return true;
    } else {
        return false;
    }
349
350
}

351
352
const mp_obj_type_t mp_type_type = {
    { &mp_type_type },
353
    .name = MP_QSTR_type,
354
    .print = type_print,
355
    .make_new = type_make_new,
356
    .call = type_call,
357
358
    .load_attr = type_load_attr,
    .store_attr = type_store_attr,
359
};
360

361
mp_obj_t mp_obj_new_type(qstr name, mp_obj_t bases_tuple, mp_obj_t locals_dict) {
362
363
    assert(MP_OBJ_IS_TYPE(bases_tuple, &tuple_type)); // Micro Python restriction, for now
    assert(MP_OBJ_IS_TYPE(locals_dict, &dict_type)); // Micro Python restriction, for now
364
    mp_obj_type_t *o = m_new0(mp_obj_type_t, 1);
365
    o->base.type = &mp_type_type;
366
    o->name = name;
367
368
    o->print = class_print;
    o->make_new = class_make_new;
369
    o->unary_op = class_unary_op;
370
    o->binary_op = class_binary_op;
371
372
    o->load_attr = class_load_attr;
    o->store_attr = class_store_attr;
373
    o->store_item = class_store_item;
374
375
    o->bases_tuple = bases_tuple;
    o->locals_dict = locals_dict;
376
377
    return o;
}
378
379

/******************************************************************************/
Damien George's avatar
Damien George committed
380
381
382
383
384
385
386
387
// super object

typedef struct _mp_obj_super_t {
    mp_obj_base_t base;
    mp_obj_t type;
    mp_obj_t obj;
} mp_obj_super_t;

388
STATIC void super_print(void (*print)(void *env, const char *fmt, ...), void *env, mp_obj_t self_in, mp_print_kind_t kind) {
Damien George's avatar
Damien George committed
389
390
391
392
393
394
395
396
    mp_obj_super_t *self = self_in;
    print(env, "<super: ");
    mp_obj_print_helper(print, env, self->type, PRINT_STR);
    print(env, ", ");
    mp_obj_print_helper(print, env, self->obj, PRINT_STR);
    print(env, ">");
}

397
STATIC mp_obj_t super_make_new(mp_obj_t type_in, uint n_args, uint n_kw, const mp_obj_t *args) {
Damien George's avatar
Damien George committed
398
399
400
    if (n_args != 2 || n_kw != 0) {
        // 0 arguments are turned into 2 in the compiler
        // 1 argument is not yet implemented
401
        nlr_jump(mp_obj_new_exception_msg(&mp_type_TypeError, "super() requires 2 arguments"));
Damien George's avatar
Damien George committed
402
403
404
405
406
    }
    return mp_obj_new_super(args[0], args[1]);
}

// for fail, do nothing; for attr, dest[0] = value; for method, dest[0] = method, dest[1] = self
407
STATIC void super_load_attr(mp_obj_t self_in, qstr attr, mp_obj_t *dest) {
Damien George's avatar
Damien George committed
408
409
410
    assert(MP_OBJ_IS_TYPE(self_in, &super_type));
    mp_obj_super_t *self = self_in;

411
    assert(MP_OBJ_IS_TYPE(self->type, &mp_type_type));
Damien George's avatar
Damien George committed
412
413
414
415
416
417
418
419
420
421
422
423

    mp_obj_type_t *type = self->type;

    // for a const struct, this entry might be NULL
    if (type->bases_tuple == MP_OBJ_NULL) {
        return;
    }

    uint len;
    mp_obj_t *items;
    mp_obj_tuple_get(type->bases_tuple, &len, &items);
    for (uint i = 0; i < len; i++) {
424
        assert(MP_OBJ_IS_TYPE(items[i], &mp_type_type));
Damien George's avatar
Damien George committed
425
426
427
428
429
430
431
432
433
434
        mp_obj_t member = mp_obj_class_lookup((mp_obj_type_t*)items[i], attr);
        if (member != MP_OBJ_NULL) {
            // XXX this and the code in class_load_attr need to be factored out
            if (mp_obj_is_callable(member)) {
                // class member is callable so build a bound method
                // check if the methods are functions, static or class methods
                // see http://docs.python.org/3.3/howto/descriptor.html
                // TODO check that this is the correct place to have this logic
                if (MP_OBJ_IS_TYPE(member, &mp_type_staticmethod)) {
                    // return just the function
435
                    dest[0] = ((mp_obj_static_class_method_t*)member)->fun;
Damien George's avatar
Damien George committed
436
437
                } else if (MP_OBJ_IS_TYPE(member, &mp_type_classmethod)) {
                    // return a bound method, with self being the type of this object
438
                    dest[0] = ((mp_obj_static_class_method_t*)member)->fun;
Damien George's avatar
Damien George committed
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
                    dest[1] = mp_obj_get_type(self->obj);
                } else {
                    // return a bound method, with self being this object
                    dest[0] = member;
                    dest[1] = self->obj;
                }
                return;
            } else {
                // class member is a value, so just return that value
                dest[0] = member;
                return;
            }
        }
    }
}

const mp_obj_type_t super_type = {
456
    { &mp_type_type },
457
    .name = MP_QSTR_super,
Damien George's avatar
Damien George committed
458
459
460
461
462
463
464
465
466
467
468
469
    .print = super_print,
    .make_new = super_make_new,
    .load_attr = super_load_attr,
};

mp_obj_t mp_obj_new_super(mp_obj_t type, mp_obj_t obj) {
    mp_obj_super_t *o = m_new_obj(mp_obj_super_t);
    *o = (mp_obj_super_t){{&super_type}, type, obj};
    return o;
}

/******************************************************************************/
470
// subclassing and built-ins specific to types
471

472
473
// object and classinfo should be type objects
// (but the function will fail gracefully if they are not)
474
bool mp_obj_is_subclass_fast(mp_const_obj_t object, mp_const_obj_t classinfo) {
475
476
    for (;;) {
        if (object == classinfo) {
477
            return true;
478
479
480
481
        }

        // not equivalent classes, keep searching base classes

482
483
484
485
486
        // object should always be a type object, but just return false if it's not
        if (!MP_OBJ_IS_TYPE(object, &mp_type_type)) {
            return false;
        }

487
        const mp_obj_type_t *self = object;
488
489
490

        // for a const struct, this entry might be NULL
        if (self->bases_tuple == MP_OBJ_NULL) {
491
            return false;
492
493
        }

494
        // get the base objects (they should be type objects)
495
496
497
498
        uint len;
        mp_obj_t *items;
        mp_obj_tuple_get(self->bases_tuple, &len, &items);
        if (len == 0) {
499
            return false;
500
        }
501
502

        // iterate through the base objects
503
        for (uint i = 0; i < len - 1; i++) {
504
            if (mp_obj_is_subclass_fast(items[i], classinfo)) {
505
                return true;
506
507
508
509
510
511
512
513
            }
        }

        // search last base (simple tail recursion elimination)
        object = items[len - 1];
    }
}

514
515
516
517
518
519
520
521
522
523
STATIC mp_obj_t mp_obj_is_subclass(mp_obj_t object, mp_obj_t classinfo) {
    uint len;
    mp_obj_t *items;
    if (MP_OBJ_IS_TYPE(classinfo, &mp_type_type)) {
        len = 1;
        items = &classinfo;
    } else if (MP_OBJ_IS_TYPE(classinfo, &tuple_type)) {
        mp_obj_tuple_get(classinfo, &len, &items);
    } else {
        nlr_jump(mp_obj_new_exception_msg(&mp_type_TypeError, "issubclass() arg 2 must be a class or a tuple of classes"));
524
525
    }

526
527
528
529
    for (uint i = 0; i < len; i++) {
        if (mp_obj_is_subclass_fast(object, items[i])) {
            return mp_const_true;
        }
530
    }
531
    return mp_const_false;
532
533
}

534
STATIC mp_obj_t mp_builtin_issubclass(mp_obj_t object, mp_obj_t classinfo) {
535
536
537
538
    if (!MP_OBJ_IS_TYPE(object, &mp_type_type)) {
        nlr_jump(mp_obj_new_exception_msg(&mp_type_TypeError, "issubclass() arg 1 must be a class"));
    }
    return mp_obj_is_subclass(object, classinfo);
539
540
}

541
542
MP_DEFINE_CONST_FUN_OBJ_2(mp_builtin_issubclass_obj, mp_builtin_issubclass);

543
STATIC mp_obj_t mp_builtin_isinstance(mp_obj_t object, mp_obj_t classinfo) {
544
    return mp_obj_is_subclass(mp_obj_get_type(object), classinfo);
545
546
547
}

MP_DEFINE_CONST_FUN_OBJ_2(mp_builtin_isinstance_obj, mp_builtin_isinstance);
548
549
550
551

/******************************************************************************/
// staticmethod and classmethod types (probably should go in a different file)

552
STATIC mp_obj_t static_class_method_make_new(mp_obj_t self_in, uint n_args, uint n_kw, const mp_obj_t *args) {
553
554
555
    assert(self_in == &mp_type_staticmethod || self_in == &mp_type_classmethod);

    if (n_args != 1 || n_kw != 0) {
556
        nlr_jump(mp_obj_new_exception_msg_varg(&mp_type_TypeError, "function takes 1 positional argument but %d were given", n_args));
557
558
559
560
561
562
563
    }

    mp_obj_static_class_method_t *o = m_new_obj(mp_obj_static_class_method_t);
    *o = (mp_obj_static_class_method_t){{(mp_obj_type_t*)self_in}, args[0]};
    return o;
}

564
const mp_obj_type_t mp_type_staticmethod = {
565
    { &mp_type_type },
566
    .name = MP_QSTR_staticmethod,
567
    .make_new = static_class_method_make_new
568
569
570
};

const mp_obj_type_t mp_type_classmethod = {
571
    { &mp_type_type },
572
    .name = MP_QSTR_classmethod,
573
    .make_new = static_class_method_make_new
574
};