diff piecrust/templating/pystacheengine.py @ 185:139179dc7abd

render: Add support for a Mustache template engine. Add unit tests for the new class.
author Ludovic Chabant <ludovic@chabant.com>
date Sun, 04 Jan 2015 14:59:12 -0800
parents
children f4b7c8f183a4
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/piecrust/templating/pystacheengine.py	Sun Jan 04 14:59:12 2015 -0800
@@ -0,0 +1,57 @@
+import logging
+import pystache
+from piecrust.templating.base import (
+        TemplateEngine, TemplateNotFoundError, TemplatingError)
+
+
+logger = logging.getLogger(__name__)
+
+
+class PystacheTemplateEngine(TemplateEngine):
+    ENGINE_NAMES = ['mustache']
+    EXTENSIONS = ['mustache']
+
+    def __init__(self):
+        self.renderer = None
+
+    def renderString(self, txt, data, filename=None):
+        self._ensureLoaded()
+        try:
+            return self.renderer.render(txt, data)
+        except pystache.TemplateNotFoundError as ex:
+            raise TemplateNotFoundError() from ex
+        except pystache.PystacheError as ex:
+            raise TemplatingError(str(ex), filename) from ex
+
+    def renderFile(self, paths, data):
+        self._ensureLoaded()
+        tpl = None
+        logger.debug("Looking for template: %s" % paths)
+        for p in paths:
+            if not p.endswith('.mustache'):
+                raise TemplatingError(
+                        "The Mustache template engine only accepts template "
+                        "filenames with a `.mustache` extension. Got: %s" %
+                        p)
+            name = p[:-9]  # strip `.mustache`
+            try:
+                tpl = self.renderer.load_template(name)
+            except Exception as ex:
+                print(p, ex)
+                pass
+
+        if tpl is None:
+            raise TemplateNotFoundError()
+
+        try:
+            return self.renderer.render(tpl, data)
+        except pystache.PystacheError as ex:
+            raise TemplatingError(str(ex)) from ex
+
+    def _ensureLoaded(self):
+        if self.renderer:
+            return
+
+        self.renderer = pystache.Renderer(
+                search_dirs=self.app.templates_dirs)
+