Parcourir la source

code改为result_code

guanhailong il y a 2 ans
Parent
commit
fbef46d20b
1 fichiers modifiés avec 11 ajouts et 5 suppressions
  1. 11 5
      MiddleWare/requestRecord.py

+ 11 - 5
MiddleWare/requestRecord.py

@@ -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