commit 21c719bd0a0acf709de4d6dfe0d830aa60427e3f
author Delio Brignoli <brignoli.delio@gmail.com> Mon Aug 22 12:28:22 2016 +0200
committer Paul Sokolovsky <pfalcon@users.sourceforge.net> Tue Sep 20 14:01:31 2016 +0300
tree b9eccf5e77711f0116bc0564877b5915be40b2f3
parent 34e01984368ff2c689808ef8de182d838cd96725

builtinimport: add the module specified by -m to sys.modules as '__main__'

py/builtinimport.c

diff --git a/py/builtinimport.c b/py/builtinimport.c
index ef3545d..3c912e2 100644
--- a/py/builtinimport.c
+++ b/py/builtinimport.c
@@ -448,6 +448,8 @@
                     mp_obj_module_t *o = MP_OBJ_TO_PTR(module_obj);
                     mp_obj_dict_store(MP_OBJ_FROM_PTR(o->globals), MP_OBJ_NEW_QSTR(MP_QSTR___name__), MP_OBJ_NEW_QSTR(MP_QSTR___main__));
                     #if MICROPY_CPYTHON_COMPAT
+                    // Store module as "__main__" in the dictionary of loaded modules (returned by sys.modules).
+                    mp_obj_dict_store(&MP_STATE_VM(mp_loaded_modules_dict), MP_OBJ_NEW_QSTR(MP_QSTR___main__), module_obj);
                     // Store real name in "__main__" attribute. Choosen semi-randonly, to reuse existing qstr's.
                     mp_obj_dict_store(MP_OBJ_FROM_PTR(o->globals), MP_OBJ_NEW_QSTR(MP_QSTR___main__), MP_OBJ_NEW_QSTR(mod_name));
                     #endif