comparison hgext/evolve.py @ 1034:5fd28e46e8f5

evolve: move OBSEXC progress in a dedicated function

This will help use to rework them.

1033:908273755215

def obsexcmsg(ui, message):
    message = 'OBSEXC: ' + message
    ui.status(message)


@command('debugobsoleterelevant',
         [],
         'REVSET')
def debugobsoleterelevant(ui, repo, *revsets):

    @util.propertycache
    def length(self):
        return len(self.getvalue())

    def read(self, size):
        self.ui.progress('OBSEXC', self.tell(), unit="bytes",
                         total=self.length)
        return StringIO.read(self, size)

    def __iter__(self):
        d = self.read(4096)
        while d:

            obsdata.ui = repo.ui
            obsexcmsg(repo.ui, "pushing %i markers (%i bytes)\n"
                               % (len(markers), len(obsdata.getvalue())))
            bundler.newpart('EVOLVE:B2X:OBSMARKERV1', data=obsdata)
            def callback(op):
                repo.ui.progress('OBSEXC', None)
                obsexcmsg(repo.ui, "DONE\n")
            return callback
    bundle2partsgenerators.append(_pushb2phases)
    bundle2partsgenerators.append(_pushb2obsmarker)

                raise util.Abort('missing support for %s' % exc)
            try:
                op = bundle2.processbundle(pushop.repo, reply)
            except bundle2.UnknownPartError, exc:
                raise util.Abort('missing support for %s' % exc)
            repo.ui.progress('OBSEXC', None)
        elif remote.capable('_evoext_pushobsmarkers_0'):
            obsdata = pushobsmarkerStringIO()
            _encodemarkersstream(obsdata, markers)
            obsdata.seek(0)
            obsdata.ui = repo.ui
            obsexcmsg(repo.ui, "pushing %i markers (%i bytes)\n"
                               % (len(markers), len(obsdata.getvalue())))
            remote.evoext_pushobsmarkers_0(obsdata)
            repo.ui.progress('OBSEXC', None)
        else:
            rslts = []
            remotedata = _pushkeyescape(markers).items()
            totalbytes = sum(len(d) for k,d in remotedata)
            sentbytes = 0
            obsexcmsg(repo.ui, "pushing %i markers in %i pushkey payload (%i bytes)\n"
                                % (len(markers), len(remotedata), totalbytes))
            for key, data in remotedata:
                repo.ui.progress('OBSEXC', sentbytes, item=key, unit="bytes",
                                 total=totalbytes)
                rslts.append(remote.pushkey('obsolete', key, '', data))
                sentbytes += len(data)
                repo.ui.progress('OBSEXC', sentbytes, item=key, unit="bytes",
                                 total=totalbytes)
            repo.ui.progress('OBSEXC', None)
            if [r for r in rslts if not r]:
                msg = _('failed to push some obsolete markers!\n')
                repo.ui.warn(msg)
        obsexcmsg(repo.ui, "DONE\n")

        obsdata = inpart.read()
    else:
        length = int(length)
        data = StringIO()
        current = 0
        op.ui.progress('OBSEXC', current, unit="bytes", total=length)
        while current < length:
            readsize = min(length-current, 4096)
            data.write(inpart.read(readsize))
            current += readsize
            op.ui.progress('OBSEXC', current, unit="bytes", total=length)
        op.ui.progress('OBSEXC', None)
        obsdata = data.getvalue()
    totalsize = len(obsdata)
    old = len(op.repo.obsstore._all)
    op.repo.obsstore.mergemarkers(tr, obsdata)
    new = len(op.repo.obsstore._all) - old

    length = int(f.read(20))
    chunk = 4096
    current = 0
    data = StringIO()
    ui = self.ui
    ui.progress('OBSEXC', current, unit="bytes", total=length)
    while current < length:
        readsize = min(length-current, chunk)
        data.write(f.read(readsize))
        current += readsize
        ui.progress('OBSEXC', current, unit="bytes", total=length)
    ui.progress('OBSEXC', None)
    data.seek(0)
    return data

@eh.addattr(localrepo.localpeer, 'evoext_pullobsmarkers_0')
def local_pullobsmarkers(self, heads=None, common=None):

