73b023dca2
Use an explicit Wagtail Column for pending_in_article in CommentViewSet list_display and add a regression test for /cms/snippets/comments/comment/. Fixes #37 Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
73 lines
2.5 KiB
Python
73 lines
2.5 KiB
Python
from typing import Any, cast
|
|
|
|
from django.db.models import Count, Q
|
|
from django.utils.translation import gettext_lazy as _
|
|
from django.utils.translation import ngettext
|
|
from wagtail import hooks
|
|
from wagtail.admin.ui.tables import BooleanColumn, Column
|
|
from wagtail.snippets.bulk_actions.snippet_bulk_action import SnippetBulkAction
|
|
from wagtail.snippets.models import register_snippet
|
|
from wagtail.snippets.permissions import get_permission_name
|
|
from wagtail.snippets.views.snippets import SnippetViewSet
|
|
|
|
from apps.comments.models import Comment
|
|
|
|
|
|
class ApproveCommentBulkAction(SnippetBulkAction):
|
|
display_name = _("Approve")
|
|
action_type = "approve"
|
|
aria_label = _("Approve selected comments")
|
|
template_name = "comments/confirm_bulk_approve.html"
|
|
action_priority = 20
|
|
models = [Comment]
|
|
|
|
def check_perm(self, snippet):
|
|
if getattr(self, "can_change_items", None) is None:
|
|
self.can_change_items = self.request.user.has_perm(get_permission_name("change", self.model))
|
|
return self.can_change_items
|
|
|
|
@classmethod
|
|
def execute_action(cls, objects, **kwargs):
|
|
updated = kwargs["self"].model.objects.filter(pk__in=[obj.pk for obj in objects], is_approved=False).update(
|
|
is_approved=True
|
|
)
|
|
return updated, 0
|
|
|
|
def get_success_message(self, num_parent_objects, num_child_objects):
|
|
return ngettext(
|
|
"%(count)d comment approved.",
|
|
"%(count)d comments approved.",
|
|
num_parent_objects,
|
|
) % {"count": num_parent_objects}
|
|
|
|
|
|
class CommentViewSet(SnippetViewSet):
|
|
model = Comment
|
|
queryset = Comment.objects.all()
|
|
icon = "comment"
|
|
list_display = [
|
|
"author_name",
|
|
"article",
|
|
BooleanColumn("is_approved"),
|
|
Column("pending_in_article", label="Pending (article)"),
|
|
"created_at",
|
|
]
|
|
list_filter = ["is_approved"]
|
|
search_fields = ["author_name", "body"]
|
|
add_to_admin_menu = True
|
|
|
|
def get_queryset(self, request):
|
|
base_qs = self.model.objects.all().select_related("article", "parent")
|
|
# mypy-django-plugin currently crashes on QuerySet.annotate() in this file.
|
|
typed_qs = cast(Any, base_qs)
|
|
return typed_qs.annotate(
|
|
pending_in_article=Count(
|
|
"article__comments",
|
|
filter=Q(article__comments__is_approved=False),
|
|
distinct=True,
|
|
)
|
|
)
|
|
|
|
register_snippet(CommentViewSet)
|
|
hooks.register("register_bulk_action", ApproveCommentBulkAction)
|