verdandi/verdandi/mixins/templatemixin.py

33 lines
841 B
Python

#!/usr/bin/env python2
import os
class TemplateMixin(object):
template = "base.html"
url = "index.html"
context = {}
def get_context(self):
return self.context
def render(self, output_directory, jinja_env):
context = self.get_context()
self.render_files(context, output_directory, jinja_env)
def render_files(self, context, output_directory, jinja_env):
self.render_to_file(self.template, self.url, context, output_directory, jinja_env)
def render_to_file(self, template, url, context, output_directory, jinja_env):
template = jinja_env.get_template(template)
result = template.render(context)
out_path = os.path.join(output_directory, url)
out_dir = os.path.dirname(out_path)
if not os.path.isdir(out_dir):
os.mkdir(out_dir)
out_file = open(out_path, "wb")
out_file.write(result)
out_file.close()