Обработка пользовательских исключений в django rest framework не возвращает ожидаемый код ошибки

Question

Я пытаюсь реализовать пользовательскую обработку ошибок в проекте, используя django rest framework, и я хочу, чтобы он возвращал json при попытке создать нового пользователя, но электронная почта уже используется (я применяю уникальность электронной почты).Я хочу вернуть код ошибки 400, а полученный json будет выглядеть следующим образом:

{
    "email": [
        "eMAil already in use."
    ]
}

Я реализовал решение, предложенное в https://medium.com/@mwhitt.w/restful-error-messages-with-django-537047892dff,, но получаю код ошибки 500 и не возвращаю json.

Это мой customexception.py

class BaseCustomException(Exception):
    status_code = None
    error_message = None
    is_an_error_response = True

    def __init__(self, error_message):
        Exception.__init__(self, error_message)
        self.error_message = error_message

    def to_dict(self):
        return {'errorMessage':self.error_message}

class ExistingEmailException(BaseCustomException):
    status_code = 400

    def __init__(self):
        BaseCustomException.__init__(self, 'eMail already in use')

Это мой middleware.py:

import traceback
from django.http import JsonResponse

def is_registered(exception):
    try:
        return exception.is_an_error_response
    except AttributeError:
        return False

class RequestExceptionHandler:
    def process_exception(self, request, exception):
        if is_registered(exception):
            status = exception.status_code
            exception_dict = exception.to_dict()
        else:
            status = 500
            exception_dict = {'errorMessage':'Unexpected Error!'}

        error_message = exception_dict['errorMessage']
        traceback.print_exc()
        return JsonResponse(exception_dict, status=status)

Это мой сериализатор:

class UserSerializer(serializers.HyperlinkedModelSerializer):
    class Meta:
        model = User
        fields = ('id', 'username', 'first_name', 'last_name', 'email', 'password')

    def create(self, validated_data):
        try:
            user = models.User.objects.get(email=validated_data.get('email'))
        except User.DoesNotExist:
            user = models.User.objects.create(**validated_data)

            user.set_password(user.password)
            user.save()
            Token.objects.create(user=user)

            return user
        else:
            raise ExistingEmailException()

И вот результат, который я получаю:

  File "/Users/hugovillalobos/Documents/Code/IntellibookProject/Intellibook/UsersManagerApp/serializers.py", line 27, in create
    raise ExistingEmailException()
GeneralApp.customexceptions.ExistingEmailException: eMail already in use
[06/Jun/2018 17:12:20] "POST /es/users_manager/users/ HTTP/1.1" 500 102163

Спасибо за вашу помощь.

Answer 1

Я думаю, причина того, что вы получаете неожиданный код ошибки, в том, что следующий пример не предназначен для использования с django-rest-framework.Хотя теоретически должно быть возможно использовать промежуточное ПО django для перехвата ответа (я смотрю на первый комментарий в ответ на этот вопрос ), но самый простой и, на мой взгляд, лучший подход - следовать Руководство DRF по пользовательским исключениям .