Start reworking the relative imports

This commit is contained in:
Dave Halter
2019-03-07 00:27:51 +01:00
parent 7374819ade
commit c1d65ff144
2 changed files with 37 additions and 16 deletions

View File

@@ -238,29 +238,34 @@ class Importer(object):
import_path = base + tuple(import_path)
else:
path = module_context.py__file__()
if path is None:
# If no path is defined in the module we have no ideas where we
# are in the file system. Therefore we cannot know what to do.
# In this case we just let the path there and ignore that it's
# a relative path. Not sure if that's a good idea.
pass
else:
import_path = list(import_path)
p = path
for i in range(level):
p = os.path.dirname(p)
dir_name = os.path.basename(p)
# This is not the proper way to do relative imports. However, since
# Jedi cannot be sure about the entry point, we just calculate an
# absolute path here.
if path is None:
# If no path is defined, our best case is that the current
# file is edited by a user on the current working
# directory. We need to add an initial path, because it
# will get removed as the name of the current file.
directory = os.getcwd()
else:
directory = os.path.dirname(path)
level_import_paths = []
for i in range(level - 1):
directory = os.path.dirname(directory)
while directory != os.path.dirname(directory):
if directory == self._evaluator.project._path:
break
dir_name = os.path.basename(directory)
if dir_name:
import_path.insert(0, dir_name)
level_import_paths.insert(0, dir_name)
directory = os.path.dirname(directory)
else:
_add_error(
module_context, import_path[-1],
message='Attempted relative import beyond top-level package.'
)
import_path = []
self.import_path = []
return
import_path = level_import_paths + import_path
self.import_path = import_path
@property

View File

@@ -281,7 +281,7 @@ def test_get_modules_containing_name(evaluator, path, goal):
def test_relative_imports_with_multiple_similar_directories(Script, path, empty_sys_path):
dir = get_example_dir('issue1209')
script = Script(
"from .",
"from . ",
path=os.path.join(dir, path),
)
# TODO pass this project to the script as a param once that's possible.
@@ -292,3 +292,19 @@ def test_relative_imports_with_multiple_similar_directories(Script, path, empty_
name, import_ = script.completions()
assert import_.name == 'import'
assert name.name == 'api_test1'
@cwd_at('test/examples/issue1209/api/whatever/')
def test_relative_imports_without_path(Script):
script = Script("from . ")
name, import_ = script.completions()
assert import_.name == 'import'
assert name.name == 'api_test1'
script = Script("from .. ")
name, import_ = script.completions()
assert import_.name == 'whatever'
assert name.name == 'import'
script = Script("from ... ")
assert [c.name for c in script.completions()] == ['api', 'import', 'whatever']