tmstats/makeanniversarytable.py at master · dssinger/tmstats · GitHub

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
39
40
41
42
43
44
45
46
47
48
49
#!/usr/bin/env python3
""" Create the anniversary date table """

import tmutil, sys
import tmglobals
myglobals = tmglobals.tmglobals()


### Insert classes and functions here.  The main program begins in the "if" statement below.

if __name__ == "__main__":

    import tmparms, csv

    # Establish parameters
    parms = tmparms.tmparms()
    parms.add_argument('--outfile', default='${workdir}/anniversary.csv')
    # Add other parameters here

    # Do global setup
    myglobals.setup(parms)
    curs = myglobals.curs
    conn = myglobals.conn
    yy = myglobals.today.year
    mm = myglobals.today.month
    dd = myglobals.today.day
    if mm == 12:
        nextfirst = myglobals.today.replace(year=yy+1,month=1,day=1)
    else:
        nextfirst = myglobals.today.replace(month=mm+1,day=1)

    monthname = nextfirst.strftime('%B, %Y')

    writer = csv.writer(open(parms.outfile, 'w'))
    writer.writerow(['Club Name', 'Club Number', 'Charter Year', 'Charter Month', 'Charter Day', 'Age at end of ' + monthname, 'Has Anniversary in ' + monthname, 'Next Anniversary', 'Age on Next Anniversary'])
    curs.execute('SELECT MAX(lastdate) FROM clubs')
    maxdate = curs.fetchone()[0]
    curs.execute('SELECT clubname, clubnumber, charterdate FROM clubs WHERE lastdate = %s ORDER BY month(charterdate), day(charterdate)', (maxdate,))
    for (clubname, clubnumber, charterdate) in curs.fetchall():
        cyy = charterdate.year
        cmm = charterdate.month
        cdd = charterdate.day
        ayy = yy + (1 if (100*mm + dd) >= (100*cmm + cdd) else 0)
        nextannage = ayy - cyy
        nextmonage = (yy - cyy) - (1 if cmm > nextfirst.month else 0)
        writer.writerow((clubname, clubnumber, cyy, cmm, cdd,  nextmonage, 'Yes' if cmm == nextfirst.month else '', '%04d-%02d-%02d' % (ayy, cmm, cdd), nextannage))