template_functions.py

from jinja2 import Environment, FileSystemLoader, select_autoescape
import os, logging
import socket, datetime


def get_jinja_env():
    return Environment(
        loader=FileSystemLoader('templates'),
        autoescape=select_autoescape(['html', 'xml'])
    )


def get_jinja_template(jinja_env, template_name):
    return jinja_env.get_template(template_name)


def build_template(jinja_env, template_name, context, output_name):
    os.makedirs(os.path.dirname(output_name), exist_ok=True)
    logging.info("Building template {} to {}".format(template_name, output_name))
    template = get_jinja_template(jinja_env, template_name)
    with open(output_name, "w") as f:
        f.write(template.render(context))


def get_context(context):
    return dict(context, **{"create_hostname": socket.getfqdn(),
                            "create_time": datetime.datetime.now().strftime("%H:%M"),
                            "create_date": datetime.datetime.now().strftime("%d/%m/%Y"),})