后台response和异常处理封装

我们自己封装的一些东西，往往放在一个utils文件夹内,以后也方便管理和导入

后台response封装

1. # 自己封装的Response对象
3. from rest_framework.response import Response
5. class APIResponse(Response):
6.     def __init__(self,code=1,msg='成功',result=None,status=None,headers=None,content_type=None,**kwargs):
7.         dic = {
8.             'code':code,
9.             'msg':msg
10.         }
11.         if result:
12.             dic['result']=result
13.         dic.update(kwargs)
14.         super().__init__(data=dic,status=status,headers=headers,content_type=content_type)

复制代码

自己封装的异常处理

1. # 这里用来放全局异常
3. # 这个方法可以从rest_framework里的settings.py里找到它的配置，然后导入就行，重写该方法
4. from rest_framework.views import exception_handler
5. from . import response # 导入自己定义的响应对象
6. from .logger import log  #导入自己定义的日志对象
9. def common_exception_handler(exc, context):
10.     # 记录以下报错信息到日志中
11.     # context['view']可以取到具体是哪个视图出了错
12.     log.error('view是：%s，错误是：%s' %(context['view'].__class__.__name__,str(exc)))
13.     # 先调用一下原来的exception_handler方法
14.     ret = exception_handler(exc, context) # ret是一个Response对象，其内部有一个data属性，放的就是错误信息
15.     # 这个表示如果drf处理不了的异常（交给django处理的异常），我们进行处理一下
16.     if not ret:
17.         # 这里面也可以放更多其他逻辑，具体对于会出现的异常进行分类处理
18.         # 比如下面这样
19.         # if isinstance(exc,KeyError):
20.         return response.APIResponse(code=0,msg='error',result=str(exc))
21.     else:
22.         # ret.data是一个字典，字典里是drf处理的异常信息
23.         return response.APIResponse(code=0,msg='error',result=ret.data)

复制代码

封装完毕之后，还需要在settings里配置一下，告诉drf采用我们封装的异常处理

1. # 放自己的全局异常
2. REST_FRAMEWORK = {
3.     'EXCEPTION_HANDLER': 'luffyapi.utils.exceptions.common_exception_handler',
4. }

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！