submodules need relative imports not absolute

This commit is contained in:
Dave Halter
2014-06-25 11:08:29 +02:00
parent 44238a9f92
commit 6e82fa31e1
3 changed files with 6 additions and 4 deletions
+3 -2
View File
@@ -187,9 +187,10 @@ class FakeStatement(pr.Statement):
class FakeImport(pr.Import):
def __init__(self, name, parent):
def __init__(self, name, parent, level=0):
p = 0, 0
super(FakeImport, self).__init__(FakeSubModule, p, p, name)
super(FakeImport, self).__init__(FakeSubModule, p, p, name,
relative_count=level)
self.parent = parent
+2 -1
View File
@@ -561,7 +561,8 @@ class ModuleWrapper(pr.Module):
mods = pkgutil.iter_modules([os.path.dirname(path)])
for module_loader, name, is_pkg in mods:
name = helpers.FakeName(name)
imp = helpers.FakeImport(name, self)
# It's obviously a relative import to the current module.
imp = helpers.FakeImport(name, self, level=1)
name.parent = imp
names.append(name)
return names