demandimport: fix import x.y.z as a when x.y is already imported.
authorMatt Mackall <mpm@selenic.com>
Sun, 17 Dec 2006 14:56:12 -0600
changeset 3921 6d0d025e125a
parent 3912 f2c4a2ee59ed
child 3922 693eec9570f3
demandimport: fix import x.y.z as a when x.y is already imported.
mercurial/demandimport.py
--- a/mercurial/demandimport.py	Sat Dec 16 14:51:22 2006 -0600
+++ b/mercurial/demandimport.py	Sun Dec 17 14:56:12 2006 -0600
@@ -45,12 +45,18 @@
             head, globals, locals, after = self._data
             mod = _origimport(head, globals, locals)
             # load submodules
+            def subload(mod, p):
+                h, t = p, None
+                if '.' in p:
+                    h, t = p.split('.', 1)
+                if not hasattr(mod, h):
+                    setattr(mod, h, _demandmod(p, mod.__dict__, mod.__dict__))
+                else:
+                    subload(getattr(mod, h), t)
+
             for x in after:
-                hx = x
-                if '.' in x:
-                    hx = x.split('.')[0]
-                if not hasattr(mod, hx):
-                    setattr(mod, hx, _demandmod(x, mod.__dict__, mod.__dict__))
+                subload(mod, x)
+
             # are we in the locals dictionary still?
             if locals and locals.get(head) == self:
                 locals[head] = mod