from jinja2 import Template from md_pdf.consts import TEMPLATES_DIR from md_pdf.build.context import get_context import os import logging logger = logging.getLogger(__file__) FILE_NAME_FORMAT = os.path.join(TEMPLATES_DIR, "{}.html") TEMPLATE_FORMAT = os.path.join(TEMPLATES_DIR, "{}-template.html") def render_page(input_file, output_file, context): logger.debug("Rendering {}...".format(os.path.splitext(os.path.basename(output_file))[0].title())) with open(input_file) as f: template = Template(f.read()) with open(output_file, "w") as f: cover = template.render(context) f.write(cover) return cover def render_templates(config, content): context = get_context(config, content) for template in [ 'cover', 'header', 'footer' ]: render_page(TEMPLATE_FORMAT.format(template), FILE_NAME_FORMAT.format(template), context) if config.get('toc', False): render_page(os.path.join(TEMPLATES_DIR, 'toc-template.xsl'), os.path.join(TEMPLATES_DIR, 'toc.xsl'), context)