gitinspector/blame.py

# coding: utf-8
#
# Copyright © 2012 Ejwa Software. All rights reserved.
#
# This file is part of gitinspector.
#
# gitinspector is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# gitinspector is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with gitinspector. If not, see <http://www.gnu.org/licenses/>.

from changes import FileDiff
import re
import sysrun
import terminal

class Blame:
	def __init__(self, repo, hard):
		self.blames = {}
		ls_tree_r = sysrun.run(repo, "git ls-tree --name-only -r HEAD")

		for i in ls_tree_r.readlines():
			if FileDiff.is_valid_extension(i):
				git_blame_r = sysrun.run(repo, "git blame -w {0} \"".format("-C -M" if hard else "") +
				                               i.strip() + "\"")

				for j in git_blame_r.readlines():
					if Blame.is_blame_line(j):
						author = Blame.get_author(j)

						if self.blames.get(author, None) == None:
							self.blames[author] = 0

						self.blames[author] += 1

	@staticmethod
	def is_blame_line(string):
		return string.find(" (") != -1

	@staticmethod
	def get_author(string):
		author = re.search(" \((.*?)\d\d\d\d-\d\d-\d\d", string)
		return re.sub("[^\w ]", "", author.group(1)).strip()

def output(repo, hard):
	blame = Blame(repo, hard)

	print "\nBelow is the number of rows from each author that have survived and"
	print "are still intact in the current revision:\n"
	terminal.printb("Author".ljust(21) + "      Rows")
	for i in sorted(blame.blames.items()):
		print i[0].ljust(20)[0:20],
		print str(i[1]).rjust(10)