From 57003f0ef22c76543cf8277c5af8954d42e7f07e Mon Sep 17 00:00:00 2001 From: David Fischer Date: Wed, 30 Oct 2013 15:47:39 +0100 Subject: [PATCH] Add delete message mixin --- django/contrib/messages/views.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/django/contrib/messages/views.py b/django/contrib/messages/views.py index a0e3bd6..1de732c 100644 --- a/django/contrib/messages/views.py +++ b/django/contrib/messages/views.py @@ -18,3 +18,19 @@ class SuccessMessageMixin(object): if hasattr(self, 'object'): cleaned_data['object'] = self.object return self.success_message % cleaned_data + + +class DeleteMessageMixin(object): + """ + Adds a delete message on successful object deletion. + """ + delete_message = '' + + def get_delete_message(self): + return self.delete_message % {'object': self.get_object()} + + def delete(self, request, *args, **kwargs): + delete_message = self.get_delete_message() + if delete_message: + messages.success(request, delete_message) + return super(DeleteMessageMixin, self).delete(request, *args, **kwargs) -- 1.8.3.2