优化全局日志中间件

MiddleWare/requestRecord.py

@@ -28,15 +28,11 @@ class RequestRecordMiddleware(MiddlewareMixin):
             else:
                 parameter = ''
             content = eval(str(response.content, 'utf-8'))  # bytes 转为 dict
-            # 请求是否成功
-            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))
+
+            # 记录响应码不为0的接口日志
+            if ('result_code' in content and content['result_code'] != 0) or \
+                    ('code' in content and content['code'] != 0):
+                LOGGER.info('请求路径:{}, 请求方式:{}, 输入数据:{}, 输出数据:{}, 响应状态:{}'.
+                            format(url, method, parameter, content, response.status_code))
         finally:
             return response