天眼查-工具包

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======')
+        if driver:
             for cookie in cookies_list:
                 driver.add_cookie(cookie)
             time.sleep(3)
             driver.refresh()
             time.sleep(3)
-        # jar = requests.cookies.RequestsCookieJar()  # 创建一个Cookie Jar对象
         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