iron-blogger/iron-blogger
1
1
Fork 0
Code Issues Pull requests Releases Wiki Activity

render.py: Wrap in a __name__ check.

Browse source
This commit is contained in:
Nelson Elhage 2009-12-30 00:34:35 -05:00
parent f3ffe2ec15
commit 128d932eff
1 changed files with 56 additions and 51 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

55
render.py
View file

@ -7,37 +7,32 @@ import sys
import os import os
from mako.template import Template 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) START = datetime.datetime(2009, 12, 21, 6)
if len(sys.argv) == 3: def render_template(path, week=None):
week = parse(sys.argv[2], default=START) with open('out/report.yml') as r:
else:
week = START
with open('out/report.yml') as r:
report = yaml.safe_load(r) report = yaml.safe_load(r)
with open('bloggers.yml') as f: with open('bloggers.yml') as f:
users = yaml.safe_load(f) users = yaml.safe_load(f)
if week:
week = parse(week, default=START)
else:
week = START
week = (week - START).days / 7 week = (week - START).days / 7
week_start = START + (week * datetime.timedelta(7)) week_start = START + (week * datetime.timedelta(7))
week_end = START + ((week + 1) * datetime.timedelta(7)) week_end = START + ((week + 1) * datetime.timedelta(7))
good = [] good = []
lame = [] lame = []
skip = [] skip = []
userlist = [] userlist = []
class User(object): class User(object):
pass pass
for (un, rec) in users.items(): for (un, rec) in users.items():
u = User() u = User()
u.username = un u.username = un
u.links = rec['links'] u.links = rec['links']
@ -46,12 +41,12 @@ for (un, rec) in users.items():
userlist.append(u) userlist.append(u)
def user_key(u): def user_key(u):
return (u.start, u.username) return (u.start, u.username)
userlist.sort(key=user_key) userlist.sort(key=user_key)
for u in userlist: for u in userlist:
user_start = parse(u.start, default=START) user_start = parse(u.start, default=START)
if user_start > week_start: if user_start > week_start:
skip.append(u) skip.append(u)
@ -60,6 +55,16 @@ for u in userlist:
else: else:
good.append(u) 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, week=week, week_start=week_start,week_end=week_end,
good=good, lame=lame, skip=skip, userlist=userlist) 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)