From 0cbb57b40fdb073c7ca09c9d1078926260c646db Mon Sep 17 00:00:00 2001
From: Tom Christie
Date: Wed, 8 Oct 2014 12:17:30 +0100
Subject: Tweak pre/post save hooks. Return instance in .update().

---
 rest_framework/mixins.py | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

(limited to 'rest_framework/mixins.py')

diff --git a/rest_framework/mixins.py b/rest_framework/mixins.py
index 03ebb034..4c62debb 100644
--- a/rest_framework/mixins.py
+++ b/rest_framework/mixins.py
@@ -20,11 +20,11 @@ class CreateModelMixin(object):
     def create(self, request, *args, **kwargs):
         serializer = self.get_serializer(data=request.data)
         serializer.is_valid(raise_exception=True)
-        self.create_valid(serializer)
+        self.perform_create(serializer)
         headers = self.get_success_headers(serializer.data)
         return Response(serializer.data, status=status.HTTP_201_CREATED, headers=headers)
 
-    def create_valid(self, serializer):
+    def perform_create(self, serializer):
         serializer.save()
 
     def get_success_headers(self, data):
@@ -67,10 +67,10 @@ class UpdateModelMixin(object):
         instance = self.get_object()
         serializer = self.get_serializer(instance, data=request.data, partial=partial)
         serializer.is_valid(raise_exception=True)
-        self.update_valid(serializer)
+        self.preform_update(serializer)
         return Response(serializer.data)
 
-    def update_valid(self, serializer):
+    def preform_update(self, serializer):
         serializer.save()
 
     def partial_update(self, request, *args, **kwargs):
@@ -84,9 +84,12 @@ class DestroyModelMixin(object):
     """
     def destroy(self, request, *args, **kwargs):
         instance = self.get_object()
-        instance.delete()
+        self.perform_destroy(instance)
         return Response(status=status.HTTP_204_NO_CONTENT)
 
+    def perform_destroy(self, instance):
+        instance.delete()
+
 
 # The AllowPUTAsCreateMixin was previously the default behaviour
 # for PUT requests. This has now been removed and must be *explicitly*
-- 
cgit v1.2.3