自定义异常处理

通常drf自带的异常处理类exception_handler可能会处理不了一些没有的异常，此时需要我们自定义
utils/exceptions.py 单独创建一个exceptions文件

from rest_framework.views import exception_handler
from django.db import DatabaseError
from rest_framework.response import Response
from rest_framework import status

import logging
logger = logging.getLogger(‘django‘)


def custom_exception_handler(exc, context):
    """
    自定义异常处理
    :param exc: 本次异常发生时的异常对象
    :param context: 本次异常发生时的上下文
    :return:
    """
    # 调用drf框架原生的异常处理方法
    response = exception_handler(exc, context)
    if response is None:
        # 发生异常的视图
        view = context[‘view‘]
        """为None可能有2种情况，没有发生异常，或者本次异常没有被drf进行处理"""
        if isinstance(exc, DatabaseError):
            # 数据库操作异常
            logger.error(‘数据库发生异常, view=%s, exc=%s‘ % (view, exc))
            response = Response({‘detail‘: ‘服务器内部错误‘},
                                status=status.HTTP_507_INSUFFICIENT_STORAGE)

        if isinstance(exc, Exception):
            # 未知异常
            logger.error(‘发生未知异常, views=%s, exc=%s‘ % (view, exc))
            response = Response({‘detail‘: ‘服务器发生未知错误‘},
                                status=status.HTTP_500_INTERNAL_SERVER_ERROR)
    return response

settings.py文件中需要指定使用我们自定义的异常类

REST_FRAMEWORK = {
    # 异常处理
    ‘EXCEPTION_HANDLER‘: ‘renranapi.utils.exception.custom_exception_handler‘,
}

自定义异常处理

原文：https://www.cnblogs.com/weiweivip666/p/13767893.html