diff options
author | Damien George <damien.p.george@gmail.com> | 2018-12-06 18:02:41 +1100 |
---|---|---|
committer | Damien George <damien.p.george@gmail.com> | 2018-12-06 18:02:41 +1100 |
commit | 113f00a9ab469a83d1004dac502077af0a8f4847 (patch) | |
tree | f3a56db2ea5fbd90fc41e94c9d0488d2feb632b2 | |
parent | da7355e213eb06cd82ec48548e8180e71f750603 (diff) |
py/objboundmeth: Support loading generic attrs from the method.
Instead of assuming that the method is a bytecode object, and only
supporting load of __name__, make the operation generic by delegating the
load to the method object itself. Saves a bit of code size and fixes the
case of attempting to load __name__ on a native method, see issue #4028.
-rw-r--r-- | py/objboundmeth.c | 7 | ||||
-rw-r--r-- | tests/basics/fun_name.py | 7 |
2 files changed, 10 insertions, 4 deletions
diff --git a/py/objboundmeth.c b/py/objboundmeth.c index b0df6a68a..8fc44f163 100644 --- a/py/objboundmeth.c +++ b/py/objboundmeth.c @@ -89,10 +89,9 @@ STATIC void bound_meth_attr(mp_obj_t self_in, qstr attr, mp_obj_t *dest) { // not load attribute return; } - if (attr == MP_QSTR___name__) { - mp_obj_bound_meth_t *o = MP_OBJ_TO_PTR(self_in); - dest[0] = MP_OBJ_NEW_QSTR(mp_obj_fun_get_name(o->meth)); - } + // Delegate the load to the method object + mp_obj_bound_meth_t *self = MP_OBJ_TO_PTR(self_in); + mp_load_method_maybe(self->meth, attr, dest); } #endif diff --git a/tests/basics/fun_name.py b/tests/basics/fun_name.py index a724f4111..53ca93561 100644 --- a/tests/basics/fun_name.py +++ b/tests/basics/fun_name.py @@ -15,3 +15,10 @@ try: except AttributeError: print('SKIP') raise SystemExit + +# __name__ of a bound native method is not implemented in uPy +# the test here is to make sure it doesn't crash +try: + str((1).to_bytes.__name__) +except AttributeError: + pass |