objexcept: Add mp_obj_exception_get_value() convenience function.

This gets "value" of exceptions in the sense as it's defined for
StopIteration.value (i.e. args[0] or None).

TODO: This really should be inline function.

2 files changed, 12 insertions, 5 deletions

diff --git a/py/obj.h b/py/obj.h
index b52b5a0c7e..0dc392b832 100644
--- a/py/obj.h
+++ b/py/obj.h
@@ -313,6 +313,7 @@ bool mp_obj_is_exception_instance(mp_obj_t self_in);
 void mp_obj_exception_clear_traceback(mp_obj_t self_in);
 void mp_obj_exception_add_traceback(mp_obj_t self_in, qstr file, machine_uint_t line, qstr block);
 void mp_obj_exception_get_traceback(mp_obj_t self_in, machine_uint_t *n, machine_uint_t **values);
+mp_obj_t mp_obj_exception_get_value(mp_obj_t self_in);
 
 // str
 extern const mp_obj_type_t str_type;
diff --git a/py/objexcept.c b/py/objexcept.c
index 0650920e7d..8be3048fa2 100644
--- a/py/objexcept.c
+++ b/py/objexcept.c
@@ -59,16 +59,22 @@ STATIC mp_obj_t mp_obj_exception_make_new(mp_obj_t type_in, uint n_args, uint n_
     return o;
 }
 
+// Get exception "value" - that is, first argument, or None
+mp_obj_t mp_obj_exception_get_value(mp_obj_t self_in) {
+    mp_obj_exception_t *self = self_in;
+    if (self->args.len == 0) {
+        return mp_const_none;
+    } else {
+        return self->args.items[0];
+    }
+}
+
 STATIC void exception_load_attr(mp_obj_t self_in, qstr attr, mp_obj_t *dest) {
     mp_obj_exception_t *self = self_in;
     if (attr == MP_QSTR_args) {
         dest[0] = &self->args;
     } else if (self->base.type == &mp_type_StopIteration && attr == MP_QSTR_value) {
-        if (self->args.len == 0) {
-            dest[0] = mp_const_none;
-        } else {
-            dest[0] = self->args.items[0];
-        }
+        dest[0] = mp_obj_exception_get_value(self);
     }
 }