import os
import os.path
import subprocess
+import re
from mako.template import Template
-START = datetime.datetime(2009, 12, 21, 6)
-HERE = os.path.dirname(__file__)
+from config import *
def get_balance(acct):
- p = subprocess.Popen(['ledger', '-f', os.path.join(HERE,'ledger'),
- '-n', 'balance', acct],
+ p = subprocess.Popen(BALANCE_CMD + [acct],
stdout=subprocess.PIPE)
(out, _) = p.communicate()
- return float(out.split()[0][1:])
+ return float(re.sub(r'\s*(\d+)\s+.*', r'\1', out))
def get_debts():
- p = subprocess.Popen(['ledger', '-f', os.path.join(HERE, 'ledger'),
- '-n', 'balance', 'Pool:Owed:'],
+ p = subprocess.Popen(DEBTS_CMD,
stdout=subprocess.PIPE)
(out, _) = p.communicate()
debts = []
if not line: continue
(val, acct) = line.split()
user = acct[len("Pool:Owed:"):]
- val = float(val[len("$"):])
+ if not user: continue
+ val = float(re.sub(r'(\D)?(\d+)$', r'\2', val))
debts.append((user, val))
return debts
+def to_week_num(date):
+ return (parse(date, default=START) - START).days / 7
+
+def parse_skip(rec):
+ spec = rec.get('skip', [])
+ out = []
+ for s in spec:
+ if isinstance(s, list):
+ out.append(map(to_week_num, s))
+ else:
+ out.append(to_week_num(s))
+ return out
+
+def should_skip(skips, week):
+ for e in skips:
+ if e == week:
+ return True
+ if isinstance(e, list) and e[0] <= week and e[1] > week:
+ return True
+ return False
+
def render_template(path, week=None, **kwargs):
with open('out/report.yml') as r:
report = yaml.safe_load(r)
lame = []
skip = []
userlist = []
+ punted = []
class User(object):
pass
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.skip = parse_skip(rec)
u.weeks = report.get(un, [])
userlist.append(u)
+ # create a subset of punted users
+ if u.end:
+ punted.append(u)
+
def user_key(u):
return (u.start, u.username)
userlist.sort(key=user_key)
+ punted.sort(key=user_key)
for u in userlist:
- user_start = parse(u.start, default=START)
- if u.end and parse(u.end, default=START) <= week_start:
+ user_start = datetime.datetime(*(u.start.timetuple()[:6]))
+ if u.end and datetime.datetime(*(u.end.timetuple()[:6])) <= week_start:
continue
-
- if week in u.skip:
+
+ if should_skip(u.skip, week):
pass
elif user_start > week_start:
skip.append(u)
debts = get_debts()
- return Template(filename=path, output_encoding='utf-8').render(
+ return Template(filename=path, input_encoding='utf-8',
+ output_encoding='utf-8',
+ default_filters=['decode.utf8']).render(
week=week, week_start=week_start,week_end=week_end,
good=good, lame=lame, skip=skip, userlist=userlist,
pool=get_balance('Pool'), paid=get_balance('Pool:Paid'),
- debts=debts, **kwargs)
+ debts=debts, punted=punted, currency=CURRENCY, fine=FINE_SIZE,
+ **kwargs)
if __name__ == '__main__':
if len(sys.argv) < 2: