Ticket #256: django_ifequal_templatetag.diff

django_src/django/core/defaulttags.py

@@ -125 +125 @@
             return content
         else:
             return ''
-
-class IfNotEqualNode(template.Node):
-    def __init__(self, var1, var2, nodelist):
+            
+class IfEqualNode(template.Node):
+    def __init__(self, var1, var2, nodelist, negate=False):
         self.var1, self.var2, self.nodelist = var1, var2, nodelist
+        self.negate = negate
 
     def __repr__(self):
-        return "<IfNotEqualNode>"
+        return "<IfEqualNode>"
 
     def render(self, context):
-        if template.resolve_variable(self.var1, context) != template.resolve_variable(self.var2, context):
+        if self.negate and template.resolve_variable(self.var1, context) != template.resolve_variable(self.var2, context):
             return self.nodelist.render(context)
+        elif not self.negate and template.resolve_variable(self.var1, context) == template.resolve_variable(self.var2, context):
+            return self.nodelist.render(context)
         else:
             return ''
 
@@ -447 +450 @@
     parser.delete_first_token()
     return ForNode(loopvar, sequence, reversed, nodelist_loop)
 
+def do_ifequal(parser, token):
+    """
+    Output the contents of the block if the two arguments equal each other.
+
+    Example::
+
+        {% ifequal user.id comment.user_id %}
+            ...
+        {% endifequal %}
+    """
+    bits = token.contents.split()
+    if len(bits) != 3:
+        raise template.TemplateSyntaxError, "'ifequal' takes two arguments"
+    nodelist = parser.parse(('endifequal',))
+    parser.delete_first_token()
+    return IfEqualNode(bits[1], bits[2], nodelist, negate=False)
+
 def do_ifnotequal(parser, token):
     """
     Output the contents of the block if the two arguments do not equal each other.
@@ -462 +482 @@
         raise template.TemplateSyntaxError, "'ifnotequal' takes two arguments"
     nodelist = parser.parse(('endifnotequal',))
     parser.delete_first_token()
-    return IfNotEqualNode(bits[1], bits[2], nodelist)
+    return IfEqualNode(bits[1], bits[2], nodelist, negate=True)
 
 def do_if(parser, token):
     """
@@ -732 +752 @@
 template.register_tag('filter', do_filter)
 template.register_tag('firstof', do_firstof)
 template.register_tag('for', do_for)
+template.register_tag('ifequal', do_ifequal)
 template.register_tag('ifnotequal', do_ifnotequal)
 template.register_tag('if', do_if)
 template.register_tag('ifchanged', do_ifchanged)