一、多窗口切换
在页面操作过程中有时候点击某个链接会弹出新的窗口,这时就需要主机切换到新打开的窗口上进行操作。WebDriver提供了switch_to.window()方法。可以实现在不同的窗口之间切换。
以百度和百度注册页为例,在两个窗口之间切换#coding:utf-8from selenium import webdriverfrom selenium.webdriver.common.keys import Keysimport timedriver=webdriver.Chrome()driver.implicitly_wait(10)driver.get("http://www.baidu.com")#获得百度搜索窗口句柄search_windows=driver.current_window_handledriver.find_element_by_link_text(u"登录").click()driver.find_element_by_link_text(u"立即注册").click()#获得当前所有打开的窗口的句柄all_handles=driver.window_handles#进入注册窗口for handle in all_handles: if handle != search_windows: driver.switch_to_window(handle) print 'now register window!' driver.find_element_by_name("userName").send_keys('username') time.sleep(2)#回到搜索窗口for handle in all_handles: if handle==search_windows: driver.switch_to_window(handle) print 'now search window!' driver.find_element_by_id('TANGRAM__PSP_4__closeBtn').click() driver.find_element_by_id('kw').send_keys('selenium') driver.find_element_by_id('kw').send_keys(Keys.ENTER) time.sleep(2)driver.quit()
脚本执行的过程:打开百度首页,通过current_window_handle获得当前窗口的句柄,并赋值给变量search_handle。接着打开登录弹窗,在登录弹窗上单击“立即注册”,从而打开新的注册窗口。通过window_handles获得当前打开的所有窗口的句柄,并赋值给变量all_handles。
第一个循环遍历all_handles,如果handles不等于search_handle,那么一定是注册窗口,因为脚本执行过程中只打开了两个窗口。所以,通过switch_to.window()切换到注册页进行注册操作。第二个循环类似,不过这一次判断如果handle等于search_handle,那么切换到百度搜索页面,然后进行搜索操作。本例中涉及的新方法:(1)current_window_handle:获得当前窗口句柄(2)window_handles:返回所有窗口的句柄到当前会话(3)switch_to.window():用于切换到相应的窗口,与switch_to.frame()类似,前者用于不同窗口的切换,后者用于不同表单之间的切换。