|
@@ -12,10 +12,10 @@ from AnsjerPush.config import LOGGER
|
|
|
|
|
|
|
|
|
class RequestRecordMiddleware(MiddlewareMixin):
|
|
|
- def Request(self, request):
|
|
|
+ def process_request(self, request):
|
|
|
request.start_time = time.time()
|
|
|
|
|
|
- def Response(self, request, response):
|
|
|
+ def process_response(self, request, response):
|
|
|
try:
|
|
|
method = request.method
|
|
|
url = request.path
|
|
@@ -29,8 +29,14 @@ class RequestRecordMiddleware(MiddlewareMixin):
|
|
|
parameter = ''
|
|
|
content = eval(str(response.content, 'utf-8')) # bytes 转为 dict
|
|
|
# 请求是否成功
|
|
|
- if content['result_code'] != 0 or content['code'] != 0:
|
|
|
- LOGGER.info('请求路径:{}, 请求方式:{}, 输入数据:{}, 输出数据:{}, 响应状态:{}'.format(url, method, parameter, content,
|
|
|
- response.status_code))
|
|
|
+ if content['code']:
|
|
|
+ old_key = 'code'
|
|
|
+ new_key = 'result_code'
|
|
|
+ new_dict = {new_key: content.pop(old_key)}
|
|
|
+ for key, value in content.items():
|
|
|
+ new_dict[key] = value
|
|
|
+ content = new_dict.copy()
|
|
|
+ if content['result_code'] != 0:
|
|
|
+ LOGGER.info('请求路径:{}, 请求方式:{}, 输入数据:{}, 输出数据:{}, 响应状态:{}'.format(url, method, parameter, content, response.status_code))
|
|
|
finally:
|
|
|
return response
|