Skip to content

Commit

Permalink
py/objexcept: Don't convert errno to str in constructor, do it in print.
Browse files Browse the repository at this point in the history
OSError's are now printed like:

    OSError: [Errno 1] EPERM

but only if the string corresponding to the errno is found.
  • Loading branch information
dpgeorge committed May 12, 2016
1 parent a314b84 commit 9a92499
Show file tree
Hide file tree
Showing 3 changed files with 14 additions and 10 deletions.
6 changes: 3 additions & 3 deletions py/moduerrno.c
Original file line number Diff line number Diff line change
Expand Up @@ -89,12 +89,12 @@ const mp_obj_module_t mp_module_uerrno = {
.globals = (mp_obj_dict_t*)&mp_module_uerrno_globals,
};

mp_obj_t mp_errno_to_str(mp_obj_t errno_val) {
qstr mp_errno_to_str(mp_obj_t errno_val) {
mp_map_elem_t *elem = mp_map_lookup((mp_map_t*)&errorcode_dict.map, errno_val, MP_MAP_LOOKUP);
if (elem == NULL) {
return errno_val;
return MP_QSTR_NULL;
} else {
return elem->value;
return MP_OBJ_QSTR_VALUE(elem->value);
}
}

Expand Down
4 changes: 1 addition & 3 deletions py/mperrno.h
Original file line number Diff line number Diff line change
Expand Up @@ -135,9 +135,7 @@
#endif

#if MICROPY_PY_UERRNO
mp_obj_t mp_errno_to_str(mp_obj_t errno_val);
#else
static inline mp_obj_t mp_errno_to_str(mp_obj_t errno_val) { return errno_val; }
qstr mp_errno_to_str(mp_obj_t errno_val);
#endif

#endif // __MICROPY_INCLUDED_PY_MPERRNO_H__
14 changes: 10 additions & 4 deletions py/objexcept.c
Original file line number Diff line number Diff line change
Expand Up @@ -108,6 +108,16 @@ STATIC void mp_obj_exception_print(const mp_print_t *print, mp_obj_t o_in, mp_pr
mp_print_str(print, "");
return;
} else if (o->args->len == 1) {
#if MICROPY_PY_UERRNO
// try to provide a nice OSError error message
if (o->base.type == &mp_type_OSError && MP_OBJ_IS_SMALL_INT(o->args->items[0])) {
qstr qst = mp_errno_to_str(o->args->items[0]);
if (qst != MP_QSTR_NULL) {
mp_printf(print, "[Errno %d] %q", MP_OBJ_SMALL_INT_VALUE(o->args->items[0]), qst);
return;
}
}
#endif
mp_obj_print_helper(print, o->args->items[0], PRINT_STR);
return;
}
Expand Down Expand Up @@ -289,10 +299,6 @@ mp_obj_t mp_obj_new_exception(const mp_obj_type_t *exc_type) {

// "Optimized" version for common(?) case of having 1 exception arg
mp_obj_t mp_obj_new_exception_arg1(const mp_obj_type_t *exc_type, mp_obj_t arg) {
// try to provide a nice string instead of numeric value for errno's
if (exc_type == &mp_type_OSError && MP_OBJ_IS_SMALL_INT(arg)) {
arg = mp_errno_to_str(arg);
}
return mp_obj_new_exception_args(exc_type, 1, &arg);
}

Expand Down

0 comments on commit 9a92499

Please sign in to comment.