from django.contrib.admin.utils import quote from django.urls import reverse from django.utils.functional import cached_property from django.utils.text import capfirst from django.utils.translation import gettext_lazy as _ from django.utils.translation import ngettext from wagtail.admin.views.generic import BeforeAfterHookMixin from wagtail.models import ReferenceIndex from wagtail.snippets.bulk_actions.snippet_bulk_action import SnippetBulkAction from wagtail.snippets.permissions import get_permission_name class DeleteBulkAction(BeforeAfterHookMixin, SnippetBulkAction): display_name = _("Delete") action_type = "delete" aria_label = _("Delete selected snippets") template_name = "wagtailsnippets/bulk_actions/confirm_bulk_delete.html" action_priority = 30 classes = {"serious"} @cached_property def _actionable_objects(self): # Cache the actionable objects so that we can use them early in the # dispatch() method, and reuse it later when needed. We don't name this # property "actionable_objects" because the method returns a tuple of # `(objects, {"items_with_no_access": items_with_no_access})` and the base # class assigns the `objects` to self.actionable_objects. return super().get_actionable_objects() def get_actionable_objects(self): return self._actionable_objects # Run the snippet deletion hooks that we've had since before bulk actions # were introduced. # These are different from the base bulk action hooks that are run by # __run_{before,after}_hooks(), as those are only called in form_valid(). # Here, we make use of the BeforeAfterHookMixin to run the before hook at # the beginning of dispatch(), and the after hook at the end of form_valid(), # which is how the delete-multiple view for snippets used to work. def run_before_hook(self): # The method returns a tuple of # `(objects, {"items_with_no_access": items_with_no_access})`, hence [0] objects = self.get_actionable_objects()[0] return self.run_hook("before_delete_snippet", self.request, objects) def run_after_hook(self): objects = self.get_actionable_objects()[0] return self.run_hook("after_delete_snippet", self.request, objects) def check_perm(self, snippet): if getattr(self, "can_delete_items", None) is None: # since snippets permissions are not enforced per object, makes sense to just check once per model request self.can_delete_items = self.request.user.has_perm( get_permission_name("delete", self.model) ) return self.can_delete_items @classmethod def execute_action(cls, objects, user=None, **kwargs): kwargs["self"].model.objects.filter( pk__in=[snippet.pk for snippet in objects] ).delete() return len(objects), 0 def get_context_data(self, **kwargs): context = super().get_context_data(**kwargs) # Add usage information to the context only if there is a single item if len(context["items"]) == 1: item_context = context["items"][0] item = item_context["item"] item_context.update( { "usage_count": ( ReferenceIndex.get_grouped_references_to(item).count() ), "usage_url": reverse( item.snippet_viewset.get_url_name("usage"), args=(quote(item.pk),), ), } ) return context def get_success_message(self, num_parent_objects, num_child_objects): if num_parent_objects == 1: return _("%(model_name)s '%(object)s' deleted.") % { "model_name": capfirst(self.model._meta.verbose_name), "object": self.actionable_objects[0], } else: return ngettext( "%(count)d %(model_name)s deleted.", "%(count)d %(model_name)s deleted.", num_parent_objects, ) % { "model_name": capfirst(self.model._meta.verbose_name_plural), "count": num_parent_objects, }