import sys
import xmlrpclib
import subprocess
+import datetime
+import yaml
-XMLRPC_ENDPOINT = 'http://iron-blogger.mit.edu/xmlrpc.php'
-USER = 'nelhage'
-BLOG_ID = 1
+from config import *
+
+dry_run = False
args = sys.argv[1:]
if args[0] == '-n':
date = args[0]
-try:
- subprocess.call(['stty', '-echo'])
- passwd = raw_input("Password for %s: " % (USER,))
- print
-finally:
- subprocess.call(['stty', 'echo'])
-
with open('ledger', 'a') as f:
f.write("\n")
f.write(render.render_template('templates/ledger', date))
-subprocess.check_call(["git", "commit", "ledger",
- "-m", "Update for %s" % (date,)])
+if not dry_run:
+ subprocess.check_call(["git", "commit", "ledger",
+ "-m", "Update for %s" % (date,)])
-text = render.render_template('templates/week.tmpl', date)
+debts = render.get_debts()
+punt = []
-lines = text.split("\n")
-title = lines[0]
-body = "\n".join(lines[1:])
+with open('ledger', 'a') as f:
+ f.write("\n")
+ for (user, debt) in debts:
+ if debt <= (FINE_SIZE * 6): continue
+ punt.append(user)
+ f.write("""\
+%(date)s Punt
+ Pool:Owed:%(user)s -%(debt)s
+ User:%(user)s
+""" % {'user': user, 'debt': debt, 'date': date})
-page = dict(title = title,
- description = body)
if not dry_run:
+ text = render.render_template('templates/week.tmpl', date, punt=punt)
+
+ lines = text.split("\n")
+ title = lines[0]
+ body = "\n".join(lines[1:])
+
+ page = dict(title = title, description = body)
+
+ try:
+ subprocess.call(['stty', '-echo'])
+ passwd = raw_input("Password for %s: " % (USER,))
+ print
+ finally:
+ subprocess.call(['stty', 'echo'])
+
x = xmlrpclib.ServerProxy(XMLRPC_ENDPOINT)
x.metaWeblog.newPost(BLOG_ID, USER, passwd, page, True)
-email = render.render_template('templates/email.txt', date)
+email = render.render_template('templates/email.txt', date, punt=punt)
if dry_run:
print email
p = subprocess.Popen(['mutt', '-H', '/dev/stdin'],
stdin=subprocess.PIPE)
p.communicate(email)
+
+if punt:
+ with open('bloggers.yml') as b:
+ bloggers = yaml.safe_load(b)
+ for p in punt:
+ if 'end' not in bloggers[p]:
+ bloggers[p]['end'] = datetime.date(*map(int, date.split("-")))
+ with open('bloggers.yml','w') as b:
+ yaml.safe_dump(bloggers, b)
+
+ if not dry_run:
+ subprocess.check_call(["git", "commit", "ledger", "bloggers.yml",
+ "-m", "Punts for %s" % (date,)])
+
+# if it's a dry run, lets set the ledger back to the beginning state
+if dry_run:
+ subprocess.check_call(["git", "checkout", "ledger"])
+
+ if punt:
+ subprocess.check_call(["git", "checkout", "bloggers.yml"])