encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-07-28 00:49:49 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add get_create_data to CreateModelMixin

Browse Source
This commit is contained in:
JEphron 2020-04-06 12:40:53 -07:00
parent 0c8eb91737
commit 702cd7d357
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
rest_framework/mixins.py
View File

@ -14,7 +14,7 @@ class CreateModelMixin:
Create a model instance. Create a model instance.
""" """
def create(self, request, *args, **kwargs): def create(self, request, *args, **kwargs):
serializer = self.get_serializer(data=request.data) serializer = self.get_serializer(data=self.get_create_data(request))
serializer.is_valid(raise_exception=True) serializer.is_valid(raise_exception=True)
self.perform_create(serializer) self.perform_create(serializer)
headers = self.get_success_headers(serializer.data) headers = self.get_success_headers(serializer.data)
@ -23,6 +23,9 @@ class CreateModelMixin:
def perform_create(self, serializer): def perform_create(self, serializer):
serializer.save() serializer.save()
def get_create_data(self, request):
return request.data
def get_success_headers(self, data): def get_success_headers(self, data):
try: try:
return {'Location': str(data[api_settings.URL_FIELD_NAME])} return {'Location': str(data[api_settings.URL_FIELD_NAME])}