VimPlug/jedi-fork
1
0
Fork 0
forked from VimPlug/jedi
Code Pull Requests Activity

Implement the case where raw object is a module

Browse Source 
Previously failing test_complete_raw_module is passed.
This commit is contained in:
Takafumi Arakaki
2013-03-02 21:03:53 +01:00
parent fb197c42ab
commit 17770356e2
1 changed files with 31 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

48
jedi/api.py
View File

@@ -533,31 +533,45 @@ class Interpreter(Script):
def _import_raw_namespace(self, raw_namespace):
for (variable, obj) in raw_namespace.items():
try:
module = obj.__module__
except AttributeError:
module = getattr(obj, '__module__', None)
if module:
fakeimport = self._make_fakeimport(module, variable)
self._parser.scope.imports.append(fakeimport)
continue
fakeimport = self._make_fakeimport(variable, module)
self._parser.scope.imports.append(fakeimport)
def _make_fakeimport(self, variable, module):
if getattr(obj, '__file__', None):
fakeimport = self._make_fakeimport(obj.__name__)
self._parser.scope.imports.append(fakeimport)
continue
def _make_fakeimport(self, module, variable=None):
submodule = self._parser.scope._sub_module
varname = pr.Name(
module=submodule,
names=[(variable, (0, 0))],
start_pos=(0, 0),
end_pos=(None, None))
if variable:
varname = pr.Name(
module=submodule,
names=[(variable, (0, 0))],
start_pos=(0, 0),
end_pos=(None, None))
else:
varname = None
modname = pr.Name(
module=submodule,
names=[(module, (0, 0))],
start_pos=(0, 0),
end_pos=(None, None))
fakeimport = pr.Import(
module=submodule,
namespace=varname,
from_ns=modname,
start_pos=(0, 0),
end_pos=(None, None))
if varname:
fakeimport = pr.Import(
module=submodule,
namespace=varname,
from_ns=modname,
start_pos=(0, 0),
end_pos=(None, None))
else:
fakeimport = pr.Import(
module=submodule,
namespace=modname,
start_pos=(0, 0),
end_pos=(None, None))
fakeimport.parent = submodule
return fakeimport