X-Git-Url: http://git.ithinksw.org/philo.git/blobdiff_plain/0d51e8dc2b15ca84b8979b94a3224b4a33f7f25e..22c4e979b5ad061e448318310fddc4bb8fc56465:/templatetags/containers.py

diff --git a/templatetags/containers.py b/templatetags/containers.py
index 9bf056f..8bb0c6b 100644
--- a/templatetags/containers.py
+++ b/templatetags/containers.py
@@ -2,51 +2,98 @@ from django import template
 from django.conf import settings
 from django.utils.safestring import SafeUnicode, mark_safe
 from django.core.exceptions import ObjectDoesNotExist
+from django.contrib.contenttypes.models import ContentType
+
 
 register = template.Library()
 
 
 class ContainerNode(template.Node):
-	def __init__(self, name, as_var=None):
+	def __init__(self, name, references=None, as_var=None):
 		self.name = name
 		self.as_var = as_var
+		self.references = references
+	
 	def render(self, context):
-		page = None
+		content = settings.TEMPLATE_STRING_IF_INVALID
 		if 'page' in context:
-			page = context['page']
-		if page:
-			contentlet = None
+			container_content = self.get_container_content(context)
+		else:
+			container_content = None
+		
+		if self.as_var:
+			context[self.as_var] = container_content
+			return ''
+		
+		if not container_content:
+			return ''
+		
+		return container_content
+	
+	def get_container_content(self, context):
+		page = context['page']
+		if self.references:
+			# Then it's a content reference.
 			try:
-				contentlet = page.contentlets.get(name__exact=self.name)
+				contentreference = page.contentreferences.get(name__exact=self.name, content_type=self.references)
+				content = contentreference.content
 			except ObjectDoesNotExist:
-				pass
-			if contentlet:
-				content = contentlet.content
-				if contentlet.dynamic:
+				content = ''
+		else:
+			# Otherwise it's a contentlet.
+			try:
+				contentlet = page.contentlets.get(name__exact=self.name)
+				if '{%' in contentlet.content:
 					try:
-						content = mark_safe(template.Template(content, name=contentlet.name).render(context))
+						content = template.Template(contentlet.content, name=contentlet.name).render(context)
 					except template.TemplateSyntaxError, error:
-						content = ''
 						if settings.DEBUG:
-							content = ('[Error parsing contentlet \'%s\': %s]' % self.name, error)
-				if self.as_var:
-					context[self.as_var] = content
-					content = ''
-				return content
-		return ''
+							content = ('[Error parsing contentlet \'%s\': %s]' % (self.name, error))
+						else:
+							content = settings.TEMPLATE_STRING_IF_INVALID
+				else:
+					content = contentlet.content
+			except ObjectDoesNotExist:
+				content = settings.TEMPLATE_STRING_IF_INVALID
+			content = mark_safe(content)
+		return content
 
 
 def do_container(parser, token):
 	"""
-	{% container <name> [as <variable>] %}
+	{% container <name> [[references <type>] as <variable>] %}
 	"""
 	params = token.split_contents()
-	if len(params) >= 2: # without as_var
+	if len(params) >= 2:
+		tag = params[0]
 		name = params[1].strip('"')
+		references = None
 		as_var = None
-		if len(params) == 4:
-			as_var = params[3]
-		return ContainerNode(name, as_var)
+		if len(params) > 2:
+			remaining_tokens = params[2:]
+			while remaining_tokens:
+				option_token = remaining_tokens.pop(0)
+				if option_token == 'references':
+					try:
+						app_label, model = remaining_tokens.pop(0).strip('"').split('.')
+						references = ContentType.objects.get(app_label=app_label, model=model)
+					except IndexError:
+						raise template.TemplateSyntaxError('"%s" template tag option "references" requires an argument specifying a content type' % tag)
+					except ValueError:
+						raise template.TemplateSyntaxError('"%s" template tag option "references" requires an argument of the form app_label.model (see django.contrib.contenttypes)' % tag)
+					except ObjectDoesNotExist:
+						raise template.TemplateSyntaxError('"%s" template tag option "references" requires an argument of the form app_label.model which refers to an installed content type (see django.contrib.contenttypes)' % tag)
+				elif option_token == 'as':
+					try:
+						as_var = remaining_tokens.pop(0)
+					except IndexError:
+						raise template.TemplateSyntaxError('"%s" template tag option "as" requires an argument specifying a variable name' % tag)
+			if references and not as_var:
+				raise template.TemplateSyntaxError('"%s" template tags using "references" option require additional use of the "as" option specifying a variable name' % tag)
+		return ContainerNode(name, references, as_var)
+		
 	else: # error
-		raise template.TemplateSyntaxError('do_container template tag provided with invalid arguments')
+		raise template.TemplateSyntaxError('"%s" template tag provided without arguments (at least one required)' % tag)
+
+
 register.tag('container', do_container)