X-Git-Url: https://projects.mako.cc/source/iron-blogger/blobdiff_plain/4b123d38b4f031e226e03d3780432f92b153e6e1..11a8aa29c39adc2ae33af0fd5791ff918288c6da:/render.py?ds=sidebyside diff --git a/render.py b/render.py index 0a7bf46..58cb92e 100755 --- a/render.py +++ b/render.py @@ -7,12 +7,53 @@ import sys import os import os.path import subprocess +import re from mako.template import Template -START = datetime.datetime(2009, 12, 21, 6) -HERE = os.path.dirname(__file__) +from config import * -def render_template(path, week=None): +def get_balance(acct): + p = subprocess.Popen(BALANCE_CMD + [acct], + stdout=subprocess.PIPE) + (out, _) = p.communicate() + return float(re.sub(r'\s*(\d+)\s+.*', r'\1', out)) + +def get_debts(): + p = subprocess.Popen(DEBTS_CMD, + 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:"):] + if not user: continue + val = float(re.sub(r'(\D)?(\d+)$', r'\2', val)) + debts.append((user, val)) + return debts + +def to_week_num(date): + return (parse(date, default=START) - START).days / 7 + +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 + +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 + +def render_template(path, week=None, **kwargs): with open('out/report.yml') as r: report = yaml.safe_load(r) @@ -31,6 +72,7 @@ def render_template(path, week=None): lame = [] skip = [] userlist = [] + punted = [] class User(object): pass @@ -41,23 +83,27 @@ def render_template(path, week=None): u.links = rec['links'] u.start = rec['start'] u.end = rec.get('end') - u.skip = [(parse(x, default=START) - START).days / 7 - for x in rec.get('skip', [])] + u.skip = parse_skip(rec) u.weeks = report.get(un, []) userlist.append(u) + # create a subset of punted users + if u.end: + punted.append(u) + def user_key(u): return (u.start, u.username) userlist.sort(key=user_key) + punted.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: + user_start = datetime.datetime(*(u.start.timetuple()[:6])) + if u.end and datetime.datetime(*(u.end.timetuple()[:6])) <= week_start: continue - - if week in u.skip: + + if should_skip(u.skip, week): pass elif user_start > week_start: skip.append(u) @@ -66,30 +112,16 @@ def render_template(path, week=None): else: good.append(u) - def get_balance(acct): - p = subprocess.Popen(['ledger', '-f', os.path.join(HERE,'ledger'), - '-n', 'balance', acct], - stdout=subprocess.PIPE) - (out, _) = p.communicate() - return int(out.split()[0][1:]) - - p = subprocess.Popen(['ledger', '-f', os.path.join(HERE,'ledger'), - '-n', '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 = int(val[len("$"):]) - debts.append((user, val)) + debts = get_debts() - return Template(filename=path, output_encoding='utf-8').render( + return Template(filename=path, input_encoding='utf-8', + output_encoding='utf-8', + default_filters=['decode.utf8']).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) + debts=debts, punted=punted, currency=CURRENCY, fine=FINE_SIZE, + **kwargs) if __name__ == '__main__': if len(sys.argv) < 2: