update the rest of the code to support the new config format

[iron-blogger] / render.py

diff --git a/render.py b/render.py
index f8cdd7a93e54ef57de46a68278066fee4a8bb664..6d6415eea204e612606aa966cb93394d2fa423d0 100755 (executable)
--- a/render.py
+++ b/render.py
@@ -5,11 +5,57 @@ import datetime
 import dateutil.tz as tz
 import sys
 import os
 import dateutil.tz as tz
 import sys
 import os

+import os.path
+import subprocess
+import re

 from mako.template import Template
 
 from mako.template import Template
 

-START = datetime.datetime(2009, 12, 21, 6)
+from config import *

 
 

-def render_template(path, week=None):
+def get_balance(acct):
+    balance_cmd_tmp = BALANCE_CMD
+    balance_cmd_tmp.append(acct)
+    p = subprocess.Popen(balance_cmd_tmp,
+                         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)
 
     with open('out/report.yml') as r:
         report = yaml.safe_load(r)
 


@@ -28,6 +74,7 @@ def render_template(path, week=None):
     lame = []
     skip = []
     userlist = []
     lame = []
     skip = []
     userlist = []

+    punted = []

 
     class User(object):
         pass
 
     class User(object):
         pass


@@ -37,27 +84,46 @@ def render_template(path, week=None):
         u.username = un
         u.links = rec['links']
         u.start = rec['start']
         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)
 
         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)
     def user_key(u):
         return (u.start, u.username)
 
     userlist.sort(key=user_key)

+    punted.sort(key=user_key)

 
     for u in userlist:
 
     for u in userlist:

-        user_start = parse(u.start, default=START)
-        if user_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 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)
 
             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(
+    debts = get_debts()
+
+    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,
         week=week, week_start=week_start,week_end=week_end,

-        good=good, lame=lame, skip=skip, userlist=userlist)
+        good=good, lame=lame, skip=skip, userlist=userlist,
+        pool=get_balance('Pool'), paid=get_balance('Pool:Paid'),
+        debts=debts, punted=punted, currency=CURRENCY, fine=FINE_SIZE,
+        **kwargs)

 
 if __name__ == '__main__':
     if len(sys.argv) < 2:
 
 if __name__ == '__main__':
     if len(sys.argv) < 2: