From f49fba5eda40fe73d3924c75251d9e317b6e4db2 Mon Sep 17 00:00:00 2001 From: Zac Medico Date: Tue, 9 Mar 2010 02:42:40 +0000 Subject: [PATCH] When portage upgrades or downgrades itself, preload lazily referenced portage submodules into memory so that imports won't fail later. (trunk r15778) svn path=/main/branches/2.1.7/; revision=15782 --- pym/portage/dbapi/vartree.py | 3 +++ pym/portage/proxy/lazyimport.py | 18 ++++++++++++++++++ 2 files changed, 21 insertions(+) diff --git a/pym/portage/dbapi/vartree.py b/pym/portage/dbapi/vartree.py index 2f09ecaa2..80b564fc9 100644 --- a/pym/portage/dbapi/vartree.py +++ b/pym/portage/dbapi/vartree.py @@ -3181,6 +3181,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 c62f90788..84b2c730b 100644 --- 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: -- 2.26.2