X-Git-Url: http://git.ithinksw.org/philo.git/blobdiff_plain/80fa50c87ed121896ddf1bdd993d5697bb5a56c3..4bd5e63372542c7a224576fc4082324bc39f13aa:/templatetags/containers.py diff --git a/templatetags/containers.py b/templatetags/containers.py index c04cb64..92e126e 100644 --- a/templatetags/containers.py +++ b/templatetags/containers.py @@ -9,48 +9,27 @@ register = template.Library() class ContainerNode(template.Node): - child_nodelists = ('nodelist_main', 'nodelist_empty',) - - def __init__(self, name, references=None, as_var=None, nodelist_main=None, nodelist_empty=None): + def __init__(self, name, references=None, as_var=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): content = settings.TEMPLATE_STRING_IF_INVALID if 'page' in context: - container_content = self.get_container_content(context['page']) + container_content = self.get_container_content(context) - if self.nodelist_main is None: - 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 - context.push() + if self.as_var: context[self.as_var] = container_content - return nodelist_main.render(context) + return '' - if nodelist_empty is not None: - return nodelist_empty.render(context) + if not container_content: + return '' - return '' + return container_content - def get_container_content(self, page): + def get_container_content(self, context): + page = context['page'] if self.references: try: contentreference = page.contentreferences.get(name__exact=self.name, content_type=self.references) @@ -65,17 +44,17 @@ class ContainerNode(template.Node): content = mark_safe(template.Template(contentlet.content, name=contentlet.name).render(context)) except template.TemplateSyntaxError, error: if settings.DEBUG: - content = ('[Error parsing contentlet \'%s\': %s]' % self.name, error) + content = ('[Error parsing contentlet \'%s\': %s]' % (self.name, error)) else: content = contentlet.content except ObjectDoesNotExist: content = '' return content + def do_container(parser, token): """ - {% container [[references ] as ] %} - {% blockcontainer [[references ] as ] %} [ {% empty %} ] {% endblockcontainer %} + {% container [[references ] as ] %} """ params = token.split_contents() if len(params) >= 2: @@ -104,19 +83,10 @@ def do_container(parser, token): 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 BlockContainerNode(name, references, as_var, nodelist_main, nodelist_empty) + return ContainerNode(name, references, as_var) else: # error raise template.TemplateSyntaxError('"%s" template tag provided without arguments (at least one required)' % tag) -register.tag('container', do_container) \ No newline at end of file + + +register.tag('container', do_container)