DRF框架（三）——exception异常模块

目录

 

自定义异常模块

源码分析

使用方法

1、配置异常模块 settings.py

2、应用下创建处理异常文件 exception.py

---

自定义异常模块

• 所有经过drf的APIView视图类产生的异常，都可以提供异常处理方案
• drf默认提供了异常处理方案（ref_framework.views.exception_handler)，但是处理范围有限
• drf提供的处理方案有两种，处理了返回异常现象，没处理返回None(后续就是服务器抛异常给前台）
• 自定义异常的目的就是解决drf没有处理的异常，让前台得到合理的异常信息返回，后台记录异常具体信息

 

源码分析

# 异常模块：APIView类的dispatch方法中
response = self.handle_exception(exc)  # 点进去

# 获取处理异常的句柄(方法)
# 一层层看源码，走的是配置文件，拿到的是rest_framework.views的exception_handler
# 自定义：直接写exception_handler函数，在自己的配置文件配置EXCEPTION_HANDLER指向自己的异常处理函数
exception_handler = self.get_exception_handler()

# 异常处理的结果
# 自定义异常就是提供exception_handler异常处理函数，处理的目的就是让response一定有值
response = exception_handler(exc, context)

 

使用方法

1、配置异常模块 settings.py

# 自定义drf配置-全局配置
REST_FRAMEWORK = {
    # 全局配置异常模块
    # 设置自定义异常文件路径，在rest_api应用下创建exception文件，exception_handler函数
    'EXCEPTION_HANDLER': 'rest_api.exception.exception_handler',
}

2、应用下创建处理异常文件 exception.py

1. 先将异常处理交给rest_framework.views的exception_handler去处理
2. 判断处理的结果（返回值）response，有值代表drf已经处理了，None代表需要自己处理

   # 自定义异常处理文件exception，重写exception_handler函数
   from rest_framework.views import  exception_handler as drf_exception_handler # drf原生处理异常函数取别名
   from rest_framework.views import Response
   from rest_framework import status
   
   
   def exception_handler(exc, context):
       # drf的exception_handler做基础处理
       response = drf_exception_handler(exc, context)
       # 为空，进行自定义二次处理
       if response is None:
           # print(exc)        # 错误原因 
           # print(context)    # 错误信息
           print('%s - %s - %s' % (context['view'], context['request'].method, exc))
           return Response({
               'detail': '服务器错误'
           }, status=status.HTTP_500_INTERNAL_SERVER_ERROR, exception=True)
       return response