1034:5fd28e46e8f5

def obsexcmsg(ui, message):
    message = 'OBSEXC: ' + message
    ui.status(message)

def obsexcprg(ui, *args, **kwargs):
    ui.progress('OBSEXC', *args, **kwargs)


@command('debugobsoleterelevant',
         [],
         'REVSET')
def debugobsoleterelevant(ui, repo, *revsets):

    @util.propertycache
    def length(self):
        return len(self.getvalue())

    def read(self, size):
        obsexcprg(self.ui, self.tell(), unit="bytes", total=self.length)
        return StringIO.read(self, size)

    def __iter__(self):
        d = self.read(4096)
        while d:

            obsdata.ui = repo.ui
            obsexcmsg(repo.ui, "pushing %i markers (%i bytes)\n"
                               % (len(markers), len(obsdata.getvalue())))
            bundler.newpart('EVOLVE:B2X:OBSMARKERV1', data=obsdata)
            def callback(op):
                obsexcprg(repo.ui, None)
                obsexcmsg(repo.ui, "DONE\n")
            return callback
    bundle2partsgenerators.append(_pushb2phases)
    bundle2partsgenerators.append(_pushb2obsmarker)

                raise util.Abort('missing support for %s' % exc)
            try:
                op = bundle2.processbundle(pushop.repo, reply)
            except bundle2.UnknownPartError, exc:
                raise util.Abort('missing support for %s' % exc)
            obsexcprg(repo.ui, None)
        elif remote.capable('_evoext_pushobsmarkers_0'):
            obsdata = pushobsmarkerStringIO()
            _encodemarkersstream(obsdata, markers)
            obsdata.seek(0)
            obsdata.ui = repo.ui
            obsexcmsg(repo.ui, "pushing %i markers (%i bytes)\n"
                               % (len(markers), len(obsdata.getvalue())))
            remote.evoext_pushobsmarkers_0(obsdata)
            obsexcprg(repo.ui, None)
        else:
            rslts = []
            remotedata = _pushkeyescape(markers).items()
            totalbytes = sum(len(d) for k,d in remotedata)
            sentbytes = 0
            obsexcmsg(repo.ui, "pushing %i markers in %i pushkey payload (%i bytes)\n"
                                % (len(markers), len(remotedata), totalbytes))
            for key, data in remotedata:
                obsexcprg(repo.ui, sentbytes, item=key, unit="bytes",
                          total=totalbytes)
                rslts.append(remote.pushkey('obsolete', key, '', data))
                sentbytes += len(data)
                obsexcprg(repo.ui, sentbytes, item=key, unit="bytes",
                          total=totalbytes)
            obsexcprg(repo.ui, None)
            if [r for r in rslts if not r]:
                msg = _('failed to push some obsolete markers!\n')
                repo.ui.warn(msg)
        obsexcmsg(repo.ui, "DONE\n")

        obsdata = inpart.read()
    else:
        length = int(length)
        data = StringIO()
        current = 0
        obsexcprg(op.repo.ui, current, unit="bytes", total=length)
        while current < length:
            readsize = min(length-current, 4096)
            data.write(inpart.read(readsize))
            current += readsize
            obsexcprg(op.repo.ui, current, unit="bytes", total=length)
        obsexcprg(op.repo.ui, None)
        obsdata = data.getvalue()
    totalsize = len(obsdata)
    old = len(op.repo.obsstore._all)
    op.repo.obsstore.mergemarkers(tr, obsdata)
    new = len(op.repo.obsstore._all) - old

    length = int(f.read(20))
    chunk = 4096
    current = 0
    data = StringIO()
    ui = self.ui
    obsexcprg(ui, current, unit="bytes", total=length)
    while current < length:
        readsize = min(length-current, chunk)
        data.write(f.read(readsize))
        current += readsize
        obsexcprg(ui, current, unit="bytes", total=length)
    obsexcprg(ui, None)
    data.seek(0)
    return data

@eh.addattr(localrepo.localpeer, 'evoext_pullobsmarkers_0')
def local_pullobsmarkers(self, heads=None, common=None):