异常模块

为什么要自定义异常模块

1）所有经过drf的APIView视图类产生的异常，都可以提供异常处理方案
2）drf默认提供了异常处理方案(rest_framework.views.exception_handler)，但是处理范围有限
3）drf提供的处理方案两种，处理了返回异常现象，没处理返回None(后续就是服务器抛异常给前台)
4）自定义异常的目的就是解决drf没有处理的异常，让前台得到合理的异常信息返回，后台记录异常具体信息ps:ORM查询时的错误drf不会自动处理

源码分析

# 异常模块：APIView类的dispatch方法中
response = self.handle_exception(exc)  # 点进去

# 获取处理异常的句柄(方法)
# 一层层看源码，走的是配置文件，拿到的是rest_framework.views的exception_handler
# 自定义：直接写exception_handler函数，在自己的配置文件配置EXCEPTION_HANDLER指向自己的
exception_handler = self.get_exception_handler()

# 异常处理的结果
# 自定义异常就是提供exception_handler异常处理函数，处理的目的就是让response一定有值
response = exception_handler(exc, context)

如何使用:

全局配置: settings.py

REST_FRAMEWORK = {
    # 全局配置异常模块
    #设置自定义异常文件路径，在api应用下创建exception文件，exception_handler函数
    "EXCEPTION_HANDLER": "api.exception.exception_handler", 
}

应用文件下创建exception.py

from rest_framework.views import exception_handler as drf_exception_handler
from rest_framework.response import Response
from rest_framework import status

def exception_handler(exc, context):
    # 1.先让drf的exception_handler做基础处理,拿到返回值
    # 2.若有返回值则drf处理了,若返回值为空说明drf没处理,需要我们手动处理
    response = drf_exception_handler(exc, context)
    print(exc)   # 错误内容 "NoneType" object has no attribute "title"
    print(context)
    # {"view": <api.views.Book object at 0x000001BBBCE05B00>, "args": (), "kwargs": {"pk": "3"}, "request": <rest_framework.request.Request object at 0x000001BBBCF33978>}
    print(response)
    # 返回值为空,做二次处理
    if response is None:
        print("%s - %s - %s" % (context["view"], context["request"].method, exc))
        # <api.views.Book object at 0x00000242DC316940> - GET - "NoneType" object has no attribute "title"
        return Response({
            "detail": "服务器错误"
        }, status=status.HTTP_500_INTERNAL_SERVER_ERROR, exception=True)
    return response

异常模块的大致流程：从dispatch中的handle_exception进入，get_exception_handler()获得处理异常方法exception_handler()，在这里也可以自定义异常方法。执行exception_handler()获取异常处理的结果。