py/objclosure: Forward function attributes for closures.

Add .attr attribute which forwards to self->fun.

A closure is intended to wrap around a function object, so forward any
requested attributes to the wrapped function object.

Signed-off-by: Michael Bentley <mikebentley15@gmail.com>
This commit is contained in:
Michael Bentley 2021-10-03 00:05:27 -06:00 committed by Damien George
parent 432b65f178
commit d68532558d
1 changed files with 11 additions and 0 deletions

View File

@ -78,6 +78,14 @@ STATIC void closure_print(const mp_print_t *print, mp_obj_t o_in, mp_print_kind_
}
#endif
#if MICROPY_PY_FUNCTION_ATTRS
STATIC void mp_obj_closure_attr(mp_obj_t self_in, qstr attr, mp_obj_t *dest) {
// forward to self_in->fun
mp_obj_closure_t *o = MP_OBJ_TO_PTR(self_in);
mp_load_method_maybe(o->fun, attr, dest);
}
#endif
const mp_obj_type_t mp_type_closure = {
{ &mp_type_type },
.flags = MP_TYPE_FLAG_BINDS_SELF,
@ -86,6 +94,9 @@ const mp_obj_type_t mp_type_closure = {
.print = closure_print,
#endif
.call = closure_call,
#if MICROPY_PY_FUNCTION_ATTRS
.attr = mp_obj_closure_attr,
#endif
};
mp_obj_t mp_obj_new_closure(mp_obj_t fun, size_t n_closed_over, const mp_obj_t *closed) {