天眼查-工具包

enterprise_tyc/classtool.py

@@ -136,7 +136,7 @@ class Driver():
         return session
 
 class Login():
-    def login(self, driver):
+    def login(self, driver=None):
         cookies = {}
         cookies_list, id_cookie, user_name = Token().get_cookies()
         if cookies_list:
@@ -146,16 +146,15 @@ class Login():
             time.sleep(30 * 60)
             return '', '', ''
         log.info(f'=====当前使用的是{user_name}的cookie======')
-        for cookie in cookies_list:
-            driver.add_cookie(cookie)
-        time.sleep(3)
-        driver.refresh()
-        time.sleep(3)
-        # jar = requests.cookies.RequestsCookieJar()  # 创建一个Cookie Jar对象
+        if driver:
+            for cookie in cookies_list:
+                driver.add_cookie(cookie)
+            time.sleep(3)
+            driver.refresh()
+            time.sleep(3)
         update_headers = {}
         for cookie in cookies_list:
             cookies[cookie['name']] = cookie['value']
-            # jar.set(cookie['name'], cookie['value'])
             if cookie['name'] == 'auth_token':
                 update_headers['X-AUTH-TOKEN'] = cookie['value']
             if cookie['name'] == 'TYCID':
@@ -163,6 +162,5 @@ class Login():
 
         s = requests.Session()
         s.cookies.update(cookies)
-        # s.cookies.update(jar)
 
         return driver, id_cookie, s, update_headers