Khung làm việc Django REST cách chỉ định mã lỗi khi tăng lỗi xác thực trong bộ nối tiếp

Question

Tôi có điểm cuối API cho phép người dùng đăng ký tài khoản. Tôi muốn trả lại HTTP 409 thay vì 400 cho tên người dùng trùng lặp.

Đây là serializer tôi:

from django.contrib.auth.models import User 
from rest_framework.serializers import ModelSerializer 

class UserSerializer(ModelSerializer): 
    username = CharField() 

    def validate_username(self, value): 
     if User.objects.filter(username=value).exists(): 
      raise NameDuplicationError() 
     return value 


class NameDuplicationError(APIException): 
    status_code = status.HTTP_409_CONFLICT 
    default_detail = u'Duplicate Username' 

Khi lỗi được kích hoạt, phản ứng là: {"detail":"Duplicate Username"}. Tôi nhận ra rằng nếu tôi phân lớp APIException, khóa detail được sử dụng thay vì username.

Tôi muốn có phản ứng này để thay thế {"username":"Duplicate Username"}

hoặc tôi muốn chỉ định một mã trạng thái khi nuôi một ValidationError:

def validate_username(self, value): 
    if User.objects.filter(username=value).exists(): 
     raise serializers.ValidationError('Duplicate Username', 
              status_code=status.HTTP_409_CONFLICT) 
    return value 

Nhưng điều này không làm việc như ValidationError chỉ trả về 400.

Có cách nào khác để thực hiện việc này không?

Answer 1

Bạn có thể tăng trường hợp ngoại lệ khác nhau như:

from rest_framework.exceptions import APIException 
from django.utils.encoding import force_text 
from rest_framework import status 


class CustomValidation(APIException): 
    status_code = status.HTTP_500_INTERNAL_SERVER_ERROR 
    default_detail = 'A server error occurred.' 

    def __init__(self, detail, field, status_code): 
     if status_code is not None:self.status_code = status_code 
     if detail is not None: 
      self.detail = {field: force_text(detail)} 
     else: self.detail = {'detail': force_text(self.default_detail)} 

bạn có thể sử dụng trong serializer của bạn như:

raise CustomValidation('Duplicate Username','username', status_code=status.HTTP_409_CONFLICT) 

hoặc

raise CustomValidation('Access denied','username', status_code=status.HTTP_403_FORBIDDEN) 

Answer 2

Sử dụng django-nghỉ ngơi-framework tùy chỉnh ngoại lệ handler http://www.django-rest-framework.org/api-guide/exceptions/

def custom_exception_handler(exc, context=None): 
    response = exception_handler(exc, context) 
    if response is not None: 
     if response.data['detail'] == 'Duplicate Username': 
      response.data['username'] = response.data.pop('detail') 
     response.status_code = status.HTTP_409_CONFLICT 
    return response