← Back to branch summary

~abentley/bzrtools/bzrtools.dev

« back to all changes in this revision

Viewing changes to diffstat.py

  • Committer: Michael Ellerman
  • Date: 2006-06-08 04:43:46 UTC
  • mto: (0.3.1 shelf-dev)
  • mto: This revision was merged to the branch mainline in revision 393.
  • Revision ID: michael@ellerman.id.au-20060608044346-2f58031501426a8f
Add a test for basic switch functionality.

Show diffs side-by-side

added added

removed removed

Lines of Context:
diffstat.py
1
 
#!/usr/bin/python
2
 
 
3
 
class DiffStat(object):
4
 
    def __init__(self, lines):
5
 
        self.maxname = 0
6
 
        self.maxtotal = 0
7
 
        self.total_adds = 0
8
 
        self.total_removes = 0
9
 
        self.stats = {}
10
 
        self.__parse(lines)
11
 
 
12
 
    def __parse(self, lines):
13
 
        import string
14
 
        adds = 0
15
 
        removes = 0
16
 
        current = None
17
 
 
18
 
        for line in lines:
19
 
            if line.startswith('+') and not line.startswith('+++'):
20
 
                adds += 1
21
 
            elif line.startswith('-') and not line.startswith('---'):
22
 
                removes += 1
23
 
            elif line.startswith('=== '):
24
 
                self.__add_stats(current, adds, removes)
25
 
 
26
 
                adds = 0
27
 
                removes = 0
28
 
                context = 0
29
 
                current = None
30
 
            elif line.startswith('--- ') and current is None:
31
 
                current = line[4:].strip()
32
 
 
33
 
        self.__add_stats(current, adds, removes)
34
 
 
35
 
    class Filestat:
36
 
        def __init__(self):
37
 
            self.adds = 0
38
 
            self.removes = 0
39
 
            self.total = 0
40
 
 
41
 
    def __add_stats(self, file, adds, removes):
42
 
        if file is None:
43
 
            return
44
 
        elif file in self.stats:
45
 
            fstat = self.stats[file]
46
 
        else:
47
 
            fstat = self.Filestat()
48
 
 
49
 
        fstat.adds += adds
50
 
        fstat.removes += removes
51
 
        fstat.total = adds + removes
52
 
        self.stats[file] = fstat
53
 
 
54
 
        self.maxname = max(self.maxname, len(file))
55
 
        self.maxtotal = max(self.maxtotal, fstat.total)
56
 
        self.total_adds += adds
57
 
        self.total_removes += removes
58
 
 
59
 
    def __str__(self):
60
 
        # Work out widths
61
 
        width = 78 - 5
62
 
        countwidth = len(str(self.maxtotal))
63
 
        graphwidth = width - countwidth - self.maxname
64
 
        factor = 1
65
 
 
66
 
        while (self.maxtotal / factor) > graphwidth:
67
 
            factor += 1
68
 
 
69
 
        s = ""
70
 
 
71
 
        for file, fstat in self.stats.iteritems():
72
 
            s += ' %-*s | %*.d ' % (self.maxname, file, countwidth, fstat.total)
73
 
 
74
 
            # If diffstat runs out of room it doesn't print anything, which
75
 
            # isn't very useful, so always print at least one + or 1
76
 
            s += '+' * max(fstat.adds / factor, 1)
77
 
            s += '-' * max(fstat.removes / factor, 1)
78
 
            s += '\n'
79
 
 
80
 
        s += ' %d files changes, %d insertions(+), %d deletions(-)' % \
81
 
                (len(self.stats), self.total_adds, self.total_removes)
82
 
        return s
83
 
 
84
 
if __name__ == '__main__':
85
 
    import sys
86
 
    ds = DiffStat(sys.stdin.readlines())
87
 
    print ds