render.py: Break out some more code
authorNelson Elhage <nelhage@ksplice.com>
Tue, 1 Jun 2010 00:40:55 +0000 (20:40 -0400)
committerNelson Elhage <nelhage@ksplice.com>
Tue, 1 Jun 2010 00:40:55 +0000 (20:40 -0400)
render.py

index 0fa0489434bcea6f0222c511f23adaea04f7c25a..f226adb906448fa945efda8ea05b7bd169f22073 100755 (executable)
--- a/render.py
+++ b/render.py
@@ -12,6 +12,27 @@ from mako.template import Template
 START = datetime.datetime(2009, 12, 21, 6)
 HERE  = os.path.dirname(__file__)
 
+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:])
+
+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
+
 def render_template(path, week=None):
     with open('out/report.yml') as r:
         report = yaml.safe_load(r)
@@ -66,24 +87,7 @@ 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 float(out.split()[0][1:])
-
-    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))
+    debts = get_debts()
 
     return Template(filename=path, output_encoding='utf-8').render(
         week=week, week_start=week_start,week_end=week_end,

Benjamin Mako Hill || Want to submit a patch?