view tests/revnamesext.py @ 51304:d91b55371d6f

git-hgext: adjust to the lack of `changelog.heads` method

We don't have a `heads` method returning nodeid, but this is very easy to get
the same result.

This was flagged by pytype.

We can note that the fact this code did not break is probably a good sign that
it is dead code.

However this is a question outside of the scop of this series.

# Dummy extension to define a namespace containing revision names


from mercurial import namespaces


def reposetup(ui, repo):
    names = {b'r%d' % rev: repo[rev].node() for rev in repo}
    namemap = lambda r, name: names.get(name)
    nodemap = lambda r, node: [b'r%d' % repo[node].rev()]

    ns = namespaces.namespace(
        b'revnames',
        templatename=b'revname',
        logname=b'revname',
        listnames=lambda r: names.keys(),
        namemap=namemap,
        nodemap=nodemap,
    )
    repo.names.addnamespace(ns)