from django.http import JsonResponse
from rest_framework.views import exception_handler as drf_exception_handler
import logging
from django.db import DatabaseError
from redis.exceptions import RedisError
import traceback
# 獲取在配置文件中定義的logger，用來(lái)記錄日志
logger = logging.getLogger('django')

def exception_handler(exc, context):
    """
    自定義異常處理
    :param exc: 異常
    :param context: 拋出異常的上下文（包含request請(qǐng)求對(duì)象和view視圖對(duì)象）
    :return: Response響應(yīng)對(duì)象
    """
    # 調(diào)用drf框架原生的異常處理方法
    response = drf_exception_handler(exc, context)
    if response is not None:
    	logger.info('REST_FRAMEWORK處理異常')
    	# 使用自定義封裝好的方法處理數(shù)據(jù)格式并返回給前端
        return response_util.success(response)
    view = context['view']
    # 打印異常類以及異常的文本信息
    logger.error('[%s] %s' % (view, exc))
    returnData = {
        'code': 500,
        'msg': '服務(wù)器錯(cuò)誤',
        'data': ""
    }
    # 捕獲其他異常，直接返回 500
    if isinstance(exc, DatabaseError) or isinstance(exc, RedisError):
        # 數(shù)據(jù)庫(kù)異常，使用traceback.format_exc()可以像系統(tǒng)報(bào)錯(cuò)沒(méi)攔截那樣子打印詳細(xì)錯(cuò)誤信息又不會(huì)導(dǎo)致程序停止
        logger.error(f'其他未知錯(cuò)誤：{traceback.format_exc()}')
        # 使用django原生的響應(yīng)對(duì)象返回，不能用DRF框架的響應(yīng)對(duì)象
        return JsonResponse(data=returnData, status=500)
    else:
        # 打印錯(cuò)誤詳細(xì)信息
        logger.error(f'其他未知錯(cuò)誤：{traceback.format_exc()}')
        # 使用django原生的響應(yīng)對(duì)象返回，不能用DRF框架的響應(yīng)對(duì)象
        return JsonResponse(data=returnData, status=500)

REST_FRAMEWORK = {
    # 異常處理
    'EXCEPTION_HANDLER': 'meiduo_service.utils.excepitons.exception_handler',
}