Mercurial > hg > mercurial-source
changeset 37888:638a241202a3
templater: add hook point to populate additional mapping items
The 'revcache' dict will be inserted by this hook.
author | Yuya Nishihara <yuya@tcha.org> |
---|---|
date | Thu, 15 Mar 2018 21:49:33 +0900 |
parents | 656ac240f392 |
children | be3f33f5e232 |
files | mercurial/formatter.py mercurial/templater.py |
diffstat | 2 files changed, 20 insertions(+), 0 deletions(-) [+] |
line wrap: on
line diff
--- a/mercurial/formatter.py +++ b/mercurial/formatter.py @@ -517,6 +517,9 @@ return None return get(self, context, mapping, key) + def populatemap(self, context, origmapping, newmapping): + return {} + def _getsome(self, context, mapping, key): v = mapping.get(key) if v is not None:
--- a/mercurial/templater.py +++ b/mercurial/templater.py @@ -577,6 +577,11 @@ def lookup(self, context, mapping, key): """Return a resource for the key if available; otherwise None""" + @abc.abstractmethod + def populatemap(self, context, origmapping, newmapping): + """Return a dict of additional mapping items which should be paired + with the given new mapping""" + class nullresourcemapper(resourcemapper): def availablekeys(self, context, mapping): return set() @@ -587,6 +592,9 @@ def lookup(self, context, mapping, key): return None + def populatemap(self, context, origmapping, newmapping): + return {} + class engine(object): '''template expansion engine. @@ -634,6 +642,8 @@ if (k in knownres # not a symbol per self.symbol() or newres.isdisjoint(self._defaultrequires(k)))} mapping.update(newmapping) + mapping.update( + self._resources.populatemap(self, origmapping, newmapping)) return mapping def _defaultrequires(self, key): @@ -689,6 +699,13 @@ mapping contains added elements for use during expansion. Is a generator.''' func, data = self._load(t) + # populate additional items only if they don't exist in the given + # mapping. this is slightly different from overlaymap() because the + # initial 'revcache' may contain pre-computed items. + extramapping = self._resources.populatemap(self, {}, mapping) + if extramapping: + extramapping.update(mapping) + mapping = extramapping return _flatten(func(self, mapping, data)) engines = {'default': engine}