comparison mercurial/dispatch.py @ 14439:80c599eee3f3

dispatch: use the request to store the ui object

and check if we got one before creating.

note that the contents of the ui object might change after
dispatch() returns (by options passed through --config for example),
to ensure it doesn't, pass a copy() of it.

14438:08bfec2ef031

import util, commands, hg, fancyopts, extensions, hook, error
import cmdutil, encoding
import ui as uimod

class request(object):
    def __init__(self, args):
        self.args = args

def run():
    "run the command in sys.argv"
    sys.exit(dispatch(request(sys.argv[1:])))

def dispatch(req):
    "run the command specified in req.args"
    try:
        u = uimod.ui()
        if '--traceback' in req.args:
            u.setconfig('ui', 'traceback', 'on')
    except util.Abort, inst:
        sys.stderr.write(_("abort: %s\n") % inst)
        if inst.hint:
            sys.stderr.write(_("(%s)\n") % inst.hint)
        return -1

            sys.stderr.write(_("hg: parse error at %s: %s\n") %
                             (inst.args[1], inst.args[0]))
        else:
            sys.stderr.write(_("hg: parse error: %s\n") % inst.args[0])
        return -1
    return _runcatch(u, req)

def _runcatch(ui, req):
    def catchterm(*args):
        raise error.SignalInterrupt

    try:
        for name in 'SIGBREAK', 'SIGHUP', 'SIGTERM':
            num = getattr(signal, name, None)
            if num:
                signal.signal(num, catchterm)

            if '--debugger' in req.args:
                ui.warn(_("entering debugger - "
                        "type c to continue starting hg or h for help\n"))
                pdb.set_trace()
            try:
                return _dispatch(ui, req)
            finally:
                ui.flush()
        except:
            # enter the debugger when we hit an exception
            if '--debugger' in req.args:

    commands.norepo = norepo
    os.chdir(cwd)

_loaded = set()
def _dispatch(ui, req):
    args = req.args
    shellaliasfn = _checkshellalias(ui, args)
    if shellaliasfn:
        return shellaliasfn()

    # read --config before doing anything else

                if args and not path: # try to infer -R from command args
                    repos = map(cmdutil.findrepo, args)
                    guess = repos[0]
                    if guess and repos.count(guess) == len(repos):
                        req.args = ['--repository', guess] + fullargs
                        return _dispatch(ui, req)
                if not path:
                    raise error.RepoError(_("no repository found in %r"
                                            " (.hg not found)") % os.getcwd())
                raise
        args.insert(0, repo)

14439:80c599eee3f3

import util, commands, hg, fancyopts, extensions, hook, error
import cmdutil, encoding
import ui as uimod

class request(object):
    def __init__(self, args, ui=None):
        self.args = args
        self.ui = ui

def run():
    "run the command in sys.argv"
    sys.exit(dispatch(request(sys.argv[1:])))

def dispatch(req):
    "run the command specified in req.args"
    try:
        if not req.ui:
            req.ui = uimod.ui()
        if '--traceback' in req.args:
            req.ui.setconfig('ui', 'traceback', 'on')
    except util.Abort, inst:
        sys.stderr.write(_("abort: %s\n") % inst)
        if inst.hint:
            sys.stderr.write(_("(%s)\n") % inst.hint)
        return -1

            sys.stderr.write(_("hg: parse error at %s: %s\n") %
                             (inst.args[1], inst.args[0]))
        else:
            sys.stderr.write(_("hg: parse error: %s\n") % inst.args[0])
        return -1
    return _runcatch(req)

def _runcatch(req):
    def catchterm(*args):
        raise error.SignalInterrupt

    ui = req.ui
    try:
        for name in 'SIGBREAK', 'SIGHUP', 'SIGTERM':
            num = getattr(signal, name, None)
            if num:
                signal.signal(num, catchterm)

            if '--debugger' in req.args:
                ui.warn(_("entering debugger - "
                        "type c to continue starting hg or h for help\n"))
                pdb.set_trace()
            try:
                return _dispatch(req)
            finally:
                ui.flush()
        except:
            # enter the debugger when we hit an exception
            if '--debugger' in req.args:

    commands.norepo = norepo
    os.chdir(cwd)

_loaded = set()
def _dispatch(req):
    args = req.args
    ui = req.ui

    shellaliasfn = _checkshellalias(ui, args)
    if shellaliasfn:
        return shellaliasfn()

    # read --config before doing anything else

                if args and not path: # try to infer -R from command args
                    repos = map(cmdutil.findrepo, args)
                    guess = repos[0]
                    if guess and repos.count(guess) == len(repos):
                        req.args = ['--repository', guess] + fullargs
                        return _dispatch(req)
                if not path:
                    raise error.RepoError(_("no repository found in %r"
                                            " (.hg not found)") % os.getcwd())
                raise
        args.insert(0, repo)