X-Git-Url: http://git.ithinksw.org/philo.git/blobdiff_plain/767cec415355aea0ac6af15580ffdc5e5787c04f..d8d6976f3a0646a05e052d33294e3c8c44c5eed3:/philo/models/pages.py

diff --git a/philo/models/pages.py b/philo/models/pages.py
index bdd9b42..0ce047e 100644
--- a/philo/models/pages.py
+++ b/philo/models/pages.py
@@ -4,6 +4,8 @@
 
 """
 
+import itertools
+
 from django.conf import settings
 from django.contrib.contenttypes.models import ContentType
 from django.contrib.contenttypes import generic
@@ -30,7 +32,7 @@ class LazyContainerFinder(object):
 	def __init__(self, nodes, extends=False):
 		self.nodes = nodes
 		self.initialized = False
-		self.contentlet_specs = set()
+		self.contentlet_specs = []
 		self.contentreference_specs = SortedDict()
 		self.blocks = {}
 		self.block_super = False
@@ -47,7 +49,7 @@ class LazyContainerFinder(object):
 			
 			if isinstance(node, ContainerNode):
 				if not node.references:
-					self.contentlet_specs.add(node.name)
+					self.contentlet_specs.append(node.name)
 				else:
 					if node.name not in self.contentreference_specs.keys():
 						self.contentreference_specs[node.name] = node.references
@@ -78,6 +80,26 @@ class LazyContainerFinder(object):
 			self.initialized = True
 
 
+def build_extension_tree(nodelist):
+	nodelists = []
+	extends = None
+	for node in nodelist:
+		if not isinstance(node, TextNode):
+			if isinstance(node, ExtendsNode):
+				extends = node
+			break
+	
+	if extends:
+		if extends.nodelist:
+			nodelists.append(LazyContainerFinder(extends.nodelist, extends=True))
+		loaded_template = getattr(extends, LOADED_TEMPLATE_ATTR)
+		nodelists.extend(build_extension_tree(loaded_template.nodelist))
+	else:
+		# Base case: root.
+		nodelists.append(LazyContainerFinder(nodelist))
+	return nodelists
+
+
 class Template(TreeModel):
 	"""Represents a database-driven django template."""
 	#: The name of the template. Used for organization and debugging.
@@ -97,35 +119,16 @@ class Template(TreeModel):
 		"""
 		template = DjangoTemplate(self.code)
 		
-		def build_extension_tree(nodelist):
-			nodelists = []
-			extends = None
-			for node in nodelist:
-				if not isinstance(node, TextNode):
-					if isinstance(node, ExtendsNode):
-						extends = node
-					break
-			
-			if extends:
-				if extends.nodelist:
-					nodelists.append(LazyContainerFinder(extends.nodelist, extends=True))
-				loaded_template = getattr(extends, LOADED_TEMPLATE_ATTR)
-				nodelists.extend(build_extension_tree(loaded_template.nodelist))
-			else:
-				# Base case: root.
-				nodelists.append(LazyContainerFinder(nodelist))
-			return nodelists
-		
 		# Build a tree of the templates we're using, placing the root template first.
-		levels = build_extension_tree(template.nodelist)[::-1]
+		levels = build_extension_tree(template.nodelist)
 		
-		contentlet_specs = set()
+		contentlet_specs = []
 		contentreference_specs = SortedDict()
 		blocks = {}
 		
-		for level in levels:
+		for level in reversed(levels):
 			level.initialize()
-			contentlet_specs |= level.contentlet_specs
+			contentlet_specs.extend(itertools.ifilter(lambda x: x not in contentlet_specs, level.contentlet_specs))
 			contentreference_specs.update(level.contentreference_specs)
 			for name, block in level.blocks.items():
 				if block.block_super:
@@ -136,7 +139,7 @@ class Template(TreeModel):
 		for block_list in blocks.values():
 			for block in block_list:
 				block.initialize()
-				contentlet_specs |= block.contentlet_specs
+				contentlet_specs.extend(itertools.ifilter(lambda x: x not in contentlet_specs, block.contentlet_specs))
 				contentreference_specs.update(block.contentreference_specs)
 		
 		return contentlet_specs, contentreference_specs