churn.py
191 lines
| 6.4 KiB
| text/x-python
|
PythonLexer
/ hgext / churn.py
|
r7065 | # churn.py - create a graph of revisions count grouped by template | ||
|
r6348 | # | ||
# Copyright 2006 Josef "Jeff" Sipek <jeffpc@josefsipek.net> | ||||
|
r7065 | # Copyright 2008 Alexander Solovyov <piranha@piranha.org.ua> | ||
|
r6348 | # | ||
|
r8225 | # This software may be used and distributed according to the terms of the | ||
|
r10263 | # GNU General Public License version 2 or any later version. | ||
|
r8228 | |||
|
r8934 | '''command to display statistics about repository history''' | ||
|
r6348 | |||
|
r7051 | from mercurial.i18n import _ | ||
|
r11265 | from mercurial import patch, cmdutil, util, templater, commands | ||
|
r10905 | import os | ||
|
r7065 | import time, datetime | ||
|
r6348 | |||
|
r7065 | def maketemplater(ui, repo, tmpl): | ||
tmpl = templater.parsestring(tmpl, quoted=False) | ||||
try: | ||||
|
r7762 | t = cmdutil.changeset_templater(ui, repo, False, None, None, False) | ||
|
r7065 | except SyntaxError, inst: | ||
raise util.Abort(inst.args[0]) | ||||
t.use_template(tmpl) | ||||
return t | ||||
|
r7870 | def changedlines(ui, repo, ctx1, ctx2, fns): | ||
|
r9669 | added, removed = 0, 0 | ||
|
r9321 | fmatch = cmdutil.matchfiles(repo, fns) | ||
|
r7870 | diff = ''.join(patch.diff(repo, ctx1.node(), ctx2.node(), fmatch)) | ||
|
r7065 | for l in diff.split('\n'): | ||
|
r9669 | if l.startswith("+") and not l.startswith("+++ "): | ||
added += 1 | ||||
elif l.startswith("-") and not l.startswith("--- "): | ||||
removed += 1 | ||||
return (added, removed) | ||||
|
r6348 | |||
|
r7065 | def countrate(ui, repo, amap, *pats, **opts): | ||
"""Calculate stats""" | ||||
if opts.get('dateformat'): | ||||
def getkey(ctx): | ||||
t, tz = ctx.date() | ||||
date = datetime.datetime(*time.gmtime(float(t) - tz)[:6]) | ||||
return date.strftime(opts['dateformat']) | ||||
else: | ||||
tmpl = opts.get('template', '{author|email}') | ||||
tmpl = maketemplater(ui, repo, tmpl) | ||||
def getkey(ctx): | ||||
ui.pushbuffer() | ||||
|
r7369 | tmpl.show(ctx) | ||
|
r7065 | return ui.popbuffer() | ||
|
r10647 | state = {'count': 0} | ||
|
r7065 | rate = {} | ||
df = False | ||||
if opts.get('date'): | ||||
df = util.matchdate(opts['date']) | ||||
|
r9652 | m = cmdutil.match(repo, pats, opts) | ||
|
r9662 | def prep(ctx, fns): | ||
|
r9654 | rev = ctx.rev() | ||
|
r9367 | if df and not df(ctx.date()[0]): # doesn't match date format | ||
|
r9662 | return | ||
|
r6348 | |||
|
r7065 | key = getkey(ctx) | ||
key = amap.get(key, key) # alias remap | ||||
|
r7070 | if opts.get('changesets'): | ||
|
r9670 | rate[key] = (rate.get(key, (0,))[0] + 1, 0) | ||
|
r7070 | else: | ||
|
r7065 | parents = ctx.parents() | ||
if len(parents) > 1: | ||||
ui.note(_('Revision %d is a merge, ignoring...\n') % (rev,)) | ||||
|
r9662 | return | ||
|
r6348 | |||
|
r7065 | ctx1 = parents[0] | ||
|
r7870 | lines = changedlines(ui, repo, ctx1, ctx, fns) | ||
|
r9669 | rate[key] = [r + l for r, l in zip(rate.get(key, (0, 0)), lines)] | ||
|
r6348 | |||
|
r10647 | state['count'] += 1 | ||
|
r10700 | ui.progress(_('analyzing'), state['count'], total=len(repo)) | ||
|
r6348 | |||
|
r9665 | for ctx in cmdutil.walkchangerevs(repo, m, opts, prep): | ||
|
r9662 | continue | ||
|
r10700 | ui.progress(_('analyzing'), None) | ||
|
r6348 | |||
|
r7065 | return rate | ||
|
r7070 | def churn(ui, repo, *pats, **opts): | ||
|
r8823 | '''histogram of changes to the repository | ||
|
r6348 | |||
|
r8823 | This command will display a histogram representing the number | ||
of changed lines or revisions, grouped according to the given | ||||
template. The default template will group changes by author. | ||||
The --dateformat option may be used to group the results by | ||||
date instead. | ||||
|
r7065 | |||
|
r8823 | Statistics are based on the number of changed lines, or | ||
alternatively the number of matching revisions if the | ||||
--changesets option is specified. | ||||
|
r7065 | |||
|
r9205 | Examples:: | ||
|
r6666 | |||
|
r7070 | # display count of changed lines for every committer | ||
hg churn -t '{author|email}' | ||||
|
r7065 | |||
# display daily activity graph | ||||
|
r7070 | hg churn -f '%H' -s -c | ||
|
r6666 | |||
|
r7065 | # display activity of developers by month | ||
|
r7070 | hg churn -f '%Y-%m' -s -c | ||
|
r6348 | |||
|
r7065 | # display count of lines changed in every year | ||
|
r7070 | hg churn -f '%Y' -s | ||
|
r8823 | It is possible to map alternate email addresses to a main address | ||
|
r9254 | by providing a file using the following format:: | ||
|
r8843 | |||
|
r11264 | <alias email> = <actual email> | ||
|
r8254 | |||
|
r9254 | Such a file may be specified with the --aliases option, otherwise | ||
a .hgchurn file will be looked for in the working directory root. | ||||
|
r8254 | ''' | ||
|
r6348 | def pad(s, l): | ||
|
r6759 | return (s + " " * l)[:l] | ||
|
r6348 | |||
amap = {} | ||||
aliases = opts.get('aliases') | ||||
|
r8254 | if not aliases and os.path.exists(repo.wjoin('.hgchurn')): | ||
aliases = repo.wjoin('.hgchurn') | ||||
|
r6348 | if aliases: | ||
|
r6759 | for l in open(aliases, "r"): | ||
|
r11264 | alias, actual = l.split('=' in l and '=' or None, 1) | ||
amap[alias.strip()] = actual.strip() | ||||
|
r6348 | |||
|
r7065 | rate = countrate(ui, repo, amap, *pats, **opts).items() | ||
if not rate: | ||||
|
r6348 | return | ||
|
r9669 | sortkey = ((not opts.get('sort')) and (lambda x: -sum(x[1])) or None) | ||
|
r9032 | rate.sort(key=sortkey) | ||
|
r7065 | |||
|
r9388 | # Be careful not to have a zero maxcount (issue833) | ||
|
r9669 | maxcount = float(max(sum(v) for k, v in rate)) or 1.0 | ||
|
r9390 | maxname = max(len(k) for k, v in rate) | ||
|
r6348 | |||
|
r7547 | ttywidth = util.termwidth() | ||
|
r9467 | ui.debug("assuming %i character terminal\n" % ttywidth) | ||
|
r9669 | width = ttywidth - maxname - 2 - 2 - 2 | ||
|
r7065 | |||
|
r9669 | if opts.get('diffstat'): | ||
width -= 15 | ||||
|
r11501 | def format(name, diffstat): | ||
added, removed = diffstat | ||||
|
r11310 | return "%s %15s %s%s\n" % (pad(name, maxname), | ||
'+%d/-%d' % (added, removed), | ||||
|
r10821 | ui.label('+' * charnum(added), | ||
'diffstat.inserted'), | ||||
ui.label('-' * charnum(removed), | ||||
'diffstat.deleted')) | ||||
|
r9669 | else: | ||
width -= 6 | ||||
def format(name, count): | ||||
|
r11310 | return "%s %6d %s\n" % (pad(name, maxname), sum(count), | ||
'*' * charnum(sum(count))) | ||||
|
r9669 | |||
def charnum(count): | ||||
|
r10282 | return int(round(count * width / maxcount)) | ||
|
r9669 | |||
for name, count in rate: | ||||
|
r11310 | ui.write(format(name, count)) | ||
|
r6348 | |||
cmdtable = { | ||||
|
r7070 | "churn": | ||
(churn, | ||||
|
r11321 | [('r', 'rev', [], | ||
_('count rate for the specified revision or range'), _('REV')), | ||||
('d', 'date', '', | ||||
_('count rate for revisions matching date spec'), _('DATE')), | ||||
|
r10282 | ('t', 'template', '{author|email}', | ||
|
r11321 | _('template to group changesets'), _('TEMPLATE')), | ||
|
r7065 | ('f', 'dateformat', '', | ||
|
r11321 | _('strftime-compatible format for grouping by date'), _('FORMAT')), | ||
|
r7070 | ('c', 'changesets', False, _('count rate by number of changesets')), | ||
|
r7065 | ('s', 'sort', False, _('sort by key (default: sort by count)')), | ||
|
r9669 | ('', 'diffstat', False, _('display added/removed lines separately')), | ||
|
r11321 | ('', 'aliases', '', | ||
_('file with email aliases'), _('FILE')), | ||||
|
r11265 | ] + commands.walkopts, | ||
|
r10647 | _("hg churn [-d DATE] [-r REV] [--aliases FILE] [FILE]")), | ||
|
r6348 | } | ||