X-Git-Url: https://projects.mako.cc/source/iron-blogger/blobdiff_plain/b60b22c20ab2666492cad025c5c3cb0e1b04d03f..d9449b6b588a35f54b61698de5f82ebde11af118:/render.py diff --git a/render.py b/render.py index 8cda0d5..cacd90d 100755 --- a/render.py +++ b/render.py @@ -12,7 +12,28 @@ from mako.template import Template START = datetime.datetime(2009, 12, 21, 6) HERE = os.path.dirname(__file__) -def render_template(path, week=None): +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, **kwargs): with open('out/report.yml') as r: report = yaml.safe_load(r) @@ -66,16 +87,13 @@ def render_template(path, week=None): 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:]) + debts = get_debts() 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, **kwargs) if __name__ == '__main__': if len(sys.argv) < 2: