X-Git-Url: https://projects.mako.cc/source/iron-blogger/blobdiff_plain/75eca5ec298ea59b5dbd3f968f5104cd2a690579..8230262ecfa6acf99522c2479297d25e067d7572:/render.py diff --git a/render.py b/render.py index d5c9e51..6e4e80c 100755 --- a/render.py +++ b/render.py @@ -5,61 +5,124 @@ import datetime import dateutil.tz as tz import sys import os +import os.path +import subprocess from mako.template import Template -if len(sys.argv) < 2: - print >>sys.stderr, "Usage: %s TEMPLATE [WEEK]" - sys.exit(1) +START = datetime.datetime(2011, 10, 24, 6) +HERE = os.path.dirname(__file__) -template = sys.argv[1] -START = datetime.datetime(2009, 12, 21, 6) +def get_balance(acct): + p = subprocess.Popen(['ledger', '-f', os.path.join(HERE,'ledger'), + '-n', 'balance', acct], + stdout=subprocess.PIPE) + (out, _) = p.communicate() + return float(out.split()[0][1:]) -if len(sys.argv) == 3: - week = parse(sys.argv[2], default=START) -else: - week = START +def get_debts(): + p = subprocess.Popen(['ledger', '-f', + os.path.join(HERE, 'ledger'), + '--no-color', '--flat', + '--no-total', 'balance', 'Pool:Owed:'], + stdout=subprocess.PIPE) + (out, _) = p.communicate() + debts = [] + for line in out.split("\n"): + if not line: continue + (val, acct) = line.split() + user = acct[len("Pool:Owed:"):] + val = float(val[len("$"):]) + debts.append((user, val)) + return debts -with open('out/report.yml') as r: - report = yaml.safe_load(r) +def to_week_num(date): + return (parse(date, default=START) - START).days / 7 -with open('bloggers.yml') as f: - users = yaml.safe_load(f) +def parse_skip(rec): + spec = rec.get('skip', []) + out = [] + for s in spec: + if isinstance(s, list): + out.append(map(to_week_num, s)) + else: + out.append(to_week_num(s)) + return out -week = (week - START).days / 7 -week_start = START + (week * datetime.timedelta(7)) -week_end = START + ((week + 1) * datetime.timedelta(7)) +def should_skip(skips, week): + for e in skips: + if e == week: + return True + if isinstance(e, list) and e[0] <= week and e[1] > week: + return True + return False -good = [] -lame = [] -skip = [] -userlist = [] +def render_template(path, week=None, **kwargs): + with open('out/report.yml') as r: + report = yaml.safe_load(r) -class User(object): - pass + with open('bloggers.yml') as f: + users = yaml.safe_load(f) + if week: + week = parse(week, default=START) + else: + week = START -for (un, rec) in users.items(): - u = User() - u.username = un - u.links = rec['links'] - u.start = rec['start'] - u.weeks = report.get(un, []) + week = (week - START).days / 7 + week_start = START + (week * datetime.timedelta(7)) + week_end = START + ((week + 1) * datetime.timedelta(7)) - userlist.append(u) + good = [] + lame = [] + skip = [] + userlist = [] -def user_key(u): - return (u.start, u.username) + class User(object): + pass -userlist.sort(key=user_key) + for (un, rec) in users.items(): + u = User() + u.username = un + u.links = rec['links'] + u.start = rec['start'] + u.end = rec.get('end') + u.skip = parse_skip(rec) + u.weeks = report.get(un, []) -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) + 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 u.end and parse(u.end, default=START) <= week_start: + continue + + if should_skip(u.skip, week): + pass + elif user_start > week_start: + skip.append(u) + elif len(u.weeks) <= week or not u.weeks[week]: + lame.append(u) + else: + good.append(u) + + debts = get_debts() + + 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, + pool=get_balance('Pool'), paid=get_balance('Pool:Paid'), + debts=debts, **kwargs) + +if __name__ == '__main__': + if len(sys.argv) < 2: + print >>sys.stderr, "Usage: %s TEMPLATE [WEEK]" + sys.exit(1) -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) + template = sys.argv[1] + week = None + if len(sys.argv) > 2: week = sys.argv[2] + print render_template(template, week)