diff options
-rw-r--r-- | py/builtinimport.c | 2 | ||||
-rw-r--r-- | py/runtime.c | 4 | ||||
-rw-r--r-- | tests/basics/import-pkg4.py | 2 | ||||
-rw-r--r-- | tests/basics/pkg2/__init__.py | 1 | ||||
-rw-r--r-- | tests/basics/pkg2/mod1.py | 1 | ||||
-rw-r--r-- | tests/basics/pkg2/mod2.py | 1 | ||||
-rw-r--r-- | unix/gccollect.c | 2 |
7 files changed, 9 insertions, 4 deletions
diff --git a/py/builtinimport.c b/py/builtinimport.c index 501ced7644..9eede65cd9 100644 --- a/py/builtinimport.c +++ b/py/builtinimport.c @@ -218,6 +218,7 @@ mp_obj_t mp_builtin___import__(uint n_args, mp_obj_t *args) { if (stat == MP_IMPORT_STAT_DIR) { DEBUG_printf("%s is dir\n", vstr_str(&path)); + mp_store_attr(module_obj, MP_QSTR___path__, mp_obj_new_str((byte*)vstr_str(&path), vstr_len(&path), false)); vstr_add_char(&path, PATH_SEP_CHAR); vstr_add_str(&path, "__init__.py"); if (mp_import_stat(vstr_str(&path)) != MP_IMPORT_STAT_FILE) { @@ -230,7 +231,6 @@ mp_obj_t mp_builtin___import__(uint n_args, mp_obj_t *args) { vstr_cut_tail_bytes(&path, sizeof("/__init__.py") - 1); // cut off /__init__.py // https://docs.python.org/3.3/reference/import.html // "Specifically, any module that contains a __path__ attribute is considered a package." - mp_store_attr(module_obj, MP_QSTR___path__, mp_obj_new_str((byte*)vstr_str(&path), vstr_len(&path), false)); } else { // MP_IMPORT_STAT_FILE do_load(module_obj, &path); // TODO: We cannot just break here, at the very least, we must execute diff --git a/py/runtime.c b/py/runtime.c index 4793f054a8..b83cdb13bc 100644 --- a/py/runtime.c +++ b/py/runtime.c @@ -1050,7 +1050,7 @@ mp_obj_t mp_import_from(mp_obj_t module, qstr name) { if (dest[1] != MP_OBJ_NULL) { // Hopefully we can't import bound method from an object import_error: - nlr_raise(mp_obj_new_exception_msg_varg(&mp_type_ImportError, "Cannot import name '%s'", qstr_str(name))); + nlr_raise(mp_obj_new_exception_msg_varg(&mp_type_ImportError, "cannot import name %s", qstr_str(name))); } if (dest[0] != MP_OBJ_NULL) { @@ -1078,7 +1078,7 @@ import_error: args[1] = mp_const_none; // TODO should be globals args[2] = mp_const_none; // TODO should be locals args[3] = mp_const_true; // Pass sentinel "non empty" value to force returning of leaf module - args[4] = 0; + args[4] = MP_OBJ_NEW_SMALL_INT(0); // TODO lookup __import__ and call that instead of going straight to builtin implementation return mp_builtin___import__(5, args); diff --git a/tests/basics/import-pkg4.py b/tests/basics/import-pkg4.py new file mode 100644 index 0000000000..90b6f2e0ee --- /dev/null +++ b/tests/basics/import-pkg4.py @@ -0,0 +1,2 @@ +# Testing that "recursive" imports (pkg2/__init__.py imports from pkg2) work +import pkg2 diff --git a/tests/basics/pkg2/__init__.py b/tests/basics/pkg2/__init__.py new file mode 100644 index 0000000000..101ac7d400 --- /dev/null +++ b/tests/basics/pkg2/__init__.py @@ -0,0 +1 @@ +from pkg2 import mod1 diff --git a/tests/basics/pkg2/mod1.py b/tests/basics/pkg2/mod1.py new file mode 100644 index 0000000000..03754a45f6 --- /dev/null +++ b/tests/basics/pkg2/mod1.py @@ -0,0 +1 @@ +from pkg2 import mod2 diff --git a/tests/basics/pkg2/mod2.py b/tests/basics/pkg2/mod2.py new file mode 100644 index 0000000000..97dadcde46 --- /dev/null +++ b/tests/basics/pkg2/mod2.py @@ -0,0 +1 @@ +print("in mod2") diff --git a/unix/gccollect.c b/unix/gccollect.c index 33748d240e..35b775bc36 100644 --- a/unix/gccollect.c +++ b/unix/gccollect.c @@ -77,7 +77,7 @@ void gc_collect(void) { //gc_dump_info(); gc_collect_start(); - // this traces .data and .bss sections + // this traces the .bss section #ifdef __CYGWIN__ #define BSS_START __bss_start__ #else |