Question

我正在尝试使用django rest框架在项目中实现自定义错误处理，我希望它在尝试创建新用户时返回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的中间件来劫持响应（我正在查看响应this SO question的第一条评论），最简单且我认为最好的方法是遵循{{3} }。

django rest框架自定义异常处理不返回预期的错误代码

1 个答案: