This commit is contained in:
jeffrey k eliasen 2017-05-16 11:44:49 +00:00 committed by GitHub
commit d9f182704f

View File

@ -18,12 +18,12 @@ class CreateModelMixin(object):
def create(self, request, *args, **kwargs):
serializer = self.get_serializer(data=request.data)
serializer.is_valid(raise_exception=True)
self.perform_create(serializer)
self.perform_create(serializer, **(kwargs.get('extra_fields', {})))
headers = self.get_success_headers(serializer.data)
return Response(serializer.data, status=status.HTTP_201_CREATED, headers=headers)
def perform_create(self, serializer):
serializer.save()
def perform_create(self, serializer, **kwargs):
serializer.save(**kwargs)
def get_success_headers(self, data):
try: