comparison mercurial/cmdutil.py @ 35662:91f0979f16c0

cat: factor out a function that populates the formatter

This will allow extensions to add data to the templater.

35661:1c0ee0befba0

        for warning in warnings:
            ui.warn(warning)

    return ret

def cat(ui, repo, ctx, matcher, basefm, fntemplate, prefix, **opts):
    err = 1
    opts = pycompat.byteskwargs(opts)

    def write(path):

            try:
                os.makedirs(os.path.dirname(filename))
            except OSError:
                pass
        with formatter.maybereopen(basefm, filename, opts) as fm:
            data = ctx[path].data()
            if opts.get('decode'):
                data = repo.wwritedata(path, data)
            fm.startitem()
            fm.write('data', '%s', data)
            fm.data(abspath=path, path=matcher.rel(path))

    # Automation often uses hg cat on single files, so special case it
    # for performance to avoid the cost of parsing the manifest.
    if len(matcher.files()) == 1 and not matcher.anypats():
        file = matcher.files()[0]

35662:91f0979f16c0

        for warning in warnings:
            ui.warn(warning)

    return ret

def _updatecatformatter(fm, ctx, matcher, path, decode):
    """Hook for adding data to the formatter used by ``hg cat``.

    Extensions (e.g., lfs) can wrap this to inject keywords/data, but must call
    this method first."""
    data = ctx[path].data()
    if decode:
        data = ctx.repo().wwritedata(path, data)
    fm.startitem()
    fm.write('data', '%s', data)
    fm.data(abspath=path, path=matcher.rel(path))

def cat(ui, repo, ctx, matcher, basefm, fntemplate, prefix, **opts):
    err = 1
    opts = pycompat.byteskwargs(opts)

    def write(path):

            try:
                os.makedirs(os.path.dirname(filename))
            except OSError:
                pass
        with formatter.maybereopen(basefm, filename, opts) as fm:
            _updatecatformatter(fm, ctx, matcher, path, opts.get('decode'))

    # Automation often uses hg cat on single files, so special case it
    # for performance to avoid the cost of parsing the manifest.
    if len(matcher.files()) == 1 and not matcher.anypats():
        file = matcher.files()[0]