1
0
forked from VimPlug/jedi

a (temporary) solution for the __file__ access issues in imported modules, see #408

This commit is contained in:
Dave Halter
2014-05-19 01:01:56 +02:00
parent 709c53a679
commit 08b48807e9
5 changed files with 41 additions and 4 deletions

View File

@@ -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)

View File

@@ -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

View File

@@ -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)