自定義異常處理
阿新 • • 發佈:2020-10-04
通常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',
}