render.py: Wrap in a __name__ check.

render.py

@@ -7,23 +7,18 @@ import sys
 import os
 from mako.template import Template
 
-if len(sys.argv) < 2:
-    print >>sys.stderr, "Usage: %s TEMPLATE [WEEK]" 
-    sys.exit(1)
-
-template = sys.argv[1]
 START = datetime.datetime(2009, 12, 21, 6)
 
-if len(sys.argv) == 3:
-    week = parse(sys.argv[2], default=START)
-else:
-    week = START
-
+def render_template(path, week=None):
     with open('out/report.yml') as r:
         report = yaml.safe_load(r)
 
     with open('bloggers.yml') as f:
         users = yaml.safe_load(f)
+    if week:
+        week = parse(week, default=START)
+    else:
+        week = START
 
     week = (week - START).days / 7
     week_start = START + (week * datetime.timedelta(7))
@@ -60,6 +55,16 @@ for u in userlist:
         else:
             good.append(u)
 
-print Template(filename=template, output_encoding='utf-8').render(
+    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)
+
+if __name__ == '__main__':
+    if len(sys.argv) < 2:
+        print >>sys.stderr, "Usage: %s TEMPLATE [WEEK]"
+        sys.exit(1)
+
+    template = sys.argv[1]
+    week = None
+    if len(sys.argv) > 2: week = sys.argv[2]
+    print render_template(template, week)