Imported Upstream version 6.4.0upstream/6.4.0

2 files changed, 13 insertions, 0 deletions

diff --git a/nikola/plugins/template/jinja.py b/nikola/plugins/template/jinja.py
index 17c33d4..f14adfe 100644
--- a/nikola/plugins/template/jinja.py
+++ b/nikola/plugins/template/jinja.py
@@ -61,6 +61,11 @@ class JinjaTemplates(TemplateSystem):
         self.lookup.loader = jinja2.FileSystemLoader(directories,
                                                      encoding='utf-8')
 
+    def set_site(self, site):
+        """Sets the site."""
+        self.site = site
+        self.lookup.filters.update(self.site.config['TEMPLATE_FILTERS'])
+
     def render_template(self, template_name, output_name, context):
         """Render the template into output_name using context."""
         if jinja2 is None:
diff --git a/nikola/plugins/template/mako.py b/nikola/plugins/template/mako.py
index 45f4335..5a23230 100644
--- a/nikola/plugins/template/mako.py
+++ b/nikola/plugins/template/mako.py
@@ -49,6 +49,7 @@ class MakoTemplates(TemplateSystem):
 
     lookup = None
     cache = {}
+    filters = {}
 
     def get_deps(self, filename):
         text = util.read_file(filename)
@@ -81,6 +82,11 @@ class MakoTemplates(TemplateSystem):
             module_directory=cache_dir,
             output_encoding='utf-8')
 
+    def set_site(self, site):
+        """Sets the site."""
+        self.site = site
+        self.filters.update(self.site.config['TEMPLATE_FILTERS'])
+
     def render_template(self, template_name, output_name, context):
         """Render the template into output_name using context."""
         context['striphtml'] = striphtml
@@ -95,6 +101,8 @@ class MakoTemplates(TemplateSystem):
     def render_template_to_string(self, template, context):
         """ Render template to a string using context. """
 
+        context = context.update(self.filters)
+
         return Template(template).render(**context)
 
     def template_deps(self, template_name):