64 lines
1.5 KiB
Python
64 lines
1.5 KiB
Python
#!/usr/bin/env python2
|
|
|
|
import os
|
|
import codecs
|
|
import copy
|
|
|
|
from dateutil import parser
|
|
|
|
class TemplateMixin(object):
|
|
template = "base.html"
|
|
url = "index.html"
|
|
context = {}
|
|
|
|
def get_context(self):
|
|
return copy.copy(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 = codecs.open(out_path, 'wb', 'utf-8')
|
|
out_file.write(result)
|
|
out_file.close()
|
|
|
|
|
|
def read_content_file(self, path):
|
|
content_file = codecs.open(path, 'r', 'utf-8')
|
|
|
|
result = {}
|
|
|
|
first_line = content_file.readline()
|
|
second_line = content_file.readline()
|
|
|
|
third_line = content_file.readline().strip()
|
|
while third_line == '':
|
|
third_line = content_file.readline().strip()
|
|
|
|
result['title'] = third_line
|
|
|
|
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
|