From 77f0b522e1a16b1dba36432d5aa7a099bfe84403 Mon Sep 17 00:00:00 2001 From: rsiddharth Date: Sat, 19 Dec 2015 17:41:41 -0500 Subject: updated lps_gen.RenderHTML - Now handles the case when the template does not exist. - HTML output is now prettified by BeautifulSoup. --- lps_gen.py | 16 ++++++++++++++-- 1 file changed, 14 insertions(+), 2 deletions(-) (limited to 'lps_gen.py') diff --git a/lps_gen.py b/lps_gen.py index e4cd0e8..89d20e6 100644 --- a/lps_gen.py +++ b/lps_gen.py @@ -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(): -- cgit v1.2.3