verdandi/mixins/templatemixin.py

55 lines
1.3 KiB
Python

#!/usr/bin/env python2
import os
from dateutil import parser
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)
print "Rendering %s" % out_path
out_file = open(out_path, "wb")
out_file.write(result.encode('utf-8'))
out_file.close()
def read_content_file(self, path):
content_file = open(path, 'r')
result = {}
first_line = content_file.readline()
second_line = content_file.readline()
result['content'] = content_file.read().decode('utf-8')
result['creation_time'] = parser.parse(first_line)
result['edit_time'] = parser.parse(second_line)
content_file.close()
return result