Переглянути джерело

MOD:更新检查是否已经登录

cxyu 3 роки тому
батько
коміт
8d1646c566
1 змінених файлів з 15 додано та 6 видалено
  1. 15 6
      wechat_action/login_ad.py

+ 15 - 6
wechat_action/login_ad.py

@@ -180,17 +180,26 @@ class LogIn:
         self.driver.get('https://a.weixin.qq.com/client')
         for _ in driver_cookies:
             self.driver.add_cookie(_)
-        self.driver.get('https://a.weixin.qq.com/client')
+        self.driver.get('https://a.weixin.qq.com/index.html')
 
+        result = False
         try:
-            WebDriverWait(self.driver, 3).until(
+            WebDriverWait(self.driver, 5).until(
                 lambda driver: self.driver.find_elements_by_xpath('//*[@class="qrcode lightBorder"]'))
+            result = True
         except:
+            # print('cookie不能使用')
             pass
-        if 'qrcode lightBorder' in self.driver.page_source:
-            return False
-        else:
-            return True
+        # print(self.driver.page_source)
+        self.driver.save_screenshot('log.png')
+        # time.sleep(100)
+        return result
+        # if 'qrcode lightBorder' in self.driver.page_source:
+        #     print('cookie 不能使用')
+        #     return False
+        # else:
+        #     print('cookie 可以使用')
+        #     return True
 
     def upadte_user_info(self):
         # TODO: 更新 用户相关信息