When portage upgrades or downgrades itself, preload lazily referenced portage

submodules into memory so that imports won't fail later.

svn path=/main/trunk/; revision=15778

diff --git a/pym/portage/dbapi/vartree.py b/pym/portage/dbapi/vartree.py
index 76a9034ebabbb18631ac0a89796c3d149cb18788..4eee0cf05b753f49cec505cc62671eb6cbbb0d7a 100644 (file)
--- a/pym/portage/dbapi/vartree.py
+++ b/pym/portage/dbapi/vartree.py
@@ -4343,6 +4343,9 @@ class dblink(object):
                if self.myroot == "/" and \
                        match_from_list(PORTAGE_PACKAGE_ATOM, [self.mycpv]) and \
                        not self.vartree.dbapi.cpv_exists(self.mycpv):
+                       # Load lazily referenced portage submodules into memory,
+                       # so imports won't fail during portage upgrade/downgrade.
+                       portage.proxy.lazyimport._preload_portage_submodules()
                        settings = self.settings
                        base_path_orig = os.path.dirname(settings["PORTAGE_BIN_PATH"])
                        from tempfile import mkdtemp

diff --git a/pym/portage/proxy/lazyimport.py b/pym/portage/proxy/lazyimport.py
index c62f9078895e24c599e8f851ae828436f877ea85..84b2c730b0f25831c9232bfb0a6156302c68d0a2 100644 (file)
--- a/pym/portage/proxy/lazyimport.py
+++ b/pym/portage/proxy/lazyimport.py
@@ -20,6 +20,24 @@ if sys.hexversion >= 0x3000000:
 _module_proxies = {}
 _module_proxies_lock = threading.RLock()
 
+def _preload_portage_submodules():
+       """
+       Load lazily referenced portage submodules into memory,
+       so imports won't fail during portage upgrade/downgrade.
+       Note that this recursively loads only the modules that
+       are lazily referenced by currently imported modules,
+       so some portage submodules may still remain unimported
+       after this function is called.
+       """
+       while True:
+               remaining = False
+               for name in list(_module_proxies):
+                       if name.startswith('portage.'):
+                               remaining = True
+                               _unregister_module_proxy(name)
+               if not remaining:
+                       break
+
 def _register_module_proxy(name, proxy):
        _module_proxies_lock.acquire()
        try: