MySession.py新增需要填写method进行请求的方法

MySession.py维护了下网页端登录方法 MySession.py维护了下客户端端登录方法

MySession.py新增需要填写method进行请求的方法
MySession.py维护了下网页端登录方法 MySession.py维护了下客户端端登录方法
lixi
1 parent 165ae2bf
Showing 2 changed files with 53 additions and 21 deletions
commons/MySession.py
report/test.log
@@ -67,9 +67,12 @@ class mysession():
         # 则在再次使用该session对象求求该网站的其他网页都会默认使用该session之前使用的cookie等参数
         self.se = requests.session()
         #更新cookie加入UAP_firmId
-        firm=account.split("_")[0]
         co = requests.cookies.RequestsCookieJar()
+        #backup:通过提前配置的方法更新市场id
+        firm=account.split("_")[0]
         co.set("UAP_firmId", self.firmid[firm])
+        #通过客户端登录的方法更新市场id
+        # co.set("UAP_firmId", str(self.get_login_info(account).json()["data"]["user"]["firmId"]))
         self.se.cookies.update(co)
         # 使用session对象的方法POST/GET等
         re = self.se.post(url=self.url, headers=self.header, data=self.body, proxies=self.proxies, **kwargs)
@@ -81,24 +84,42 @@ class mysession():
         # 返回session对象，供其他接口使用
         return self.se
-    def get_session_client(self, account, **kwargs):
-        "get_session和get_session_client的方法只能用一个"
-        "如下代码，可以通过配置文件来控制登录的账户session"
+    def get_login_info(self, account, **kwargs):
+        "用于获取用户信息"
         self.body_client.update({"userName":com.get_global_config("global_data", "account", account).split("&")[0]})
         self.body_client.update({"password":pwdCry(com.get_global_config("global_data", "account", account).split("&")[1])})
         # requests.session()会话保持,比如使用session成功的登录了某个网站，
         # 则在再次使用该session对象求求该网站的其他网页都会默认使用该session之前使用的cookie等参数
+        tmp = requests.post(url=self.url_client, headers=self.header_client, json=self.body_client, proxies=self.proxies, **kwargs)
+        # 使用session对象的方法POST/GET等
+        #获取关键信息供其他接口header使用
+        # self.UAP_accessToken=tmp.json()["data"]["accessToken"]
+        # self.UAP_refreshToken=tmp.json()["data"]["refreshToken"]
+        # 返回session对象，供其他接口使用
+        return tmp
+
+    def get_session_client(self, account, **kwargs):
+        "get_session和get_session_client的方法只能用一个"
+        self.body_client.update({"userName": com.get_global_config("global_data", "account", account).split("&")[0]})
+        self.body_client.update( {"password": pwdCry(com.get_global_config("global_data", "account", account).split("&")[1])})
+        # requests.session()会话保持,比如使用session成功的登录了某个网站，
+        # 则在再次使用该session对象求求该网站的其他网页都会默认使用该session之前使用的cookie等参数
         self.se = requests.session()
         # 使用session对象的方法POST/GET等
-        re = self.se.post(url=self.url_client, headers=self.header_client, json=self.body_client, proxies=self.proxies, **kwargs)
+        self.re = self.se.post(url=self.url_client, headers=self.header_client, json=self.body_client, proxies=self.proxies, **kwargs)
+        #组装请求的cookie
+        co = requests.cookies.RequestsCookieJar()
+        co.set("UAP_firmId", str(self.re.json()["data"]["user"]["firmId"]))
+        co.set("UAP_accessToken", self.re.json()["data"]["accessToken"])
+        co.set("UAP_refreshToken", self.re.json()["data"]["refreshToken"])
+        self.se.cookies.update(co)
         #获取关键信息供其他接口header使用
