templatekw.py
320 lines
| 10.3 KiB
| text/x-python
|
PythonLexer
/ mercurial / templatekw.py
|
r10053 | # templatekw.py - common changeset template keywords | ||
# | ||||
# Copyright 2005-2009 Matt Mackall <mpm@selenic.com> | ||||
# | ||||
# This software may be used and distributed according to the terms of the | ||||
|
r10264 | # GNU General Public License version 2 or any later version. | ||
|
r10053 | |||
|
r10055 | from node import hex | ||
|
r13593 | import encoding, patch, util, error, help | ||
|
r10053 | |||
|
r10260 | def showlist(name, values, plural=None, **args): | ||
|
r10053 | '''expand set of values. | ||
name is name of key in template map. | ||||
values is list of strings or dicts. | ||||
plural is plural of name, if not simply name + 's'. | ||||
expansion works like this, given name 'foo'. | ||||
if values is empty, expand 'no_foos'. | ||||
if 'foo' not in template map, return values as a string, | ||||
joined by space. | ||||
expand 'start_foos'. | ||||
for each value, expand 'foo'. if 'last_foo' in template | ||||
map, expand it instead of 'foo' for last key. | ||||
expand 'end_foos'. | ||||
''' | ||||
|
r10260 | templ = args['templ'] | ||
|
r10282 | if plural: | ||
names = plural | ||||
|
r10053 | else: names = name + 's' | ||
if not values: | ||||
noname = 'no_' + names | ||||
if noname in templ: | ||||
yield templ(noname, **args) | ||||
return | ||||
if name not in templ: | ||||
if isinstance(values[0], str): | ||||
yield ' '.join(values) | ||||
else: | ||||
for v in values: | ||||
yield dict(v, **args) | ||||
return | ||||
startname = 'start_' + names | ||||
if startname in templ: | ||||
yield templ(startname, **args) | ||||
vargs = args.copy() | ||||
def one(v, tag=name): | ||||
try: | ||||
vargs.update(v) | ||||
except (AttributeError, ValueError): | ||||
try: | ||||
for a, b in v: | ||||
vargs[a] = b | ||||
except ValueError: | ||||
vargs[name] = v | ||||
return templ(tag, **vargs) | ||||
lastname = 'last_' + name | ||||
if lastname in templ: | ||||
last = values.pop() | ||||
else: | ||||
last = None | ||||
for v in values: | ||||
yield one(v) | ||||
if last is not None: | ||||
yield one(last, tag=lastname) | ||||
endname = 'end_' + names | ||||
if endname in templ: | ||||
yield templ(endname, **args) | ||||
|
r10056 | def getfiles(repo, ctx, revcache): | ||
if 'files' not in revcache: | ||||
|
r13878 | revcache['files'] = repo.status(ctx.p1().node(), ctx.node())[:3] | ||
|
r10056 | return revcache['files'] | ||
|
r10057 | def getlatesttags(repo, ctx, cache): | ||
'''return date, distance and name for the latest tag of rev''' | ||||
if 'latesttags' not in cache: | ||||
# Cache mapping from rev to a tuple with tag date, tag | ||||
# distance and tag name | ||||
cache['latesttags'] = {-1: (0, 0, 'null')} | ||||
latesttags = cache['latesttags'] | ||||
rev = ctx.rev() | ||||
todo = [rev] | ||||
while todo: | ||||
rev = todo.pop() | ||||
if rev in latesttags: | ||||
continue | ||||
ctx = repo[rev] | ||||
tags = [t for t in ctx.tags() if repo.tagtype(t) == 'global'] | ||||
if tags: | ||||
latesttags[rev] = ctx.date()[0], 0, ':'.join(sorted(tags)) | ||||
continue | ||||
try: | ||||
# The tuples are laid out so the right one can be found by | ||||
# comparison. | ||||
pdate, pdist, ptag = max( | ||||
latesttags[p.rev()] for p in ctx.parents()) | ||||
except KeyError: | ||||
# Cache miss - recurse | ||||
todo.append(rev) | ||||
todo.extend(p.rev() for p in ctx.parents()) | ||||
continue | ||||
latesttags[rev] = pdate, pdist + 1, ptag | ||||
return latesttags[rev] | ||||
|
r10060 | def getrenamedfn(repo, endrev=None): | ||
rcache = {} | ||||
if endrev is None: | ||||
endrev = len(repo) | ||||
def getrenamed(fn, rev): | ||||
'''looks up all renames for a file (up to endrev) the first | ||||
time the file is given. It indexes on the changerev and only | ||||
parses the manifest if linkrev != changerev. | ||||
Returns rename info for fn at changerev rev.''' | ||||
if fn not in rcache: | ||||
rcache[fn] = {} | ||||
fl = repo.file(fn) | ||||
for i in fl: | ||||
lr = fl.linkrev(i) | ||||
renamed = fl.renamed(fl.node(i)) | ||||
rcache[fn][lr] = renamed | ||||
if lr >= endrev: | ||||
break | ||||
if rev in rcache[fn]: | ||||
return rcache[fn][rev] | ||||
# If linkrev != rev (i.e. rev not found in rcache) fallback to | ||||
# filectx logic. | ||||
try: | ||||
return repo[rev][fn].renamed() | ||||
except error.LookupError: | ||||
return None | ||||
return getrenamed | ||||
|
r10055 | def showauthor(repo, ctx, templ, **args): | ||
|
r13585 | """:author: String. The unmodified author of the changeset.""" | ||
|
r10054 | return ctx.user() | ||
|
r13156 | def showbranch(**args): | ||
|
r13585 | """:branch: String. The name of the branch on which the changeset was | ||
committed. | ||||
""" | ||||
|
r13156 | return args['ctx'].branch() | ||
|
r10260 | def showbranches(**args): | ||
|
r13585 | """:branches: List of strings. The name of the branch on which the | ||
changeset was committed. Will be empty if the branch name was | ||||
default. | ||||
""" | ||||
|
r10260 | branch = args['ctx'].branch() | ||
|
r10054 | if branch != 'default': | ||
|
r10260 | return showlist('branch', [branch], plural='branches', **args) | ||
|
r10054 | |||
|
r13386 | def showbookmarks(**args): | ||
|
r13592 | """:bookmarks: List of strings. Any bookmarks associated with the | ||
changeset. | ||||
""" | ||||
|
r13386 | bookmarks = args['ctx'].bookmarks() | ||
return showlist('bookmark', bookmarks, **args) | ||||
|
r11655 | def showchildren(**args): | ||
|
r13585 | """:children: List of strings. The children of the changeset.""" | ||
|
r11655 | ctx = args['ctx'] | ||
childrevs = ['%d:%s' % (cctx, cctx) for cctx in ctx.children()] | ||||
return showlist('children', childrevs, **args) | ||||
|
r10055 | def showdate(repo, ctx, templ, **args): | ||
|
r13585 | """:date: Date information. The date when the changeset was committed.""" | ||
|
r10054 | return ctx.date() | ||
|
r10055 | def showdescription(repo, ctx, templ, **args): | ||
|
r13585 | """:desc: String. The text of the changeset description.""" | ||
|
r10054 | return ctx.description().strip() | ||
|
r10055 | def showdiffstat(repo, ctx, templ, **args): | ||
|
r13585 | """:diffstat: String. Statistics of changes with the following format: | ||
"modified files: +added/-removed lines" | ||||
""" | ||||
|
r10055 | files, adds, removes = 0, 0, 0 | ||
|
r13114 | for i in patch.diffstatdata(util.iterlines(ctx.diff())): | ||
|
r10055 | files += 1 | ||
adds += i[1] | ||||
removes += i[2] | ||||
return '%s: +%s/-%s' % (files, adds, removes) | ||||
|
r10260 | def showextras(**args): | ||
templ = args['templ'] | ||||
for key, value in sorted(args['ctx'].extra().items()): | ||||
|
r10054 | args = args.copy() | ||
args.update(dict(key=key, value=value)) | ||||
yield templ('extra', **args) | ||||
|
r10260 | def showfileadds(**args): | ||
|
r13585 | """:file_adds: List of strings. Files added by this changeset.""" | ||
|
r10260 | repo, ctx, revcache = args['repo'], args['ctx'], args['revcache'] | ||
return showlist('file_add', getfiles(repo, ctx, revcache)[1], **args) | ||||
|
r10056 | |||
|
r10260 | def showfilecopies(**args): | ||
|
r13585 | """:file_copies: List of strings. Files copied in this changeset with | ||
their sources. | ||||
""" | ||||
|
r10394 | cache, ctx = args['cache'], args['ctx'] | ||
|
r10260 | copies = args['revcache'].get('copies') | ||
|
r10060 | if copies is None: | ||
if 'getrenamed' not in cache: | ||||
|
r10260 | cache['getrenamed'] = getrenamedfn(args['repo']) | ||
|
r10060 | copies = [] | ||
getrenamed = cache['getrenamed'] | ||||
for fn in ctx.files(): | ||||
rename = getrenamed(fn, ctx.rev()) | ||||
if rename: | ||||
copies.append((fn, rename[0])) | ||||
|
r10282 | |||
|
r10060 | c = [{'name': x[0], 'source': x[1]} for x in copies] | ||
|
r10260 | return showlist('file_copy', c, plural='file_copies', **args) | ||
|
r10060 | |||
# showfilecopiesswitch() displays file copies only if copy records are | ||||
# provided before calling the templater, usually with a --copies | ||||
# command line switch. | ||||
|
r10260 | def showfilecopiesswitch(**args): | ||
|
r13585 | """:file_copies_switch: List of strings. Like "file_copies" but displayed | ||
only if the --copied switch is set. | ||||
""" | ||||
|
r10260 | copies = args['revcache'].get('copies') or [] | ||
|
r10060 | c = [{'name': x[0], 'source': x[1]} for x in copies] | ||
|
r10260 | return showlist('file_copy', c, plural='file_copies', **args) | ||
|
r10058 | |||
|
r10260 | def showfiledels(**args): | ||
|
r13585 | """:file_dels: List of strings. Files removed by this changeset.""" | ||
|
r10260 | repo, ctx, revcache = args['repo'], args['ctx'], args['revcache'] | ||
return showlist('file_del', getfiles(repo, ctx, revcache)[2], **args) | ||||
|
r10056 | |||
|
r10260 | def showfilemods(**args): | ||
|
r13585 | """:file_mods: List of strings. Files modified by this changeset.""" | ||
|
r10260 | repo, ctx, revcache = args['repo'], args['ctx'], args['revcache'] | ||
return showlist('file_mod', getfiles(repo, ctx, revcache)[0], **args) | ||||
|
r10056 | |||
|
r10260 | def showfiles(**args): | ||
|
r13585 | """:files: List of strings. All files modified, added, or removed by this | ||
changeset. | ||||
""" | ||||
|
r10260 | return showlist('file', args['ctx'].files(), **args) | ||
|
r10054 | |||
|
r10057 | def showlatesttag(repo, ctx, templ, cache, **args): | ||
|
r13585 | """:latesttag: String. Most recent global tag in the ancestors of this | ||
changeset. | ||||
""" | ||||
|
r10057 | return getlatesttags(repo, ctx, cache)[2] | ||
def showlatesttagdistance(repo, ctx, templ, cache, **args): | ||||
|
r13585 | """:latesttagdistance: Integer. Longest path to the latest tag.""" | ||
|
r10057 | return getlatesttags(repo, ctx, cache)[1] | ||
|
r10260 | def showmanifest(**args): | ||
repo, ctx, templ = args['repo'], args['ctx'], args['templ'] | ||||
|
r10055 | args = args.copy() | ||
args.update(dict(rev=repo.manifest.rev(ctx.changeset()[0]), | ||||
node=hex(ctx.changeset()[0]))) | ||||
return templ('manifest', **args) | ||||
def shownode(repo, ctx, templ, **args): | ||||
|
r13585 | """:node: String. The changeset identification hash, as a 40 hexadecimal | ||
digit string. | ||||
""" | ||||
|
r10054 | return ctx.hex() | ||
|
r10055 | def showrev(repo, ctx, templ, **args): | ||
|
r13585 | """:rev: Integer. The repository-local changeset revision number.""" | ||
|
r10054 | return ctx.rev() | ||
|
r10260 | def showtags(**args): | ||
|
r13585 | """:tags: List of strings. Any tags associated with the changeset.""" | ||
|
r10260 | return showlist('tag', args['ctx'].tags(), **args) | ||
|
r10054 | |||
|
r10260 | # keywords are callables like: | ||
# fn(repo, ctx, templ, cache, revcache, **args) | ||||
# with: | ||||
# repo - current repository instance | ||||
# ctx - the changectx being displayed | ||||
# templ - the templater instance | ||||
# cache - a cache dictionary for the whole templater run | ||||
# revcache - a cache dictionary for the current revision | ||||
|
r10054 | keywords = { | ||
'author': showauthor, | ||||
|
r13156 | 'branch': showbranch, | ||
|
r10054 | 'branches': showbranches, | ||
|
r13386 | 'bookmarks': showbookmarks, | ||
|
r11655 | 'children': showchildren, | ||
|
r10054 | 'date': showdate, | ||
'desc': showdescription, | ||||
|
r10055 | 'diffstat': showdiffstat, | ||
|
r10054 | 'extras': showextras, | ||
|
r10056 | 'file_adds': showfileadds, | ||
|
r10058 | 'file_copies': showfilecopies, | ||
|
r10060 | 'file_copies_switch': showfilecopiesswitch, | ||
|
r10056 | 'file_dels': showfiledels, | ||
'file_mods': showfilemods, | ||||
|
r10054 | 'files': showfiles, | ||
|
r10057 | 'latesttag': showlatesttag, | ||
'latesttagdistance': showlatesttagdistance, | ||||
|
r10055 | 'manifest': showmanifest, | ||
|
r10054 | 'node': shownode, | ||
'rev': showrev, | ||||
'tags': showtags, | ||||
} | ||||
|
r13585 | def makedoc(topic, doc): | ||
|
r13593 | return help.makeitemsdoc(topic, doc, '.. keywordsmarker', keywords) | ||
|
r13585 | |||
# tell hggettext to extract docstrings from these functions: | ||||
i18nfunctions = keywords.values() | ||||