Small email template update

[iron-blogger] / render.py

diff --git a/render.py b/render.py
index f8cdd7a93e54ef57de46a68278066fee4a8bb664..8cda0d52b3808d8bead38e3229242d17b9ca3632 100755 (executable)
--- a/render.py
+++ b/render.py
@@ -5,9 +5,12 @@ import datetime
 import dateutil.tz as tz
 import sys
 import os
+import os.path
+import subprocess
 from mako.template import Template
 
 START = datetime.datetime(2009, 12, 21, 6)
+HERE  = os.path.dirname(__file__)
 
 def render_template(path, week=None):
     with open('out/report.yml') as r:
@@ -37,6 +40,9 @@ def render_template(path, week=None):
         u.username = un
         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.weeks = report.get(un, [])
 
         userlist.append(u)
@@ -48,16 +54,28 @@ def render_template(path, week=None):
 
     for u in userlist:
         user_start = parse(u.start, default=START)
-        if user_start > week_start:
+        if u.end and parse(u.end, default=START) <= week_start:
+            continue
+
+        if week in u.skip:
+            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)
 
+    p = subprocess.Popen(['ledger', '-f', os.path.join(HERE,'ledger'),
+                          '-n', 'balance', 'Pool'],
+                         stdout=subprocess.PIPE)
+    (out, _) = p.communicate()
+    pool = int(out.split()[0][1:])
+
     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)
+        good=good, lame=lame, skip=skip, userlist=userlist,
+        pool=pool)
 
 if __name__ == '__main__':
     if len(sys.argv) < 2: