You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
38 lines
1.2 KiB
38 lines
1.2 KiB
#!/usr/bin/env python
|
|
# -*- coding:utf-8 -*-
|
|
# project: 1月
|
|
# author: NinEveN
|
|
# date: 2022/1/13
|
|
import logging
|
|
|
|
from rest_framework.exceptions import Throttled
|
|
from rest_framework.views import exception_handler
|
|
|
|
from common.core.response import ApiResponse
|
|
|
|
logger = logging.getLogger(__file__)
|
|
|
|
|
|
def common_exception_handler(exc, context):
|
|
# context['view'] 是TextView的对象，想拿出这个对象对应的类名
|
|
ret = exception_handler(exc, context) # 是Response对象，它内部有个data
|
|
logger.error(f'{context["view"].__class__.__name__} ERROR: {exc} ret:{ret}')
|
|
|
|
if isinstance(exc, Throttled):
|
|
second = f' {exc.wait} 秒之后'
|
|
if not exc.wait:
|
|
second = '稍后'
|
|
ret.data = {
|
|
'code': 999,
|
|
'detail': f'您手速太快啦，请{second}再次访问',
|
|
}
|
|
|
|
if not ret: # drf内置处理不了，丢给django 的，我们自己来处理
|
|
return ApiResponse(msg='error', result=str(exc), code=5000)
|
|
else:
|
|
msg = ret.data.get('msg')
|
|
if not msg:
|
|
msg = 'error'
|
|
else:
|
|
del ret.data['msg']
|
|
return ApiResponse(msg=msg, status=ret.status_code, **ret.data)
|
|
|