Ticket #8297: 8297.withblock.diff

django/django/template/defaulttags.py

@@ -405 +405 @@
         context.pop()
         return output
 
+class WithBlockNode(Node):
+    def __init__(self, with_nodelists, in_nodelist):
+        self.with_nodelists, self.in_nodelist = with_nodelists, in_nodelist
+
+    def __repr__(self):
+        return "<WithBlock node>"
+
+    def render(self, context):
+        in_context = Context()
+        for (var_name, nodelist) in self.with_nodelists:
+            in_context[var_name] = nodelist.render(context)
+        in_context.update(context)
+        return self.in_nodelist.render(in_context)
+
 #@register.tag
 def autoescape(parser, token):
     """
@@ -1106 +1120 @@
     parser.delete_first_token()
     return WithNode(var, name, nodelist)
 do_with = register.tag('with', do_with)
+
+def do_withblock(parser, token):
+    bits = token.contents.split()
+    if not (len(bits) == 3 and bits[1] == 'as'):
+        raise TemplateSyntaxError("'withblock' statement requires 2 arguments: `as` followed by a variable name")
+    with_nodelists = []
+    var_name = bits[2]
+    while True:
+        nodelist = parser.parse(('and', 'in'))
+        with_nodelists.append((var_name, nodelist))
+        token = parser.next_token()
+        if token.contents == 'in':
+            break
+        else:
+            bits = token.contents.split()
+            if not (len(bits) == 3 and bits[1] == 'as'):
+                raise TemplateSyntaxError("'and' statement requires 2 arguments: `as` followed by a variable name")
+            var_name = bits[2]
+    in_nodelist = parser.parse(('endwithblock'))
+    parser.delete_first_token()
+    return WithBlockNode(with_nodelists, in_nodelist)
+do_withblock = register.tag("withblock", do_withblock)
+ No newline at end of file