encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2024-11-23 01:57:00 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add pre_save hook in generic views

Browse Source
This commit is contained in:
Tom Christie 2012-10-25 12:15:31 +01:00
parent 3e751ccd8a
commit 8c360770c1
1 changed files with 10 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
rest_framework/mixins.py
View File

@ -20,10 +20,14 @@ class CreateModelMixin(object):
def create(self, request, *args, **kwargs): def create(self, request, *args, **kwargs):
serializer = self.get_serializer(data=request.DATA) serializer = self.get_serializer(data=request.DATA)
if serializer.is_valid(): if serializer.is_valid():
self.pre_save(serializer.object)
self.object = serializer.save() self.object = serializer.save()
return Response(serializer.data, status=status.HTTP_201_CREATED) return Response(serializer.data, status=status.HTTP_201_CREATED)
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST) return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
def pre_save(self, obj):
pass
class ListModelMixin(object): class ListModelMixin(object):
""" """
@ -46,7 +50,8 @@ class ListModelMixin(object):
# which may be `None` to disable pagination. # which may be `None` to disable pagination.
page_size = self.get_paginate_by(self.object_list) page_size = self.get_paginate_by(self.object_list)
if page_size: if page_size:
paginator, page, queryset, is_paginated = self.paginate_queryset(self.object_list, page_size) packed = self.paginate_queryset(self.object_list, page_size)
paginator, page, queryset, is_paginated = packed
serializer = self.get_pagination_serializer(page) serializer = self.get_pagination_serializer(page)
else: else:
serializer = self.get_serializer(instance=self.object_list) serializer = self.get_serializer(instance=self.object_list)
@ -79,20 +84,17 @@ class UpdateModelMixin(object):
serializer = self.get_serializer(data=request.DATA, instance=self.object) serializer = self.get_serializer(data=request.DATA, instance=self.object)
if serializer.is_valid(): if serializer.is_valid():
if self.object is None: self.pre_save(serializer.object)
# If PUT occurs to a non existant object, we need to set any
# attributes on the object that are implicit in the URL.
self.update_urlconf_attributes(serializer.object)
self.object = serializer.save() self.object = serializer.save()
return Response(serializer.data) return Response(serializer.data)
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST) return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
def update_urlconf_attributes(self, obj): def pre_save(self, obj):
""" """
When update (re)creates an object, we need to set any attributes that Set any attributes on the object that are implicit in the request.
are tied to the URLconf.
""" """
# pk and/or slug attributes are implicit in the URL.
pk = self.kwargs.get(self.pk_url_kwarg, None) pk = self.kwargs.get(self.pk_url_kwarg, None)
if pk: if pk:
setattr(obj, 'pk', pk) setattr(obj, 'pk', pk)