Merge branch 'main' into sqlalchemy

jackluson · Sep 1, 2022 · 0f22a6e · 0f22a6e
2 parents d685a60 + c3775a2
commit 0f22a6e
Showing 1 changed file with 31 additions and 29 deletions.
diff --git a/src/utils/login.py b/src/utils/login.py
@@ -63,41 +63,43 @@ def mock_login_site(chrome_driver, site_url, redirect_url=None):
     from selenium.webdriver.support import expected_conditions as EC
     username = chrome_driver.find_element_by_id('emailTxt')
     password = chrome_driver.find_element_by_id('pwdValue')
-    check_code = chrome_driver.find_element_by_id('txtCheckCode')
     env_username = os.getenv('morning_star_username')
     env_password = os.getenv('morning_star_password')
     username.send_keys(env_username)
     password.send_keys(env_password)
-    count = 1
-    flag = True
-    while count < 10 and flag:
-        code = identify_verification_code(chrome_driver)
-        check_code.clear()
-        time.sleep(1)
-        check_code.send_keys(code)
-        time.sleep(3)
-        submit = chrome_driver.find_element_by_id('loginGo')
-        submit.click()
-        # 通过弹窗判断验证码是否正确
-        time.sleep(3)
-        from selenium.webdriver.common.by import By
-        # message_container = chrome_driver.find_element_by_id('message-container')
-        try:
-            message_box = chrome_driver.find_element_by_id(
-                'message-container')
-            flag = message_box.is_displayed()
-            if flag:
-                close_btn = message_box.find_element(
-                    By.CLASS_NAME, "modal-close")
-                close_btn.click()
-                time.sleep(1)
-            print('flag', flag)
+
+    # check_code = chrome_driver.find_element_by_id('txtCheckCode')
+    # count = 1
+    # flag = True
+    # while count < 10 and flag:
+    #     code = identify_verification_code(chrome_driver)
+    #     check_code.clear()
+    #     time.sleep(1)
+    #     check_code.send_keys(code)
+    #     time.sleep(3)
+    #     submit = chrome_driver.find_element_by_id('loginGo')
+    #     submit.click()
+    #     # 通过弹窗判断验证码是否正确
+    #     time.sleep(3)
+    #     from selenium.webdriver.common.by import By
+    #     # message_container = chrome_driver.find_element_by_id('message-container')
+    #     try:
+    #         message_box = chrome_driver.find_element_by_id(
+    #             'message-container')
+    #         flag = message_box.is_displayed()
+    #         if flag:
+    #             close_btn = message_box.find_element(
+    #                 By.CLASS_NAME, "modal-close")
+    #             close_btn.click()
+    #             time.sleep(1)
+    #         print('flag', flag)
 
-        except:
-            return True
+    #     except:
+    #         return True
 
-    if count > 10:
-        return False
+    # if count > 10:
+    #     return False
+    time.sleep(5)
     return True