Просмотр исходного кода

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

cxyu 3 лет назад
Родитель
Сommit
b3c31d14ee
1 измененных файлов с 7 добавлено и 18 удалено
  1. 7 18
      wechat_action/login_ad.py

+ 7 - 18
wechat_action/login_ad.py

@@ -39,11 +39,11 @@ class LogIn:
         prefs = {"profile.managed_default_content_settings.images": 2, 'permissions.default.stylesheet': 2}
         options.add_experimental_option("prefs", prefs)
 
-        driver = webdriver.Remote(
-            # command_executor='http://192.168.1.100/wd/hub',
-            command_executor='http://118.31.53.105:4555/wd/hub',
-            options=options)
-        # driver = webdriver.Chrome(options=options)
+        # driver = webdriver.Remote(
+        #     # command_executor='http://192.168.1.100/wd/hub',
+        #     command_executor='http://118.31.53.105:4555/wd/hub',
+        #     options=options)
+        driver = webdriver.Chrome(options=options)
         driver.maximize_window()
         return driver
 
@@ -185,21 +185,10 @@ class LogIn:
         result = False
         try:
             WebDriverWait(self.driver, 5).until(
-                lambda driver: self.driver.find_elements_by_xpath('//*[@class="qrcode lightBorder"]'))
-            result = True
+                lambda driver: self.driver.find_elements_by_xpath('//*[@class="headerInner"]'))
         except:
-            # print('cookie不能使用')
-            pass
-        # print(self.driver.page_source)
-        self.driver.save_screenshot('log.png')
-        # time.sleep(100)
+            result = True
         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: 更新 用户相关信息