hook.py
214 lines
| 7.6 KiB
| text/x-python
|
PythonLexer
/ mercurial / hook.py
|
r4622 | # hook.py - hook support for mercurial | ||
# | ||||
# Copyright 2007 Matt Mackall <mpm@selenic.com> | ||||
# | ||||
|
r8225 | # This software may be used and distributed according to the terms of the | ||
|
r10263 | # GNU General Public License version 2 or any later version. | ||
|
r4622 | |||
|
r25953 | from __future__ import absolute_import | ||
import os | ||||
import sys | ||||
import time | ||||
from .i18n import _ | ||||
from . import ( | ||||
demandimport, | ||||
error, | ||||
extensions, | ||||
util, | ||||
) | ||||
|
r4622 | |||
def _pythonhook(ui, repo, name, hname, funcname, args, throw): | ||||
'''call python hook. hook is callable object, looked up as | ||||
name in python module. if callable returns "true", hook | ||||
fails, else passes. if hook raises exception, treated as | ||||
hook failure. exception propagates if throw is "true". | ||||
reason for "true" meaning "hook failed" is so that | ||||
unmodified commands (e.g. mercurial.commands.update) can | ||||
be run as hooks without wrappers to convert return values.''' | ||||
|
r21797 | if callable(funcname): | ||
|
r20548 | obj = funcname | ||
funcname = obj.__module__ + "." + obj.__name__ | ||||
else: | ||||
|
r4622 | d = funcname.rfind('.') | ||
if d == -1: | ||||
|
r26587 | raise error.Abort(_('%s hook is invalid ("%s" not in ' | ||
|
r4622 | 'a module)') % (hname, funcname)) | ||
modname = funcname[:d] | ||||
|
r10103 | oldpaths = sys.path | ||
|
r14941 | if util.mainfrozen(): | ||
|
r9332 | # binary installs require sys.path manipulation | ||
|
r10103 | modpath, modfile = os.path.split(modname) | ||
if modpath and modfile: | ||||
sys.path = sys.path[:] + [modpath] | ||||
modname = modfile | ||||
|
r25328 | with demandimport.deactivated(): | ||
|
r4622 | try: | ||
|
r25328 | obj = __import__(modname) | ||
|
r4622 | except ImportError: | ||
|
r25328 | e1 = sys.exc_type, sys.exc_value, sys.exc_traceback | ||
try: | ||||
# extensions are loaded with hgext_ prefix | ||||
obj = __import__("hgext_%s" % modname) | ||||
except ImportError: | ||||
e2 = sys.exc_type, sys.exc_value, sys.exc_traceback | ||||
if ui.tracebackflag: | ||||
ui.warn(_('exception from first failed import ' | ||||
'attempt:\n')) | ||||
ui.traceback(e1) | ||||
if ui.tracebackflag: | ||||
ui.warn(_('exception from second failed import ' | ||||
'attempt:\n')) | ||||
ui.traceback(e2) | ||||
|
r26587 | raise error.Abort(_('%s hook is invalid ' | ||
|
r25328 | '(import of "%s" failed)') % | ||
(hname, modname)) | ||||
|
r9332 | sys.path = oldpaths | ||
|
r4622 | try: | ||
for p in funcname.split('.')[1:]: | ||||
obj = getattr(obj, p) | ||||
|
r7280 | except AttributeError: | ||
|
r26587 | raise error.Abort(_('%s hook is invalid ' | ||
|
r4622 | '("%s" is not defined)') % | ||
(hname, funcname)) | ||||
|
r21797 | if not callable(obj): | ||
|
r26587 | raise error.Abort(_('%s hook is invalid ' | ||
|
r4622 | '("%s" is not callable)') % | ||
(hname, funcname)) | ||||
|
r20547 | |||
ui.note(_("calling hook %s: %s\n") % (hname, funcname)) | ||||
starttime = time.time() | ||||
|
r4622 | try: | ||
|
r25084 | # redirect IO descriptors to the ui descriptors so hooks | ||
# that write directly to these don't mess up the command | ||||
# protocol when running through the command server | ||||
old = sys.stdout, sys.stderr, sys.stdin | ||||
sys.stdout, sys.stderr, sys.stdin = ui.fout, ui.ferr, ui.fin | ||||
|
r14889 | |||
|
r25084 | r = obj(ui=ui, repo=repo, hooktype=name, **args) | ||
|
r25660 | except Exception as exc: | ||
|
r26587 | if isinstance(exc, error.Abort): | ||
|
r25084 | ui.warn(_('error: %s hook failed: %s\n') % | ||
(hname, exc.args[0])) | ||||
else: | ||||
ui.warn(_('error: %s hook raised an exception: ' | ||||
'%s\n') % (hname, exc)) | ||||
if throw: | ||||
|
r4622 | raise | ||
|
r25084 | ui.traceback() | ||
return True | ||||
|
r14889 | finally: | ||
sys.stdout, sys.stderr, sys.stdin = old | ||||
|
r18671 | duration = time.time() - starttime | ||
|
r18691 | ui.log('pythonhook', 'pythonhook-%s: %s finished in %0.2f seconds\n', | ||
|
r20548 | name, funcname, duration) | ||
|
r4622 | if r: | ||
if throw: | ||||
|
r23415 | raise error.HookAbort(_('%s hook failed') % hname) | ||
|
r4622 | ui.warn(_('warning: %s hook failed\n') % hname) | ||
return r | ||||
def _exthook(ui, repo, name, cmd, args, throw): | ||||
ui.note(_("running hook %s: %s\n") % (name, cmd)) | ||||
|
r7787 | |||
|
r18671 | starttime = time.time() | ||
|
r7787 | env = {} | ||
for k, v in args.iteritems(): | ||||
|
r21797 | if callable(v): | ||
|
r7787 | v = v() | ||
|
r13207 | if isinstance(v, dict): | ||
# make the dictionary element order stable across Python | ||||
# implementations | ||||
v = ('{' + | ||||
', '.join('%r: %r' % i for i in sorted(v.iteritems())) + | ||||
'}') | ||||
|
r7787 | env['HG_' + k.upper()] = v | ||
|
r5869 | if repo: | ||
cwd = repo.root | ||||
else: | ||||
cwd = os.getcwd() | ||||
|
r23270 | r = ui.system(cmd, environ=env, cwd=cwd) | ||
|
r18671 | |||
duration = time.time() - starttime | ||||
|
r18691 | ui.log('exthook', 'exthook-%s: %s finished in %0.2f seconds\n', | ||
|
r18671 | name, cmd, duration) | ||
|
r4622 | if r: | ||
|
r14234 | desc, r = util.explainexit(r) | ||
|
r4622 | if throw: | ||
|
r23415 | raise error.HookAbort(_('%s hook %s') % (name, desc)) | ||
|
r4622 | ui.warn(_('warning: %s hook %s\n') % (name, desc)) | ||
return r | ||||
|
r15896 | def _allhooks(ui): | ||
hooks = [] | ||||
for name, cmd in ui.configitems('hooks'): | ||||
if not name.startswith('priority'): | ||||
priority = ui.configint('hooks', 'priority.%s' % name, 0) | ||||
hooks.append((-priority, len(hooks), name, cmd)) | ||||
return [(k, v) for p, o, k, v in sorted(hooks)] | ||||
|
r5833 | _redirect = False | ||
def redirect(state): | ||||
|
r6266 | global _redirect | ||
|
r5833 | _redirect = state | ||
|
r4622 | def hook(ui, repo, name, throw=False, **args): | ||
|
r17048 | if not ui.callhooks: | ||
return False | ||||
|
r4622 | r = False | ||
|
r9658 | oldstdout = -1 | ||
|
r5833 | |||
|
r7416 | try: | ||
|
r15896 | for hname, cmd in _allhooks(ui): | ||
|
r7416 | if hname.split('.')[0] != name or not cmd: | ||
continue | ||||
|
r17963 | |||
if oldstdout == -1 and _redirect: | ||||
try: | ||||
stdoutno = sys.__stdout__.fileno() | ||||
stderrno = sys.__stderr__.fileno() | ||||
# temporarily redirect stdout to stderr, if possible | ||||
if stdoutno >= 0 and stderrno >= 0: | ||||
sys.__stdout__.flush() | ||||
oldstdout = os.dup(stdoutno) | ||||
os.dup2(stderrno, stdoutno) | ||||
|
r17964 | except (OSError, AttributeError): | ||
# files seem to be bogus, give up on redirecting (WSGI, etc) | ||||
|
r17963 | pass | ||
|
r21797 | if callable(cmd): | ||
|
r7416 | r = _pythonhook(ui, repo, name, hname, cmd, args, throw) or r | ||
elif cmd.startswith('python:'): | ||||
|
r9332 | if cmd.count(':') >= 2: | ||
path, cmd = cmd[7:].rsplit(':', 1) | ||||
|
r13118 | path = util.expandpath(path) | ||
|
r13119 | if repo: | ||
path = os.path.join(repo.root, path) | ||||
|
r17217 | try: | ||
mod = extensions.loadpath(path, 'hghook.%s' % hname) | ||||
except Exception: | ||||
ui.write(_("loading %s hook failed:\n") % hname) | ||||
raise | ||||
|
r7916 | hookfn = getattr(mod, cmd) | ||
else: | ||||
hookfn = cmd[7:].strip() | ||||
r = _pythonhook(ui, repo, name, hname, hookfn, args, throw) or r | ||||
|
r7416 | else: | ||
r = _exthook(ui, repo, hname, cmd, args, throw) or r | ||||
|
r24716 | |||
# The stderr is fully buffered on Windows when connected to a pipe. | ||||
# A forcible flush is required to make small stderr data in the | ||||
# remote side available to the client immediately. | ||||
sys.stderr.flush() | ||||
|
r7416 | finally: | ||
|
r9658 | if _redirect and oldstdout >= 0: | ||
os.dup2(oldstdout, stdoutno) | ||||
|
r7416 | os.close(oldstdout) | ||
|
r6266 | |||
return r | ||||