From 11abd92d2b5d743aa4c6af25473caab7b37fb29e Mon Sep 17 00:00:00 2001 From: Takafumi Arakaki Date: Sun, 3 Mar 2013 04:49:53 +0100 Subject: [PATCH] Support alias in _make_fakeimport --- jedi/api.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/jedi/api.py b/jedi/api.py index 612dfa3e..6bbe52d7 100644 --- a/jedi/api.py +++ b/jedi/api.py @@ -545,7 +545,7 @@ class Interpreter(Script): scope.add_import(fakeimport) continue - def _make_fakeimport(self, module, variable=None): + def _make_fakeimport(self, module, variable=None, alias=None): submodule = self._parser.scope._sub_module if variable: varname = pr.Name( @@ -560,17 +560,27 @@ class Interpreter(Script): names=[(module, (0, 0))], start_pos=(0, 0), end_pos=(None, None)) + if alias: + aliasname = pr.Name( + module=submodule, + names=[(alias, (0, 0))], + start_pos=(0, 0), + end_pos=(None, None)) + else: + aliasname = None if varname: fakeimport = pr.Import( module=submodule, namespace=varname, from_ns=modname, + alias=aliasname, start_pos=(0, 0), end_pos=(None, None)) else: fakeimport = pr.Import( module=submodule, namespace=modname, + alias=aliasname, start_pos=(0, 0), end_pos=(None, None)) return fakeimport