Ticket #11730: ifequal-template-bug.diff

django/template/defaulttags.py

@@ -170 +170 @@
         self._varlist = varlist
         self._id = str(id(self))
 
+    def __iter__(self):
+        for node in self.nodelist_true:
+            yield node
+        for node in self.nodelist_false:
+            yield node
+
+    def get_nodes_by_type(self, nodetype):
+        nodes = []
+        if isinstance(self, nodetype):
+            nodes.append(self)
+        nodes.extend(self.nodelist_true.get_nodes_by_type(nodetype))
+        nodes.extend(self.nodelist_false.get_nodes_by_type(nodetype))
+        return nodes
+
     def render(self, context):
         if 'forloop' in context and self._id not in context['forloop']:
             self._last_seen = None
@@ -202 +216 @@
     def __repr__(self):
         return "<IfEqualNode>"
 
+    def __iter__(self):
+        for node in self.nodelist_true:
+            yield node
+        for node in self.nodelist_false:
+            yield node
+
+    def get_nodes_by_type(self, nodetype):
+        nodes = []
+        if isinstance(self, nodetype):
+            nodes.append(self)
+        nodes.extend(self.nodelist_true.get_nodes_by_type(nodetype))
+        nodes.extend(self.nodelist_false.get_nodes_by_type(nodetype))
+        return nodes
+
     def render(self, context):
         val1 = self.var1.resolve(context, True)
         val2 = self.var2.resolve(context, True)