Django 如何在REST框架中以SON格式返回错误，而不是HTML格式返回错误

发布于04月17日

I want to make an error handler exception to return the error 404 instead of this html. how can i do this ?

我try 了下面的代码，但它对我不起作用

from rest_framework.views import exception_handler
from rest_framework.response import Response
from rest_framework import status
from django.http import Http404

def custom_exception_handler(exc, context):
    # Call the default exception handler first,
    # to get the standard error response.
    response = exception_handler(exc, context)

    # Now add the HTTP 404 handling
    if isinstance(exc, Http404):
        custom_response_data = { "error": "page not found" }
        return Response(custom_response_data, status=status.HTTP_404_NOT_FOUND)

    # Return the default response if the exception handled is not a Http404
    return response

推荐答案

我创造了同样的东西，它对我来说效果很好

目录路径100

from rest_framework.views import exception_handler


def custom_exception_handler(exc, context):
    response = exception_handler(exc, context)
    if response.status_code == 404:
        response.data = {
            "success": False,
            "message": "Not found❗",
            "data": {}
        }
        return response
    elif response.status_code == 500:
        response.data = {
            "success": False,
            "message": "Internal server error! 🌐",
            "data": {}
        }
    # elif response.status_code == 400:
    #     response.data = {
    #         "success": False,
    #         "message": "Bad request!",
    #         "data": {}
    #     }
    elif response.status_code == 401:
        response.data = {
            "success": False,
            "message": "Login required! 🗝️",
            "data": {}
        }
    elif response.status_code == 403:
        response.data = {
            "success": False,
            "message": "Permission denied!",
            "data": {}
        }

    return response

settings.py

REST_FRAMEWORK = {

    'DEFAULT_AUTHENTICATION_CLASSES': [
        'rest_framework_simplejwt.authentication.JWTAuthentication',
    ],

    'DEFAULT_FILTER_BACKENDS': [
        'django_filters.rest_framework.DjangoFilterBackend',
    ],

    'EXCEPTION_HANDLER': 'myapp.custom_exception.custom_exception_handler',
}