feat: reomve identify_verification_code in login process

main
jackluson 2 years ago
parent 88dbaa0249
commit c3775a2d28

@ -63,41 +63,43 @@ def mock_login_site(chrome_driver, site_url, redirect_url=None):
from selenium.webdriver.support import expected_conditions as EC from selenium.webdriver.support import expected_conditions as EC
username = chrome_driver.find_element_by_id('emailTxt') username = chrome_driver.find_element_by_id('emailTxt')
password = chrome_driver.find_element_by_id('pwdValue') 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_username = os.getenv('morning_star_username')
env_password = os.getenv('morning_star_password') env_password = os.getenv('morning_star_password')
username.send_keys(env_username) username.send_keys(env_username)
password.send_keys(env_password) password.send_keys(env_password)
count = 1
flag = True # check_code = chrome_driver.find_element_by_id('txtCheckCode')
while count < 10 and flag: # count = 1
code = identify_verification_code(chrome_driver) # flag = True
check_code.clear() # while count < 10 and flag:
time.sleep(1) # code = identify_verification_code(chrome_driver)
check_code.send_keys(code) # check_code.clear()
time.sleep(3) # time.sleep(1)
submit = chrome_driver.find_element_by_id('loginGo') # check_code.send_keys(code)
submit.click() # time.sleep(3)
# 通过弹窗判断验证码是否正确 # submit = chrome_driver.find_element_by_id('loginGo')
time.sleep(3) # submit.click()
from selenium.webdriver.common.by import By # # 通过弹窗判断验证码是否正确
# message_container = chrome_driver.find_element_by_id('message-container') # time.sleep(3)
try: # from selenium.webdriver.common.by import By
message_box = chrome_driver.find_element_by_id( # # message_container = chrome_driver.find_element_by_id('message-container')
'message-container') # try:
flag = message_box.is_displayed() # message_box = chrome_driver.find_element_by_id(
if flag: # 'message-container')
close_btn = message_box.find_element( # flag = message_box.is_displayed()
By.CLASS_NAME, "modal-close") # if flag:
close_btn.click() # close_btn = message_box.find_element(
time.sleep(1) # By.CLASS_NAME, "modal-close")
print('flag', flag) # close_btn.click()
# time.sleep(1)
# print('flag', flag)
except: # except:
return True # return True
if count > 10: # if count > 10:
return False # return False
time.sleep(5)
return True return True

Loading…
Cancel
Save