render.py: Wrap in a __name__ check.
This commit is contained in:
parent
f3ffe2ec15
commit
128d932eff
27
render.py
27
render.py
|
@ -7,23 +7,18 @@ 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)
|
|
||||||
else:
|
|
||||||
week = START
|
|
||||||
|
|
||||||
with open('out/report.yml') as r:
|
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))
|
||||||
|
@ -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)
|
||||||
|
|
Loading…
Reference in a new issue