##// END OF EJS Templates
hgweb: pre-init mimetypes module (fixes ugly bug in python-2.6.2 mimetypes)...
hgweb: pre-init mimetypes module (fixes ugly bug in python-2.6.2 mimetypes) The ugly initialization hack in mimetypes' initialization procedure can make hg serve hit the maximum recursion depth when in threaded mode.

File last commit:

r8206:cce63ef1 default
r8224:1075f5c1 default
Show More
extensions.py
119 lines | 3.4 KiB | text/x-python | PythonLexer
Matt Mackall
Create a separate module for managing extensions
r4544 # extensions.py - extension handling for mercurial
#
Thomas Arendsen Hein
Updated copyright notices and add "and others" to "hg version"
r4635 # Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
Matt Mackall
Create a separate module for managing extensions
r4544 #
# This software may be used and distributed according to the terms
# of the GNU General Public License, incorporated herein by reference.
Brendan Cully
extensions: load modules in module/__init__.py form....
r4569 import imp, os
Matt Mackall
extensions: add wrapping functions
r7215 import util, cmdutil
Matt Mackall
Create a separate module for managing extensions
r4544 from i18n import _
_extensions = {}
Alexis S. L. Carvalho
Move cmdtable and reposetup handling out of extensions.py...
r5192 _order = []
def extensions():
for name in _order:
module = _extensions[name]
if module:
yield name, module
Matt Mackall
Create a separate module for managing extensions
r4544
def find(name):
'''return module with given extension name'''
try:
return _extensions[name]
except KeyError:
for k, v in _extensions.iteritems():
Matt Mackall
extensions: fix lookup of hgext.foo modules
r4560 if k.endswith('.' + name) or k.endswith('/' + name):
return v
Matt Mackall
Create a separate module for managing extensions
r4544 raise KeyError(name)
Alexander Solovyov
ability to load hooks from arbitrary python module
r7916 def loadpath(path, module_name):
module_name = module_name.replace('.', '_')
path = os.path.expanduser(path)
if os.path.isdir(path):
# module/__init__.py style
Alexander Solovyov
extensions loading: don't fail if path to extension ends with a '/'
r7960 d, f = os.path.split(path.rstrip('/'))
Alexander Solovyov
ability to load hooks from arbitrary python module
r7916 fd, fpath, desc = imp.find_module(f, [d])
return imp.load_module(module_name, fd, fpath, desc)
else:
return imp.load_source(module_name, path)
Matt Mackall
Create a separate module for managing extensions
r4544 def load(ui, name, path):
Benoit Boissinot
Do not try to load extensions twice (issue811)
r7011 if name.startswith('hgext.') or name.startswith('hgext/'):
Bryan O'Sullivan
extensions: don't get confused by aliasing between "foo" and "hgext.foo"
r5031 shortname = name[6:]
else:
shortname = name
if shortname in _extensions:
Matt Mackall
Create a separate module for managing extensions
r4544 return
Brendan Cully
Cache extension load failures....
r5087 _extensions[shortname] = None
Matt Mackall
Create a separate module for managing extensions
r4544 if path:
# the module will be loaded in sys.modules
# choose an unique name so that it doesn't
# conflicts with other modules
Alexander Solovyov
ability to load hooks from arbitrary python module
r7916 mod = loadpath(path, 'hgext.%s' % name)
Matt Mackall
Create a separate module for managing extensions
r4544 else:
def importh(name):
mod = __import__(name)
components = name.split('.')
for comp in components[1:]:
mod = getattr(mod, comp)
return mod
try:
mod = importh("hgext.%s" % name)
except ImportError:
mod = importh(name)
Bryan O'Sullivan
extensions: don't get confused by aliasing between "foo" and "hgext.foo"
r5031 _extensions[shortname] = mod
Alexis S. L. Carvalho
Move cmdtable and reposetup handling out of extensions.py...
r5192 _order.append(shortname)
Matt Mackall
Create a separate module for managing extensions
r4544
uisetup = getattr(mod, 'uisetup', None)
if uisetup:
uisetup(ui)
def loadall(ui):
Matt Mackall
extensions: pull extension-aware bits out of ui
r4617 result = ui.configitems("extensions")
Peter Arrenbrecht
cleanup: drop enumerate() when index is not used
r7876 for (name, path) in result:
Matt Mackall
extensions: pull extension-aware bits out of ui
r4617 if path:
Steve Borho
Allow explicit disabling of extensions...
r5469 if path[0] == '!':
continue
Matt Mackall
Create a separate module for managing extensions
r4544 try:
load(ui, name, path)
Matt Mackall
error: move SignalInterrupt...
r7644 except KeyboardInterrupt:
Matt Mackall
Create a separate module for managing extensions
r4544 raise
except Exception, inst:
Jesse Glick
When failing to load an extension, show where Hg tried to load it from.
r6204 if path:
ui.warn(_("*** failed to import extension %s from %s: %s\n")
% (name, path, inst))
else:
ui.warn(_("*** failed to import extension %s: %s\n")
% (name, inst))
Matt Mackall
ui: print_exc() -> traceback()
r8206 if ui.traceback():
Matt Mackall
Create a separate module for managing extensions
r4544 return 1
Matt Mackall
extensions: add wrapping functions
r7215 def wrapcommand(table, command, wrapper):
aliases, entry = cmdutil.findcmd(command, table)
for alias, e in table.iteritems():
if e is entry:
key = alias
break
origfn = entry[0]
def wrap(*args, **kwargs):
Matt Mackall
dispatch: generalize signature checking for extension command wrapping
r7388 return util.checksignature(wrapper)(
util.checksignature(origfn), *args, **kwargs)
Matt Mackall
extensions: add wrapping functions
r7215
wrap.__doc__ = getattr(origfn, '__doc__')
Dirkjan Ochtman
extensions: copy __module__ for wrapped commands
r7373 wrap.__module__ = getattr(origfn, '__module__')
Matt Mackall
extensions: add wrapping functions
r7215
newentry = list(entry)
newentry[0] = wrap
table[key] = tuple(newentry)
return entry
def wrapfunction(container, funcname, wrapper):
def wrap(*args, **kwargs):
return wrapper(origfn, *args, **kwargs)
origfn = getattr(container, funcname)
setattr(container, funcname, wrap)
return origfn