comparison mercurial/cmdutil.py @ 24356:a38f384f2a57

record: consolidate ui.write wrapping in a function

24355:ca4b89683078

import lock as lockmod
import crecord as crecordmod

def parsealiases(cmd):
    return cmd.lstrip("^").split("|")

def recordfilter(ui, originalhunks):
    usecurses =  ui.configbool('experimental', 'crecord', False)
    if usecurses:
        testfile = ui.config('experimental', 'crecordtest', None)

                if tobackup:
                    os.rmdir(backupdir)
            except OSError:
                pass

    # wrap ui.write so diff output can be labeled/colorized
    def wrapwrite(orig, *args, **kw):
        label = kw.pop('label', '')
        for chunk, l in patch.difflabel(lambda: args):
            orig(chunk, label=label + l)

    oldwrite = ui.write
    def wrap(*args, **kwargs):
        return wrapwrite(oldwrite, *args, **kwargs)
    setattr(ui, 'write', wrap)

    try:
        return commit(ui, repo, recordfunc, pats, opts)
    finally:
        ui.write = oldwrite

24356:a38f384f2a57

import lock as lockmod
import crecord as crecordmod

def parsealiases(cmd):
    return cmd.lstrip("^").split("|")

def setupwrapcolorwrite(ui):
    # wrap ui.write so diff output can be labeled/colorized
    def wrapwrite(orig, *args, **kw):
        label = kw.pop('label', '')
        for chunk, l in patch.difflabel(lambda: args):
            orig(chunk, label=label + l)

    oldwrite = ui.write
    def wrap(*args, **kwargs):
        return wrapwrite(oldwrite, *args, **kwargs)
    setattr(ui, 'write', wrap)
    return oldwrite

def recordfilter(ui, originalhunks):
    usecurses =  ui.configbool('experimental', 'crecord', False)
    if usecurses:
        testfile = ui.config('experimental', 'crecordtest', None)

                if tobackup:
                    os.rmdir(backupdir)
            except OSError:
                pass

    oldwrite = setupwrapcolorwrite(ui)
    try:
        return commit(ui, repo, recordfunc, pats, opts)
    finally:
        ui.write = oldwrite