Selenium 窗口切换
来源:SegmentFault 思否社区
作者:Leofighting
在浏览网页的时候,有时点击一个链接或者按钮,会弹出一个新的窗口。
使用 Selenium 进行 Web 自动化测试,如果弹出新窗口时,没有对窗口进行切换,那么,WebDriver 对象 的焦点对应的,依然是旧窗口,后续的自动化操作,将继续在旧窗口中进行。
实例验证:
# 模拟场景:打开百度,点击左上角“视频”链接,打印出当前`WebDriver 对象` 对应的窗口信息
from selenium import webdriver
from time import sleep
driver = webdriver.Chrome() # 打开浏览器
driver.maximize_window() # 浏览器最大化
driver.get("https://www.baidu.com/") # 跳转至百度首页
sleep(1)
element = driver.find_element_by_xpath("//div[@id='s-top-left']/a[4]") # 定位“视频”元素
element.click() # 点击
sleep(1)
print(driver.title) # 百度一下,你就知道
print(driver.current_url) # https://www.baidu.com/
driver.quit() # 关闭浏览器
根据程序的打印结果可知,窗口依然停留在百度首页,当前的 url :https://www.baidu.com/,由此可推断,Selenium 没有按照我们预期那样,自动切换到新的窗口。
考虑到后续的操作步骤都需要在新窗口进行,就需要对进行窗口切换,Selenium 提供的窗口切换方法是:
WebDriver.switch_to.window()
WebDriver 对象有 window_handles 属性,返回一个列表,里面记录了当前浏览器所有的窗口句柄。
句柄:对应窗口的 id
# 获取当前浏览器上所有窗口句柄
from selenium import webdriver
from time import sleep
driver = webdriver.Chrome() # 打开浏览器
driver.get("https://www.baidu.com/") # 跳转至百度首页
sleep(1)
element = driver.find_element_by_xpath("//div[@id='s-top-left']/a[4]") # 定位“视频”元素
element.click() # 点击
sleep(1)
print(driver.current_window_handle) # 当前浏览器句柄:CDwindow-1FABF3D9B0B190F9883E66F25481738E
print(driver.window_handles) # 获取所有句柄:['CDwindow-1FABF3D9B0B190F9883E66F25481738E', 'CDwindow-0871993F618FDBE540426562AEA2FB32']
for handle in driver.window_handles:
print(handle)
driver.quit() # 关闭浏览器
因为 window_handles 返回的是一个列表,所以,可以通过列表下标来更换当前浏览器的窗口句柄,来实现窗口的自由切换:
# 模拟场景:通过百度首页,打开 hao123,再通过 hao123 点击京东,进入京东首页,搜索“笔记本电脑”
from selenium import webdriver
from time import sleep
driver = webdriver.Chrome() # 打开浏览器
driver.get("https://www.baidu.com/") # 跳转至百度首页
sleep(1)
element = driver.find_element_by_xpath("//div[@id='s-top-left']/a[2]") # 定位“hao123”元素
element.click() # 点击
sleep(1)
windows1 = driver.window_handles # 获取所有窗口句柄
print(windows1) # 打印句柄列表
print(driver.current_window_handle) # 打印切换前的句柄
driver.switch_to.window(windows1[-1]) # 切换到最后一个句柄
print(driver.current_window_handle) # 打印切换后的句柄
element1 = driver.find_element_by_xpath("//li[@data-id='11']//a[1]") # 在 hao123 定位 京东
element1.click() # 点击
windows2 = driver.window_handles # 第二次获取所有窗口句柄
print(windows2) # 打印句柄列表
print(driver.current_window_handle) # 打印切换前的句柄
driver.switch_to.window(windows2[-1]) # 第二次切换到最后一个句柄
print(driver.current_window_handle) # 打印切换后的句柄
element2 = driver.find_element_by_xpath("//input[@aria-label='搜索']") # 定位搜索输入框
element2.send_keys("笔记本电脑") # 输入内容
element3 = driver.find_element_by_class_name("button") # 定位搜索按钮
element3.click() # 点击
sleep(3)
driver.quit() # 关闭浏览器
通过这个实例操作,可以看到,每一次切换窗口,都需要对获取到最新的句柄位置。
同样,由于 WebDriver.window_handles 返回的是一个列表,也可以通过下标值来切换到旧的窗口。
总结
评论