Commit dde3b1df by Tom Christie

Use api_settings.URL_FIELD_NAME

parent d20c49da
...@@ -11,6 +11,7 @@ from django.http import Http404 ...@@ -11,6 +11,7 @@ from django.http import Http404
from rest_framework import status from rest_framework import status
from rest_framework.response import Response from rest_framework.response import Response
from rest_framework.request import clone_request from rest_framework.request import clone_request
from rest_framework.settings import api_settings
import warnings import warnings
...@@ -60,7 +61,7 @@ class CreateModelMixin(object): ...@@ -60,7 +61,7 @@ class CreateModelMixin(object):
def get_success_headers(self, data): def get_success_headers(self, data):
try: try:
return {'Location': data['url']} return {'Location': data[api_settings.URL_FIELD_NAME]}
except (TypeError, KeyError): except (TypeError, KeyError):
return {} return {}
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment