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):
+ child_nodelists = ('nodelist_main', 'nodelist_empty',)
+
+ def __init__(self, name, references=None, as_var=None, nodelist_main=None, nodelist_empty=None):
self.name = name
self.as_var = as_var
+ self.references = references
+
+ if nodelist_main is None:
+ self.nodelist_main = template.NodeList()
+ else:
+ self.nodelist_main = nodelist_main
+
+ if nodelist_empty is None:
+ self.nodelist_empty = template.NodeList()
+ else:
+ self.nodelist_empty = nodelist_empty
+
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)
+
+ if not self.nodelist_main:
+ if container_content and self.as_var:
+ context[self.as_var] = container_content
+ return ''
+ return container_content
+
+ if container_content:
+ if self.as_var is None:
+ self.as_var = self.name
+
+ #make a new context
+ context.push()
+ context[self.as_var] = container_content
+ nodelist = template.NodeList()
+ for node in self.nodelist_main:
+ nodelist.append(node.render(context))
+ context.pop()
+ return nodelist.render(context)
+
+ if self.nodelist_empty is not None:
+ return self.nodelist_empty.render(context)
+
+ return ''
+
+ def get_container_content(self, context):
+ page = context['page']
+ if self.references:
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
+ content = ''
+ else:
+ try:
+ contentlet = page.contentlets.get(name__exact=self.name)
if contentlet.dynamic:
try:
- content = mark_safe(template.Template(content, name=contentlet.name).render(context))
+ content = mark_safe(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 = contentlet.content
+ except ObjectDoesNotExist:
+ content = ''
+ return content
def do_container(parser, token):
"""
- {% container <name> [as <variable>] %}
+ {% container <name> [[references <type>] as <variable>] %}
+ {% blockcontainer <name> [[references <type>] as <variable>] %} [ {% empty %} ] {% endblockcontainer %}
"""
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)
+ if tag == 'container':
+ return ContainerNode(name, references, as_var)
+
+ nodelist_main = parser.parse(('empty','endblockcontainer',))
+ token = parser.next_token()
+
+ if token.contents == 'empty':
+ nodelist_empty = parser.parse(('endblockcontainer',))
+ parser.delete_first_token()
+ else:
+ nodelist_empty = None
+ return ContainerNode(name, references, as_var, nodelist_main, nodelist_empty)
+
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)
+register.tag('blockcontainer', do_container)
git.ithinksw.org / philo.git / blobdiff