jebbatime
/
micropython
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

py/objgetitemiter: Allow to work with native objects. 

Previously, this class expected to be fed with obj.__getitem__ bound method.
But native objects instead have .subscr slot method. So, make it possible
to reuse this iterator even for such objects, if called with {obj, NULL}
structure.
getitemiter-native
Paul Sokolovsky 2015-09-13 21:07:17 +03:00
parent a5deadf082
commit db00e6a5c2
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
py/objgetitemiter.c
View File

@ -41,7 +41,12 @@ STATIC mp_obj_t it_iternext(mp_obj_t self_in) {
nlr_buf_t nlr;
if (nlr_push(&nlr) == 0) {
// try to get next item
mp_obj_t value = mp_call_method_n_kw(1, 0, self->args);
mp_obj_t value;
if (self->args[1] == NULL) {
value = mp_obj_subscr(self->args[0], self->args[2], MP_OBJ_SENTINEL);
} else {
value = mp_call_method_n_kw(1, 0, self->args);
}
self->args[2] = MP_OBJ_NEW_SMALL_INT(MP_OBJ_SMALL_INT_VALUE(self->args[2]) + 1);
nlr_pop();
return value;