comparison hgext/rebase.py @ 45548:25e365d5aa8f

rebase: add dryrun property to rebaseruntime

I want to be able to check the property in
`rebaseruntime._rebasenode()`. Passing the value via the runtime is a
convenient way.

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

45547:2dcf595f6985

class rebaseruntime(object):
    """This class is a container for rebase runtime state"""

    def __init__(self, repo, ui, inmemory=False, opts=None):
        if opts is None:
            opts = {}

        # prepared: whether we have rebasestate prepared or not. Currently it
        # decides whether "self.repo" is unfiltered or not.

            repo.ui, b'rebase'
        )
        self.obsoletenotrebased = {}
        self.obsoletewithoutsuccessorindestination = set()
        self.inmemory = inmemory
        self.stateobj = statemod.cmdstate(repo, b'rebasestate')

    @property
    def repo(self):
        if self.prepared:

    else:
        return _dorebase(ui, repo, action, opts)


def _dryrunrebase(ui, repo, action, opts):
    rbsrt = rebaseruntime(repo, ui, inmemory=True, opts=opts)
    confirm = opts.get(b'confirm')
    if confirm:
        ui.status(_(b'starting in-memory rebase\n'))
    else:
        ui.status(

        needsabort = True
        try:
            overrides = {(b'rebase', b'singletransaction'): True}
            with ui.configoverride(overrides, b'rebase'):
                _origrebase(
                    ui, repo, action, opts, rbsrt, leaveunfinished=True,
                )
        except error.InMemoryMergeConflictsError:
            ui.status(_(b'hit a merge conflict\n'))
            return 1
        except error.Abort:

                    dryrun=opts.get(b'dry_run'),
                )


def _dorebase(ui, repo, action, opts, inmemory=False):
    rbsrt = rebaseruntime(repo, ui, inmemory, opts)
    return _origrebase(ui, repo, action, opts, rbsrt)


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

            dsguard = None
            if singletr and not rbsrt.inmemory:
                dsguard = dirstateguard.dirstateguard(repo, b'rebase')
            with util.acceptintervention(dsguard):
                rbsrt._performrebase(tr)
                if not leaveunfinished:
                    rbsrt._finishrebase()


def _definedestmap(ui, repo, inmemory, destf, srcf, basef, revf, destspace):
    """use revisions argument to define destmap {srcrev: destrev}"""

45548:25e365d5aa8f

class rebaseruntime(object):
    """This class is a container for rebase runtime state"""

    def __init__(self, repo, ui, inmemory=False, dryrun=False, opts=None):
        if opts is None:
            opts = {}

        # prepared: whether we have rebasestate prepared or not. Currently it
        # decides whether "self.repo" is unfiltered or not.

            repo.ui, b'rebase'
        )
        self.obsoletenotrebased = {}
        self.obsoletewithoutsuccessorindestination = set()
        self.inmemory = inmemory
        self.dryrun = dryrun
        self.stateobj = statemod.cmdstate(repo, b'rebasestate')

    @property
    def repo(self):
        if self.prepared:

    else:
        return _dorebase(ui, repo, action, opts)


def _dryrunrebase(ui, repo, action, opts):
    rbsrt = rebaseruntime(repo, ui, inmemory=True, dryrun=True, opts=opts)
    confirm = opts.get(b'confirm')
    if confirm:
        ui.status(_(b'starting in-memory rebase\n'))
    else:
        ui.status(

        needsabort = True
        try:
            overrides = {(b'rebase', b'singletransaction'): True}
            with ui.configoverride(overrides, b'rebase'):
                _origrebase(
                    ui, repo, action, opts, rbsrt,
                )
        except error.InMemoryMergeConflictsError:
            ui.status(_(b'hit a merge conflict\n'))
            return 1
        except error.Abort:

                    dryrun=opts.get(b'dry_run'),
                )


def _dorebase(ui, repo, action, opts, inmemory=False):
    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'):

            dsguard = None
            if singletr and not rbsrt.inmemory:
                dsguard = dirstateguard.dirstateguard(repo, b'rebase')
            with util.acceptintervention(dsguard):
                rbsrt._performrebase(tr)
                if not rbsrt.dryrun:
                    rbsrt._finishrebase()


def _definedestmap(ui, repo, inmemory, destf, srcf, basef, revf, destspace):
    """use revisions argument to define destmap {srcrev: destrev}"""