Add comment revision count field

judge/comments.py

@@ -169,7 +169,6 @@ class CommentedDetailView(TemplateResponseMixin, SingleObjectMixin, View):
                 queryset.select_related("author__user")
                 .filter(hidden=False)
                 .defer("author__about")
-                .annotate(revisions=Count("versions", distinct=True))
             )
         else:
             queryset = self.object.comments
@@ -180,7 +179,6 @@ class CommentedDetailView(TemplateResponseMixin, SingleObjectMixin, View):
                 .filter(hidden=False)
                 .annotate(
                     count_replies=Count("replies", distinct=True),
-                    revisions=Count("versions", distinct=True),
                 )[:DEFAULT_OFFSET]
             )
 

judge/migrations/0176_comment_revision_count.py

@@ -0,0 +1,30 @@
+# Generated by Django 3.2.18 on 2023-12-06 01:28
+
+from django.db import migrations, models
+
+
+# Run this in shell
+def migrate_revision(apps, schema_editor):
+    Comment = apps.get_model("judge", "Comment")
+
+    for c in Comment.objects.all():
+        c.revision_count = c.versions.count()
+        c.save()
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("judge", "0175_add_profile_index"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="comment",
+            name="revision_count",
+            field=models.PositiveIntegerField(default=1),
+        ),
+        # migrations.RunPython(
+        #     migrate_revision, migrations.RunPython.noop, atomic=True
+        # ),
+    ]

judge/models/comment.py

@@ -56,6 +56,7 @@ class Comment(MPTTModel):
         related_name="replies",
         on_delete=CASCADE,
     )
+    revision_count = models.PositiveIntegerField(default=1)
 
     versions = VersionRelation()
 
@@ -118,10 +119,6 @@ class Comment(MPTTModel):
         query = Comment.filter(parent=self)
         return len(query)
 
-    @cached_property
-    def get_revisions(self):
-        return self.versions.count()
-
     @cached_property
     def page_title(self):
         if isinstance(self.linked_object, Problem):

judge/views/comment.py

@@ -147,7 +147,6 @@ def get_comments(request, limit=10):
         .defer("author__about")
         .annotate(
             count_replies=Count("replies", distinct=True),
-            revisions=Count("versions", distinct=True),
         )[offset : offset + limit]
     )
     profile = None
@@ -241,7 +240,8 @@ class CommentEditAjax(LoginRequiredMixin, CommentMixin, UpdateView):
         # update notifications
         comment = form.instance
         add_mention_notifications(comment)
-
+        comment.revision_count = comment.versions.count() + 1
+        comment.save(update_fields=["revision_count"])
         with transaction.atomic(), revisions.create_revision():
             revisions.set_comment(_("Edited from site"))
             revisions.set_user(self.request.user)