clean up from the previous bug (remove post punt charges)

[iron-blogger] / weekly-update.py

diff --git a/weekly-update.py b/weekly-update.py
index ad6ef1a43095fce7926a0ef25d8562020cd9d90c..7a44dffe36cd2115147348f4ef159d177b51e3ec 100755 (executable)
--- a/weekly-update.py
+++ b/weekly-update.py
@@ -19,14 +19,14 @@ if args[0] == '-n':
     args = args[1:]
 
 date = args[0]
-today = str(datetime.date.today())
 
 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,)])
 
 debts = render.get_debts()
 punt = []
@@ -37,10 +37,10 @@ with open('ledger', 'a') as f:
         if debt < 30: continue
         punt.append(user)
         f.write("""\
-%(today)s Punt
+%(date)s Punt
   Pool:Owed:%(user)s  $-%(debt)s
   User:%(user)s
-""" % {'user': user, 'debt': debt, 'today': today})
+""" % {'user': user, 'debt': debt, 'date': date})
 
 
 if not dry_run:
@@ -76,9 +76,17 @@ if punt:
         bloggers = yaml.safe_load(b)
     for p in punt:
         if 'end' not in bloggers[p]:
-            bloggers[p]['end'] = today
+            bloggers[p]['end'] = date
     with open('bloggers.yml','w') as b:
         yaml.safe_dump(bloggers, b)
 
-    subprocess.check_call(["git", "commit", "ledger", "bloggers.yml",
-                           "-m", "Punts for %s" % (today,)])
+    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"])