如何在Selenium中验证电子邮件用户输入？

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
import re

# 启动浏览器驱动
driver = webdriver.Chrome()

# 导航到电子邮件登录页面
driver.get("https://example.com/login")

# 找到电子邮件输入框并输入待验证的电子邮件地址
email_input = driver.find_element(By.ID, "email-input")
email_input.send_keys("example@example.com")

# 找到登录按钮并点击
login_button = driver.find_element(By.ID, "login-button")
login_button.click()

# 等待页面加载完成
wait = WebDriverWait(driver, 10)
result_element = wait.until(EC.presence_of_element_located((By.ID, "result-element")))

# 获取验证结果的文本内容
result_text = result_element.text

# 使用正则表达式验证电子邮件地址的格式
email_pattern = r'^[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}$'
is_valid_email = re.match(email_pattern, result_text)

# 输出验证结果
if is_valid_email:
    print("输入的电子邮件地址有效")
else:
    print("输入的电子邮件地址无效")

# 关闭浏览器
driver.quit()