diff options
author | Ben Lewis <benjimin@users.noreply.github.com> | 2019-09-11 10:09:47 (GMT) |
---|---|---|
committer | Brett Cannon <54418+brettcannon@users.noreply.github.com> | 2019-09-11 10:09:47 (GMT) |
commit | 92420b3e679959a7d0ce875875601a4cee45231e (patch) | |
tree | 8e73ad151c044a4233f9b8cfa94127b477a2a33a /Python/import.c | |
parent | 6472ece5a0fe82809d3aa0ffb281796fcd252d76 (diff) | |
download | cpython-92420b3e679959a7d0ce875875601a4cee45231e.zip cpython-92420b3e679959a7d0ce875875601a4cee45231e.tar.gz cpython-92420b3e679959a7d0ce875875601a4cee45231e.tar.bz2 |
bpo-37409: fix relative import with no parent (#14956)
Relative imports use resolve_name to get the absolute target name,
which first seeks the current module's absolute package name from the globals:
If __package__ (and __spec__.parent) are missing then
import uses __name__, truncating the last segment if
the module is a submodule rather than a package __init__.py
(which it guesses from whether __path__ is defined).
The __name__ attempt should fail if there is no parent package (top level modules),
if __name__ is '__main__' (-m entry points), or both (scripts).
That is, if both __name__ has no subcomponents and the module does not seem
to be a package __init__ module then import should fail.
Diffstat (limited to 'Python/import.c')
-rw-r--r-- | Python/import.c | 24 |
1 files changed, 13 insertions, 11 deletions
diff --git a/Python/import.c b/Python/import.c index 41c2f34..5be4d19 100644 --- a/Python/import.c +++ b/Python/import.c @@ -1646,23 +1646,20 @@ resolve_name(PyThreadState *tstate, PyObject *name, PyObject *globals, int level if (dot == -2) { goto error; } - - if (dot >= 0) { - PyObject *substr = PyUnicode_Substring(package, 0, dot); - if (substr == NULL) { - goto error; - } - Py_SETREF(package, substr); + else if (dot == -1) { + goto no_parent_error; } + PyObject *substr = PyUnicode_Substring(package, 0, dot); + if (substr == NULL) { + goto error; + } + Py_SETREF(package, substr); } } last_dot = PyUnicode_GET_LENGTH(package); if (last_dot == 0) { - _PyErr_SetString(tstate, PyExc_ImportError, - "attempted relative import " - "with no known parent package"); - goto error; + goto no_parent_error; } for (level_up = 1; level_up < level; level_up += 1) { @@ -1688,6 +1685,11 @@ resolve_name(PyThreadState *tstate, PyObject *name, PyObject *globals, int level Py_DECREF(base); return abs_name; + no_parent_error: + _PyErr_SetString(tstate, PyExc_ImportError, + "attempted relative import " + "with no known parent package"); + error: Py_XDECREF(package); return NULL; |