##// END OF EJS Templates
win32mbcs: fix a `str` type conditional for py3...
win32mbcs: fix a `str` type conditional for py3 Differential Revision: https://phab.mercurial-scm.org/D7535

File last commit:

r44106:da5ccc59 default
r44181:66210a20 stable
Show More
extensions.py
944 lines | 29.1 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 #
Martin Geisler
updated license to be explicit about GPL version 2
r8225 # This software may be used and distributed according to the terms of the
Matt Mackall
Update license to GPLv2+
r10263 # GNU General Public License version 2 or any later version.
Matt Mackall
Create a separate module for managing extensions
r4544
Gregory Szorc
extensions: use absolute_import
r25946 from __future__ import absolute_import
Yuya Nishihara
extensions: peek command table of disabled extensions without importing...
r38180 import ast
import collections
Jun Wu
wrapfunction: use functools.partial if possible...
r34089 import functools
Gregory Szorc
extensions: use absolute_import
r25946 import imp
Augie Fackler
extensions: use inspect module instead of func_code.co_argcount...
r31263 import inspect
Gregory Szorc
extensions: use absolute_import
r25946 import os
from .i18n import (
_,
gettext,
)
Gregory Szorc
py3: manually import pycompat.setattr where it is needed...
r43357 from .pycompat import (
Gregory Szorc
py3: manually import getattr where it is needed...
r43359 getattr,
Gregory Szorc
py3: manually import pycompat.setattr where it is needed...
r43357 open,
setattr,
)
Gregory Szorc
extensions: use absolute_import
r25946
from . import (
cmdutil,
configitems: add an official API for extensions to register config item...
r33127 configitems,
Gregory Szorc
extensions: use absolute_import
r25946 error,
Pulkit Goyal
py3: use pycompat.sysstr() in __import__()...
r30570 pycompat,
Gregory Szorc
extensions: use absolute_import
r25946 util,
)
Matt Mackall
Create a separate module for managing extensions
r4544
Augie Fackler
formatting: blacken the codebase...
r43346 from .utils import stringutil
Yuya Nishihara
stringutil: bulk-replace call sites to point to new module...
r37102
Matt Mackall
Create a separate module for managing extensions
r4544 _extensions = {}
liscju
help: show content for explicitly disabled extension (issue5228)
r29895 _disabledextensions = {}
Gregory Szorc
extensions: support callbacks after another extension loads...
r24065 _aftercallbacks = {}
Alexis S. L. Carvalho
Move cmdtable and reposetup handling out of extensions.py...
r5192 _order = []
Boris Feld
extensions: expand the builtins extensions declaration...
r33526 _builtin = {
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 b'hbisect',
b'bookmarks',
b'color',
b'parentrevspec',
b'progress',
b'interhg',
b'inotify',
b'hgcia',
b'shelve',
Boris Feld
extensions: expand the builtins extensions declaration...
r33526 }
Alexis S. L. Carvalho
Move cmdtable and reposetup handling out of extensions.py...
r5192
Augie Fackler
formatting: blacken the codebase...
r43346
FUJIWARA Katsunori
extensions: list up only enabled extensions, if "ui" is specified...
r19777 def extensions(ui=None):
if ui:
Augie Fackler
formatting: blacken the codebase...
r43346
FUJIWARA Katsunori
extensions: list up only enabled extensions, if "ui" is specified...
r19777 def enabled(name):
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 for format in [b'%s', b'hgext.%s']:
conf = ui.config(b'extensions', format % name)
if conf is not None and not conf.startswith(b'!'):
FUJIWARA Katsunori
extensions: list up only enabled extensions, if "ui" is specified...
r19777 return True
Augie Fackler
formatting: blacken the codebase...
r43346
FUJIWARA Katsunori
extensions: list up only enabled extensions, if "ui" is specified...
r19777 else:
enabled = lambda name: True
Alexis S. L. Carvalho
Move cmdtable and reposetup handling out of extensions.py...
r5192 for name in _order:
module = _extensions[name]
FUJIWARA Katsunori
extensions: list up only enabled extensions, if "ui" is specified...
r19777 if module and enabled(name):
Alexis S. L. Carvalho
Move cmdtable and reposetup handling out of extensions.py...
r5192 yield name, module
Matt Mackall
Create a separate module for managing extensions
r4544
Augie Fackler
formatting: blacken the codebase...
r43346
Matt Mackall
Create a separate module for managing extensions
r4544 def find(name):
'''return module with given extension name'''
Idan Kamara
extensions: raise when trying to find an extension that failed to load...
r14415 mod = None
Matt Mackall
Create a separate module for managing extensions
r4544 try:
timeless
cleanup: remove superfluous space after space after equals (python)
r27637 mod = _extensions[name]
Matt Mackall
Create a separate module for managing extensions
r4544 except KeyError:
Gregory Szorc
py3: finish porting iteritems() to pycompat and remove source transformer...
r43376 for k, v in pycompat.iteritems(_extensions):
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 if k.endswith(b'.' + name) or k.endswith(b'/' + name):
Idan Kamara
extensions: raise when trying to find an extension that failed to load...
r14415 mod = v
break
if not mod:
Matt Mackall
Create a separate module for managing extensions
r4544 raise KeyError(name)
Idan Kamara
extensions: raise when trying to find an extension that failed to load...
r14415 return mod
Matt Mackall
Create a separate module for managing extensions
r4544
Augie Fackler
formatting: blacken the codebase...
r43346
Alexander Solovyov
ability to load hooks from arbitrary python module
r7916 def loadpath(path, module_name):
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 module_name = module_name.replace(b'.', b'_')
Ed Morley
extensions: use normpath to allow trailing '\' on Windows (issue4187)...
r20645 path = util.normpath(util.expandpath(path))
Pulkit Goyal
py3: use pycompat.fsdecode() to pass to imp.* functions...
r30575 module_name = pycompat.fsdecode(module_name)
path = pycompat.fsdecode(path)
Alexander Solovyov
ability to load hooks from arbitrary python module
r7916 if os.path.isdir(path):
# module/__init__.py style
Ed Morley
extensions: use normpath to allow trailing '\' on Windows (issue4187)...
r20645 d, f = os.path.split(path)
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:
Simon Heimberg
hooks: print out more information when loading a python hook fails...
r17217 try:
return imp.load_source(module_name, path)
Gregory Szorc
global: mass rewrite to use modern exception syntax...
r25660 except IOError as exc:
Simon Heimberg
hooks: print out more information when loading a python hook fails...
r17217 if not exc.filename:
Augie Fackler
formatting: blacken the codebase...
r43346 exc.filename = path # python does not fill this
Simon Heimberg
hooks: print out more information when loading a python hook fails...
r17217 raise
Alexander Solovyov
ability to load hooks from arbitrary python module
r7916
Augie Fackler
formatting: blacken the codebase...
r43346
Pierre-Yves David
extensions: extract the 'importh' closure as normal function...
r28505 def _importh(name):
"""import and return the <name> module"""
Pulkit Goyal
py3: use pycompat.sysstr() in __import__()...
r30570 mod = __import__(pycompat.sysstr(name))
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 components = name.split(b'.')
Pierre-Yves David
extensions: extract the 'importh' closure as normal function...
r28505 for comp in components[1:]:
mod = getattr(mod, comp)
return mod
Augie Fackler
formatting: blacken the codebase...
r43346
Jun Wu
extensions: move the "import" logic out from "load"...
r30058 def _importext(name, path=None, reportfunc=None):
if path:
# the module will be loaded in sys.modules
# choose an unique name so that it doesn't
# conflicts with other modules
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 mod = loadpath(path, b'hgext.%s' % name)
Jun Wu
extensions: move the "import" logic out from "load"...
r30058 else:
try:
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 mod = _importh(b"hgext.%s" % name)
Jun Wu
extensions: move the "import" logic out from "load"...
r30058 except ImportError as err:
if reportfunc:
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 reportfunc(err, b"hgext.%s" % name, b"hgext3rd.%s" % name)
Jun Wu
extensions: move the "import" logic out from "load"...
r30058 try:
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 mod = _importh(b"hgext3rd.%s" % name)
Jun Wu
extensions: move the "import" logic out from "load"...
r30058 except ImportError as err:
if reportfunc:
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 reportfunc(err, b"hgext3rd.%s" % name, name)
Jun Wu
extensions: move the "import" logic out from "load"...
r30058 mod = _importh(name)
return mod
Augie Fackler
formatting: blacken the codebase...
r43346
Pierre-Yves David
extensions: factor import error reporting out...
r28506 def _reportimporterror(ui, err, failed, next):
Yuya Nishihara
extensions: use ui.log() interface to provide detailed loading information...
r41032 # note: this ui.log happens before --debug is processed,
Pierre-Yves David
extensions: add a note about debug output during extensions search...
r30028 # Use --config ui.debug=1 to see them.
Augie Fackler
formatting: blacken the codebase...
r43346 ui.log(
b'extension',
b' - could not import %s (%s): trying %s\n',
failed,
stringutil.forcebytestr(err),
next,
)
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 if ui.debugflag and ui.configbool(b'devel', b'debug.extensions'):
Yuya Nishihara
extensions: use ui.log() interface to provide detailed loading information...
r41032 ui.traceback()
Pierre-Yves David
extensions: factor import error reporting out...
r28506
Augie Fackler
formatting: blacken the codebase...
r43346
Yuya Nishihara
extensions: reject any unicode strings in tables before loading...
r36283 def _rejectunicode(name, xs):
if isinstance(xs, (list, set, tuple)):
for x in xs:
_rejectunicode(name, x)
elif isinstance(xs, dict):
for k, v in xs.items():
_rejectunicode(name, k)
Yuya Nishihara
stringutil: bulk-replace call sites to point to new module...
r37102 _rejectunicode(b'%s.%s' % (name, stringutil.forcebytestr(k)), v)
Yuya Nishihara
extensions: reject any unicode strings in tables before loading...
r36283 elif isinstance(xs, type(u'')):
Augie Fackler
formatting: blacken the codebase...
r43346 raise error.ProgrammingError(
b"unicode %r found in %s" % (xs, name),
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 hint=b"use b'' to make it byte string",
Augie Fackler
formatting: blacken the codebase...
r43346 )
Yuya Nishihara
extensions: reject any unicode strings in tables before loading...
r36283
Yuya Nishihara
extensions: prohibit registration of command without using @command (API)...
r32342 # attributes set by registrar.command
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 _cmdfuncattrs = (b'norepo', b'optionalrepo', b'inferrepo')
Yuya Nishihara
extensions: prohibit registration of command without using @command (API)...
r32342
Augie Fackler
formatting: blacken the codebase...
r43346
Yuya Nishihara
extensions: show deprecation warning for the use of cmdutil.command...
r32343 def _validatecmdtable(ui, cmdtable):
Yuya Nishihara
extensions: prohibit registration of command without using @command (API)...
r32342 """Check if extension commands have required attributes"""
Gregory Szorc
py3: finish porting iteritems() to pycompat and remove source transformer...
r43376 for c, e in pycompat.iteritems(cmdtable):
Yuya Nishihara
extensions: prohibit registration of command without using @command (API)...
r32342 f = e[0]
missing = [a for a in _cmdfuncattrs if not util.safehasattr(f, a)]
if not missing:
continue
raise error.ProgrammingError(
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 b'missing attributes: %s' % b', '.join(missing),
hint=b"use @command decorator to register '%s'" % c,
Augie Fackler
formatting: blacken the codebase...
r43346 )
Yuya Nishihara
extensions: prohibit registration of command without using @command (API)...
r32342
Yuya Nishihara
extensions: reject any unicode strings in tables before loading...
r36283 def _validatetables(ui, mod):
"""Sanity check for loadable tables provided by extension module"""
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 for t in [b'cmdtable', b'colortable', b'configtable']:
Yuya Nishihara
extensions: reject any unicode strings in tables before loading...
r36283 _rejectunicode(t, getattr(mod, t, {}))
Augie Fackler
formatting: blacken the codebase...
r43346 for t in [
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 b'filesetpredicate',
b'internalmerge',
b'revsetpredicate',
b'templatefilter',
b'templatefunc',
b'templatekeyword',
Augie Fackler
formatting: blacken the codebase...
r43346 ]:
Yuya Nishihara
extensions: reject any unicode strings in tables before loading...
r36283 o = getattr(mod, t, None)
if o:
_rejectunicode(t, o._table)
_validatecmdtable(ui, getattr(mod, 'cmdtable', {}))
Augie Fackler
formatting: blacken the codebase...
r43346
Yuya Nishihara
extensions: use ui.log() interface to provide detailed loading information...
r41032 def load(ui, name, path, loadingtime=None):
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 if name.startswith(b'hgext.') or name.startswith(b'hgext/'):
Bryan O'Sullivan
extensions: don't get confused by aliasing between "foo" and "hgext.foo"
r5031 shortname = name[6:]
else:
shortname = name
Bryan O'Sullivan
extensions: rename _ignore to _builtin, add descriptive comment...
r27111 if shortname in _builtin:
Matt Mackall
extensions: add an ignore list for old extensions
r13349 return None
Bryan O'Sullivan
extensions: don't get confused by aliasing between "foo" and "hgext.foo"
r5031 if shortname in _extensions:
Erik Zielke
extensions.load: return module...
r12779 return _extensions[shortname]
Yuya Nishihara
extensions: use ui.log() interface to provide detailed loading information...
r41032 ui.log(b'extension', b' - loading extension: %s\n', shortname)
Brendan Cully
Cache extension load failures....
r5087 _extensions[shortname] = None
Augie Fackler
cleanup: hgdemandimport.tracing accepts strings, not bytes...
r43532 with util.timedcm('load extension %s', shortname) as stats:
Martijn Pieters
extensions: add detailed loading information...
r38834 mod = _importext(name, path, bind(_reportimporterror, ui))
Yuya Nishihara
extensions: use ui.log() interface to provide detailed loading information...
r41032 ui.log(b'extension', b' > %s extension loaded in %s\n', shortname, stats)
Boris Feld
extension: add a summary of total loading time per extension...
r39547 if loadingtime is not None:
loadingtime[shortname] += stats.elapsed
Gregory Szorc
extensions: refuse to load extensions if minimum hg version not met...
r27142
# Before we do anything with the extension, check against minimum stated
# compatibility. This gives extension authors a mechanism to have their
# extensions short circuit when loaded with a known incompatible version
# of Mercurial.
minver = getattr(mod, 'minimumhgversion', None)
if minver and util.versiontuple(minver, 2) > util.versiontuple(n=2):
Augie Fackler
formatting: blacken the codebase...
r43346 msg = _(
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 b'(third party extension %s requires version %s or newer '
b'of Mercurial (current: %s); disabling)\n'
Augie Fackler
formatting: blacken the codebase...
r43346 )
Boris Feld
extensions: include current version in "invalid version" message...
r40499 ui.warn(msg % (shortname, minver, util.version()))
Gregory Szorc
extensions: refuse to load extensions if minimum hg version not met...
r27142 return
Yuya Nishihara
extensions: use ui.log() interface to provide detailed loading information...
r41032 ui.log(b'extension', b' - validating extension tables: %s\n', shortname)
Yuya Nishihara
extensions: reject any unicode strings in tables before loading...
r36283 _validatetables(ui, mod)
Gregory Szorc
extensions: refuse to load extensions if minimum hg version not met...
r27142
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)
Augie Fackler
formatting: blacken the codebase...
r43346 ui.log(
b'extension', b' - invoking registered callbacks: %s\n', shortname
)
Augie Fackler
cleanup: hgdemandimport.tracing accepts strings, not bytes...
r43532 with util.timedcm('callbacks extension %s', shortname) as stats:
Martijn Pieters
extensions: add detailed loading information...
r38834 for fn in _aftercallbacks.get(shortname, []):
fn(loaded=True)
Yuya Nishihara
extensions: use ui.log() interface to provide detailed loading information...
r41032 ui.log(b'extension', b' > callbacks completed in %s\n', stats)
Erik Zielke
extensions.load: return module...
r12779 return mod
Matt Mackall
Create a separate module for managing extensions
r4544
Augie Fackler
formatting: blacken the codebase...
r43346
Jun Wu
extensions: move uisetup and extsetup to standalone functions...
r29461 def _runuisetup(name, ui):
uisetup = getattr(_extensions[name], 'uisetup', None)
if uisetup:
Augie Fackler
extensions: catch uisetup and extsetup failures and don't let them break hg...
r32724 try:
uisetup(ui)
except Exception as inst:
Martin von Zweigbergk
extensions: always include traceback when extension setup fails...
r34846 ui.traceback(force=True)
Yuya Nishihara
stringutil: bulk-replace call sites to point to new module...
r37102 msg = stringutil.forcebytestr(inst)
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 ui.warn(_(b"*** failed to set up extension %s: %s\n") % (name, msg))
Augie Fackler
extensions: catch uisetup and extsetup failures and don't let them break hg...
r32724 return False
return True
Jun Wu
extensions: move uisetup and extsetup to standalone functions...
r29461
Augie Fackler
formatting: blacken the codebase...
r43346
Jun Wu
extensions: move uisetup and extsetup to standalone functions...
r29461 def _runextsetup(name, ui):
extsetup = getattr(_extensions[name], 'extsetup', None)
if extsetup:
try:
Matt Harbison
extensions: drop support for extsetup() without `ui` argument (API)
r42522 extsetup(ui)
Augie Fackler
extensions: catch uisetup and extsetup failures and don't let them break hg...
r32724 except Exception as inst:
Martin von Zweigbergk
extensions: always include traceback when extension setup fails...
r34846 ui.traceback(force=True)
Yuya Nishihara
stringutil: bulk-replace call sites to point to new module...
r37102 msg = stringutil.forcebytestr(inst)
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 ui.warn(_(b"*** failed to set up extension %s: %s\n") % (name, msg))
Augie Fackler
extensions: catch uisetup and extsetup failures and don't let them break hg...
r32724 return False
return True
Jun Wu
extensions: move uisetup and extsetup to standalone functions...
r29461
Augie Fackler
formatting: blacken the codebase...
r43346
Jun Wu
extensions: allow loading a whitelisted subset of extensions...
r32416 def loadall(ui, whitelist=None):
Boris Feld
extension: add a summary of total loading time per extension...
r39547 loadingtime = collections.defaultdict(int)
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 result = ui.configitems(b"extensions")
Jun Wu
profiling: allow loading profiling extension before everything else...
r32417 if whitelist is not None:
Jun Wu
extensions: allow loading a whitelisted subset of extensions...
r32416 result = [(k, v) for (k, v) in result if k in whitelist]
Martin Geisler
extensions: load and configure extensions in well-defined phases...
r9410 newindex = len(_order)
Augie Fackler
formatting: blacken the codebase...
r43346 ui.log(
b'extension',
b'loading %sextensions\n',
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 b'additional ' if newindex else b'',
Augie Fackler
formatting: blacken the codebase...
r43346 )
Yuya Nishihara
extensions: use ui.log() interface to provide detailed loading information...
r41032 ui.log(b'extension', b'- processing %d entries\n', len(result))
Augie Fackler
cleanup: hgdemandimport.tracing accepts strings, not bytes...
r43532 with util.timedcm('load all extensions') as stats:
Martijn Pieters
extensions: add detailed loading information...
r38834 for (name, path) in result:
Jesse Glick
When failing to load an extension, show where Hg tried to load it from.
r6204 if path:
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 if path[0:1] == b'!':
Martijn Pieters
extensions: add detailed loading information...
r38834 if name not in _disabledextensions:
Augie Fackler
formatting: blacken the codebase...
r43346 ui.log(
b'extension',
b' - skipping disabled extension: %s\n',
name,
)
Martijn Pieters
extensions: add detailed loading information...
r38834 _disabledextensions[name] = path[1:]
continue
try:
Yuya Nishihara
extensions: use ui.log() interface to provide detailed loading information...
r41032 load(ui, name, path, loadingtime)
Martijn Pieters
extensions: add detailed loading information...
r38834 except Exception as inst:
msg = stringutil.forcebytestr(inst)
if path:
Augie Fackler
formatting: blacken the codebase...
r43346 ui.warn(
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 _(b"*** failed to import extension %s from %s: %s\n")
Augie Fackler
formatting: blacken the codebase...
r43346 % (name, path, msg)
)
Martijn Pieters
extensions: add detailed loading information...
r38834 else:
Augie Fackler
formatting: blacken the codebase...
r43346 ui.warn(
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 _(b"*** failed to import extension %s: %s\n")
Augie Fackler
formatting: blacken the codebase...
r43346 % (name, msg)
)
Martijn Pieters
extensions: add detailed loading information...
r38834 if isinstance(inst, error.Hint) and inst.hint:
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 ui.warn(_(b"*** (%s)\n") % inst.hint)
Martijn Pieters
extensions: add detailed loading information...
r38834 ui.traceback()
Augie Fackler
formatting: blacken the codebase...
r43346 ui.log(
b'extension',
b'> loaded %d extensions, total time %s\n',
len(_order) - newindex,
stats,
)
Boris Feld
extensions: register config item early...
r34188 # list of (objname, loadermod, loadername) tuple:
# - objname is the name of an object in extension module,
# from which extra information is loaded
# - loadermod is the module where loader is placed
# - loadername is the name of the function,
# which takes (ui, extensionname, extraobj) arguments
#
# This one is for the list of item that must be run before running any setup
earlyextraloaders = [
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 (b'configtable', configitems, b'loadconfigtable'),
Boris Feld
extensions: register config item early...
r34188 ]
Martijn Pieters
extensions: add detailed loading information...
r38834
Yuya Nishihara
extensions: use ui.log() interface to provide detailed loading information...
r41032 ui.log(b'extension', b'- loading configtable attributes\n')
Boris Feld
extensions: register config item early...
r34188 _loadextra(ui, newindex, earlyextraloaders)
Matt Mackall
Create a separate module for managing extensions
r4544
Augie Fackler
extensions: catch uisetup and extsetup failures and don't let them break hg...
r32724 broken = set()
Yuya Nishihara
extensions: use ui.log() interface to provide detailed loading information...
r41032 ui.log(b'extension', b'- executing uisetup hooks\n')
Augie Fackler
cleanup: hgdemandimport.tracing accepts strings, not bytes...
r43532 with util.timedcm('all uisetup') as alluisetupstats:
Boris Feld
extensions: trace the total time of running all uisetup callbacks...
r39544 for name in _order[newindex:]:
Yuya Nishihara
extensions: use ui.log() interface to provide detailed loading information...
r41032 ui.log(b'extension', b' - running uisetup for %s\n', name)
Augie Fackler
cleanup: hgdemandimport.tracing accepts strings, not bytes...
r43532 with util.timedcm('uisetup %s', name) as stats:
Boris Feld
extensions: trace the total time of running all uisetup callbacks...
r39544 if not _runuisetup(name, ui):
Augie Fackler
formatting: blacken the codebase...
r43346 ui.log(
b'extension',
b' - the %s extension uisetup failed\n',
name,
)
Boris Feld
extensions: trace the total time of running all uisetup callbacks...
r39544 broken.add(name)
Yuya Nishihara
extensions: use ui.log() interface to provide detailed loading information...
r41032 ui.log(b'extension', b' > uisetup for %s took %s\n', name, stats)
Boris Feld
extension: add a summary of total loading time per extension...
r39547 loadingtime[name] += stats.elapsed
Yuya Nishihara
extensions: use ui.log() interface to provide detailed loading information...
r41032 ui.log(b'extension', b'> all uisetup took %s\n', alluisetupstats)
Martin Geisler
extensions: load and configure extensions in well-defined phases...
r9410
Yuya Nishihara
extensions: use ui.log() interface to provide detailed loading information...
r41032 ui.log(b'extension', b'- executing extsetup hooks\n')
Augie Fackler
cleanup: hgdemandimport.tracing accepts strings, not bytes...
r43532 with util.timedcm('all extsetup') as allextetupstats:
Boris Feld
extensions: trace the total time of running all extsetup callbacks...
r39545 for name in _order[newindex:]:
if name in broken:
continue
Yuya Nishihara
extensions: use ui.log() interface to provide detailed loading information...
r41032 ui.log(b'extension', b' - running extsetup for %s\n', name)
Augie Fackler
cleanup: hgdemandimport.tracing accepts strings, not bytes...
r43532 with util.timedcm('extsetup %s', name) as stats:
Boris Feld
extensions: trace the total time of running all extsetup callbacks...
r39545 if not _runextsetup(name, ui):
Augie Fackler
formatting: blacken the codebase...
r43346 ui.log(
b'extension',
b' - the %s extension extsetup failed\n',
name,
)
Boris Feld
extensions: trace the total time of running all extsetup callbacks...
r39545 broken.add(name)
Yuya Nishihara
extensions: use ui.log() interface to provide detailed loading information...
r41032 ui.log(b'extension', b' > extsetup for %s took %s\n', name, stats)
Boris Feld
extension: add a summary of total loading time per extension...
r39547 loadingtime[name] += stats.elapsed
Yuya Nishihara
extensions: use ui.log() interface to provide detailed loading information...
r41032 ui.log(b'extension', b'> all extsetup took %s\n', allextetupstats)
Augie Fackler
extensions: catch uisetup and extsetup failures and don't let them break hg...
r32724
for name in broken:
Yuya Nishihara
extensions: use ui.log() interface to provide detailed loading information...
r41032 ui.log(b'extension', b' - disabling broken %s extension\n', name)
Augie Fackler
extensions: catch uisetup and extsetup failures and don't let them break hg...
r32724 _extensions[name] = None
Yuya Nishihara
extensions: changed to call extsetup() from extensions.loadall()...
r9660
Gregory Szorc
extensions: support callbacks after another extension loads...
r24065 # Call aftercallbacks that were never met.
Yuya Nishihara
extensions: use ui.log() interface to provide detailed loading information...
r41032 ui.log(b'extension', b'- executing remaining aftercallbacks\n')
Augie Fackler
cleanup: hgdemandimport.tracing accepts strings, not bytes...
r43532 with util.timedcm('aftercallbacks') as stats:
Martijn Pieters
extensions: add detailed loading information...
r38834 for shortname in _aftercallbacks:
if shortname in _extensions:
continue
Gregory Szorc
extensions: support callbacks after another extension loads...
r24065
Martijn Pieters
extensions: add detailed loading information...
r38834 for fn in _aftercallbacks[shortname]:
Augie Fackler
formatting: blacken the codebase...
r43346 ui.log(
b'extension',
b' - extension %s not loaded, notify callbacks\n',
shortname,
)
Martijn Pieters
extensions: add detailed loading information...
r38834 fn(loaded=False)
Yuya Nishihara
extensions: use ui.log() interface to provide detailed loading information...
r41032 ui.log(b'extension', b'> remaining aftercallbacks completed in %s\n', stats)
Gregory Szorc
extensions: support callbacks after another extension loads...
r24065
Gregory Szorc
extensions: clear aftercallbacks after execution (issue4646)...
r24950 # loadall() is called multiple times and lingering _aftercallbacks
# entries could result in double execution. See issue4646.
_aftercallbacks.clear()
FUJIWARA Katsunori
extensions: register functions always at loading extension (issue5601)...
r33052 # delay importing avoids cyclic dependency (especially commands)
from . import (
color,
commands,
FUJIWARA Katsunori
filemerge: move decorator definition for internal merge tools to registrar...
r33663 filemerge,
FUJIWARA Katsunori
extensions: register functions always at loading extension (issue5601)...
r33052 fileset,
revset,
templatefilters,
Yuya Nishihara
templater: split template functions to new module...
r36940 templatefuncs,
FUJIWARA Katsunori
extensions: register functions always at loading extension (issue5601)...
r33052 templatekw,
)
# list of (objname, loadermod, loadername) tuple:
# - objname is the name of an object in extension module,
# from which extra information is loaded
# - loadermod is the module where loader is placed
# - loadername is the name of the function,
# which takes (ui, extensionname, extraobj) arguments
Yuya Nishihara
extensions: use ui.log() interface to provide detailed loading information...
r41032 ui.log(b'extension', b'- loading extension registration objects\n')
FUJIWARA Katsunori
extensions: register functions always at loading extension (issue5601)...
r33052 extraloaders = [
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 (b'cmdtable', commands, b'loadcmdtable'),
(b'colortable', color, b'loadcolortable'),
(b'filesetpredicate', fileset, b'loadpredicate'),
(b'internalmerge', filemerge, b'loadinternalmerge'),
(b'revsetpredicate', revset, b'loadpredicate'),
(b'templatefilter', templatefilters, b'loadfilter'),
(b'templatefunc', templatefuncs, b'loadfunction'),
(b'templatekeyword', templatekw, b'loadkeyword'),
FUJIWARA Katsunori
extensions: register functions always at loading extension (issue5601)...
r33052 ]
Augie Fackler
cleanup: hgdemandimport.tracing accepts strings, not bytes...
r43532 with util.timedcm('load registration objects') as stats:
Martijn Pieters
extensions: add detailed loading information...
r38834 _loadextra(ui, newindex, extraloaders)
Augie Fackler
formatting: blacken the codebase...
r43346 ui.log(
b'extension',
b'> extension registration object loading took %s\n',
stats,
)
Boris Feld
extension: add a summary of total loading time per extension...
r39547
# Report per extension loading time (except reposetup)
for name in sorted(loadingtime):
Augie Fackler
formatting: blacken the codebase...
r43346 ui.log(
b'extension',
b'> extension %s take a total of %s to load\n',
name,
util.timecount(loadingtime[name]),
)
Boris Feld
extension: add a summary of total loading time per extension...
r39547
Yuya Nishihara
extensions: use ui.log() interface to provide detailed loading information...
r41032 ui.log(b'extension', b'extension loading complete\n')
FUJIWARA Katsunori
extensions: register functions always at loading extension (issue5601)...
r33052
Augie Fackler
formatting: blacken the codebase...
r43346
Boris Feld
extensions: factor extra data loading out...
r34187 def _loadextra(ui, newindex, extraloaders):
FUJIWARA Katsunori
extensions: register functions always at loading extension (issue5601)...
r33052 for name in _order[newindex:]:
module = _extensions[name]
if not module:
Augie Fackler
formatting: blacken the codebase...
r43346 continue # loading this module failed
FUJIWARA Katsunori
extensions: register functions always at loading extension (issue5601)...
r33052
for objname, loadermod, loadername in extraloaders:
extraobj = getattr(module, objname, None)
if extraobj is not None:
getattr(loadermod, loadername)(ui, name, extraobj)
Augie Fackler
formatting: blacken the codebase...
r43346
Gregory Szorc
extensions: support callbacks after another extension loads...
r24065 def afterloaded(extension, callback):
'''Run the specified function after a named extension is loaded.
If the named extension is already loaded, the callback will be called
immediately.
If the named extension never loads, the callback will be called after
all extensions have been loaded.
The callback receives the named argument ``loaded``, which is a boolean
indicating whether the dependent extension actually loaded.
'''
if extension in _extensions:
Adam Simpkins
extensions: call afterloaded() with loaded=False for disabled extensions...
r33014 # Report loaded as False if the extension is disabled
Augie Fackler
formatting: blacken the codebase...
r43346 loaded = _extensions[extension] is not None
Adam Simpkins
extensions: call afterloaded() with loaded=False for disabled extensions...
r33014 callback(loaded=loaded)
Gregory Szorc
extensions: support callbacks after another extension loads...
r24065 else:
_aftercallbacks.setdefault(extension, []).append(callback)
Augie Fackler
formatting: blacken the codebase...
r43346
Yuya Nishihara
extensions: add "uipopulate" hook, called per instance, not per process...
r40760 def populateui(ui):
"""Run extension hooks on the given ui to populate additional members,
extend the class dynamically, etc.
This will be called after the configuration is loaded, and/or extensions
are loaded. In general, it's once per ui instance, but in command-server
and hgweb, this may be called more than once with the same ui.
"""
for name, mod in extensions(ui):
hook = getattr(mod, 'uipopulate', None)
if not hook:
continue
try:
hook(ui)
except Exception as inst:
ui.traceback(force=True)
Augie Fackler
formatting: blacken the codebase...
r43346 ui.warn(
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 _(b'*** failed to populate ui by extension %s: %s\n')
Augie Fackler
formatting: blacken the codebase...
r43346 % (name, stringutil.forcebytestr(inst))
)
Yuya Nishihara
extensions: add "uipopulate" hook, called per instance, not per process...
r40760
Eric Sumner
extensions: extract partial application into a bind() function...
r24734 def bind(func, *args):
'''Partial function application
Returns a new function that is the partial application of args and kwargs
to func. For example,
f(1, 2, bar=3) === bind(f, 1)(2, bar=3)'''
assert callable(func)
Augie Fackler
formatting: blacken the codebase...
r43346
Eric Sumner
extensions: extract partial application into a bind() function...
r24734 def closure(*a, **kw):
return func(*(args + a), **kw)
Augie Fackler
formatting: blacken the codebase...
r43346
Eric Sumner
extensions: extract partial application into a bind() function...
r24734 return closure
Augie Fackler
formatting: blacken the codebase...
r43346
Jun Wu
extensions: set attributes to wrappers so we can trace them back...
r29763 def _updatewrapper(wrap, origfn, unboundwrapper):
'''Copy and add some useful attributes to wrapper'''
Yuya Nishihara
extensions: fix wrapcommand/function of class instance...
r34130 try:
wrap.__name__ = origfn.__name__
except AttributeError:
pass
Yuya Nishihara
extensions: extract function that copies function attributes to wrapper...
r28310 wrap.__module__ = getattr(origfn, '__module__')
wrap.__doc__ = getattr(origfn, '__doc__')
Yuya Nishihara
extensions: copy extra __dict__ of original function...
r28312 wrap.__dict__.update(getattr(origfn, '__dict__', {}))
Jun Wu
extensions: set attributes to wrappers so we can trace them back...
r29763 wrap._origfunc = origfn
wrap._unboundwrapper = unboundwrapper
Yuya Nishihara
extensions: extract function that copies function attributes to wrapper...
r28310
Augie Fackler
formatting: blacken the codebase...
r43346
Ryan McElroy
extensions: allow extending command synopsis and docstring...
r24124 def wrapcommand(table, command, wrapper, synopsis=None, docstring=None):
Dan Villiom Podlaski Christiansen
extensions: add docstring for wrapcommand().
r11519 '''Wrap the command named `command' in table
Replace command in the command table with wrapper. The wrapped command will
be inserted into the command table specified by the table argument.
The wrapper will be called like
wrapper(orig, *args, **kwargs)
where orig is the original (wrapped) function, and *args, **kwargs
are the arguments passed to it.
Ryan McElroy
extensions: allow extending command synopsis and docstring...
r24124
Optionally append to the command synopsis and docstring, used for help.
For example, if your extension wraps the ``bookmarks`` command to add the
flags ``--remote`` and ``--all`` you might call this function like so:
synopsis = ' [-a] [--remote]'
docstring = """
The ``remotenames`` extension adds the ``--remote`` and ``--all`` (``-a``)
flags to the bookmarks command. Either flag will show the remote bookmarks
Mads Kiilerich
spelling: trivial spell checking
r26781 known to the repository; ``--remote`` will also suppress the output of the
Ryan McElroy
extensions: allow extending command synopsis and docstring...
r24124 local bookmarks.
"""
extensions.wrapcommand(commands.table, 'bookmarks', exbookmarks,
synopsis, docstring)
Dan Villiom Podlaski Christiansen
extensions: add docstring for wrapcommand().
r11519 '''
Augie Fackler
extensions: restore use of callable() since it was readded in Python 3.2
r21795 assert callable(wrapper)
Matt Mackall
extensions: add wrapping functions
r7215 aliases, entry = cmdutil.findcmd(command, table)
Gregory Szorc
py3: finish porting iteritems() to pycompat and remove source transformer...
r43376 for alias, e in pycompat.iteritems(table):
Matt Mackall
extensions: add wrapping functions
r7215 if e is entry:
key = alias
break
origfn = entry[0]
Augie Fackler
formatting: blacken the codebase...
r43346 wrap = functools.partial(
util.checksignature(wrapper), util.checksignature(origfn)
)
Jun Wu
extensions: set attributes to wrappers so we can trace them back...
r29763 _updatewrapper(wrap, origfn, wrapper)
Ryan McElroy
extensions: allow extending command synopsis and docstring...
r24124 if docstring is not None:
Yuya Nishihara
extensions: extract function that copies function attributes to wrapper...
r28310 wrap.__doc__ += docstring
Ryan McElroy
extensions: allow extending command synopsis and docstring...
r24124
Matt Mackall
extensions: add wrapping functions
r7215 newentry = list(entry)
newentry[0] = wrap
Ryan McElroy
extensions: allow extending command synopsis and docstring...
r24124 if synopsis is not None:
newentry[2] += synopsis
Matt Mackall
extensions: add wrapping functions
r7215 table[key] = tuple(newentry)
return entry
Augie Fackler
formatting: blacken the codebase...
r43346
Augie Fackler
extensions: move wrapfilecache function from fsmonitor...
r32722 def wrapfilecache(cls, propname, wrapper):
"""Wraps a filecache property.
These can't be wrapped using the normal wrapfunction.
"""
Augie Fackler
extensions: if on py3 and propname is a bytestr, convert to sysstr...
r33835 propname = pycompat.sysstr(propname)
Augie Fackler
extensions: move wrapfilecache function from fsmonitor...
r32722 assert callable(wrapper)
for currcls in cls.__mro__:
if propname in currcls.__dict__:
origfn = currcls.__dict__[propname].func
assert callable(origfn)
Augie Fackler
formatting: blacken the codebase...
r43346
Augie Fackler
extensions: move wrapfilecache function from fsmonitor...
r32722 def wrap(*args, **kwargs):
return wrapper(origfn, *args, **kwargs)
Augie Fackler
formatting: blacken the codebase...
r43346
Augie Fackler
extensions: move wrapfilecache function from fsmonitor...
r32722 currcls.__dict__[propname].func = wrap
break
if currcls is object:
Augie Fackler
formatting: blacken the codebase...
r43346 raise AttributeError(
r"type '%s' has no property '%s'" % (cls, propname)
)
Augie Fackler
extensions: move wrapfilecache function from fsmonitor...
r32722
Martin von Zweigbergk
extensions: add wrappedfunction() context manager...
r34016 class wrappedfunction(object):
'''context manager for temporarily wrapping a function'''
def __init__(self, container, funcname, wrapper):
assert callable(wrapper)
self._container = container
self._funcname = funcname
self._wrapper = wrapper
def __enter__(self):
wrapfunction(self._container, self._funcname, self._wrapper)
def __exit__(self, exctype, excvalue, traceback):
unwrapfunction(self._container, self._funcname, self._wrapper)
Augie Fackler
formatting: blacken the codebase...
r43346
Matt Mackall
extensions: add wrapping functions
r7215 def wrapfunction(container, funcname, wrapper):
Greg Ward
extensions: recommend against using wrapfunction for repo methods...
r11402 '''Wrap the function named funcname in container
Dan Villiom Podlaski Christiansen
extensions: improve language for wrapfunction() docstring.
r11520 Replace the funcname member in the given container with the specified
wrapper. The container is typically a module, class, or instance.
Greg Ward
extensions: recommend against using wrapfunction for repo methods...
r11402
The wrapper will be called like
wrapper(orig, *args, **kwargs)
where orig is the original (wrapped) function, and *args, **kwargs
are the arguments passed to it.
Wrapping methods of the repository object is not recommended since
it conflicts with extensions that extend the repository by
subclassing. All extensions that need to extend methods of
localrepository should use this subclassing trick: namely,
reposetup() should look like
def reposetup(ui, repo):
class myrepo(repo.__class__):
def whatever(self, *args, **kwargs):
[...extension stuff...]
super(myrepo, self).whatever(*args, **kwargs)
[...extension stuff...]
repo.__class__ = myrepo
In general, combining wrapfunction() with subclassing does not
work. Since you cannot control what other extensions are loaded by
your end users, you should play nicely with others by using the
subclass trick.
'''
Augie Fackler
extensions: restore use of callable() since it was readded in Python 3.2
r21795 assert callable(wrapper)
Matt Mackall
extensions: add wrapping functions
r7215
origfn = getattr(container, funcname)
Augie Fackler
extensions: restore use of callable() since it was readded in Python 3.2
r21795 assert callable(origfn)
Jun Wu
wrapfunction: use functools.partial if possible...
r34089 if inspect.ismodule(container):
# origfn is not an instance or class method. "partial" can be used.
# "partial" won't insert a frame in traceback.
wrap = functools.partial(wrapper, origfn)
else:
# "partial" cannot be safely used. Emulate its effect by using "bind".
# The downside is one more frame in traceback.
wrap = bind(wrapper, origfn)
Jun Wu
extensions: set attributes to wrappers so we can trace them back...
r29763 _updatewrapper(wrap, origfn, wrapper)
Yuya Nishihara
extensions: copy attributes to wrapper by wrapfunction()...
r28311 setattr(container, funcname, wrap)
Matt Mackall
extensions: add wrapping functions
r7215 return origfn
Cédric Duval
extensions: move extensions listing functions from mercurial.help...
r8871
Augie Fackler
formatting: blacken the codebase...
r43346
Jun Wu
extensions: add unwrapfunction to undo wrapfunction...
r29765 def unwrapfunction(container, funcname, wrapper=None):
'''undo wrapfunction
If wrappers is None, undo the last wrap. Otherwise removes the wrapper
from the chain of wrappers.
Return the removed wrapper.
Raise IndexError if wrapper is None and nothing to unwrap; ValueError if
wrapper is not None but is not found in the wrapper chain.
'''
chain = getwrapperchain(container, funcname)
origfn = chain.pop()
if wrapper is None:
wrapper = chain[0]
chain.remove(wrapper)
setattr(container, funcname, origfn)
for w in reversed(chain):
wrapfunction(container, funcname, w)
return wrapper
Augie Fackler
formatting: blacken the codebase...
r43346
Jun Wu
extensions: add getwrapperchain to get a list of wrappers...
r29764 def getwrapperchain(container, funcname):
'''get a chain of wrappers of a function
Return a list of functions: [newest wrapper, ..., oldest wrapper, origfunc]
The wrapper functions are the ones passed to wrapfunction, whose first
argument is origfunc.
'''
result = []
fn = getattr(container, funcname)
while fn:
assert callable(fn)
result.append(getattr(fn, '_unboundwrapper', fn))
fn = getattr(fn, '_origfunc', None)
return result
Augie Fackler
formatting: blacken the codebase...
r43346
Yuya Nishihara
extensions: remove strip_init=True from _disabledpaths()...
r38181 def _disabledpaths():
'''find paths of disabled extensions. returns a dict of {name: path}'''
Dirkjan Ochtman
extensions: simplify by selecting primary hgext
r8872 import hgext
Augie Fackler
formatting: blacken the codebase...
r43346
Pulkit Goyal
py3: use pycompat.fsencode() to convert __file__ to bytes...
r31074 extpath = os.path.dirname(
Augie Fackler
formatting: blacken the codebase...
r43346 os.path.abspath(pycompat.fsencode(hgext.__file__))
)
try: # might not be a filesystem path
Cédric Duval
extensions: catch OSError when hgext is not accessible (issue1708)...
r8964 files = os.listdir(extpath)
except OSError:
Brodie Rao
extensions: refactor disabled()
r10363 return {}
Cédric Duval
extensions: catch OSError when hgext is not accessible (issue1708)...
r8964
Cédric Duval
extensions: move extensions listing functions from mercurial.help...
r8871 exts = {}
Cédric Duval
extensions: catch OSError when hgext is not accessible (issue1708)...
r8964 for e in files:
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 if e.endswith(b'.py'):
name = e.rsplit(b'.', 1)[0]
Dirkjan Ochtman
extensions: simplify by selecting primary hgext
r8872 path = os.path.join(extpath, e)
else:
name = e
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 path = os.path.join(extpath, e, b'__init__.py')
Cédric Duval
extensions: check for path existence only when necessary
r8877 if not os.path.exists(path):
continue
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 if name in exts or name in _order or name == b'__init__':
Brodie Rao
extensions: refactor disabled()
r10363 continue
exts[name] = path
Gregory Szorc
py3: finish porting iteritems() to pycompat and remove source transformer...
r43376 for name, path in pycompat.iteritems(_disabledextensions):
Martin von Zweigbergk
dispatch: fix typo suggestion for disabled extension...
r33327 # If no path was provided for a disabled extension (e.g. "color=!"),
# don't replace the path we already found by the scan above.
if path:
exts[name] = path
Brodie Rao
extensions: refactor disabled()
r10363 return exts
Dirkjan Ochtman
extensions: simplify by selecting primary hgext
r8872
Augie Fackler
formatting: blacken the codebase...
r43346
Matt Mackall
extensions: move moduledoc to break import loop with help
r14317 def _moduledoc(file):
'''return the top-level python documentation for the given file
Loosely inspired by pydoc.source_synopsis(), but rewritten to
handle triple quotes and to return the whole text instead of just
the synopsis'''
result = []
line = file.readline()
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 while line[:1] == b'#' or not line.strip():
Matt Mackall
extensions: move moduledoc to break import loop with help
r14317 line = file.readline()
if not line:
break
start = line[:3]
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 if start == b'"""' or start == b"'''":
Matt Mackall
extensions: move moduledoc to break import loop with help
r14317 line = line[3:]
while line:
if line.rstrip().endswith(start):
line = line.split(start)[0]
if line:
result.append(line)
break
elif not line:
Augie Fackler
formatting: blacken the codebase...
r43346 return None # unmatched delimiter
Matt Mackall
extensions: move moduledoc to break import loop with help
r14317 result.append(line)
line = file.readline()
else:
return None
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 return b''.join(result)
Matt Mackall
extensions: move moduledoc to break import loop with help
r14317
Augie Fackler
formatting: blacken the codebase...
r43346
Brodie Rao
extensions: refactor disabled()
r10363 def _disabledhelp(path):
'''retrieve help synopsis of a disabled extension (without importing)'''
try:
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 with open(path, b'rb') as src:
Yuya Nishihara
extensions: use context manger for open()
r38363 doc = _moduledoc(src)
Brodie Rao
extensions: refactor disabled()
r10363 except IOError:
return
Augie Fackler
formatting: blacken the codebase...
r43346 if doc: # extracting localized synopsis
Pulkit Goyal
help: show help for disabled extensions (issue5228)...
r30306 return gettext(doc)
Brodie Rao
extensions: refactor disabled()
r10363 else:
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 return _(b'(no help text available)')
Brodie Rao
extensions: refactor disabled()
r10363
Augie Fackler
formatting: blacken the codebase...
r43346
Brodie Rao
extensions: refactor disabled()
r10363 def disabled():
Yuya Nishihara
extensions: update doc of enabled() and disabled() according to d5b525697ddb
r14530 '''find disabled extensions from hgext. returns a dict of {name: desc}'''
Yuya Nishihara
extensions: make disabled()/disabledext() load prebuilt index if available...
r14539 try:
from hgext import __index__
Augie Fackler
formatting: blacken the codebase...
r43346
return dict(
(name, gettext(desc))
Gregory Szorc
py3: finish porting iteritems() to pycompat and remove source transformer...
r43376 for name, desc in pycompat.iteritems(__index__.docs)
Augie Fackler
formatting: blacken the codebase...
r43346 if name not in _order
)
Thomas Arendsen Hein
setup.py, make: avoid problems with outdated, existing hgext/__index__.py*...
r21229 except (ImportError, AttributeError):
Yuya Nishihara
extensions: make disabled()/disabledext() load prebuilt index if available...
r14539 pass
Brodie Rao
extensions: refactor disabled()
r10363 paths = _disabledpaths()
if not paths:
Augie Fackler
extensions.disabled: return {} instead of None no extensions are disabled
r16709 return {}
Brodie Rao
extensions: refactor disabled()
r10363
exts = {}
Gregory Szorc
py3: finish porting iteritems() to pycompat and remove source transformer...
r43376 for name, path in pycompat.iteritems(paths):
Brodie Rao
extensions: refactor disabled()
r10363 doc = _disabledhelp(path)
Matt Mackall
extensions: drop maxlength from enabled and disabled...
r14316 if doc:
Pulkit Goyal
help: show help for disabled extensions (issue5228)...
r30306 exts[name] = doc.splitlines()[0]
Cédric Duval
extensions: move extensions listing functions from mercurial.help...
r8871
Matt Mackall
extensions: drop maxlength from enabled and disabled...
r14316 return exts
Cédric Duval
extensions: move extensions listing functions from mercurial.help...
r8871
Augie Fackler
formatting: blacken the codebase...
r43346
Brodie Rao
dispatch: provide help for disabled extensions and commands...
r10364 def disabledext(name):
'''find a specific disabled extension from hgext. returns desc'''
Yuya Nishihara
extensions: make disabled()/disabledext() load prebuilt index if available...
r14539 try:
from hgext import __index__
Augie Fackler
formatting: blacken the codebase...
r43346
Yuya Nishihara
extensions: make disabled()/disabledext() load prebuilt index if available...
r14539 if name in _order: # enabled
return
else:
return gettext(__index__.docs.get(name))
Thomas Arendsen Hein
setup.py, make: avoid problems with outdated, existing hgext/__index__.py*...
r21229 except (ImportError, AttributeError):
Yuya Nishihara
extensions: make disabled()/disabledext() load prebuilt index if available...
r14539 pass
Brodie Rao
dispatch: provide help for disabled extensions and commands...
r10364 paths = _disabledpaths()
if name in paths:
return _disabledhelp(paths[name])
Augie Fackler
formatting: blacken the codebase...
r43346
Yuya Nishihara
extensions: peek command table of disabled extensions without importing...
r38180 def _walkcommand(node):
"""Scan @command() decorators in the tree starting at node"""
todo = collections.deque([node])
while todo:
node = todo.popleft()
if not isinstance(node, ast.FunctionDef):
todo.extend(ast.iter_child_nodes(node))
continue
for d in node.decorator_list:
if not isinstance(d, ast.Call):
continue
if not isinstance(d.func, ast.Name):
continue
if d.func.id != r'command':
continue
yield d
Augie Fackler
formatting: blacken the codebase...
r43346
Yuya Nishihara
extensions: peek command table of disabled extensions without importing...
r38180 def _disabledcmdtable(path):
"""Construct a dummy command table without loading the extension module
This may raise IOError or SyntaxError.
"""
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 with open(path, b'rb') as src:
Yuya Nishihara
extensions: peek command table of disabled extensions without importing...
r38180 root = ast.parse(src.read(), path)
cmdtable = {}
for node in _walkcommand(root):
if not node.args:
continue
a = node.args[0]
if isinstance(a, ast.Str):
name = pycompat.sysbytes(a.s)
elif pycompat.ispy3 and isinstance(a, ast.Bytes):
name = a.s
else:
continue
cmdtable[name] = (None, [], b'')
return cmdtable
Augie Fackler
formatting: blacken the codebase...
r43346
Yuya Nishihara
extensions: extract closure that looks for commands from disabled module...
r37995 def _finddisabledcmd(ui, cmd, name, path, strict):
try:
Yuya Nishihara
extensions: peek command table of disabled extensions without importing...
r38180 cmdtable = _disabledcmdtable(path)
except (IOError, SyntaxError):
Yuya Nishihara
extensions: extract closure that looks for commands from disabled module...
r37995 return
try:
Yuya Nishihara
extensions: peek command table of disabled extensions without importing...
r38180 aliases, entry = cmdutil.findcmd(cmd, cmdtable, strict)
Yuya Nishihara
extensions: extract closure that looks for commands from disabled module...
r37995 except (error.AmbiguousCommand, error.UnknownCommand):
return
for c in aliases:
if c.startswith(cmd):
cmd = c
break
else:
cmd = aliases[0]
Yuya Nishihara
extensions: peek command table of disabled extensions without importing...
r38180 doc = _disabledhelp(path)
Yuya Nishihara
help: load module doc of disabled extension in extensions.disabledcmd()...
r37996 return (cmd, name, doc)
Yuya Nishihara
extensions: extract closure that looks for commands from disabled module...
r37995
Augie Fackler
formatting: blacken the codebase...
r43346
Mads Kiilerich
extensions: warn about invalid extensions when listing disabled commands...
r13191 def disabledcmd(ui, cmd, strict=False):
Yuya Nishihara
extensions: peek command table of disabled extensions without importing...
r38180 '''find cmd from disabled extensions without importing.
Yuya Nishihara
help: load module doc of disabled extension in extensions.disabledcmd()...
r37996 returns (cmdname, extname, doc)'''
Brodie Rao
dispatch: provide help for disabled extensions and commands...
r10364
Yuya Nishihara
extensions: peek command table of disabled extensions without importing...
r38180 paths = _disabledpaths()
Brodie Rao
dispatch: provide help for disabled extensions and commands...
r10364 if not paths:
raise error.UnknownCommand(cmd)
Martin Geisler
extensions: don't suggest commands from deprecated extensions
r16667 ext = None
Brodie Rao
dispatch: provide help for disabled extensions and commands...
r10364 # first, search for an extension with the same name as the command
path = paths.pop(cmd, None)
if path:
Yuya Nishihara
extensions: extract closure that looks for commands from disabled module...
r37995 ext = _finddisabledcmd(ui, cmd, cmd, path, strict=strict)
Martin Geisler
extensions: don't suggest commands from deprecated extensions
r16667 if not ext:
# otherwise, interrogate each extension until there's a match
Gregory Szorc
py3: finish porting iteritems() to pycompat and remove source transformer...
r43376 for name, path in pycompat.iteritems(paths):
Yuya Nishihara
extensions: extract closure that looks for commands from disabled module...
r37995 ext = _finddisabledcmd(ui, cmd, name, path, strict=strict)
Martin Geisler
extensions: don't suggest commands from deprecated extensions
r16667 if ext:
break
Yuya Nishihara
extensions: drop dead code trying to exclude deprecated disabled commands...
r37994 if ext:
Martin Geisler
extensions: don't suggest commands from deprecated extensions
r16667 return ext
Brodie Rao
dispatch: provide help for disabled extensions and commands...
r10364
raise error.UnknownCommand(cmd)
Augie Fackler
formatting: blacken the codebase...
r43346
FUJIWARA Katsunori
help: use full name of extensions to look up them for keyword search...
r19769 def enabled(shortname=True):
Yuya Nishihara
extensions: update doc of enabled() and disabled() according to d5b525697ddb
r14530 '''return a dict of {name: desc} of extensions'''
Cédric Duval
extensions: move extensions listing functions from mercurial.help...
r8871 exts = {}
for ename, ext in extensions():
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 doc = gettext(ext.__doc__) or _(b'(no help text available)')
FUJIWARA Katsunori
help: use full name of extensions to look up them for keyword search...
r19769 if shortname:
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 ename = ename.split(b'.')[-1]
Nicolas Dumazet
for calls expecting bool args, pass bool instead of int...
r9136 exts[ename] = doc.splitlines()[0].strip()
Cédric Duval
extensions: move extensions listing functions from mercurial.help...
r8871
Matt Mackall
extensions: drop maxlength from enabled and disabled...
r14316 return exts
anatoly techtonik
version: show enabled extensions (issue4209)...
r21848
Augie Fackler
formatting: blacken the codebase...
r43346
Jun Wu
extensions: add notloaded method to return extensions failed to load...
r28155 def notloaded():
'''return short names of extensions that failed to load'''
Gregory Szorc
py3: finish porting iteritems() to pycompat and remove source transformer...
r43376 return [
name for name, mod in pycompat.iteritems(_extensions) if mod is None
]
Jun Wu
extensions: add notloaded method to return extensions failed to load...
r28155
Augie Fackler
formatting: blacken the codebase...
r43346
anatoly techtonik
version: show enabled extensions (issue4209)...
r21848 def moduleversion(module):
'''return version information from given module as a string'''
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 if util.safehasattr(module, b'getversion') and callable(module.getversion):
anatoly techtonik
version: show enabled extensions (issue4209)...
r21848 version = module.getversion()
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 elif util.safehasattr(module, b'__version__'):
anatoly techtonik
version: show enabled extensions (issue4209)...
r21848 version = module.__version__
else:
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 version = b''
anatoly techtonik
version: show enabled extensions (issue4209)...
r21848 if isinstance(version, (list, tuple)):
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 version = b'.'.join(pycompat.bytestr(o) for o in version)
anatoly techtonik
version: show enabled extensions (issue4209)...
r21848 return version
liscju
version: verbose list internal and external extension source (issue4731)
r27990
Augie Fackler
formatting: blacken the codebase...
r43346
liscju
version: verbose list internal and external extension source (issue4731)
r27990 def ismoduleinternal(module):
exttestedwith = getattr(module, 'testedwith', None)
Augie Fackler
formatting: byteify all mercurial/ and hgext/ string literals...
r43347 return exttestedwith == b"ships-with-hg-core"