09 異常模塊

爲何要自定義異常模塊

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)

如何使用

在配置文件中聲明自定義的異常處理

REST_FRAMEWORK = {
    # 全局配置異常模塊
    #設置自定義異常文件路徑，在api應用下建立exception文件，exception_handler函數
    'EXCEPTION_HANDLER': 'api.exception.exception_handler', 
}

若是未聲明，會採用默認的方式，以下

REST_FRAMEWORK = {
    'EXCEPTION_HANDLER': 'rest_framework.views.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

REST framework定義的異常

• APIException 全部異常的父類
• ParseError 解析錯誤
• AuthenticationFailed 認證失敗
• NotAuthenticated 還沒有認證
• PermissionDenied 權限決絕
• NotFound 未找到
• MethodNotAllowed 請求方式不支持
• NotAcceptable 要獲取的數據格式不支持
• Throttled 超過限流次數
• ValidationError 校驗失敗
  異常模塊的大體流程：從dispatch中的handle_exception進入，get_exception_handler()得到處理異常方法exception_handler()，在這裏也能夠自定義異常方法。執行exception_handler()獲取異常處理的結果。