X-Git-Url: https://projects.mako.cc/source/iron-blogger/blobdiff_plain/d9449b6b588a35f54b61698de5f82ebde11af118..8230262ecfa6acf99522c2479297d25e067d7572:/render.py diff --git a/render.py b/render.py index cacd90d..6e4e80c 100755 --- a/render.py +++ b/render.py @@ -9,7 +9,7 @@ import os.path import subprocess from mako.template import Template -START = datetime.datetime(2009, 12, 21, 6) +START = datetime.datetime(2011, 10, 24, 6) HERE = os.path.dirname(__file__) def get_balance(acct): @@ -20,8 +20,10 @@ def get_balance(acct): return float(out.split()[0][1:]) def get_debts(): - p = subprocess.Popen(['ledger', '-f', os.path.join(HERE, 'ledger'), - '-n', 'balance', 'Pool:Owed:'], + 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 = [] @@ -33,6 +35,27 @@ def get_debts(): 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) @@ -62,8 +85,7 @@ def render_template(path, week=None, **kwargs): 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) @@ -78,7 +100,7 @@ def render_template(path, week=None, **kwargs): if u.end and parse(u.end, default=START) <= week_start: continue - if week in u.skip: + if should_skip(u.skip, week): pass elif user_start > week_start: skip.append(u)