comparison hgext/rebase.py @ 47459:de54d11040e7

rebase: keep str-keyed opts long enough to make `action` a str

This is just another little py3 cleanup.

Differential Revision: https://phab.mercurial-scm.org/D10891

47458:50cd14dbd3b3

    Returns 0 on success, 1 if nothing to rebase or there are
    unresolved conflicts.

    """
    opts = pycompat.byteskwargs(opts)
    inmemory = ui.configbool(b'rebase', b'experimental.inmemory')
    action = cmdutil.check_at_most_one_arg(opts, b'abort', b'stop', b'continue')
    if action:
        cmdutil.check_incompatible_arguments(
            opts, action, [b'confirm', b'dry_run']
        )
        cmdutil.check_incompatible_arguments(
            opts, action, [b'rev', b'source', b'base', b'dest']
        )
    cmdutil.check_at_most_one_arg(opts, b'confirm', b'dry_run')
    cmdutil.check_at_most_one_arg(opts, b'rev', b'source', b'base')

    if action or repo.currenttransaction() is not None:
        # in-memory rebase is not compatible with resuming rebases.
        # (Or if it is run within a transaction, since the restart logic can
        # fail the entire transaction.)
        inmemory = False

    if opts.get(b'auto_orphans'):
        disallowed_opts = set(opts) - {b'auto_orphans'}
        cmdutil.check_incompatible_arguments(
            opts, b'auto_orphans', disallowed_opts
        )

        userrevs = list(repo.revs(opts.get(b'auto_orphans')))
        opts[b'rev'] = [revsetlang.formatspec(b'%ld and orphan()', userrevs)]
        opts[b'dest'] = b'_destautoorphanrebase(SRC)'

    if opts.get(b'dry_run') or opts.get(b'confirm'):
        return _dryrunrebase(ui, repo, action, opts)
    elif action == b'stop':
        rbsrt = rebaseruntime(repo, ui)
        with repo.wlock(), repo.lock():
            rbsrt.restorestatus()
            if rbsrt.collapsef:
                raise error.StateError(_(b"cannot stop in --collapse session"))

    rbsrt = rebaseruntime(repo, ui, inmemory, opts=opts)
    return _origrebase(ui, repo, action, opts, rbsrt)


def _origrebase(ui, repo, action, opts, rbsrt):
    assert action != b'stop'
    with repo.wlock(), repo.lock():
        if opts.get(b'interactive'):
            try:
                if extensions.find(b'histedit'):
                    enablehistedit = b''

        if action:
            if rbsrt.collapsef:
                raise error.InputError(
                    _(b'cannot use collapse with continue or abort')
                )
            if action == b'abort' and opts.get(b'tool', False):
                ui.warn(_(b'tool option will be ignored\n'))
            if action == b'continue':
                ms = mergestatemod.mergestate.read(repo)
                mergeutil.checkunresolved(ms)

            retcode = rbsrt._prepareabortorcontinue(
                isabort=(action == b'abort')
            )
            if retcode is not None:
                return retcode
        else:
            # search default destination in this space
            # used in the 'hg pull --rebase' case, see issue 5214.

47459:de54d11040e7

    Returns 0 on success, 1 if nothing to rebase or there are
    unresolved conflicts.

    """
    inmemory = ui.configbool(b'rebase', b'experimental.inmemory')
    action = cmdutil.check_at_most_one_arg(opts, 'abort', 'stop', 'continue')
    if action:
        cmdutil.check_incompatible_arguments(
            opts, action, ['confirm', 'dry_run']
        )
        cmdutil.check_incompatible_arguments(
            opts, action, ['rev', 'source', 'base', 'dest']
        )
    cmdutil.check_at_most_one_arg(opts, 'confirm', 'dry_run')
    cmdutil.check_at_most_one_arg(opts, 'rev', 'source', 'base')

    if action or repo.currenttransaction() is not None:
        # in-memory rebase is not compatible with resuming rebases.
        # (Or if it is run within a transaction, since the restart logic can
        # fail the entire transaction.)
        inmemory = False

    if opts.get('auto_orphans'):
        disallowed_opts = set(opts) - {'auto_orphans'}
        cmdutil.check_incompatible_arguments(
            opts, 'auto_orphans', disallowed_opts
        )

        userrevs = list(repo.revs(opts.get('auto_orphans')))
        opts['rev'] = [revsetlang.formatspec(b'%ld and orphan()', userrevs)]
        opts['dest'] = b'_destautoorphanrebase(SRC)'

    opts = pycompat.byteskwargs(opts)
    if opts.get(b'dry_run') or opts.get(b'confirm'):
        return _dryrunrebase(ui, repo, action, opts)
    elif action == 'stop':
        rbsrt = rebaseruntime(repo, ui)
        with repo.wlock(), repo.lock():
            rbsrt.restorestatus()
            if rbsrt.collapsef:
                raise error.StateError(_(b"cannot stop in --collapse session"))

    rbsrt = rebaseruntime(repo, ui, inmemory, opts=opts)
    return _origrebase(ui, repo, action, opts, rbsrt)


def _origrebase(ui, repo, action, opts, rbsrt):
    assert action != 'stop'
    with repo.wlock(), repo.lock():
        if opts.get(b'interactive'):
            try:
                if extensions.find(b'histedit'):
                    enablehistedit = b''

        if action:
            if rbsrt.collapsef:
                raise error.InputError(
                    _(b'cannot use collapse with continue or abort')
                )
            if action == 'abort' and opts.get(b'tool', False):
                ui.warn(_(b'tool option will be ignored\n'))
            if action == 'continue':
                ms = mergestatemod.mergestate.read(repo)
                mergeutil.checkunresolved(ms)

            retcode = rbsrt._prepareabortorcontinue(isabort=(action == 'abort'))
            if retcode is not None:
                return retcode
        else:
            # search default destination in this space
            # used in the 'hg pull --rebase' case, see issue 5214.