diff options
author | Damien George <damien.p.george@gmail.com> | 2014-04-11 09:13:30 +0100 |
---|---|---|
committer | Damien George <damien.p.george@gmail.com> | 2014-04-11 09:13:30 +0100 |
commit | 686afc5c0aaf2bc5a8d2547b703ab3177e0ea569 (patch) | |
tree | af911b84d516a18014258f64f49b917427c026ca /py | |
parent | be019ce0636398d1cbdee007d5238160dce8eda7 (diff) | |
download | micropython-686afc5c0aaf2bc5a8d2547b703ab3177e0ea569.tar.gz micropython-686afc5c0aaf2bc5a8d2547b703ab3177e0ea569.zip |
py: Check that sequence has 2 elements for dict iterable constructor.
Diffstat (limited to 'py')
-rw-r--r-- | py/obj.c | 2 | ||||
-rw-r--r-- | py/objdict.c | 6 |
2 files changed, 5 insertions, 3 deletions
@@ -280,7 +280,7 @@ void mp_obj_get_array_fixed_n(mp_obj_t o, uint len, mp_obj_t **items) { mp_obj_list_get(o, &seq_len, items); } if (seq_len != len) { - nlr_raise(mp_obj_new_exception_msg_varg(&mp_type_IndexError, "requested length %d but object has length %d", len, seq_len)); + nlr_raise(mp_obj_new_exception_msg_varg(&mp_type_ValueError, "requested length %d but object has length %d", len, seq_len)); } } else { nlr_raise(mp_obj_new_exception_msg_varg(&mp_type_TypeError, "object '%s' is not a tuple or list", mp_obj_get_type_str(o))); diff --git a/py/objdict.c b/py/objdict.c index 4dffa53da9..963e188074 100644 --- a/py/objdict.c +++ b/py/objdict.c @@ -50,9 +50,11 @@ STATIC mp_obj_t dict_make_new(mp_obj_t type_in, uint n_args, uint n_kw, const mp mp_obj_t iterable = mp_getiter(args[0]); mp_obj_t dict = mp_obj_new_dict(0); // TODO: support arbitrary seq as a pair - mp_obj_tuple_t *item; + mp_obj_t item; while ((item = mp_iternext(iterable)) != MP_OBJ_NULL) { - mp_obj_dict_store(dict, item->items[0], item->items[1]); + mp_obj_t *sub_items; + mp_obj_get_array_fixed_n(item, 2, &sub_items); + mp_obj_dict_store(dict, sub_items[0], sub_items[1]); } return dict; } |