drf框架的异常处理

drf框架对本身内部定义的方法作了特殊处理 好比:频率控制组件 返回特殊的信息

如何在drf异常处理的基础上再从新包装异常

from rest_framework.views import exception_handler
from rest_framework.response import Response
import logging

error_log = logging.getLogger('')


def custom_exception_handler(exc, context):
    '''重定义异常(使异常返回值符合规范)'''
    # 先让drf内部作处理
    response = exception_handler(exc, context)
    # drf处理过得异常 能够按照本身的规范来返回
    if response:
        '''drf内部错误'''
        print(response.data)
        detail = response.data.get('detail')
        non_field_errors = response.data.get('non_field_errors')
        if detail:
            return Response({"code": 1001, "msg": detail})

        if non_field_errors:
            return Response({"code": 1001, "msg": non_field_errors[0]})
        return response
    # 没有通过drf处理的异常好比代码报错 咱们能够捕捉下来写入日志
    else:
        '''异常写入日志'''
        error_views = context['view']
        error_info = repr(exc)
        error_line = []
        trace = exc.__traceback__
        while trace:
            error = '"%s"  line %s' % (trace.tb_frame.f_code.co_filename, trace.tb_lineno)
            print('\033[31;0mERROR: %s \033[0m' % error)
            error_line.append(error)
            trace = trace.tb_next
        print('\033[31;0m%s \033[0m' % error_info)
        # error_log.error(f'error_views:{error_views} | error_info:{error_info} | error_line:{error_line[::-1]}')
        error = '服务器内部错误'

        return Response({"code": 1001, "msg": error}, status=500)

 

 

# 在配置文件中配置生效

REST_FRAMEWORK = {
    'EXCEPTION_HANDLER': 'background.utils.exception.custom_exception_handler',
}