##// END OF EJS Templates
Add some modules to the demandimport ignore list for Windows
Matt Mackall -
r3929:33988aaa default
parent child Browse files
Show More
@@ -1,115 +1,115
1 1 # demandimport.py - global demand-loading of modules for Mercurial
2 2 #
3 3 # Copyright 2006 Matt Mackall <mpm@selenic.com>
4 4 #
5 5 # This software may be used and distributed according to the terms
6 6 # of the GNU General Public License, incorporated herein by reference.
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 _origimport = __import__
28 28
29 29 class _demandmod(object):
30 30 """module demand-loader and proxy"""
31 31 def __init__(self, name, globals, locals):
32 32 if '.' in name:
33 33 head, rest = name.split('.', 1)
34 34 after = [rest]
35 35 else:
36 36 head = name
37 37 after = []
38 38 object.__setattr__(self, "_data", (head, globals, locals, after))
39 39 object.__setattr__(self, "_module", None)
40 40 def _extend(self, name):
41 41 """add to the list of submodules to load"""
42 42 self._data[3].append(name)
43 43 def _load(self):
44 44 if not self._module:
45 45 head, globals, locals, after = self._data
46 46 mod = _origimport(head, globals, locals)
47 47 # load submodules
48 48 def subload(mod, p):
49 49 h, t = p, None
50 50 if '.' in p:
51 51 h, t = p.split('.', 1)
52 52 if not hasattr(mod, h):
53 53 setattr(mod, h, _demandmod(p, mod.__dict__, mod.__dict__))
54 54 elif t:
55 55 subload(getattr(mod, h), t)
56 56
57 57 for x in after:
58 58 subload(mod, x)
59 59
60 60 # are we in the locals dictionary still?
61 61 if locals and locals.get(head) == self:
62 62 locals[head] = mod
63 63 object.__setattr__(self, "_module", mod)
64 64 def __repr__(self):
65 65 return "<unloaded module '%s'>" % self._data[0]
66 66 def __call__(self, *args, **kwargs):
67 67 raise TypeError("'unloaded module' object is not callable")
68 68 def __getattribute__(self, attr):
69 69 if attr in ('_data', '_extend', '_load', '_module'):
70 70 return object.__getattribute__(self, attr)
71 71 self._load()
72 72 return getattr(self._module, attr)
73 73 def __setattr__(self, attr, val):
74 74 self._load()
75 75 setattr(self._module, attr, val)
76 76
77 77 def _demandimport(name, globals=None, locals=None, fromlist=None):
78 78 if not locals or name in ignore or fromlist == ('*',):
79 79 # these cases we can't really delay
80 80 return _origimport(name, globals, locals, fromlist)
81 81 elif not fromlist:
82 82 # import a [as b]
83 83 if '.' in name: # a.b
84 84 base, rest = name.split('.', 1)
85 85 # email.__init__ loading email.mime
86 86 if globals and globals.get('__name__', None) == base:
87 87 return _origimport(name, globals, locals, fromlist)
88 88 # if a is already demand-loaded, add b to its submodule list
89 89 if base in locals:
90 90 if isinstance(locals[base], _demandmod):
91 91 locals[base]._extend(rest)
92 92 return locals[base]
93 93 return _demandmod(name, globals, locals)
94 94 else:
95 95 # from a import b,c,d
96 96 mod = _origimport(name, globals, locals)
97 97 # recurse down the module chain
98 98 for comp in name.split('.')[1:]:
99 99 mod = getattr(mod, comp)
100 100 for x in fromlist:
101 101 # set requested submodules for demand load
102 102 if not(hasattr(mod, x)):
103 103 setattr(mod, x, _demandmod(x, mod.__dict__, mod.__dict__))
104 104 return mod
105 105
106 ignore = ['_hashlib']
106 ignore = ['_hashlib', 'fcntl', 'win32com.gen_py']
107 107
108 108 def enable():
109 109 "enable global demand-loading of modules"
110 110 __builtins__["__import__"] = _demandimport
111 111
112 112 def disable():
113 113 "disable global demand-loading of modules"
114 114 __builtins__["__import__"] = _origimport
115 115
General Comments 0
You need to be logged in to leave comments. Login now