churn.py
206 lines
| 5.4 KiB
| text/x-python
|
PythonLexer
/ hgext / churn.py
|
r6348 | # churn.py - create a graph showing who changed the most lines | ||
# | ||||
# Copyright 2006 Josef "Jeff" Sipek <jeffpc@josefsipek.net> | ||||
# | ||||
# This software may be used and distributed according to the terms | ||||
# of the GNU General Public License, incorporated herein by reference. | ||||
# | ||||
# | ||||
# Aliases map file format is simple one alias per line in the following | ||||
# format: | ||||
# | ||||
# <alias email> <actual email> | ||||
from mercurial.i18n import gettext as _ | ||||
from mercurial import mdiff, cmdutil, util, node | ||||
import os, sys | ||||
def get_tty_width(): | ||||
if 'COLUMNS' in os.environ: | ||||
try: | ||||
return int(os.environ['COLUMNS']) | ||||
except ValueError: | ||||
pass | ||||
try: | ||||
import termios, array, fcntl | ||||
for dev in (sys.stdout, sys.stdin): | ||||
try: | ||||
fd = dev.fileno() | ||||
if not os.isatty(fd): | ||||
continue | ||||
arri = fcntl.ioctl(fd, termios.TIOCGWINSZ, '\0' * 8) | ||||
return array.array('h', arri)[1] | ||||
except ValueError: | ||||
pass | ||||
except ImportError: | ||||
pass | ||||
return 80 | ||||
def __gather(ui, repo, node1, node2): | ||||
def dirtywork(f, mmap1, mmap2): | ||||
lines = 0 | ||||
to = mmap1 and repo.file(f).read(mmap1[f]) or None | ||||
tn = mmap2 and repo.file(f).read(mmap2[f]) or None | ||||
diff = mdiff.unidiff(to, "", tn, "", f, f).split("\n") | ||||
for line in diff: | ||||
if not line: | ||||
continue # skip EOF | ||||
if line.startswith(" "): | ||||
continue # context line | ||||
if line.startswith("--- ") or line.startswith("+++ "): | ||||
continue # begining of diff | ||||
if line.startswith("@@ "): | ||||
continue # info line | ||||
# changed lines | ||||
lines += 1 | ||||
return lines | ||||
## | ||||
lines = 0 | ||||
changes = repo.status(node1, node2, None, util.always)[:5] | ||||
modified, added, removed, deleted, unknown = changes | ||||
who = repo.changelog.read(node2)[1] | ||||
who = util.email(who) # get the email of the person | ||||
mmap1 = repo.manifest.read(repo.changelog.read(node1)[0]) | ||||
mmap2 = repo.manifest.read(repo.changelog.read(node2)[0]) | ||||
for f in modified: | ||||
lines += dirtywork(f, mmap1, mmap2) | ||||
for f in added: | ||||
lines += dirtywork(f, None, mmap2) | ||||
for f in removed: | ||||
lines += dirtywork(f, mmap1, None) | ||||
for f in deleted: | ||||
lines += dirtywork(f, mmap1, mmap2) | ||||
for f in unknown: | ||||
lines += dirtywork(f, mmap1, mmap2) | ||||
return (who, lines) | ||||
def gather_stats(ui, repo, amap, revs=None, progress=False): | ||||
stats = {} | ||||
cl = repo.changelog | ||||
if not revs: | ||||
revs = range(0, cl.count()) | ||||
nr_revs = len(revs) | ||||
cur_rev = 0 | ||||
for rev in revs: | ||||
cur_rev += 1 # next revision | ||||
node2 = cl.node(rev) | ||||
node1 = cl.parents(node2)[0] | ||||
if cl.parents(node2)[1] != node.nullid: | ||||
ui.note(_('Revision %d is a merge, ignoring...\n') % (rev,)) | ||||
continue | ||||
who, lines = __gather(ui, repo, node1, node2) | ||||
# remap the owner if possible | ||||
if who in amap: | ||||
ui.note("using '%s' alias for '%s'\n" % (amap[who], who)) | ||||
who = amap[who] | ||||
if not who in stats: | ||||
stats[who] = 0 | ||||
stats[who] += lines | ||||
ui.note("rev %d: %d lines by %s\n" % (rev, lines, who)) | ||||
if progress: | ||||
nr_revs = max(nr_revs, 1) | ||||
if int(100.0*(cur_rev - 1)/nr_revs) < int(100.0*cur_rev/nr_revs): | ||||
ui.write("\rGenerating stats: %d%%" % (int(100.0*cur_rev/nr_revs),)) | ||||
sys.stdout.flush() | ||||
if progress: | ||||
ui.write("\r") | ||||
sys.stdout.flush() | ||||
return stats | ||||
def churn(ui, repo, **opts): | ||||
"Graphs the number of lines changed" | ||||
def pad(s, l): | ||||
if len(s) < l: | ||||
return s + " " * (l-len(s)) | ||||
return s[0:l] | ||||
def graph(n, maximum, width, char): | ||||
maximum = max(1, maximum) | ||||
n = int(n * width / float(maximum)) | ||||
return char * (n) | ||||
def get_aliases(f): | ||||
aliases = {} | ||||
for l in f.readlines(): | ||||
l = l.strip() | ||||
|
r6421 | alias, actual = l.split() | ||
|
r6348 | aliases[alias] = actual | ||
return aliases | ||||
amap = {} | ||||
aliases = opts.get('aliases') | ||||
if aliases: | ||||
try: | ||||
f = open(aliases,"r") | ||||
except OSError, e: | ||||
print "Error: " + e | ||||
return | ||||
amap = get_aliases(f) | ||||
f.close() | ||||
revs = [int(r) for r in cmdutil.revrange(repo, opts['rev'])] | ||||
revs.sort() | ||||
stats = gather_stats(ui, repo, amap, revs, opts.get('progress')) | ||||
# make a list of tuples (name, lines) and sort it in descending order | ||||
ordered = stats.items() | ||||
if not ordered: | ||||
return | ||||
ordered.sort(lambda x, y: cmp(y[1], x[1])) | ||||
max_churn = ordered[0][1] | ||||
tty_width = get_tty_width() | ||||
ui.note(_("assuming %i character terminal\n") % tty_width) | ||||
tty_width -= 1 | ||||
max_user_width = max([len(user) for user, churn in ordered]) | ||||
graph_width = tty_width - max_user_width - 1 - 6 - 2 - 2 | ||||
for user, churn in ordered: | ||||
print "%s %6d %s" % (pad(user, max_user_width), | ||||
churn, | ||||
graph(churn, max_churn, graph_width, '*')) | ||||
cmdtable = { | ||||
"churn": | ||||
(churn, | ||||
[('r', 'rev', [], _('limit statistics to the specified revisions')), | ||||
('', 'aliases', '', _('file with email aliases')), | ||||
('', 'progress', None, _('show progress'))], | ||||
|
r6841 | 'hg churn [-r REVISIONS] [--aliases FILE] [--progress]'), | ||
|
r6348 | } | ||