drf框架的异常处理
drf框架对自己内部定义的方法做了特殊处理 比如:频率控制组件 返回特殊的信息
如何在drf异常处理的基础上再重新包装异常
from rest_framework.views import exception_handler from rest_framework.response import Response import logging error_log = logging.getLogger('') def custom_exception_handler(exc, context): '''重定义异常(使异常返回值符合规范)''' # 先让drf内部做处理 response = exception_handler(exc, context) # drf处理过得异常 可以按照自己的规范来返回 if response: '''drf内部错误''' print(response.data) detail = response.data.get('detail') non_field_errors = response.data.get('non_field_errors') if detail: return Response({"code": 1001, "msg": detail}) if non_field_errors: return Response({"code": 1001, "msg": non_field_errors[0]}) return response # 没有经过drf处理的异常比如代码报错 我们可以捕捉下来写入日志 else: '''异常写入日志''' error_views = context['view'] error_info = repr(exc) error_line = [] trace = exc.__traceback__ while trace: error = '"%s" line %s' % (trace.tb_frame.f_code.co_filename, trace.tb_lineno) print('\033[31;0mERROR: %s \033[0m' % error) error_line.append(error) trace = trace.tb_next print('\033[31;0m%s \033[0m' % error_info) # error_log.error(f'error_views:{error_views} | error_info:{error_info} | error_line:{error_line[::-1]}') error = '服务器内部错误' return Response({"code": 1001, "msg": error}, status=500)
# 在配置文件中配置生效 REST_FRAMEWORK = { 'EXCEPTION_HANDLER': 'background.utils.exception.custom_exception_handler', }
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步