获取see_web返回的用户信息

src/inspect_report/utils/account.py

@@ -3,21 +3,39 @@ import logging
 
 import requests
 
+from requests.cookies import RequestsCookieJar
+
 from functools import wraps
 from rest_framework.request import Request
 
 logger = logging.getLogger(__name__)
 
 
+def transfer_cookie(cookies):
+    """
+    将cookie加入cookie_jar对象
+    :param cookies: dict
+    :return: cookie_jar
+    """
+    cookie_jar = RequestsCookieJar()
+    for (key, value) in cookies:
+        cookie_jar.set(key, value)
+    return cookie_jar
+
+
 def get_account_info(func):
+    """
+    装饰器，对请求进行权限验证
+    :param func:
+    :return:
+    """
     @wraps(func)
     def account_info(obj, req: Request, *args, **kw):
-        # info = requests.get('http://172.17.1.74:8248/getUserInfo', headers={
-        #     'Cookie': 'aiforce=a038baa0-9a23-45f6-8042-a01a2309f1a8; JSESSIONID=fef7de70-7157-4b62-abf3-6fe9d2357350; csrfToken=wFVUJX9rT2jyDderTzHKUjaq; EGG_SESS=pRumla5Jc0N4k1qJnqyfP0SIlONA60YZCFEMh2sO_ywQrldAhhiX4Y2exluQfYfm'})
-        # print(info.json())
         logger.info('req中的data为: ' + json.dumps(req.COOKIES))
+        cookie_jar = transfer_cookie(json.dumps(req.COOKIES))
+        info = requests.get('http://172.17.1.74:8248/getUserInfo', cookies=cookie_jar)
         # logger.info('req中的query_params为: ', req.query_params)
-        # logger.info('req中的query_params为: ', req.parsers())
+        logger.info('info中的text为: ' + info.content)
         return func(obj, req, *args, **kw)
 
     return account_info