-        self.UAP_accessToken=re.json()["data"]["accessToken"]
-        self.UAP_refreshToken=re.json()["data"]["refreshToken"]
-        print(self.UAP_accessToken)
-        print(self.UAP_refreshToken)
+        self.UAP_accessToken=self.se.cookies["UAP_accessToken"]
+        self.UAP_refreshToken=self.se.cookies["UAP_refreshToken"]
         # 返回session对象，供其他接口使用
         return self.se
+
     def close_session(self):
         "关闭session"
         self.se.close()
@@ -140,6 +161,19 @@ class mysession():
             res = requests.request(method=method, url=url, data=data, headers=self.webHeaders, **kwargs)
         return res
+
+    def request(self, method ,url, **kwargs):
+        """:param method: method for the new :class:`Request` object:
+        ``GET``, ``OPTIONS``, ``HEAD``, ``POST``, ``PUT``, ``PATCH``, or ``DELETE``.
+        """
+        # 记录日志
+        log.info("{0:=^86}".format(''))
+        log.info("{}\n{}\n".format(url, kwargs))
+        # 进行请求
+        re = self.se.request(method , url, **kwargs, proxies=self.proxies, timeout=self.timeout)
+        return re
+
+
     def get(self, url, **kwargs):
         """Sends a GET request. Returns :class:`Response` object.
@@ -252,10 +286,9 @@ class mysession():
 my = mysession()
 my.set_mark()
 my.cliLogin()
-s1 = my.get_session("sg_user01")
-# s2 = my.get_session("sg_user02")
-# print(s1.cookies)
-# s3 = my.get_session_client("sg_user02")
-# print(s3.cookies)
-# sg1=my.get_session("sg_user01")
-# sy1=my.get_session("sy_user01")
+s1 = my.get_session_client("sg_user01")
+s2 = my.get_session_client("sg_user02")
+# s3 = my.get_session("sg_user01")
+# s3= my.get_session_client("sg_user02")
+# sg=my.get_session_client("sg_user01")
+# sy=my.get_session_client("sy_user01")
 \ No newline at end of file
-[2021-07-22 15:46:06] [INFO] : ======================================================================================
-[2021-07-22 15:46:06] [INFO] : http://test.card.diligrp.com:8386/accountQuery/page.action
-rows=10&page=1&sort=card_create_time&order=desc
-None
-{'headers': {'Host': 'test.card.diligrp.com:8386', 'Connection': 'keep-alive', 'Content-Length': '47', 'Accept': 'application/json,text/javascript,*/*;q=0.01', 'Origin': 'http://test.card.diligrp.com:8386', 'X-Requested-With': 'XMLHttpRequest', 'User-Agent': 'Mozilla/5.0(WindowsNT6.2;WOW64)AppleWebKit/537.36(KHTML,likeGecko)Chrome/84.0.4147.105Safari/537.36', 'Content-Type': 'application/x-www-form-urlencoded', 'Accept-Language': 'zh-CN,zh;q=0.9', 'Referer': 'http://test.card.diligrp.com:8386/accountQuery/list.html', 'Accept-Encoding': 'gzip,deflate'}}
+[2021-07-22 17:11:13] [INFO] : ======================================================================================
+[2021-07-22 17:11:13] [INFO] : http://test.card.diligrp.com:8386/accountQuery/page.action
+{'headers': {'Host': 'test.card.diligrp.com:8386', 'Connection': 'keep-alive', 'Content-Length': '47', 'Accept': 'application/json,text/javascript,*/*;q=0.01', 'Origin': 'http://test.card.diligrp.com:8386', 'X-Requested-With': 'XMLHttpRequest', 'User-Agent': 'Mozilla/5.0(WindowsNT6.2;WOW64)AppleWebKit/537.36(KHTML,likeGecko)Chrome/84.0.4147.105Safari/537.36', 'Content-Type': 'application/x-www-form-urlencoded', 'Accept-Language': 'zh-CN,zh;q=0.9', 'Referer': 'http://test.card.diligrp.com:8386/accountQuery/list.html', 'Accept-Encoding': 'gzip,deflate'}, 'data': 'rows=10&page=1&sort=card_create_time&order=desc'}
+