python+selenium（四）

1、多窗口切换

本文来介绍如何处理driver在多窗口之间切换，想一下这样的场景，在页面A点击一个连接，会触发在新Tab或者新窗口打开页面B，由于之前的driver实例对象在页面A，但是你接下来的脚本是操作页面B的元素，这样就造成了找不到元素的报错。本来介绍selenium中switch_to.window()方法来处理这个问题。

测试场景：打开百度新闻（页面A），点击热点新闻中第一个新闻链接（一般是国家领导人的新闻），会在第二个窗口打开这个新闻的具体详情页（页面B），测试需要去判断你点击这个这个新闻，在打开的详情页是否正确。

问题拆分：

1. 我们已经知道switch_to.window()方法可以处理窗口切换的问题

2. 在页面A跳转到页面B之前，我们需要用一个变量保存这个新闻的标题

3. 切换到页面B后，我们获取这个新闻标题，然后和前面这个变量保存的值去对比，如果相等，那么就测试通过。

我们分两个步骤去解答这个测试需求：

1. 先实现页面A切换到页面B

2. 页面A和页面B两个新闻标题进行对比

先看看窗口切换的脚本代码：
 

# coding=utf-8
import time
from selenium import webdriver
 
 
driver = webdriver.Chrome()
driver.maximize_window()
driver.get('http://news.baidu.com')
time.sleep(1)
 
driver.find_element_by_xpath("//*[@id='pane-news']/div/ul/li[1]/strong/a").click()
print (driver.current_window_handle) # 输出当前窗口句柄
handles = driver.window_handles # 获取当前全部窗口句柄集合
print (handles) # 输出句柄集合
 
for handle in handles:# 切换窗口
    if handle != driver.current_window_handle:
        print ('switch to second window',handle)
        driver.close() # 关闭第一个窗口
        driver.switch_to.window(handle) #切换到第二个窗口

2、处理iframe切换

本文介绍如何利用Selenium中方法，在不同iframe中切换。

问题：有时候我们发现元素定位没问题，在测试回放的过程，发现就是找不到元素报错。

由于没有找到合适的iframe网站，这里不好用代码举例，简单文字加图片来介绍。

自己写了一个网页，利用iframe网页框架，把网页分成上下两个不同iframe控制的页面，把下面html代码保存到一个记事本，然后修改名称为iframe-demo.html
 

driver.switch_to.frame("iframe1")
# 操作目标元素，这个目标元素在 iframe1里面，这里就是百度文本输入框输入文字
driver.switch_to.default_content() 

3、处理Alert弹窗

本文来介绍如何通过Selenium方法去处理网页Alert弹窗，和处理iframe类似，都是通过switch_to方法。这里还是没有找到合适的alert弹窗网站，我们就自己创建一个吧，前面文章介绍了如何通过执行JS来增加一个弹窗。

# coding=utf-8
import time
from selenium import webdriver
 
 
driver = webdriver.Chrome()
driver.maximize_window()
driver.implicitly_wait(6)
driver.get("https://www.baidu.com")
time.sleep(1)
 
driver.execute_script("window.alert('这是一个测试Alert弹窗');")
time.sleep(2)
driver.switch_to_alert().accept()  # 点击弹出里面的确定按钮
#driver.switch_to_alert().dismiss() # 点击弹出上面的X按钮

4、获取页面全部图片信息

# coding=utf-8
import time
from selenium import webdriver
 
 
driver = webdriver.Chrome()
driver.maximize_window()
driver.implicitly_wait(6)
driver.get("http://news.baidu.com")
time.sleep(1)
 
for image in driver.find_elements_by_tag_name("img"):
    print (image.text)
    print (image.size)
    print (image.tag_name)

5、获取页面元素的href属性

本文来介绍如何通过Selenium获取页面元素的某一个属性。一个元素可能有多个属性，例如 class, id, name, text, href, vale等等。这里我们举例一个爬虫中经常需要处理的链接问题：找出当前页面所有的超链接。

已百度首页为例，打印所有包含href的元素的链接。
 

# coding=utf-8
 
from selenium import webdriver
 
driver = webdriver.Chrome()
driver.maximize_window()
driver.get('http://news.baidu.com')
driver.implicitly_wait(8)
 
for i in driver.find_elements_by_xpath("//*/input[@type='radio']"):
    i.click()

6、如何截图并保存

# coding=utf-8
import time
from selenium import webdriver
 
 
driver = webdriver.Chrome()
driver.maximize_window()
driver.implicitly_wait(6)
driver.get("https://www.baidu.com")
time.sleep(1)
 
driver.get_screenshot_as_file("C:\\Users\\你的账户名\\Desktop\\baidu.png")
driver.quit()