##// END OF EJS Templates
use parent.__setattr__ instead of __dict__
Benoit Boissinot -
r3896:3b628b5d default
parent child Browse files
Show More
@@ -1,104 +1,104 b''
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 self.__dict__["_data"] = (head, globals, locals, after)
39 self.__dict__["_module"] = None
38 object.__setattr__(self, "_data", (head, globals, locals, after))
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 for x in after:
49 49 hx = x
50 50 if '.' in x:
51 51 hx = x.split('.')[0]
52 52 if not hasattr(mod, hx):
53 53 setattr(mod, hx, _demandmod(x, mod.__dict__, mod.__dict__))
54 54 # are we in the locals dictionary still?
55 55 if locals and locals.get(head) == self:
56 56 locals[head] = mod
57 self.__dict__["_module"] = mod
57 object.__setattr__(self, "_module", mod)
58 58 def __repr__(self):
59 59 return "<unloaded module '%s'>" % self._data[0]
60 60 def __call__(self, *args, **kwargs):
61 61 raise TypeError("'unloaded module' object is not callable")
62 62 def __getattr__(self, attr):
63 63 self._load()
64 64 return getattr(self._module, attr)
65 65 def __setattr__(self, attr, val):
66 66 self._load()
67 67 setattr(self._module, attr, val)
68 68
69 69 def _demandimport(name, globals=None, locals=None, fromlist=None):
70 70 if not locals or name in ignore or fromlist == ('*',):
71 71 # these cases we can't really delay
72 72 return _origimport(name, globals, locals, fromlist)
73 73 elif not fromlist:
74 74 # import a [as b]
75 75 if '.' in name: # a.b
76 76 base, rest = name.split('.', 1)
77 77 # if a is already demand-loaded, add b to its submodule list
78 78 if base in locals:
79 79 if isinstance(locals[base], _demandmod):
80 80 locals[base]._extend(rest)
81 81 return locals[base]
82 82 return _demandmod(name, globals, locals)
83 83 else:
84 84 # from a import b,c,d
85 85 mod = _origimport(name, globals, locals)
86 86 # recurse down the module chain
87 87 for comp in name.split('.')[1:]:
88 88 mod = getattr(mod, comp)
89 89 for x in fromlist:
90 90 # set requested submodules for demand load
91 91 if not(hasattr(mod, x)):
92 92 setattr(mod, x, _demandmod(x, mod.__dict__, mod.__dict__))
93 93 return mod
94 94
95 95 ignore = []
96 96
97 97 def enable():
98 98 "enable global demand-loading of modules"
99 99 __builtins__["__import__"] = _demandimport
100 100
101 101 def disable():
102 102 "disable global demand-loading of modules"
103 103 __builtins__["__import__"] = _origimport
104 104
General Comments 0
You need to be logged in to leave comments. Login now