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，用來記錄日志
logger = logging.getLogger('django')

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

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