summaryrefslogtreecommitdiffstatshomepage
path: root/py
diff options
context:
space:
mode:
authorDamien George <damien.p.george@gmail.com>2014-04-11 09:13:30 +0100
committerDamien George <damien.p.george@gmail.com>2014-04-11 09:13:30 +0100
commit686afc5c0aaf2bc5a8d2547b703ab3177e0ea569 (patch)
treeaf911b84d516a18014258f64f49b917427c026ca /py
parentbe019ce0636398d1cbdee007d5238160dce8eda7 (diff)
downloadmicropython-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.c2
-rw-r--r--py/objdict.c6
2 files changed, 5 insertions, 3 deletions
diff --git a/py/obj.c b/py/obj.c
index 844ec41216..623b396422 100644
--- a/py/obj.c
+++ b/py/obj.c
@@ -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;
}