TonBERRY/micropython
3
0
Fork 0
Code Pull Requests Activity

py/obj: Add support for __int__ special method.

Browse Source 
Based on the discussion, this special method is available unconditionally,
as converting to int is a common operation.
This commit is contained in:
Paul Sokolovsky
2018-08-29 17:59:36 +03:00
committed by Damien George
parent 113f00a9ab
commit b1d08726ee
5 changed files with 36 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
py/obj.c
View File

@@ -235,12 +235,8 @@ mp_int_t mp_obj_get_int(mp_const_obj_t arg) {
} else if (MP_OBJ_IS_TYPE(arg, &mp_type_int)) {
return mp_obj_int_get_checked(arg);
} else {
if (MICROPY_ERROR_REPORTING == MICROPY_ERROR_REPORTING_TERSE) {
mp_raise_TypeError("can't convert to int");
} else {
nlr_raise(mp_obj_new_exception_msg_varg(&mp_type_TypeError,
"can't convert %s to int", mp_obj_get_type_str(arg)));
}
mp_obj_t res = mp_unary_op(MP_UNARY_OP_INT, (mp_obj_t)arg);
return mp_obj_int_get_checked(res);
}
}