Merge pull request #4 from WnP/custom-template-class

Ability to definied a custom Template class

Author: iamjazzar

djangomako/backends.py

@@ -92,6 +92,9 @@ def __init__(self, parameters):
         # particular template URI
         options.setdefault('directories', self.template_dirs)
 
+        self.template_class = import_string(options.pop(
+            'template_class', 'djangomako.backends.Template'))
+
         self.engine = MakoEngine(**options)
 
     def from_string(self, template_code):
@@ -104,7 +107,7 @@ def from_string(self, template_code):
         :return: Returns a compiled Mako template.
         """
         try:
-            return Template(self.engine.from_string(template_code))
+            return self.template_class(self.engine.from_string(template_code))
         except mako_exceptions.SyntaxException as exc:
             raise TemplateSyntaxError(exc.args)
 
@@ -118,7 +121,7 @@ def get_template(self, template_name):
         :return: Compiled Template.
         """
         try:
-            return Template(self.engine.get_template(template_name))
+            return self.template_class(self.engine.get_template(template_name))
         except mako_exceptions.TemplateLookupException as exc:
             raise TemplateDoesNotExist(exc.args)
         except mako_exceptions.CompileException as exc:
@@ -159,4 +162,4 @@ def render(self, context=None, request=None):
             context['static'] = static
             context['url'] = reverse
 
-        return self.template.render(**context)
+        return self.template.render(**context)