Added custom template loader to CMS in order to enable {% include %} and {% import %} in jinja temp…

sitescripts/cms/converters.py

 # coding: utf-8
 
 # This file is part of the Adblock Plus web scripts,
 # Copyright (C) 2006-2013 Eyeo GmbH
 #
 # Adblock Plus is free software: you can redistribute it and/or modify
 # it under the terms of the GNU General Public License version 3 as
 # published by the Free Software Foundation.
 #
 # Adblock Plus is distributed in the hope that it will be useful,
@@ skipping 154 matching lines @@
     md = markdown.Markdown(output="html5", extensions=["attr_list"])
     md.preprocessors["html_block"].markdown_in_raw = True
 
     result = self.insert_localized_strings(source, escapes)
     result = md.convert(result)
     result = re.sub(r"&#(\d+);", remove_unnecessary_entities, result)
     result = self.process_links(result)
     return result
 
 class TemplateConverter(Converter):
+  class _SourceLoader(jinja2.BaseLoader):
+    def __init__(self, source):
+      self.source = source
+
+    def get_source(self, environment, template):
+      try:
+        return self.source.read_include(template, "tmpl"), None, None
+      except Exception:
+        raise jinja2.TemplateNotFound(template)
+
   def __init__(self, *args, **kwargs):
     Converter.__init__(self, *args, **kwargs)
 
     filters = {
       "translate": self.translate,
       "linkify": self.linkify,
       "toclist": self.toclist,
     }
 
     for filename in self._params["source"].list_files("filters"):
       root, ext = os.path.splitext(filename)
       if ext.lower() != ".py":
         continue
 
       path = "%s/%s" % ("filters", filename)
       code = self._params["source"].read_file(path)
       module = imp.new_module(root.replace("/", "."))
       exec code in module.__dict__
 
       func = os.path.basename(root)
       if not hasattr(module, func):
         raise Exception("Expected function %s not found in filter file %s" % (func, filename))
       filters[func] = getattr(module, func)
       filters[func].module_ref = module  # Prevent garbage collection
 
-    self._env = get_custom_template_environment(filters)
+    self._env = get_custom_template_environment(filters, self._SourceLoader(self._params["source"]))
 
   def get_html(self, source):
     template = self._env.from_string(source)
     return template.render(self._params)
 
   def translate(self, name, page=None, links=[]):
     if page == None:
       localedata = self._params["localedata"]
     else:
       localedata = self._params["source"].read_locale(self._params["locale"], page)
@@ skipping 26 matching lines @@
         stack.pop()
       stack[-1]["subitems"].append(item)
       stack.append(item)
     return structured
 
 converters = {
   "raw": RawConverter,
   "md": MarkdownConverter,
   "tmpl": TemplateConverter,
 }