Ticket #582: template_eggs-2.patch

django/core/template_eggs.py

@@ +1 @@
+"""Wrapper for loading templates from eggs via pkg_resources.resource_string."""
+
+try:
+    from pkg_resources import resource_string
+except ImportError:
+    # python2.3 compatibility hack
+    def resource_string(*args):
+        raise AssertionError, 'pkg_resource required for template_eggs'
+
+from django.conf.settings import INSTALLED_APPS, TEMPLATE_FILE_EXTENSION
+from django.core.template import TemplateDoesNotExist
+
+def load_template_source(name, dirs=None):
+    """I load templates from Python eggs via pkg_resource.resource_string.
+
+    For every install app I try to get the resource (app, name)."""
+
+    pkg_name = 'templates/' + name + TEMPLATE_FILE_EXTENSION
+
+    for app in INSTALLED_APPS:
+        try:
+            return resource_string(app, pkg_name)
+        except:
+            pass
+
+    raise TemplateDoesNotExist, name

django/core/template_loader.py

@@ -1 +1 @@
 "Wrapper for loading templates from storage of some sort (e.g. files or db)"
 import template
-from template_file import load_template_source
+from django.conf.settings import TEMPLATE_SOURCE_LOADERS
 
+def load_template_source(name, dirs=None):
+    for loader in TEMPLATE_SOURCE_LOADERS:
+        try:
+            return loader(name, dirs)
+        except:
+            pass
+    raise template.TemplateDoesNotExist, name
+
 class ExtendsError(Exception):
     pass
 

django/conf/global_settings.py

@@ -58 +58 @@
 # Extension on all templates.
 TEMPLATE_FILE_EXTENSION = '.html'
 
+# Callables which know how to import template sources from various
+# sources; the expected interface is callable(name, [dirs]), where dirs
+# is a list of directories to search instead of TEMPLATE_DIRS.
+import template_file
+import template_eggs
+TEMPLATE_SOURCE_LOADERS = [
+    template_file.load_template_source,
+    template_eggs.load_template_source,
+]
+del template_file
+del template_eggs
+
 # URL prefix for admin media -- CSS, JavaScript and images. Make sure to use a
 # trailing slash.
 # Examples: "http://foo.com/media/", "/media/".