comparison hgext/rebase.py @ 29359:d2168849539d

rebase: move local variable 'target' to the RR class

29358:6e83f5bbed8d

        # Mapping between the old revision id and either what is the new rebased
        # revision or what needs to be done with the old revision. The state
        # dict will be what contains most of the rebase progress state.
        self.state = {}
        self.activebookmark = None

@command('rebase',
    [('s', 'source', '',
     _('rebase the specified changeset and descendants'), _('REV')),
    ('b', 'base', '',

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

    """
    rbsrt = rebaseruntime()
    target = None
    skipped = set()
    targetancestors = set()


    lock = wlock = None

                    _('abort and continue do not allow specifying revisions'))
            if abortf and opts.get('tool', False):
                ui.warn(_('tool option will be ignored\n'))

            try:
                (rbsrt.originalwd, target, rbsrt.state, skipped,
                 collapsef, keepf, keepbranchesf, rbsrt.external,
                 rbsrt.activebookmark) = restorestatus(repo)
                collapsemsg = restorecollapsemsg(repo)
            except error.RepoLookupError:
                if abortf:

                else:
                    msg = _('cannot continue inconsistent rebase')
                    hint = _('use "hg rebase --abort" to clear broken state')
                    raise error.Abort(msg, hint=hint)
            if abortf:
                return abort(repo, rbsrt.originalwd, target, rbsrt.state,
                             activebookmark=rbsrt.activebookmark)

            obsoletenotrebased = {}
            if ui.configbool('experimental', 'rebaseskipobsolete',
                             default=True):
                rebaseobsrevs = set([r for r, st in rbsrt.state.items()
                                     if st == revprecursor])
                rebasesetrevs = set(rbsrt.state.keys())
                obsoletenotrebased = _computeobsoletenotrebased(repo,
                                        rebaseobsrevs, target)
                rebaseobsskipped = set(obsoletenotrebased)
                _checkobsrebase(repo, ui, rebaseobsrevs, rebasesetrevs,
                                rebaseobsskipped)
        else:
            dest, rebaseset = _definesets(ui, repo, destf, srcf, basef, revf,

            if not keepf and not repo[root].mutable():
                raise error.Abort(_("can't rebase public changeset %s")
                                 % repo[root],
                                 hint=_('see "hg help phases" for details'))

            (rbsrt.originalwd, target, rbsrt.state) = result
            if collapsef:
                targetancestors = repo.changelog.ancestors([target],
                                                           inclusive=True)
                rbsrt.external = externalparent(repo, rbsrt.state,
                                                       targetancestors)

            if dest.closesbranch() and not keepbranchesf:

                        raise error.Abort(_('cannot collapse multiple named '
                            'branches'))

        # Rebase
        if not targetancestors:
            targetancestors = repo.changelog.ancestors([target], inclusive=True)

        # Keep track of the current bookmarks in order to reset them later
        currentbookmarks = repo._bookmarks.copy()
        rbsrt.activebookmark = rbsrt.activebookmark or repo._activebookmark
        if rbsrt.activebookmark:

            pos += 1
            if rbsrt.state[rev] == revtodo:
                ui.status(_('rebasing %s\n') % desc)
                ui.progress(_("rebasing"), pos, ("%d:%s" % (rev, ctx)),
                            _('changesets'), total)
                p1, p2, base = defineparents(repo, rev, target, rbsrt.state,
                                             targetancestors,
                                             obsoletenotrebased)
                storestatus(repo, rbsrt.originalwd, target, rbsrt.state,
                            collapsef, keepf, keepbranchesf,
                            rbsrt.external, rbsrt.activebookmark)
                storecollapsemsg(repo, collapsemsg)
                if len(repo[None].parents()) == 2:
                    repo.ui.debug('resuming interrupted rebase\n')
                else:
                    try:
                        ui.setconfig('ui', 'forcemerge', opts.get('tool', ''),
                                     'rebase')
                        stats = rebasenode(repo, rev, p1, base, rbsrt.state,
                                           collapsef, target)
                        if stats and stats[3] > 0:
                            raise error.InterventionRequired(
                                _('unresolved conflicts (see hg '
                                  'resolve, then hg rebase --continue)'))
                    finally:

        ui.progress(_('rebasing'), None)
        ui.note(_('rebase merging completed\n'))

        if collapsef and not keepopen:
            p1, p2, _base = defineparents(repo, min(rbsrt.state), target,
                                          rbsrt.state, targetancestors,
                                          obsoletenotrebased)
            editopt = opts.get('edit')
            editform = 'rebase.collapse'
            if collapsemsg:
                commitmsg = collapsemsg
            else:

                                   commitmsg=commitmsg,
                                   extrafn=extrafn, editor=editor,
                                   keepbranches=keepbranchesf,
                                   date=date)
            if newnode is None:
                newrev = target
            else:
                newrev = repo[newnode].rev()
            for oldrev in rbsrt.state.iterkeys():
                if rbsrt.state[oldrev] > nullmerge:
                    rbsrt.state[oldrev] = newrev

            for k, v in rbsrt.state.iteritems():
                if v > nullmerge:
                    nstate[repo[k].node()] = repo[v].node()
            # XXX this is the same as dest.node() for the non-continue path --
            # this should probably be cleaned up
            targetnode = repo[target].node()

        # restore original working directory
        # (we do this before stripping)
        newwd = rbsrt.state.get(rbsrt.originalwd, rbsrt.originalwd)
        if newwd < 0:

29359:d2168849539d

        # Mapping between the old revision id and either what is the new rebased
        # revision or what needs to be done with the old revision. The state
        # dict will be what contains most of the rebase progress state.
        self.state = {}
        self.activebookmark = None
        self.target = None

@command('rebase',
    [('s', 'source', '',
     _('rebase the specified changeset and descendants'), _('REV')),
    ('b', 'base', '',

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

    """
    rbsrt = rebaseruntime()
    skipped = set()
    targetancestors = set()


    lock = wlock = None

                    _('abort and continue do not allow specifying revisions'))
            if abortf and opts.get('tool', False):
                ui.warn(_('tool option will be ignored\n'))

            try:
                (rbsrt.originalwd, rbsrt.target, rbsrt.state, skipped,
                 collapsef, keepf, keepbranchesf, rbsrt.external,
                 rbsrt.activebookmark) = restorestatus(repo)
                collapsemsg = restorecollapsemsg(repo)
            except error.RepoLookupError:
                if abortf:

                else:
                    msg = _('cannot continue inconsistent rebase')
                    hint = _('use "hg rebase --abort" to clear broken state')
                    raise error.Abort(msg, hint=hint)
            if abortf:
                return abort(repo, rbsrt.originalwd, rbsrt.target,
                             rbsrt.state,
                             activebookmark=rbsrt.activebookmark)

            obsoletenotrebased = {}
            if ui.configbool('experimental', 'rebaseskipobsolete',
                             default=True):
                rebaseobsrevs = set([r for r, st in rbsrt.state.items()
                                     if st == revprecursor])
                rebasesetrevs = set(rbsrt.state.keys())
                obsoletenotrebased = _computeobsoletenotrebased(repo,
                                        rebaseobsrevs, rbsrt.target)
                rebaseobsskipped = set(obsoletenotrebased)
                _checkobsrebase(repo, ui, rebaseobsrevs, rebasesetrevs,
                                rebaseobsskipped)
        else:
            dest, rebaseset = _definesets(ui, repo, destf, srcf, basef, revf,

            if not keepf and not repo[root].mutable():
                raise error.Abort(_("can't rebase public changeset %s")
                                 % repo[root],
                                 hint=_('see "hg help phases" for details'))

            (rbsrt.originalwd, rbsrt.target, rbsrt.state) = result
            if collapsef:
                targetancestors = repo.changelog.ancestors([rbsrt.target],
                                                           inclusive=True)
                rbsrt.external = externalparent(repo, rbsrt.state,
                                                       targetancestors)

            if dest.closesbranch() and not keepbranchesf:

                        raise error.Abort(_('cannot collapse multiple named '
                            'branches'))

        # Rebase
        if not targetancestors:
            targetancestors = repo.changelog.ancestors([rbsrt.target],
                                                       inclusive=True)

        # Keep track of the current bookmarks in order to reset them later
        currentbookmarks = repo._bookmarks.copy()
        rbsrt.activebookmark = rbsrt.activebookmark or repo._activebookmark
        if rbsrt.activebookmark:

            pos += 1
            if rbsrt.state[rev] == revtodo:
                ui.status(_('rebasing %s\n') % desc)
                ui.progress(_("rebasing"), pos, ("%d:%s" % (rev, ctx)),
                            _('changesets'), total)
                p1, p2, base = defineparents(repo, rev, rbsrt.target,
                                             rbsrt.state,
                                             targetancestors,
                                             obsoletenotrebased)
                storestatus(repo, rbsrt.originalwd, rbsrt.target,
                            rbsrt.state, collapsef, keepf, keepbranchesf,
                            rbsrt.external, rbsrt.activebookmark)
                storecollapsemsg(repo, collapsemsg)
                if len(repo[None].parents()) == 2:
                    repo.ui.debug('resuming interrupted rebase\n')
                else:
                    try:
                        ui.setconfig('ui', 'forcemerge', opts.get('tool', ''),
                                     'rebase')
                        stats = rebasenode(repo, rev, p1, base, rbsrt.state,
                                           collapsef, rbsrt.target)
                        if stats and stats[3] > 0:
                            raise error.InterventionRequired(
                                _('unresolved conflicts (see hg '
                                  'resolve, then hg rebase --continue)'))
                    finally:

        ui.progress(_('rebasing'), None)
        ui.note(_('rebase merging completed\n'))

        if collapsef and not keepopen:
            p1, p2, _base = defineparents(repo, min(rbsrt.state),
                                          rbsrt.target, rbsrt.state,
                                          targetancestors, obsoletenotrebased)
            editopt = opts.get('edit')
            editform = 'rebase.collapse'
            if collapsemsg:
                commitmsg = collapsemsg
            else:

                                   commitmsg=commitmsg,
                                   extrafn=extrafn, editor=editor,
                                   keepbranches=keepbranchesf,
                                   date=date)
            if newnode is None:
                newrev = rbsrt.target
            else:
                newrev = repo[newnode].rev()
            for oldrev in rbsrt.state.iterkeys():
                if rbsrt.state[oldrev] > nullmerge:
                    rbsrt.state[oldrev] = newrev

            for k, v in rbsrt.state.iteritems():
                if v > nullmerge:
                    nstate[repo[k].node()] = repo[v].node()
            # XXX this is the same as dest.node() for the non-continue path --
            # this should probably be cleaned up
            targetnode = repo[rbsrt.target].node()

        # restore original working directory
        # (we do this before stripping)
        newwd = rbsrt.state.get(rbsrt.originalwd, rbsrt.originalwd)
        if newwd < 0: