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): def create(self, request, *args, **kwargs):
serializer = self.get_serializer(data=request.data) serializer = self.get_serializer(data=request.data)
serializer.is_valid(raise_exception=True) 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) headers = self.get_success_headers(serializer.data)
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, **kwargs):
serializer.save() serializer.save(**kwargs)
def get_success_headers(self, data): def get_success_headers(self, data):
try: try: