X-Git-Url: https://projects.mako.cc/source/iron-blogger/blobdiff_plain/e1ebd5c9544ef58bb29c9b9e68b469e6b2d3e60f..4b123d38b4f031e226e03d3780432f92b153e6e1:/render.py

diff --git a/render.py b/render.py
index dc4ad3d..0a7bf46 100755
--- a/render.py
+++ b/render.py
@@ -41,6 +41,8 @@ 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.weeks = report.get(un, [])
 
         userlist.append(u)
@@ -55,23 +57,39 @@ def render_template(path, week=None):
         if u.end and parse(u.end, default=START) <= week_start:
             continue
 
-        if user_start > week_start:
+        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)
 
+    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'],
+                          '-n', 'balance', 'Pool:Owed:'],
                          stdout=subprocess.PIPE)
     (out, _) = p.communicate()
-    pool = int(out.split()[0][1:])
+    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))
 
     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=pool)
+        pool=get_balance('Pool'), paid=get_balance('Pool:Paid'),
+        debts=debts)
 
 if __name__ == '__main__':
     if len(sys.argv) < 2: