Ticket #15904: render_comments_form_query_count.diff

django/contrib/comments/templatetags/comments.py

@@ -122 +122 @@
     """Insert a form for the comment model into the context."""
 
     def get_form(self, context):
-        ctype, object_pk = self.get_target_ctype_pk(context)
-        if object_pk:
-            return comments.get_form()(ctype.get_object_for_this_type(pk=object_pk))
+        obj = self.get_object(context)
+        if obj:
+            return comments.get_form()(obj)
         else:
             return None
 
+    def get_object(self, context):
+        if self.object_expr:
+            try:
+                return self.object_expr.resolve(context)
+            except template.VariableDoesNotExist:
+                return None
+        else:
+            object_pk = self.object_pk_expr.resolve(context,
+                    ignore_failures=True)
+            return self.ctype.get_object_for_this_type(pk=object_pk)
+
     def render(self, context):
         context[self.as_varname] = self.get_form(context)
         return ''

tests/regressiontests/comment_tests/tests/templatetag_tests.py

@@ -40 +40 @@
     def testRenderCommentFormFromObject(self):
         self.testRenderCommentForm("{% render_comment_form for a %}")
 
+    def testRenderCommentFormFromObjectWithQueryCount(self):
+        def test():
+            self.testRenderCommentFormFromObject()
+        self.assertNumQueries(1, test)
+
     def testGetCommentCount(self, tag=None):
         self.createSomeComments()
         t = "{% load comments %}" + (tag or "{% get_comment_count for comment_tests.article a.id as cc %}") + "{{ cc }}"