diff --git a/render.py b/render.py index d5c9e51..f8cdd7a 100755 --- a/render.py +++ b/render.py @@ -7,59 +7,64 @@ import sys import os from mako.template import Template -if len(sys.argv) < 2: - print >>sys.stderr, "Usage: %s TEMPLATE [WEEK]" - sys.exit(1) - -template = sys.argv[1] START = datetime.datetime(2009, 12, 21, 6) -if len(sys.argv) == 3: - week = parse(sys.argv[2], default=START) -else: - week = START +def render_template(path, week=None): + with open('out/report.yml') as r: + report = yaml.safe_load(r) -with open('out/report.yml') as r: - report = yaml.safe_load(r) - -with open('bloggers.yml') as f: - users = yaml.safe_load(f) - -week = (week - START).days / 7 -week_start = START + (week * datetime.timedelta(7)) -week_end = START + ((week + 1) * datetime.timedelta(7)) - -good = [] -lame = [] -skip = [] -userlist = [] - -class User(object): - pass - -for (un, rec) in users.items(): - u = User() - u.username = un - u.links = rec['links'] - u.start = rec['start'] - u.weeks = report.get(un, []) - - userlist.append(u) - -def user_key(u): - return (u.start, u.username) - -userlist.sort(key=user_key) - -for u in userlist: - user_start = parse(u.start, default=START) - if user_start > week_start: - skip.append(u) - elif len(u.weeks) <= week or not u.weeks[week]: - lame.append(u) + with open('bloggers.yml') as f: + users = yaml.safe_load(f) + if week: + week = parse(week, default=START) else: - good.append(u) + week = START -print Template(filename=template, output_encoding='utf-8').render( - week=week, week_start=week_start,week_end=week_end, - good=good, lame=lame, skip=skip, userlist=userlist) + week = (week - START).days / 7 + week_start = START + (week * datetime.timedelta(7)) + week_end = START + ((week + 1) * datetime.timedelta(7)) + + good = [] + lame = [] + skip = [] + userlist = [] + + class User(object): + pass + + for (un, rec) in users.items(): + u = User() + u.username = un + u.links = rec['links'] + u.start = rec['start'] + u.weeks = report.get(un, []) + + userlist.append(u) + + def user_key(u): + return (u.start, u.username) + + userlist.sort(key=user_key) + + for u in userlist: + user_start = parse(u.start, default=START) + if user_start > week_start: + skip.append(u) + elif len(u.weeks) <= week or not u.weeks[week]: + lame.append(u) + else: + good.append(u) + + return Template(filename=path, output_encoding='utf-8').render( + week=week, week_start=week_start,week_end=week_end, + good=good, lame=lame, skip=skip, userlist=userlist) + +if __name__ == '__main__': + if len(sys.argv) < 2: + print >>sys.stderr, "Usage: %s TEMPLATE [WEEK]" + sys.exit(1) + + template = sys.argv[1] + week = None + if len(sys.argv) > 2: week = sys.argv[2] + print render_template(template, week)