From: Nelson Elhage Date: Tue, 1 Jun 2010 00:40:55 +0000 (-0400) Subject: render.py: Break out some more code X-Git-Url: https://projects.mako.cc/source/iron-blogger/commitdiff_plain/81be85441f9314f889d3bccb73c2f9e5fa70fb4a render.py: Break out some more code --- diff --git a/render.py b/render.py index 0fa0489..f226adb 100755 --- 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,