Mercurial > hg

diff hgdemandimport/demandimportpy2.py @ 32447:252d2260c74e

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
demandimport: look for 'mod' suffix as alternative name for module reference It's widely used in our codebase.
author Yuya Nishihara <yuya@tcha.org>
date Fri, 28 Apr 2017 23:46:16 +0900
parents 63365e9621d6
children 91a2ec8e7fa0
line wrap: on
line diff
--- a/hgdemandimport/demandimportpy2.py	Mon May 01 14:26:56 2017 +0900
+++ b/hgdemandimport/demandimportpy2.py	Fri Apr 28 23:46:16 2017 +0900
@@ -132,8 +132,11 @@
                 subload(mod, x)
 
             # Replace references to this proxy instance with the actual module.
-            if locals and locals.get(head) is self:
-                locals[head] = mod
+            if locals:
+                if locals.get(head) is self:
+                    locals[head] = mod
+                elif locals.get(head + r'mod') is self:
+                    locals[head + r'mod'] = mod
 
             for modname in modrefs:
                 modref = sys.modules.get(modname, None)