Adding Mako TurboGears plugin support.

2 files changed, 43 insertions, 0 deletions

diff --git a/lib/mako/plugin.py b/lib/mako/plugin.py
new file mode 100644
index 0000000..d579eed
--- /dev/null
+++ b/lib/mako/plugin.py
@@ -0,0 +1,39 @@
+import re
+from mako.lookup import TemplateLookup
+
+class TGPlugin(object):
+    """TurboGears compatible Template Plugin."""
+    extension = 'mak'
+
+    def __init__(self, extra_vars_func=None, options=None):
+        self.extra_vars_func = extra_vars_func
+        if not options:
+            options = {}
+
+        # Pull the options out and initialize the lookup
+        tmpl_options = {}
+        for k, v in options.iteritems():
+            if k.startswith('mako.'):
+                tmpl_options[k[5:]] = v
+        self.lookup = TemplateLookup(**tmpl_options)
+
+    def load_template(self, templatename):
+        """Unused method for TG plug-in API compat"""
+        pass
+
+    def render(self, info, format="html", fragment=False, template=None):
+        # Translate TG dot notation to normal / template path
+        if '/' and '.' not in template:
+            template = '/' + template.replace('.', '/') + '.' + self.extension
+
+        if not template.startswith('/'):
+            template = '/' + template
+
+        # Load extra vars func if provided
+        if self.extra_vars_func:
+            info.update(self.extra_vars_func())
+
+        # Lookup template
+        template = self.lookup.get_template(template)
+        return template.render(**info)
+
diff --git a/setup.py b/setup.py
index f729093..0dc1afd 100644
--- a/setup.py
+++ b/setup.py
@@ -37,4 +37,8 @@ The latest version is available in a `Subversion repository
       install_requires=[
           'MyghtyUtils',
       ],
+      entry_points="""
+      [python.templating.engines]
+      mako = mako.plugin:TGPlugin
+      """,
 )