summaryrefslogtreecommitdiff
path: root/csv.py
blob: 34ea10aba6b91f0b0f751707feaccc350b848b6f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
#
# aggregate per-month statistics for people
#
import sys, datetime

class CSVStat:
    def __init__ (self, name, employer, date):
        self.name = name
        self.employer = employer
        self.added = self.removed = 0
        self.date = date
    def accumulate (self, p):
        self.added = self.added + p.added
        self.removed = self.removed + p.removed

PeriodCommitHash = { }

def AccumulatePatch (p):
    date = "%.2d-%.2d-01"%(p.date.year, p.date.month)
    authdatekey = "%s-%s"%(p.author.name, date)
    if authdatekey not in PeriodCommitHash:
        empl = p.author.emailemployer (p.email, p.date)
        stat = CSVStat (p.author.name, empl, date)
        PeriodCommitHash[authdatekey] = stat
    else:
        stat = PeriodCommitHash[authdatekey]
    stat.accumulate (p)

def OutputCSV (file):
    if file is None:
        return
    file.write ("Name\tAffliation\tDate\tAdded\tRemoved\n")
    for date, stat in PeriodCommitHash.items():
        # sanitise names " is common and \" sometimes too
        empl_name = stat.employer.name.replace ("\"", ".").replace ("\\", ".")
        author_name = stat.name.replace ("\"", ".").replace ("\\", ".")
        file.write ("\"%s\"\t\"%s\"\t%s\t%d\t%d\n"%(author_name, empl_name, stat.date, \
                                                    stat.added, stat.removed))