selenium切换弹窗

from selenium import webdriver
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC


# 创建一个Chrome会话
driver = webdriver.Chrome()
# 访问某网站
driver.get("http://xxx.com")
# 做一个操作，出现alert弹窗
driver.find_element_by_id('xxx').click()


# 方式一：
# 切换到alert弹窗
alert_1 = driver.switch_to.alert

# 方式二：
# EC条件，alert弹框已出现，并切换到alert弹框中
alert_2 = WebDriverWait(driver, 10).until(EC.alert_is_present())


# 三种弹框，分别的操作方式
# 点击确定，关闭弹窗
alert_1.accept()
alert_2.accept()
# 点击取消
alert_1.dismiss()
alert_2.dismiss()
# 输入文本内容
alert_1.send_keys('xxx')
alert_2.send_keys('xxx')

# 关闭浏览器
driver.quit()