Ticket #582: template_eggs.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
+import template_file
+import template_eggs
 
+template_source_loaders = [
+    template_file.load_template_source,
+    template_eggs.load_template_source,
+]
+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