|
1 # extensions.py - extension handling for mercurial |
|
2 # |
|
3 # Copyright 2005, 2006 Matt Mackall <mpm@selenic.com> |
|
4 # |
|
5 # This software may be used and distributed according to the terms |
|
6 # of the GNU General Public License, incorporated herein by reference. |
|
7 |
|
8 import imp, commands, hg, util |
|
9 from i18n import _ |
|
10 |
|
11 _extensions = {} |
|
12 |
|
13 def find(name): |
|
14 '''return module with given extension name''' |
|
15 try: |
|
16 return _extensions[name] |
|
17 except KeyError: |
|
18 for k, v in _extensions.iteritems(): |
|
19 if k.endswith('.' + name) or k.endswith('/' + name) or v == name: |
|
20 return sys.modules[v] |
|
21 raise KeyError(name) |
|
22 |
|
23 def load(ui, name, path): |
|
24 if name in _extensions: |
|
25 return |
|
26 if path: |
|
27 # the module will be loaded in sys.modules |
|
28 # choose an unique name so that it doesn't |
|
29 # conflicts with other modules |
|
30 module_name = "hgext_%s" % name.replace('.', '_') |
|
31 mod = imp.load_source(module_name, path) |
|
32 else: |
|
33 def importh(name): |
|
34 mod = __import__(name) |
|
35 components = name.split('.') |
|
36 for comp in components[1:]: |
|
37 mod = getattr(mod, comp) |
|
38 return mod |
|
39 try: |
|
40 mod = importh("hgext.%s" % name) |
|
41 except ImportError: |
|
42 mod = importh(name) |
|
43 _extensions[name] = mod |
|
44 |
|
45 uisetup = getattr(mod, 'uisetup', None) |
|
46 if uisetup: |
|
47 uisetup(ui) |
|
48 reposetup = getattr(mod, 'reposetup', None) |
|
49 if reposetup: |
|
50 hg.repo_setup_hooks.append(reposetup) |
|
51 cmdtable = getattr(mod, 'cmdtable', {}) |
|
52 overrides = [cmd for cmd in cmdtable if cmd in commands.table] |
|
53 if overrides: |
|
54 ui.warn(_("extension '%s' overrides commands: %s\n") |
|
55 % (name, " ".join(overrides))) |
|
56 commands.table.update(cmdtable) |
|
57 |
|
58 def loadall(ui): |
|
59 for name, path in ui.extensions(): |
|
60 try: |
|
61 load(ui, name, path) |
|
62 except (util.SignalInterrupt, KeyboardInterrupt): |
|
63 raise |
|
64 except Exception, inst: |
|
65 ui.warn(_("*** failed to import extension %s: %s\n") % |
|
66 (name, inst)) |
|
67 if ui.print_exc(): |
|
68 return 1 |
|
69 |