通常 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',
- }
来源: http://www.bubuko.com/infodetail-3679964.html