Ticket #16319: SuccessMessageMixin.diff

django/contrib/messages/views.py

@@ +1 @@
+from django.core.exceptions import ImproperlyConfigured
+from django.views.generic import FormMixin
+from django.contrib import messages
+
+
+class SuccessMessageMixin(FormMixin):
+    """
+    A mixin that add a success message when a form is completed
+    """
+    success_message = None
+
+    def is_valid(self, form):
+        success_message = self.get_success_message()
+        if success_message:
+            messages.success(self.request, success_message)
+        return super(SuccessMessageMixin, self).is_valid(form)
+
+    def get_success_message(self):
+        if has_attr(self, 'object'):
+            return self.success_message % self.object.__dict__
+        return self.success_message