allow user to pass additional fields to constructor

This commit is contained in:
jeffrey k eliasen 2015-06-19 14:43:34 -10:00
parent 8c5fc2a94b
commit b933d595d2

View File

@ -17,12 +17,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: