2014-05-03 23:27:38 +01:00
|
|
|
/*
|
2017-06-30 08:22:17 +01:00
|
|
|
* This file is part of the MicroPython project, http://micropython.org/
|
2014-05-03 23:27:38 +01:00
|
|
|
*
|
|
|
|
* The MIT License (MIT)
|
|
|
|
*
|
py: Automatically provide weak links from "foo" to "ufoo" module name.
This commit implements automatic module weak links for all built-in
modules, by searching for "ufoo" in the built-in module list if "foo"
cannot be found. This means that all modules named "ufoo" are always
available as "foo". Also, a port can no longer add any other weak links,
which makes strict the definition of a weak link.
It saves some code size (about 100-200 bytes) on ports that previously had
lots of weak links.
Some changes from the previous behaviour:
- It doesn't intern the non-u module names (eg "foo" is not interned),
which saves code size, but will mean that "import foo" creates a new qstr
(namely "foo") in RAM (unless the importing module is frozen).
- help('modules') no longer lists non-u module names, only the u-variants;
this reduces duplication in the help listing.
Weak links are effectively the same as having a set of symbolic links on
the filesystem that is searched last. So an "import foo" will search
built-in modules first, then all paths in sys.path, then weak links last,
importing "ufoo" if it exists. Thus a file called "foo.py" somewhere in
sys.path will still have precedence over the weak link of "foo" to "ufoo".
See issues: #1740, #4449, #5229, #5241.
2019-10-21 15:06:34 +01:00
|
|
|
* Copyright (c) 2013-2019 Damien P. George
|
2014-05-13 06:44:45 +01:00
|
|
|
* Copyright (c) 2014 Paul Sokolovsky
|
2014-05-03 23:27:38 +01:00
|
|
|
*
|
|
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
|
|
* of this software and associated documentation files (the "Software"), to deal
|
|
|
|
* in the Software without restriction, including without limitation the rights
|
|
|
|
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
|
|
* copies of the Software, and to permit persons to whom the Software is
|
|
|
|
* furnished to do so, subject to the following conditions:
|
|
|
|
*
|
|
|
|
* The above copyright notice and this permission notice shall be included in
|
|
|
|
* all copies or substantial portions of the Software.
|
|
|
|
*
|
|
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
|
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
|
|
* THE SOFTWARE.
|
|
|
|
*/
|
|
|
|
|
2014-01-03 14:03:48 +00:00
|
|
|
#include <stdio.h>
|
|
|
|
#include <string.h>
|
|
|
|
#include <assert.h>
|
|
|
|
|
2015-01-01 20:27:54 +00:00
|
|
|
#include "py/compile.h"
|
|
|
|
#include "py/objmodule.h"
|
2016-11-16 00:55:41 +00:00
|
|
|
#include "py/persistentcode.h"
|
2015-01-01 20:27:54 +00:00
|
|
|
#include "py/runtime.h"
|
|
|
|
#include "py/builtin.h"
|
2015-01-20 09:52:12 +00:00
|
|
|
#include "py/frozenmod.h"
|
2014-01-03 14:03:48 +00:00
|
|
|
|
2017-07-24 17:55:14 +01:00
|
|
|
#if MICROPY_DEBUG_VERBOSE // print debugging info
|
2014-04-11 21:08:29 +01:00
|
|
|
#define DEBUG_PRINT (1)
|
|
|
|
#define DEBUG_printf DEBUG_printf
|
|
|
|
#else // don't print debugging info
|
2014-11-05 21:16:41 +00:00
|
|
|
#define DEBUG_PRINT (0)
|
2014-04-11 21:08:29 +01:00
|
|
|
#define DEBUG_printf(...) (void)0
|
|
|
|
#endif
|
|
|
|
|
2018-02-20 07:00:44 +00:00
|
|
|
#if MICROPY_ENABLE_EXTERNAL_IMPORT
|
|
|
|
|
2014-02-05 23:57:48 +00:00
|
|
|
#define PATH_SEP_CHAR '/'
|
|
|
|
|
2014-10-25 19:04:13 +01:00
|
|
|
bool mp_obj_is_package(mp_obj_t module) {
|
|
|
|
mp_obj_t dest[2];
|
|
|
|
mp_load_method_maybe(module, MP_QSTR___path__, dest);
|
|
|
|
return dest[0] != MP_OBJ_NULL;
|
|
|
|
}
|
|
|
|
|
2016-05-21 19:33:42 +01:00
|
|
|
// Stat either frozen or normal module by a given path
|
|
|
|
// (whatever is available, if at all).
|
|
|
|
STATIC mp_import_stat_t mp_import_stat_any(const char *path) {
|
2016-05-23 12:42:23 +01:00
|
|
|
#if MICROPY_MODULE_FROZEN
|
2016-05-21 19:33:42 +01:00
|
|
|
mp_import_stat_t st = mp_frozen_stat(path);
|
|
|
|
if (st != MP_IMPORT_STAT_NO_EXIST) {
|
|
|
|
return st;
|
|
|
|
}
|
2016-05-21 20:23:08 +01:00
|
|
|
#endif
|
2016-05-21 19:33:42 +01:00
|
|
|
return mp_import_stat(path);
|
|
|
|
}
|
|
|
|
|
2016-12-13 04:09:48 +00:00
|
|
|
STATIC mp_import_stat_t stat_file_py_or_mpy(vstr_t *path) {
|
2016-05-21 19:33:42 +01:00
|
|
|
mp_import_stat_t stat = mp_import_stat_any(vstr_null_terminated_str(path));
|
2014-02-05 23:57:48 +00:00
|
|
|
if (stat == MP_IMPORT_STAT_FILE) {
|
|
|
|
return stat;
|
2014-01-19 22:03:34 +00:00
|
|
|
}
|
2015-11-02 21:57:42 +00:00
|
|
|
|
|
|
|
#if MICROPY_PERSISTENT_CODE_LOAD
|
|
|
|
vstr_ins_byte(path, path->len - 2, 'm');
|
2016-05-21 19:33:42 +01:00
|
|
|
stat = mp_import_stat_any(vstr_null_terminated_str(path));
|
2015-11-02 21:57:42 +00:00
|
|
|
if (stat == MP_IMPORT_STAT_FILE) {
|
|
|
|
return stat;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2014-02-05 23:57:48 +00:00
|
|
|
return MP_IMPORT_STAT_NO_EXIST;
|
|
|
|
}
|
2014-01-19 22:03:34 +00:00
|
|
|
|
2016-12-13 04:09:48 +00:00
|
|
|
STATIC mp_import_stat_t stat_dir_or_file(vstr_t *path) {
|
|
|
|
mp_import_stat_t stat = mp_import_stat_any(vstr_null_terminated_str(path));
|
|
|
|
DEBUG_printf("stat %s: %d\n", vstr_str(path), stat);
|
|
|
|
if (stat == MP_IMPORT_STAT_DIR) {
|
|
|
|
return stat;
|
|
|
|
}
|
|
|
|
|
|
|
|
// not a directory, add .py and try as a file
|
|
|
|
vstr_add_str(path, ".py");
|
|
|
|
return stat_file_py_or_mpy(path);
|
|
|
|
}
|
|
|
|
|
2014-06-11 19:55:46 +01:00
|
|
|
STATIC mp_import_stat_t find_file(const char *file_str, uint file_len, vstr_t *dest) {
|
2014-11-05 20:02:33 +00:00
|
|
|
#if MICROPY_PY_SYS
|
2014-02-05 23:57:48 +00:00
|
|
|
// extract the list of paths
|
2017-03-25 08:35:08 +00:00
|
|
|
size_t path_num;
|
2014-02-04 22:47:06 +00:00
|
|
|
mp_obj_t *path_items;
|
2014-04-13 04:43:18 +01:00
|
|
|
mp_obj_list_get(mp_sys_path, &path_num, &path_items);
|
2014-02-04 22:47:06 +00:00
|
|
|
|
2020-02-26 04:31:13 +00:00
|
|
|
if (path_num != 0) {
|
2014-02-05 23:57:48 +00:00
|
|
|
// go through each path looking for a directory or file
|
2017-07-04 14:44:22 +01:00
|
|
|
for (size_t i = 0; i < path_num; i++) {
|
2014-02-05 23:57:48 +00:00
|
|
|
vstr_reset(dest);
|
2017-03-25 08:48:18 +00:00
|
|
|
size_t p_len;
|
2014-02-08 18:17:23 +00:00
|
|
|
const char *p = mp_obj_str_get_data(path_items[i], &p_len);
|
2014-02-04 22:47:06 +00:00
|
|
|
if (p_len > 0) {
|
2014-02-08 18:17:23 +00:00
|
|
|
vstr_add_strn(dest, p, p_len);
|
2014-02-05 23:57:48 +00:00
|
|
|
vstr_add_char(dest, PATH_SEP_CHAR);
|
2014-02-04 22:47:06 +00:00
|
|
|
}
|
2014-02-05 23:57:48 +00:00
|
|
|
vstr_add_strn(dest, file_str, file_len);
|
|
|
|
mp_import_stat_t stat = stat_dir_or_file(dest);
|
|
|
|
if (stat != MP_IMPORT_STAT_NO_EXIST) {
|
|
|
|
return stat;
|
2014-02-04 22:47:06 +00:00
|
|
|
}
|
|
|
|
}
|
2014-02-05 23:57:48 +00:00
|
|
|
|
|
|
|
// could not find a directory or file
|
|
|
|
return MP_IMPORT_STAT_NO_EXIST;
|
2014-02-04 22:47:06 +00:00
|
|
|
}
|
2014-11-05 20:02:33 +00:00
|
|
|
#endif
|
2020-02-26 04:31:13 +00:00
|
|
|
|
|
|
|
// mp_sys_path is empty, so just use the given file name
|
|
|
|
vstr_add_strn(dest, file_str, file_len);
|
|
|
|
return stat_dir_or_file(dest);
|
2014-02-05 23:57:48 +00:00
|
|
|
}
|
|
|
|
|
2018-12-13 12:10:35 +00:00
|
|
|
#if MICROPY_MODULE_FROZEN_STR || MICROPY_ENABLE_COMPILER
|
2017-03-14 00:16:31 +00:00
|
|
|
STATIC void do_load_from_lexer(mp_obj_t module_obj, mp_lexer_t *lex) {
|
2014-07-25 09:00:15 +01:00
|
|
|
#if MICROPY_PY___FILE__
|
2014-12-05 19:35:18 +00:00
|
|
|
qstr source_name = lex->source_name;
|
2014-10-05 20:13:34 +01:00
|
|
|
mp_store_attr(module_obj, MP_QSTR___file__, MP_OBJ_NEW_QSTR(source_name));
|
2014-07-25 09:00:15 +01:00
|
|
|
#endif
|
2014-01-03 14:03:48 +00:00
|
|
|
|
2014-10-05 20:13:34 +01:00
|
|
|
// parse, compile and execute the module in its context
|
|
|
|
mp_obj_dict_t *mod_globals = mp_obj_module_get_globals(module_obj);
|
|
|
|
mp_parse_compile_execute(lex, MP_PARSE_FILE_INPUT, mod_globals, mod_globals);
|
2014-02-05 23:57:48 +00:00
|
|
|
}
|
2015-12-18 12:35:44 +00:00
|
|
|
#endif
|
2014-02-05 23:57:48 +00:00
|
|
|
|
2016-01-31 22:24:16 +00:00
|
|
|
#if MICROPY_PERSISTENT_CODE_LOAD || MICROPY_MODULE_FROZEN_MPY
|
2019-07-08 10:26:20 +01:00
|
|
|
STATIC void do_execute_raw_code(mp_obj_t module_obj, mp_raw_code_t *raw_code, const char *source_name) {
|
|
|
|
(void)source_name;
|
|
|
|
|
2015-11-02 21:57:42 +00:00
|
|
|
#if MICROPY_PY___FILE__
|
2019-07-08 10:26:20 +01:00
|
|
|
mp_store_attr(module_obj, MP_QSTR___file__, MP_OBJ_NEW_QSTR(qstr_from_str(source_name)));
|
2015-11-02 21:57:42 +00:00
|
|
|
#endif
|
|
|
|
|
|
|
|
// execute the module in its context
|
|
|
|
mp_obj_dict_t *mod_globals = mp_obj_module_get_globals(module_obj);
|
|
|
|
|
|
|
|
// save context
|
|
|
|
mp_obj_dict_t *volatile old_globals = mp_globals_get();
|
|
|
|
mp_obj_dict_t *volatile old_locals = mp_locals_get();
|
|
|
|
|
|
|
|
// set new context
|
|
|
|
mp_globals_set(mod_globals);
|
|
|
|
mp_locals_set(mod_globals);
|
|
|
|
|
|
|
|
nlr_buf_t nlr;
|
|
|
|
if (nlr_push(&nlr) == 0) {
|
|
|
|
mp_obj_t module_fun = mp_make_function_from_raw_code(raw_code, MP_OBJ_NULL, MP_OBJ_NULL);
|
|
|
|
mp_call_function_0(module_fun);
|
|
|
|
|
|
|
|
// finish nlr block, restore context
|
|
|
|
nlr_pop();
|
|
|
|
mp_globals_set(old_globals);
|
|
|
|
mp_locals_set(old_locals);
|
|
|
|
} else {
|
|
|
|
// exception; restore context and re-raise same exception
|
|
|
|
mp_globals_set(old_globals);
|
|
|
|
mp_locals_set(old_locals);
|
2015-11-27 17:01:44 +00:00
|
|
|
nlr_jump(nlr.ret_val);
|
2015-11-02 21:57:42 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2015-01-20 09:52:12 +00:00
|
|
|
STATIC void do_load(mp_obj_t module_obj, vstr_t *file) {
|
2018-12-13 12:10:35 +00:00
|
|
|
#if MICROPY_MODULE_FROZEN || MICROPY_ENABLE_COMPILER || (MICROPY_PERSISTENT_CODE_LOAD && MICROPY_HAS_FILE_READER)
|
2015-01-29 13:57:23 +00:00
|
|
|
char *file_str = vstr_null_terminated_str(file);
|
2016-01-31 22:24:16 +00:00
|
|
|
#endif
|
2015-12-18 12:35:44 +00:00
|
|
|
|
2016-05-23 12:42:23 +01:00
|
|
|
// If we support frozen modules (either as str or mpy) then try to find the
|
|
|
|
// requested filename in the list of frozen module filenames.
|
|
|
|
#if MICROPY_MODULE_FROZEN
|
|
|
|
void *modref;
|
|
|
|
int frozen_type = mp_find_frozen_module(file_str, file->len, &modref);
|
|
|
|
#endif
|
|
|
|
|
|
|
|
// If we support frozen str modules and the compiler is enabled, and we
|
|
|
|
// found the filename in the list of frozen files, then load and execute it.
|
|
|
|
#if MICROPY_MODULE_FROZEN_STR
|
|
|
|
if (frozen_type == MP_FROZEN_STR) {
|
2017-03-14 00:16:31 +00:00
|
|
|
do_load_from_lexer(module_obj, modref);
|
2016-05-23 12:42:23 +01:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
|
|
// If we support frozen mpy modules and we found a corresponding file (and
|
|
|
|
// its data) in the list of frozen files, execute it.
|
|
|
|
#if MICROPY_MODULE_FROZEN_MPY
|
|
|
|
if (frozen_type == MP_FROZEN_MPY) {
|
2019-07-08 10:26:20 +01:00
|
|
|
do_execute_raw_code(module_obj, modref, file_str);
|
2016-05-23 12:42:23 +01:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
|
|
// If we support loading .mpy files then check if the file extension is of
|
|
|
|
// the correct format and, if so, load and execute the file.
|
2018-12-13 12:10:35 +00:00
|
|
|
#if MICROPY_HAS_FILE_READER && MICROPY_PERSISTENT_CODE_LOAD
|
2015-11-02 21:57:42 +00:00
|
|
|
if (file_str[file->len - 3] == 'm') {
|
|
|
|
mp_raw_code_t *raw_code = mp_raw_code_load_file(file_str);
|
2019-07-08 10:26:20 +01:00
|
|
|
do_execute_raw_code(module_obj, raw_code, file_str);
|
2015-12-18 12:35:44 +00:00
|
|
|
return;
|
|
|
|
}
|
2015-11-02 21:57:42 +00:00
|
|
|
#endif
|
2015-12-18 12:35:44 +00:00
|
|
|
|
2016-05-23 12:42:23 +01:00
|
|
|
// If we can compile scripts then load the file and compile and execute it.
|
2015-12-18 12:35:44 +00:00
|
|
|
#if MICROPY_ENABLE_COMPILER
|
2015-11-02 21:57:42 +00:00
|
|
|
{
|
2016-05-23 12:42:23 +01:00
|
|
|
mp_lexer_t *lex = mp_lexer_new_from_file(file_str);
|
2017-03-14 00:16:31 +00:00
|
|
|
do_load_from_lexer(module_obj, lex);
|
2016-05-23 12:42:23 +01:00
|
|
|
return;
|
2015-11-02 21:57:42 +00:00
|
|
|
}
|
2017-07-01 00:23:29 +01:00
|
|
|
#else
|
2016-05-23 12:42:23 +01:00
|
|
|
// If we get here then the file was not frozen and we can't compile scripts.
|
2020-03-02 11:35:22 +00:00
|
|
|
mp_raise_msg(&mp_type_ImportError, MP_ERROR_TEXT("script compilation not supported"));
|
2017-07-01 00:23:29 +01:00
|
|
|
#endif
|
2015-01-20 09:52:12 +00:00
|
|
|
}
|
|
|
|
|
2015-02-16 10:10:13 +00:00
|
|
|
STATIC void chop_component(const char *start, const char **end) {
|
|
|
|
const char *p = *end;
|
|
|
|
while (p > start) {
|
|
|
|
if (*--p == '.') {
|
|
|
|
*end = p;
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
*end = p;
|
|
|
|
}
|
|
|
|
|
2016-01-03 14:21:40 +00:00
|
|
|
mp_obj_t mp_builtin___import__(size_t n_args, const mp_obj_t *args) {
|
2014-04-11 21:08:29 +01:00
|
|
|
#if DEBUG_PRINT
|
2014-09-23 10:59:05 +01:00
|
|
|
DEBUG_printf("__import__:\n");
|
2017-07-04 14:44:22 +01:00
|
|
|
for (size_t i = 0; i < n_args; i++) {
|
2014-09-23 10:59:05 +01:00
|
|
|
DEBUG_printf(" ");
|
2014-02-19 22:29:54 +00:00
|
|
|
mp_obj_print(args[i], PRINT_REPR);
|
2014-09-23 10:59:05 +01:00
|
|
|
DEBUG_printf("\n");
|
2014-02-05 23:57:48 +00:00
|
|
|
}
|
2014-04-11 21:08:29 +01:00
|
|
|
#endif
|
2014-02-05 23:57:48 +00:00
|
|
|
|
2014-04-12 15:46:54 +01:00
|
|
|
mp_obj_t module_name = args[0];
|
2014-02-19 22:29:54 +00:00
|
|
|
mp_obj_t fromtuple = mp_const_none;
|
2014-10-03 18:44:14 +01:00
|
|
|
mp_int_t level = 0;
|
2014-02-19 22:29:54 +00:00
|
|
|
if (n_args >= 4) {
|
|
|
|
fromtuple = args[3];
|
2014-02-20 23:15:20 +00:00
|
|
|
if (n_args >= 5) {
|
|
|
|
level = MP_OBJ_SMALL_INT_VALUE(args[4]);
|
2017-01-16 05:54:56 +00:00
|
|
|
if (level < 0) {
|
|
|
|
mp_raise_ValueError(NULL);
|
|
|
|
}
|
2014-02-20 23:15:20 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-03-25 08:48:18 +00:00
|
|
|
size_t mod_len;
|
2014-08-30 14:19:41 +01:00
|
|
|
const char *mod_str = mp_obj_str_get_data(module_name, &mod_len);
|
2014-04-12 15:46:54 +01:00
|
|
|
|
2014-02-20 23:15:20 +00:00
|
|
|
if (level != 0) {
|
2014-04-12 15:46:54 +01:00
|
|
|
// What we want to do here is to take name of current module,
|
|
|
|
// chop <level> trailing components, and concatenate with passed-in
|
2017-05-29 08:08:14 +01:00
|
|
|
// module name, thus resolving relative import name into absolute.
|
2014-04-12 15:46:54 +01:00
|
|
|
// This even appears to be correct per
|
|
|
|
// http://legacy.python.org/dev/peps/pep-0328/#relative-imports-and-name
|
|
|
|
// "Relative imports use a module's __name__ attribute to determine that
|
|
|
|
// module's position in the package hierarchy."
|
2015-02-16 10:10:13 +00:00
|
|
|
level--;
|
2015-11-27 17:01:44 +00:00
|
|
|
mp_obj_t this_name_q = mp_obj_dict_get(MP_OBJ_FROM_PTR(mp_globals_get()), MP_OBJ_NEW_QSTR(MP_QSTR___name__));
|
2014-04-12 15:46:54 +01:00
|
|
|
assert(this_name_q != MP_OBJ_NULL);
|
2015-06-28 22:21:36 +01:00
|
|
|
#if MICROPY_CPYTHON_COMPAT
|
|
|
|
if (MP_OBJ_QSTR_VALUE(this_name_q) == MP_QSTR___main__) {
|
|
|
|
// This is a module run by -m command-line switch, get its real name from backup attribute
|
2015-11-27 17:01:44 +00:00
|
|
|
this_name_q = mp_obj_dict_get(MP_OBJ_FROM_PTR(mp_globals_get()), MP_OBJ_NEW_QSTR(MP_QSTR___main__));
|
2015-06-28 22:21:36 +01:00
|
|
|
}
|
|
|
|
#endif
|
2015-11-27 17:01:44 +00:00
|
|
|
mp_map_t *globals_map = &mp_globals_get()->map;
|
2015-02-16 10:10:13 +00:00
|
|
|
mp_map_elem_t *elem = mp_map_lookup(globals_map, MP_OBJ_NEW_QSTR(MP_QSTR___path__), MP_MAP_LOOKUP);
|
|
|
|
bool is_pkg = (elem != NULL);
|
|
|
|
|
2014-04-12 15:46:54 +01:00
|
|
|
#if DEBUG_PRINT
|
2015-02-16 10:10:13 +00:00
|
|
|
DEBUG_printf("Current module/package: ");
|
2014-04-12 15:46:54 +01:00
|
|
|
mp_obj_print(this_name_q, PRINT_REPR);
|
2015-02-16 10:10:13 +00:00
|
|
|
DEBUG_printf(", is_package: %d", is_pkg);
|
2014-09-23 10:59:05 +01:00
|
|
|
DEBUG_printf("\n");
|
2014-04-12 15:46:54 +01:00
|
|
|
#endif
|
2014-02-19 22:29:54 +00:00
|
|
|
|
2017-03-25 08:48:18 +00:00
|
|
|
size_t this_name_l;
|
2014-08-30 14:19:41 +01:00
|
|
|
const char *this_name = mp_obj_str_get_data(this_name_q, &this_name_l);
|
2014-04-12 15:46:54 +01:00
|
|
|
|
2015-02-16 10:10:13 +00:00
|
|
|
const char *p = this_name + this_name_l;
|
|
|
|
if (!is_pkg) {
|
|
|
|
// We have module, but relative imports are anchored at package, so
|
|
|
|
// go there.
|
|
|
|
chop_component(this_name, &p);
|
|
|
|
}
|
|
|
|
|
|
|
|
while (level--) {
|
|
|
|
chop_component(this_name, &p);
|
2014-04-12 15:46:54 +01:00
|
|
|
}
|
|
|
|
|
2017-01-16 05:54:56 +00:00
|
|
|
// We must have some component left over to import from
|
|
|
|
if (p == this_name) {
|
2020-04-09 08:22:25 +01:00
|
|
|
mp_raise_ValueError(MP_ERROR_TEXT("can't perform relative import"));
|
2014-04-12 15:46:54 +01:00
|
|
|
}
|
|
|
|
|
2015-01-16 17:47:07 +00:00
|
|
|
uint new_mod_l = (mod_len == 0 ? (size_t)(p - this_name) : (size_t)(p - this_name) + 1 + mod_len);
|
2017-11-26 12:37:19 +00:00
|
|
|
char *new_mod = mp_local_alloc(new_mod_l);
|
2014-04-12 15:46:54 +01:00
|
|
|
memcpy(new_mod, this_name, p - this_name);
|
|
|
|
if (mod_len != 0) {
|
|
|
|
new_mod[p - this_name] = '.';
|
|
|
|
memcpy(new_mod + (p - this_name) + 1, mod_str, mod_len);
|
|
|
|
}
|
|
|
|
|
|
|
|
qstr new_mod_q = qstr_from_strn(new_mod, new_mod_l);
|
2017-11-26 12:37:19 +00:00
|
|
|
mp_local_free(new_mod);
|
2015-06-26 22:34:53 +01:00
|
|
|
DEBUG_printf("Resolved base name for relative import: '%s'\n", qstr_str(new_mod_q));
|
2014-04-12 15:46:54 +01:00
|
|
|
module_name = MP_OBJ_NEW_QSTR(new_mod_q);
|
2017-11-26 12:37:19 +00:00
|
|
|
mod_str = qstr_str(new_mod_q);
|
2014-04-12 15:46:54 +01:00
|
|
|
mod_len = new_mod_l;
|
|
|
|
}
|
2014-02-16 00:53:44 +00:00
|
|
|
|
2019-11-22 13:06:49 +00:00
|
|
|
if (mod_len == 0) {
|
|
|
|
mp_raise_ValueError(NULL);
|
|
|
|
}
|
|
|
|
|
2014-02-05 23:57:48 +00:00
|
|
|
// check if module already exists
|
2015-01-21 21:14:46 +00:00
|
|
|
qstr module_name_qstr = mp_obj_str_get_qstr(module_name);
|
|
|
|
mp_obj_t module_obj = mp_module_get(module_name_qstr);
|
2014-02-05 23:57:48 +00:00
|
|
|
if (module_obj != MP_OBJ_NULL) {
|
2014-04-11 21:08:29 +01:00
|
|
|
DEBUG_printf("Module already loaded\n");
|
2014-02-16 00:53:44 +00:00
|
|
|
// If it's not a package, return module right away
|
|
|
|
char *p = strchr(mod_str, '.');
|
|
|
|
if (p == NULL) {
|
|
|
|
return module_obj;
|
|
|
|
}
|
2014-02-19 22:29:54 +00:00
|
|
|
// If fromlist is not empty, return leaf module
|
|
|
|
if (fromtuple != mp_const_none) {
|
|
|
|
return module_obj;
|
|
|
|
}
|
2014-02-16 00:53:44 +00:00
|
|
|
// Otherwise, we need to return top-level package
|
|
|
|
qstr pkg_name = qstr_from_strn(mod_str, p - mod_str);
|
2014-03-25 14:18:18 +00:00
|
|
|
return mp_module_get(pkg_name);
|
2014-02-05 23:57:48 +00:00
|
|
|
}
|
2014-04-11 21:08:29 +01:00
|
|
|
DEBUG_printf("Module not yet loaded\n");
|
2014-02-05 23:57:48 +00:00
|
|
|
|
|
|
|
uint last = 0;
|
2014-05-21 20:32:59 +01:00
|
|
|
VSTR_FIXED(path, MICROPY_ALLOC_PATH_MAX)
|
2014-02-05 23:57:48 +00:00
|
|
|
module_obj = MP_OBJ_NULL;
|
2014-02-16 00:53:44 +00:00
|
|
|
mp_obj_t top_module_obj = MP_OBJ_NULL;
|
|
|
|
mp_obj_t outer_module_obj = MP_OBJ_NULL;
|
2014-02-05 23:57:48 +00:00
|
|
|
uint i;
|
|
|
|
for (i = 1; i <= mod_len; i++) {
|
|
|
|
if (i == mod_len || mod_str[i] == '.') {
|
|
|
|
// create a qstr for the module name up to this depth
|
|
|
|
qstr mod_name = qstr_from_strn(mod_str, i);
|
2014-04-11 21:08:29 +01:00
|
|
|
DEBUG_printf("Processing module: %s\n", qstr_str(mod_name));
|
2015-02-16 10:10:13 +00:00
|
|
|
DEBUG_printf("Previous path: =%.*s=\n", vstr_len(&path), vstr_str(&path));
|
2014-02-05 23:57:48 +00:00
|
|
|
|
|
|
|
// find the file corresponding to the module name
|
|
|
|
mp_import_stat_t stat;
|
2014-02-06 21:11:19 +00:00
|
|
|
if (vstr_len(&path) == 0) {
|
2014-02-05 23:57:48 +00:00
|
|
|
// first module in the dotted-name; search for a directory or file
|
2014-02-06 21:11:19 +00:00
|
|
|
stat = find_file(mod_str, i, &path);
|
2014-02-05 23:57:48 +00:00
|
|
|
} else {
|
|
|
|
// latter module in the dotted-name; append to path
|
2014-02-06 21:11:19 +00:00
|
|
|
vstr_add_char(&path, PATH_SEP_CHAR);
|
|
|
|
vstr_add_strn(&path, mod_str + last, i - last);
|
|
|
|
stat = stat_dir_or_file(&path);
|
2014-02-05 23:57:48 +00:00
|
|
|
}
|
2015-01-28 23:43:01 +00:00
|
|
|
DEBUG_printf("Current path: %.*s\n", vstr_len(&path), vstr_str(&path));
|
2014-02-05 23:57:48 +00:00
|
|
|
|
|
|
|
if (stat == MP_IMPORT_STAT_NO_EXIST) {
|
py: Automatically provide weak links from "foo" to "ufoo" module name.
This commit implements automatic module weak links for all built-in
modules, by searching for "ufoo" in the built-in module list if "foo"
cannot be found. This means that all modules named "ufoo" are always
available as "foo". Also, a port can no longer add any other weak links,
which makes strict the definition of a weak link.
It saves some code size (about 100-200 bytes) on ports that previously had
lots of weak links.
Some changes from the previous behaviour:
- It doesn't intern the non-u module names (eg "foo" is not interned),
which saves code size, but will mean that "import foo" creates a new qstr
(namely "foo") in RAM (unless the importing module is frozen).
- help('modules') no longer lists non-u module names, only the u-variants;
this reduces duplication in the help listing.
Weak links are effectively the same as having a set of symbolic links on
the filesystem that is searched last. So an "import foo" will search
built-in modules first, then all paths in sys.path, then weak links last,
importing "ufoo" if it exists. Thus a file called "foo.py" somewhere in
sys.path will still have precedence over the weak link of "foo" to "ufoo".
See issues: #1740, #4449, #5229, #5241.
2019-10-21 15:06:34 +01:00
|
|
|
module_obj = MP_OBJ_NULL;
|
2014-10-12 11:46:04 +01:00
|
|
|
#if MICROPY_MODULE_WEAK_LINKS
|
|
|
|
// check if there is a weak link to this module
|
|
|
|
if (i == mod_len) {
|
py: Automatically provide weak links from "foo" to "ufoo" module name.
This commit implements automatic module weak links for all built-in
modules, by searching for "ufoo" in the built-in module list if "foo"
cannot be found. This means that all modules named "ufoo" are always
available as "foo". Also, a port can no longer add any other weak links,
which makes strict the definition of a weak link.
It saves some code size (about 100-200 bytes) on ports that previously had
lots of weak links.
Some changes from the previous behaviour:
- It doesn't intern the non-u module names (eg "foo" is not interned),
which saves code size, but will mean that "import foo" creates a new qstr
(namely "foo") in RAM (unless the importing module is frozen).
- help('modules') no longer lists non-u module names, only the u-variants;
this reduces duplication in the help listing.
Weak links are effectively the same as having a set of symbolic links on
the filesystem that is searched last. So an "import foo" will search
built-in modules first, then all paths in sys.path, then weak links last,
importing "ufoo" if it exists. Thus a file called "foo.py" somewhere in
sys.path will still have precedence over the weak link of "foo" to "ufoo".
See issues: #1740, #4449, #5229, #5241.
2019-10-21 15:06:34 +01:00
|
|
|
module_obj = mp_module_search_umodule(mod_str);
|
|
|
|
if (module_obj != MP_OBJ_NULL) {
|
|
|
|
// found weak linked module
|
|
|
|
mp_module_call_init(mod_name, module_obj);
|
2014-10-12 11:46:04 +01:00
|
|
|
}
|
py: Automatically provide weak links from "foo" to "ufoo" module name.
This commit implements automatic module weak links for all built-in
modules, by searching for "ufoo" in the built-in module list if "foo"
cannot be found. This means that all modules named "ufoo" are always
available as "foo". Also, a port can no longer add any other weak links,
which makes strict the definition of a weak link.
It saves some code size (about 100-200 bytes) on ports that previously had
lots of weak links.
Some changes from the previous behaviour:
- It doesn't intern the non-u module names (eg "foo" is not interned),
which saves code size, but will mean that "import foo" creates a new qstr
(namely "foo") in RAM (unless the importing module is frozen).
- help('modules') no longer lists non-u module names, only the u-variants;
this reduces duplication in the help listing.
Weak links are effectively the same as having a set of symbolic links on
the filesystem that is searched last. So an "import foo" will search
built-in modules first, then all paths in sys.path, then weak links last,
importing "ufoo" if it exists. Thus a file called "foo.py" somewhere in
sys.path will still have precedence over the weak link of "foo" to "ufoo".
See issues: #1740, #4449, #5229, #5241.
2019-10-21 15:06:34 +01:00
|
|
|
}
|
2014-10-12 11:46:04 +01:00
|
|
|
#endif
|
py: Automatically provide weak links from "foo" to "ufoo" module name.
This commit implements automatic module weak links for all built-in
modules, by searching for "ufoo" in the built-in module list if "foo"
cannot be found. This means that all modules named "ufoo" are always
available as "foo". Also, a port can no longer add any other weak links,
which makes strict the definition of a weak link.
It saves some code size (about 100-200 bytes) on ports that previously had
lots of weak links.
Some changes from the previous behaviour:
- It doesn't intern the non-u module names (eg "foo" is not interned),
which saves code size, but will mean that "import foo" creates a new qstr
(namely "foo") in RAM (unless the importing module is frozen).
- help('modules') no longer lists non-u module names, only the u-variants;
this reduces duplication in the help listing.
Weak links are effectively the same as having a set of symbolic links on
the filesystem that is searched last. So an "import foo" will search
built-in modules first, then all paths in sys.path, then weak links last,
importing "ufoo" if it exists. Thus a file called "foo.py" somewhere in
sys.path will still have precedence over the weak link of "foo" to "ufoo".
See issues: #1740, #4449, #5229, #5241.
2019-10-21 15:06:34 +01:00
|
|
|
if (module_obj == MP_OBJ_NULL) {
|
2014-10-12 11:46:04 +01:00
|
|
|
// couldn't find the file, so fail
|
2021-04-22 03:13:58 +01:00
|
|
|
#if MICROPY_ERROR_REPORTING <= MICROPY_ERROR_REPORTING_TERSE
|
2020-03-02 11:35:22 +00:00
|
|
|
mp_raise_msg(&mp_type_ImportError, MP_ERROR_TEXT("module not found"));
|
2019-09-26 13:52:04 +01:00
|
|
|
#else
|
2020-03-02 11:35:22 +00:00
|
|
|
mp_raise_msg_varg(&mp_type_ImportError, MP_ERROR_TEXT("no module named '%q'"), mod_name);
|
2019-09-26 13:52:04 +01:00
|
|
|
#endif
|
2014-10-12 11:46:04 +01:00
|
|
|
}
|
|
|
|
} else {
|
|
|
|
// found the file, so get the module
|
|
|
|
module_obj = mp_module_get(mod_name);
|
2014-02-05 23:57:48 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
if (module_obj == MP_OBJ_NULL) {
|
|
|
|
// module not already loaded, so load it!
|
|
|
|
|
|
|
|
module_obj = mp_obj_new_module(mod_name);
|
|
|
|
|
2014-10-25 19:16:24 +01:00
|
|
|
// if args[3] (fromtuple) has magic value False, set up
|
|
|
|
// this module for command-line "-m" option (set module's
|
2017-05-09 12:22:21 +01:00
|
|
|
// name to __main__ instead of real name). Do this only
|
|
|
|
// for *modules* however - packages never have their names
|
|
|
|
// replaced, instead they're -m'ed using a special __main__
|
|
|
|
// submodule in them. (This all apparently is done to not
|
|
|
|
// touch package name itself, which is important for future
|
|
|
|
// imports).
|
|
|
|
if (i == mod_len && fromtuple == mp_const_false && stat != MP_IMPORT_STAT_DIR) {
|
2015-11-27 17:01:44 +00:00
|
|
|
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__));
|
2015-06-28 22:21:36 +01:00
|
|
|
#if MICROPY_CPYTHON_COMPAT
|
2016-08-22 11:28:22 +01:00
|
|
|
// Store module as "__main__" in the dictionary of loaded modules (returned by sys.modules).
|
2016-09-20 15:55:42 +01:00
|
|
|
mp_obj_dict_store(MP_OBJ_FROM_PTR(&MP_STATE_VM(mp_loaded_modules_dict)), MP_OBJ_NEW_QSTR(MP_QSTR___main__), module_obj);
|
2017-05-29 08:08:14 +01:00
|
|
|
// Store real name in "__main__" attribute. Chosen semi-randonly, to reuse existing qstr's.
|
2015-11-27 17:01:44 +00:00
|
|
|
mp_obj_dict_store(MP_OBJ_FROM_PTR(o->globals), MP_OBJ_NEW_QSTR(MP_QSTR___main__), MP_OBJ_NEW_QSTR(mod_name));
|
2015-06-28 22:21:36 +01:00
|
|
|
#endif
|
2014-10-25 19:16:24 +01:00
|
|
|
}
|
|
|
|
|
2014-02-05 23:57:48 +00:00
|
|
|
if (stat == MP_IMPORT_STAT_DIR) {
|
2015-01-28 23:43:01 +00:00
|
|
|
DEBUG_printf("%.*s is dir\n", vstr_len(&path), vstr_str(&path));
|
2014-06-05 18:51:03 +01:00
|
|
|
// https://docs.python.org/3/reference/import.html
|
2014-05-10 16:46:02 +01:00
|
|
|
// "Specifically, any module that contains a __path__ attribute is considered a package."
|
2017-11-16 02:17:51 +00:00
|
|
|
mp_store_attr(module_obj, MP_QSTR___path__, mp_obj_new_str(vstr_str(&path), vstr_len(&path)));
|
2017-01-08 09:17:23 +00:00
|
|
|
size_t orig_path_len = path.len;
|
2014-02-06 21:11:19 +00:00
|
|
|
vstr_add_char(&path, PATH_SEP_CHAR);
|
|
|
|
vstr_add_str(&path, "__init__.py");
|
2016-12-13 04:09:48 +00:00
|
|
|
if (stat_file_py_or_mpy(&path) != MP_IMPORT_STAT_FILE) {
|
2016-07-02 12:45:49 +01:00
|
|
|
// mp_warning("%s is imported as namespace package", vstr_str(&path));
|
2014-04-14 23:23:40 +01:00
|
|
|
} else {
|
|
|
|
do_load(module_obj, &path);
|
2014-02-05 23:57:48 +00:00
|
|
|
}
|
2017-01-08 09:17:23 +00:00
|
|
|
path.len = orig_path_len;
|
2014-02-05 23:57:48 +00:00
|
|
|
} else { // MP_IMPORT_STAT_FILE
|
2014-02-06 21:11:19 +00:00
|
|
|
do_load(module_obj, &path);
|
2017-01-16 05:40:47 +00:00
|
|
|
// This should be the last component in the import path. If there are
|
|
|
|
// remaining components then it's an ImportError because the current path
|
|
|
|
// (the module that was just loaded) is not a package. This will be caught
|
|
|
|
// on the next iteration because the file will not exist.
|
2014-02-05 23:57:48 +00:00
|
|
|
}
|
|
|
|
}
|
2014-02-16 00:53:44 +00:00
|
|
|
if (outer_module_obj != MP_OBJ_NULL) {
|
|
|
|
qstr s = qstr_from_strn(mod_str + last, i - last);
|
2014-03-30 13:35:08 +01:00
|
|
|
mp_store_attr(outer_module_obj, s, module_obj);
|
2014-02-16 00:53:44 +00:00
|
|
|
}
|
|
|
|
outer_module_obj = module_obj;
|
|
|
|
if (top_module_obj == MP_OBJ_NULL) {
|
|
|
|
top_module_obj = module_obj;
|
|
|
|
}
|
|
|
|
last = i + 1;
|
2014-02-05 23:57:48 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-02-19 22:29:54 +00:00
|
|
|
// If fromlist is not empty, return leaf module
|
|
|
|
if (fromtuple != mp_const_none) {
|
|
|
|
return module_obj;
|
|
|
|
}
|
|
|
|
// Otherwise, we need to return top-level package
|
2014-02-16 00:53:44 +00:00
|
|
|
return top_module_obj;
|
2014-01-03 14:03:48 +00:00
|
|
|
}
|
2018-02-20 07:00:44 +00:00
|
|
|
|
|
|
|
#else // MICROPY_ENABLE_EXTERNAL_IMPORT
|
|
|
|
|
|
|
|
mp_obj_t mp_builtin___import__(size_t n_args, const mp_obj_t *args) {
|
|
|
|
// Check that it's not a relative import
|
|
|
|
if (n_args >= 5 && MP_OBJ_SMALL_INT_VALUE(args[4]) != 0) {
|
2020-03-02 11:35:22 +00:00
|
|
|
mp_raise_NotImplementedError(MP_ERROR_TEXT("relative import"));
|
2018-02-20 07:00:44 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// Check if module already exists, and return it if it does
|
|
|
|
qstr module_name_qstr = mp_obj_str_get_qstr(args[0]);
|
|
|
|
mp_obj_t module_obj = mp_module_get(module_name_qstr);
|
|
|
|
if (module_obj != MP_OBJ_NULL) {
|
|
|
|
return module_obj;
|
|
|
|
}
|
|
|
|
|
|
|
|
#if MICROPY_MODULE_WEAK_LINKS
|
|
|
|
// Check if there is a weak link to this module
|
py: Automatically provide weak links from "foo" to "ufoo" module name.
This commit implements automatic module weak links for all built-in
modules, by searching for "ufoo" in the built-in module list if "foo"
cannot be found. This means that all modules named "ufoo" are always
available as "foo". Also, a port can no longer add any other weak links,
which makes strict the definition of a weak link.
It saves some code size (about 100-200 bytes) on ports that previously had
lots of weak links.
Some changes from the previous behaviour:
- It doesn't intern the non-u module names (eg "foo" is not interned),
which saves code size, but will mean that "import foo" creates a new qstr
(namely "foo") in RAM (unless the importing module is frozen).
- help('modules') no longer lists non-u module names, only the u-variants;
this reduces duplication in the help listing.
Weak links are effectively the same as having a set of symbolic links on
the filesystem that is searched last. So an "import foo" will search
built-in modules first, then all paths in sys.path, then weak links last,
importing "ufoo" if it exists. Thus a file called "foo.py" somewhere in
sys.path will still have precedence over the weak link of "foo" to "ufoo".
See issues: #1740, #4449, #5229, #5241.
2019-10-21 15:06:34 +01:00
|
|
|
module_obj = mp_module_search_umodule(qstr_str(module_name_qstr));
|
|
|
|
if (module_obj != MP_OBJ_NULL) {
|
2018-02-20 07:00:44 +00:00
|
|
|
// Found weak-linked module
|
py: Automatically provide weak links from "foo" to "ufoo" module name.
This commit implements automatic module weak links for all built-in
modules, by searching for "ufoo" in the built-in module list if "foo"
cannot be found. This means that all modules named "ufoo" are always
available as "foo". Also, a port can no longer add any other weak links,
which makes strict the definition of a weak link.
It saves some code size (about 100-200 bytes) on ports that previously had
lots of weak links.
Some changes from the previous behaviour:
- It doesn't intern the non-u module names (eg "foo" is not interned),
which saves code size, but will mean that "import foo" creates a new qstr
(namely "foo") in RAM (unless the importing module is frozen).
- help('modules') no longer lists non-u module names, only the u-variants;
this reduces duplication in the help listing.
Weak links are effectively the same as having a set of symbolic links on
the filesystem that is searched last. So an "import foo" will search
built-in modules first, then all paths in sys.path, then weak links last,
importing "ufoo" if it exists. Thus a file called "foo.py" somewhere in
sys.path will still have precedence over the weak link of "foo" to "ufoo".
See issues: #1740, #4449, #5229, #5241.
2019-10-21 15:06:34 +01:00
|
|
|
mp_module_call_init(module_name_qstr, module_obj);
|
|
|
|
return module_obj;
|
2018-02-20 07:00:44 +00:00
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
|
|
// Couldn't find the module, so fail
|
2021-04-22 03:13:58 +01:00
|
|
|
#if MICROPY_ERROR_REPORTING <= MICROPY_ERROR_REPORTING_TERSE
|
2020-03-02 11:35:22 +00:00
|
|
|
mp_raise_msg(&mp_type_ImportError, MP_ERROR_TEXT("module not found"));
|
2019-09-26 13:52:04 +01:00
|
|
|
#else
|
2020-03-02 11:35:22 +00:00
|
|
|
mp_raise_msg_varg(&mp_type_ImportError, MP_ERROR_TEXT("no module named '%q'"), module_name_qstr);
|
2019-09-26 13:52:04 +01:00
|
|
|
#endif
|
2018-02-20 07:00:44 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
#endif // MICROPY_ENABLE_EXTERNAL_IMPORT
|
|
|
|
|
2014-02-03 22:46:17 +00:00
|
|
|
MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(mp_builtin___import___obj, 1, 5, mp_builtin___import__);
|