forked from VimPlug/jedi
a (temporary) solution for the __file__ access issues in imported modules, see #408
This commit is contained in:
@@ -206,6 +206,8 @@ class NameFinder(object):
|
||||
typ = er.Class(evaluator, typ)
|
||||
elif isinstance(typ, pr.Function):
|
||||
typ = er.Function(evaluator, typ)
|
||||
elif isinstance(typ, pr.Module):
|
||||
typ = er.ModuleWrapper(evaluator, typ)
|
||||
if typ.isinstance(er.Function) and resolve_decorator:
|
||||
typ = typ.get_decorated_func()
|
||||
types.append(typ)
|
||||
|
||||
@@ -473,7 +473,13 @@ def follow_imports(evaluator, scopes):
|
||||
result = []
|
||||
for s in scopes:
|
||||
if isinstance(s, pr.Import):
|
||||
result += ImportWrapper(evaluator, s).follow()
|
||||
for r in ImportWrapper(evaluator, s).follow():
|
||||
if isinstance(r, pr.Module) and not isinstance(r,
|
||||
StarImportModule):
|
||||
# TODO This check is strange and feels wrong somehow. Change it.
|
||||
from jedi.evaluate import representation as er
|
||||
r = er.ModuleWrapper(evaluator, r)
|
||||
result.append(r)
|
||||
else:
|
||||
result.append(s)
|
||||
return result
|
||||
|
||||
@@ -521,3 +521,23 @@ class FunctionExecution(Executable):
|
||||
|
||||
def __repr__(self):
|
||||
return "<%s of %s>" % (type(self).__name__, self.base)
|
||||
|
||||
|
||||
class ModuleWrapper(pr.Module):
|
||||
def __init__(self, evaluator, module):
|
||||
self._evaluator = evaluator
|
||||
self._module = module
|
||||
|
||||
@memoize_default()
|
||||
def get_defined_names(self):
|
||||
names = ['__file__', '__package__', '__doc__', '__name__', '__version__']
|
||||
# All the additional module attributes are strings.
|
||||
parent = Instance(self._evaluator, compiled.create(self._evaluator, str))
|
||||
module_attributes = [helpers.FakeName(n, parent) for n in names]
|
||||
return self._module.get_defined_names() + module_attributes
|
||||
|
||||
def __getattr__(self, name):
|
||||
return getattr(self._module, name)
|
||||
|
||||
def __repr__(self):
|
||||
return "<%s: %s>" % (type(self).__name__, self._module)
|
||||
|
||||
Reference in New Issue
Block a user