##// END OF EJS Templates
demandimport: add __future__ to ignore list...
Gregory Szorc -
r25934:e283c5d9 default
parent child Browse files
Show More
@@ -1,195 +1,196
1 1 # demandimport.py - global demand-loading of modules for Mercurial
2 2 #
3 3 # Copyright 2006, 2007 Matt Mackall <mpm@selenic.com>
4 4 #
5 5 # This software may be used and distributed according to the terms of the
6 6 # GNU General Public License version 2 or any later version.
7 7
8 8 '''
9 9 demandimport - automatic demandloading of modules
10 10
11 11 To enable this module, do:
12 12
13 13 import demandimport; demandimport.enable()
14 14
15 15 Imports of the following forms will be demand-loaded:
16 16
17 17 import a, b.c
18 18 import a.b as c
19 19 from a import b,c # a will be loaded immediately
20 20
21 21 These imports will not be delayed:
22 22
23 23 from a import *
24 24 b = __import__(a)
25 25 '''
26 26
27 27 import os, sys
28 28 from contextlib import contextmanager
29 29
30 30 # __builtin__ in Python 2, builtins in Python 3.
31 31 try:
32 32 import __builtin__ as builtins
33 33 except ImportError:
34 34 import builtins
35 35
36 36 _origimport = __import__
37 37
38 38 nothing = object()
39 39
40 40 # Python 3 doesn't have relative imports nor level -1.
41 41 level = -1
42 42 if sys.version_info[0] >= 3:
43 43 level = 0
44 44 _import = _origimport
45 45
46 46 def _hgextimport(importfunc, name, globals, *args):
47 47 try:
48 48 return importfunc(name, globals, *args)
49 49 except ImportError:
50 50 if not globals:
51 51 raise
52 52 # extensions are loaded with "hgext_" prefix
53 53 hgextname = 'hgext_%s' % name
54 54 nameroot = hgextname.split('.', 1)[0]
55 55 contextroot = globals.get('__name__', '').split('.', 1)[0]
56 56 if nameroot != contextroot:
57 57 raise
58 58 # retry to import with "hgext_" prefix
59 59 return importfunc(hgextname, globals, *args)
60 60
61 61 class _demandmod(object):
62 62 """module demand-loader and proxy"""
63 63 def __init__(self, name, globals, locals, level=level):
64 64 if '.' in name:
65 65 head, rest = name.split('.', 1)
66 66 after = [rest]
67 67 else:
68 68 head = name
69 69 after = []
70 70 object.__setattr__(self, "_data",
71 71 (head, globals, locals, after, level))
72 72 object.__setattr__(self, "_module", None)
73 73 def _extend(self, name):
74 74 """add to the list of submodules to load"""
75 75 self._data[3].append(name)
76 76 def _load(self):
77 77 if not self._module:
78 78 head, globals, locals, after, level = self._data
79 79 mod = _hgextimport(_import, head, globals, locals, None, level)
80 80 # load submodules
81 81 def subload(mod, p):
82 82 h, t = p, None
83 83 if '.' in p:
84 84 h, t = p.split('.', 1)
85 85 if getattr(mod, h, nothing) is nothing:
86 86 setattr(mod, h, _demandmod(p, mod.__dict__, mod.__dict__))
87 87 elif t:
88 88 subload(getattr(mod, h), t)
89 89
90 90 for x in after:
91 91 subload(mod, x)
92 92
93 93 # are we in the locals dictionary still?
94 94 if locals and locals.get(head) == self:
95 95 locals[head] = mod
96 96 object.__setattr__(self, "_module", mod)
97 97
98 98 def __repr__(self):
99 99 if self._module:
100 100 return "<proxied module '%s'>" % self._data[0]
101 101 return "<unloaded module '%s'>" % self._data[0]
102 102 def __call__(self, *args, **kwargs):
103 103 raise TypeError("%s object is not callable" % repr(self))
104 104 def __getattribute__(self, attr):
105 105 if attr in ('_data', '_extend', '_load', '_module'):
106 106 return object.__getattribute__(self, attr)
107 107 self._load()
108 108 return getattr(self._module, attr)
109 109 def __setattr__(self, attr, val):
110 110 self._load()
111 111 setattr(self._module, attr, val)
112 112
113 113 def _demandimport(name, globals=None, locals=None, fromlist=None, level=level):
114 114 if not locals or name in ignore or fromlist == ('*',):
115 115 # these cases we can't really delay
116 116 return _hgextimport(_import, name, globals, locals, fromlist, level)
117 117 elif not fromlist:
118 118 # import a [as b]
119 119 if '.' in name: # a.b
120 120 base, rest = name.split('.', 1)
121 121 # email.__init__ loading email.mime
122 122 if globals and globals.get('__name__', None) == base:
123 123 return _import(name, globals, locals, fromlist, level)
124 124 # if a is already demand-loaded, add b to its submodule list
125 125 if base in locals:
126 126 if isinstance(locals[base], _demandmod):
127 127 locals[base]._extend(rest)
128 128 return locals[base]
129 129 return _demandmod(name, globals, locals, level)
130 130 else:
131 131 if level != -1:
132 132 # from . import b,c,d or from .a import b,c,d
133 133 return _origimport(name, globals, locals, fromlist, level)
134 134 # from a import b,c,d
135 135 mod = _hgextimport(_origimport, name, globals, locals)
136 136 # recurse down the module chain
137 137 for comp in name.split('.')[1:]:
138 138 if getattr(mod, comp, nothing) is nothing:
139 139 setattr(mod, comp, _demandmod(comp, mod.__dict__, mod.__dict__))
140 140 mod = getattr(mod, comp)
141 141 for x in fromlist:
142 142 # set requested submodules for demand load
143 143 if getattr(mod, x, nothing) is nothing:
144 144 setattr(mod, x, _demandmod(x, mod.__dict__, locals))
145 145 return mod
146 146
147 147 ignore = [
148 '__future__',
148 149 '_hashlib',
149 150 '_xmlplus',
150 151 'fcntl',
151 152 'win32com.gen_py',
152 153 '_winreg', # 2.7 mimetypes needs immediate ImportError
153 154 'pythoncom',
154 155 # imported by tarfile, not available under Windows
155 156 'pwd',
156 157 'grp',
157 158 # imported by profile, itself imported by hotshot.stats,
158 159 # not available under Windows
159 160 'resource',
160 161 # this trips up many extension authors
161 162 'gtk',
162 163 # setuptools' pkg_resources.py expects "from __main__ import x" to
163 164 # raise ImportError if x not defined
164 165 '__main__',
165 166 '_ssl', # conditional imports in the stdlib, issue1964
166 167 'rfc822',
167 168 'mimetools',
168 169 # setuptools 8 expects this module to explode early when not on windows
169 170 'distutils.msvc9compiler'
170 171 ]
171 172
172 173 def isenabled():
173 174 return builtins.__import__ == _demandimport
174 175
175 176 def enable():
176 177 "enable global demand-loading of modules"
177 178 if os.environ.get('HGDEMANDIMPORT') != 'disable':
178 179 builtins.__import__ = _demandimport
179 180
180 181 def disable():
181 182 "disable global demand-loading of modules"
182 183 builtins.__import__ = _origimport
183 184
184 185 @contextmanager
185 186 def deactivated():
186 187 "context manager for disabling demandimport in 'with' blocks"
187 188 demandenabled = isenabled()
188 189 if demandenabled:
189 190 disable()
190 191
191 192 try:
192 193 yield
193 194 finally:
194 195 if demandenabled:
195 196 enable()
General Comments 0
You need to be logged in to leave comments. Login now