diff options
| author | Paul Sokolovsky <pfalcon@users.sourceforge.net> | 2015-12-05 00:09:10 +0200 |
|---|---|---|
| committer | Paul Sokolovsky <pfalcon@users.sourceforge.net> | 2015-12-05 00:13:29 +0200 |
| commit | 1a1d11fa32ba043d22995d28cbc039cfa5f3cc46 (patch) | |
| tree | ebe82caf230957315c45e26f4873f85cbe0b0d19 /py/objmodule.c | |
| parent | 5ae3ddcc9a8fab9c845d98d58b26b6d258809475 (diff) | |
py/modsys: Implement sys.modules.
This for example will allow people to reload modules which didn't load
successfully (e.g. due to syntax error).
Diffstat (limited to 'py/objmodule.c')
| -rw-r--r-- | py/objmodule.c | 13 |
1 files changed, 8 insertions, 5 deletions
diff --git a/py/objmodule.c b/py/objmodule.c index 46c508160..1f1b7ab1f 100644 --- a/py/objmodule.c +++ b/py/objmodule.c @@ -95,7 +95,8 @@ const mp_obj_type_t mp_type_module = { }; mp_obj_t mp_obj_new_module(qstr module_name) { - mp_map_elem_t *el = mp_map_lookup(&MP_STATE_VM(mp_loaded_modules_map), MP_OBJ_NEW_QSTR(module_name), MP_MAP_LOOKUP_ADD_IF_NOT_FOUND); + mp_map_t *mp_loaded_modules_map = &MP_STATE_VM(mp_loaded_modules_dict).map; + mp_map_elem_t *el = mp_map_lookup(mp_loaded_modules_map, MP_OBJ_NEW_QSTR(module_name), MP_MAP_LOOKUP_ADD_IF_NOT_FOUND); // We could error out if module already exists, but let C extensions // add new members to existing modules. if (el->value != MP_OBJ_NULL) { @@ -200,17 +201,18 @@ STATIC const mp_rom_map_elem_t mp_builtin_module_table[] = { STATIC MP_DEFINE_CONST_MAP(mp_builtin_module_map, mp_builtin_module_table); void mp_module_init(void) { - mp_map_init(&MP_STATE_VM(mp_loaded_modules_map), 3); + mp_obj_dict_init(&MP_STATE_VM(mp_loaded_modules_dict), 3); } void mp_module_deinit(void) { - mp_map_deinit(&MP_STATE_VM(mp_loaded_modules_map)); + //mp_map_deinit(&MP_STATE_VM(mp_loaded_modules_map)); } // returns MP_OBJ_NULL if not found mp_obj_t mp_module_get(qstr module_name) { + mp_map_t *mp_loaded_modules_map = &MP_STATE_VM(mp_loaded_modules_dict).map; // lookup module - mp_map_elem_t *el = mp_map_lookup(&MP_STATE_VM(mp_loaded_modules_map), MP_OBJ_NEW_QSTR(module_name), MP_MAP_LOOKUP); + mp_map_elem_t *el = mp_map_lookup(mp_loaded_modules_map, MP_OBJ_NEW_QSTR(module_name), MP_MAP_LOOKUP); if (el == NULL) { // module not found, look for builtin module names @@ -236,5 +238,6 @@ mp_obj_t mp_module_get(qstr module_name) { } void mp_module_register(qstr qst, mp_obj_t module) { - mp_map_lookup(&MP_STATE_VM(mp_loaded_modules_map), MP_OBJ_NEW_QSTR(qst), MP_MAP_LOOKUP_ADD_IF_NOT_FOUND)->value = module; + mp_map_t *mp_loaded_modules_map = &MP_STATE_VM(mp_loaded_modules_dict).map; + mp_map_lookup(mp_loaded_modules_map, MP_OBJ_NEW_QSTR(qst), MP_MAP_LOOKUP_ADD_IF_NOT_FOUND)->value = module; } |