X-Git-Url: https://projects.mako.cc/source/iron-blogger/blobdiff_plain/75eca5ec298ea59b5dbd3f968f5104cd2a690579..c20c8ed5383102aaedd97704bc925851ca8270ec:/render.py

diff --git a/render.py b/render.py
index d5c9e51..0e320bf 100755
--- a/render.py
+++ b/render.py
@@ -5,61 +5,122 @@ 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)
-
-template = sys.argv[1]
 START = datetime.datetime(2009, 12, 21, 6)
+HERE  = os.path.dirname(__file__)
 
-if len(sys.argv) == 3:
-    week = parse(sys.argv[2], default=START)
-else:
-    week = START
+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:])
 
-with open('out/report.yml') as r:
-    report = yaml.safe_load(r)
+def get_debts():
+    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  = float(val[len("$"):])
+        debts.append((user, val))
+    return debts
 
-with open('bloggers.yml') as f:
-    users = yaml.safe_load(f)
+def to_week_num(date):
+    return (parse(date, default=START) - START).days / 7
 
-week = (week - START).days / 7
-week_start = START + (week * datetime.timedelta(7))
-week_end   = START + ((week + 1) * datetime.timedelta(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
 
-good = []
-lame = []
-skip = []
-userlist = []
+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
 
-class User(object):
-    pass
+def render_template(path, week=None, **kwargs):
+    with open('out/report.yml') as r:
+        report = yaml.safe_load(r)
 
-for (un, rec) in users.items():
-    u = User()
-    u.username = un
-    u.links = rec['links']
-    u.start = rec['start']
-    u.weeks = report.get(un, [])
+    with open('bloggers.yml') as f:
+        users = yaml.safe_load(f)
+    if week:
+        week = parse(week, default=START)
+    else:
+        week = START
 
-    userlist.append(u)
+    week = (week - START).days / 7
+    week_start = START + (week * datetime.timedelta(7))
+    week_end   = START + ((week + 1) * datetime.timedelta(7))
 
-def user_key(u):
-    return (u.start, u.username)
+    good = []
+    lame = []
+    skip = []
+    userlist = []
 
-userlist.sort(key=user_key)
+    class User(object):
+        pass
 
-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)
+    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, [])
+
+        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)