return object after saving in perform_create and perform_update

This commit is contained in:
Druhin Bala 2017-05-04 16:34:50 -05:00
parent 8f6173cd8a
commit 078243a5d1

View File

@ -23,7 +23,7 @@ class CreateModelMixin(object):
return Response(serializer.data, status=status.HTTP_201_CREATED, headers=headers) return Response(serializer.data, status=status.HTTP_201_CREATED, headers=headers)
def perform_create(self, serializer): def perform_create(self, serializer):
serializer.save() return serializer.save()
def get_success_headers(self, data): def get_success_headers(self, data):
try: try:
@ -77,7 +77,7 @@ class UpdateModelMixin(object):
return Response(serializer.data) return Response(serializer.data)
def perform_update(self, serializer): def perform_update(self, serializer):
serializer.save() return serializer.save()
def partial_update(self, request, *args, **kwargs): def partial_update(self, request, *args, **kwargs):
kwargs['partial'] = True kwargs['partial'] = True