diff options
author | rsiddharth <rsd@gnu.org> | 2015-12-19 17:41:41 -0500 |
---|---|---|
committer | rsiddharth <rsd@gnu.org> | 2015-12-19 17:41:41 -0500 |
commit | 77f0b522e1a16b1dba36432d5aa7a099bfe84403 (patch) | |
tree | 799755555aa8dc9e7f219c65619a4f74e1650d79 /lps_gen.py | |
parent | bf36773eeb0c858d064e13c96a2fd2e93757f710 (diff) |
updated lps_gen.RenderHTML
- Now handles the case when the template does not exist.
- HTML output is now prettified by BeautifulSoup.
Diffstat (limited to 'lps_gen.py')
-rw-r--r-- | lps_gen.py | 16 |
1 files changed, 14 insertions, 2 deletions
@@ -25,7 +25,9 @@ from argparse import ArgumentParser from collections import OrderedDict from os import path +from bs4 import BeautifulSoup from jinja2 import Environment, PackageLoader +from jinja2.exceptions import TemplateNotFound from mistune import Renderer, Markdown @@ -146,9 +148,19 @@ def RenderHTML(lps_dict, year): env = Environment(loader=PackageLoader('lps_gen', 'templates'), trim_blocks=True, lstrip_blocks=True) - template = env.get_template('lp-sch-%s.jinja2' % year) - return template.render(schedule=lps_dict) + template_name = 'lp-sch-%s.jinja2' % year + template = None + + try: + template = env.get_template(template_name) + except TemplateNotFound as e: + print "Template %s not found." % template_name + return + + lps_html = template.render(schedule=lps_dict) + + return BeautifulSoup(lps_html, 'html.parser').prettify() def main(): |