comparison hgext/evolve.py @ 1301:896d66841f7f

evolve: stop conditional wrapping of _getbundleobsmarkerpart

We support down to Mercurial 3.3, all such version have this function.

1300:4a27123e0696

        common = boundaries['common']
        if common != [nullid]:
            kwargs['evo_obscommon'] = common
    return ret

if getattr(exchange, '_getbundleobsmarkerpart', None) is not None:
    @eh.wrapfunction(exchange, '_getbundleobsmarkerpart')
    def _getbundleobsmarkerpart(orig, bundler, repo, source, **kwargs):
        if 'evo_obscommon' not in kwargs:
            return orig(bundler, repo, source, **kwargs)

        heads = kwargs.get('heads')
        if kwargs.get('obsmarkers', False):
            if heads is None:
                heads = repo.heads()
            obscommon = kwargs.get('evo_obscommon', ())
            assert obscommon
            obsset = repo.unfiltered().set('::%ln - ::%ln', heads, obscommon)
            subset = [c.node() for c in obsset]
            markers = repo.obsstore.relevantmarkers(subset)
            exchange.buildobsmarkerspart(bundler, markers)

    @eh.uisetup
    def installgetbundlepartgen(ui):
        origfunc = exchange.getbundle2partsmapping['obsmarkers']
        def newfunc(*args, **kwargs):
            return _getbundleobsmarkerpart(origfunc, *args, **kwargs)
        exchange.getbundle2partsmapping['obsmarkers'] = newfunc


@eh.wrapfunction(exchange, '_pullobsolete')
def _pullobsolete(orig, pullop):
    if not obsolete.isenabled(pullop.repo, obsolete.exchangeopt):
        return None

1301:896d66841f7f

        common = boundaries['common']
        if common != [nullid]:
            kwargs['evo_obscommon'] = common
    return ret

@eh.wrapfunction(exchange, '_getbundleobsmarkerpart')
def _getbundleobsmarkerpart(orig, bundler, repo, source, **kwargs):
    if 'evo_obscommon' not in kwargs:
        return orig(bundler, repo, source, **kwargs)

    heads = kwargs.get('heads')
    if kwargs.get('obsmarkers', False):
        if heads is None:
            heads = repo.heads()
        obscommon = kwargs.get('evo_obscommon', ())
        assert obscommon
        obsset = repo.unfiltered().set('::%ln - ::%ln', heads, obscommon)
        subset = [c.node() for c in obsset]
        markers = repo.obsstore.relevantmarkers(subset)
        exchange.buildobsmarkerspart(bundler, markers)

@eh.uisetup
def installgetbundlepartgen(ui):
    origfunc = exchange.getbundle2partsmapping['obsmarkers']
    def newfunc(*args, **kwargs):
        return _getbundleobsmarkerpart(origfunc, *args, **kwargs)
    exchange.getbundle2partsmapping['obsmarkers'] = newfunc

@eh.wrapfunction(exchange, '_pullobsolete')
def _pullobsolete(orig, pullop):
    if not obsolete.isenabled(pullop.repo, obsolete.exchangeopt):
        return None