From: Nelson Elhage <nelhage@ksplice.com>
Date: Mon, 24 May 2010 00:33:53 +0000 (-0400)
Subject: Include more monetary information in reports.
X-Git-Url: https://projects.mako.cc/source/iron-blogger/commitdiff_plain/4b123d38b4f031e226e03d3780432f92b153e6e1

Include more monetary information in reports.
---

diff --git a/render.py b/render.py
index 8cda0d5..0a7bf46 100755
--- a/render.py
+++ b/render.py
@@ -66,16 +66,30 @@ def render_template(path, week=None):
         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:
diff --git a/templates/email.txt b/templates/email.txt
index aab3428..e82b15f 100644
--- a/templates/email.txt
+++ b/templates/email.txt
@@ -21,3 +21,9 @@ People who have not yet started:
 Beer pool:
 This week: $${5 * len(lame)}.00
 Total:     $${pool}.00
+Paid:      $${paid}.00
+
+Individual debts:
+% for (u, v) in sorted(debts, key=lambda p:p[1], reverse=True):
+${"%20s $%d" % (u, v)}
+% endfor
diff --git a/templates/week.tmpl b/templates/week.tmpl
index 68f6474..4ed842f 100644
--- a/templates/week.tmpl
+++ b/templates/week.tmpl
@@ -31,5 +31,13 @@ Results for week beginning ${week_start.strftime("%F")}
 <h2>Beer pool:</h2>
 <table>
   <tr> <td> This week: </td> <td> $${5 * len(lame)}.00 </td> </tr>
-  <tr> <td> Total: </td> <td> ${pool}.00 </td> </tr>
+  <tr> <td> Total: </td> <td> $${pool}.00 </td> </tr>
+  <tr> <td> Paid: </td> <td> $${paid}.00 </td> </tr>
+</table>
+
+<h2>Debts:</h2>
+<table>
+% for (u, v) in sorted(debts, key=lambda p:p[1], reverse=True):
+<tr><td>${u}</td> <td>${v}</td></tr>
+% endfor
 </table>