comparison hgext/graphlog.py @ 9368:8a4773bcbaec

graphlog: extract some setup code out of common functions

9367:1ef630452e0b

from mercurial import bundlerepo, changegroup, cmdutil, commands, extensions
from mercurial import hg, url, util, graphmod

ASCIIDATA = 'ASC'

def asciiformat(ui, repo, revdag, opts, parentrepo=None):
    """formats a changelog DAG walk for ASCII output"""
    if parentrepo is None:
        parentrepo = repo
    showparents = [ctx.node() for ctx in parentrepo[None].parents()]
    displayer = show_changeset(ui, repo, opts, buffered=True)
    for (id, type, ctx, parentids) in revdag:
        if type != graphmod.CHANGESET:
            continue
        displayer.show(ctx)
        lines = displayer.hunk.pop(ctx.rev()).split('\n')[:-1]
        char = ctx.node() in showparents and '@' or 'o'
        yield (id, ASCIIDATA, (char, lines), parentids)

def asciiedges(nodes):
    """adds edge info to changelog DAG walk suitable for ascii()"""
    seen = []

    if path: # could be reset in canonpath
        revdag = graphmod.filerevs(repo, path, start, stop)
    else:
        revdag = graphmod.revisions(repo, start, stop)

    fmtdag = asciiformat(ui, repo, revdag, opts)
    ascii(ui, asciiedges(fmtdag))

def graphrevs(repo, nodes, opts):
    limit = cmdutil.loglimit(opts)
    nodes.reverse()

        ui.status(_("no changes found\n"))
        return

    o = repo.changelog.nodesbetween(o, revs)[0]
    revdag = graphrevs(repo, o, opts)
    fmtdag = asciiformat(ui, repo, revdag, opts)
    ascii(ui, asciiedges(fmtdag))

def gincoming(ui, repo, source="default", **opts):
    """show the incoming changesets alongside an ASCII revision graph

                # use the created uncompressed bundlerepo
                other = bundlerepo.bundlerepository(ui, repo.root, fname)

        chlist = other.changelog.nodesbetween(incoming, revs)[0]
        revdag = graphrevs(other, chlist, opts)
        fmtdag = asciiformat(ui, other, revdag, opts, parentrepo=repo)
        ascii(ui, asciiedges(fmtdag))

    finally:
        if hasattr(other, 'close'):
            other.close()

9368:8a4773bcbaec

from mercurial import bundlerepo, changegroup, cmdutil, commands, extensions
from mercurial import hg, url, util, graphmod

ASCIIDATA = 'ASC'

def asciiformat(revdag, displayer, parents):
    """formats a changelog DAG walk for ASCII output"""
    for (id, type, ctx, parentids) in revdag:
        if type != graphmod.CHANGESET:
            continue
        displayer.show(ctx)
        lines = displayer.hunk.pop(ctx.rev()).split('\n')[:-1]
        char = ctx.node() in parents and '@' or 'o'
        yield (id, ASCIIDATA, (char, lines), parentids)

def asciiedges(nodes):
    """adds edge info to changelog DAG walk suitable for ascii()"""
    seen = []

    if path: # could be reset in canonpath
        revdag = graphmod.filerevs(repo, path, start, stop)
    else:
        revdag = graphmod.revisions(repo, start, stop)

    displayer = show_changeset(ui, repo, opts, buffered=True)
    showparents = [ctx.node() for ctx in repo[None].parents()]
    fmtdag = asciiformat(revdag, displayer, showparents)
    ascii(ui, asciiedges(fmtdag))

def graphrevs(repo, nodes, opts):
    limit = cmdutil.loglimit(opts)
    nodes.reverse()

        ui.status(_("no changes found\n"))
        return

    o = repo.changelog.nodesbetween(o, revs)[0]
    revdag = graphrevs(repo, o, opts)
    displayer = show_changeset(ui, repo, opts, buffered=True)
    showparents = [ctx.node() for ctx in repo[None].parents()]
    fmtdag = asciiformat(revdag, displayer, showparents)
    ascii(ui, asciiedges(fmtdag))

def gincoming(ui, repo, source="default", **opts):
    """show the incoming changesets alongside an ASCII revision graph

                # use the created uncompressed bundlerepo
                other = bundlerepo.bundlerepository(ui, repo.root, fname)

        chlist = other.changelog.nodesbetween(incoming, revs)[0]
        revdag = graphrevs(other, chlist, opts)
        displayer = show_changeset(ui, other, opts, buffered=True)
        showparents = [ctx.node() for ctx in repo[None].parents()]
        fmtdag = asciiformat(revdag, displayer, showparents)
        ascii(ui, asciiedges(fmtdag))

    finally:
        if hasattr(other, 'close'):
            other.close()