如何在python中使用selenium进行两步身份验证登录

pip install selenium

from selenium import webdriver
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.chrome.options import Options

# 创建Chrome浏览器实例
options = Options()
options.add_argument("--headless")  # 无界面模式，可选
driver = webdriver.Chrome(options=options)

# 打开登录页面
driver.get("https://example.com/login")

# 定位并输入用户名
username_input = driver.find_element(By.ID, "username")
username_input.send_keys("your_username")

# 定位并输入密码
password_input = driver.find_element(By.ID, "password")
password_input.send_keys("your_password")

# 提交表单
password_input.submit()

# 定位并输入验证码
code_input = WebDriverWait(driver, 10).until(
    EC.presence_of_element_located((By.ID, "code"))
)
code_input.send_keys("your_verification_code")

# 提交验证码
code_input.submit()

# 等待登录成功页面加载完成
WebDriverWait(driver, 10).until(
    EC.title_contains("Welcome")
)

# 执行其他操作，比如访问需要登录的页面或执行特定的任务
# ...

# 关闭浏览器实例
driver.quit()