zk2epub/template.py

from jinja2 import FileSystemLoader, Environment

class Template:
    dir = ""
    env = None

    def __init__(self, template_dir):
        self.dir = template_dir
        self.env = Environment(loader=FileSystemLoader(template_dir))

    def get(self, file,  **kwargs):
        try:
            t = self.env.get_template(file)
            return t.render(**kwargs)
        except:
            return False