django rest_framework异常处理

from rest_framework.response import Response


class JsonResponse(Response):
    def __init__(self, data=None, code=None, msg=None, status=None,
                 template_name=None, headers=None,
                 exception=False, content_type=None):
        rsp_data = {"code": code, "message": msg, "data": data}
        super(JsonResponse, self).__init__(data=rsp_data, status=status, template_name=template_name,
                                                 headers=headers,
                                                 exception=exception, content_type=content_type)

from rest_framework import status
from rest_framework.views import exception_handler
from libs.response import JsonResponse


class DataException(Exception):

    def __init__(self, message="", code=0, status=status.HTTP_400_BAD_REQUEST, data=None):
        self.code = code
        self.status = status
        self.detail = message
        self.data = data if data else {}

        def __str__(self):
            return self.message


def custom_exception_handler(exc, context):
    data = exc.data if hasattr(exc, "data") else {}
    return JsonResponse(msg=exc.detail, status=exc.status_code, data=data, code=exc.status_code)

REST_FRAMEWORK = {
    'EXCEPTION_HANDLER': 'libs.exceptions.custom_exception_handler',
}