1. 程式人生 > 實用技巧 >自定義異常處理

自定義異常處理

通常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',
}