Python+Selenium之断言对应的元素是否获取以及基础知识回顾

 # coding=utf-8
 from selenium import webdriver
 driver = webdriver.Firefox()
 driver.maximize_window ()
 driver.get("https://www.baidu.com")
 try:
     driver.find_element_by_id("kwf")
     print("id is find")
 except Exception as e:
     print("exception find ",format(e))
 driver.quit()

运行结果：

注：该断言判断是否获取对应的元素，再自动化脚本执行时，可以定位出现的问题

2.利用link text定位元素

 try:
     driver.find_element_by_link_text("新闻")
     print ('test pass: element found by link text')
 except Exception as e:
     print ("Exception found", format(e))  

3.利用partial link text定位元素

try:
    driver.find_element_by_partial_link_text("主页").click()
    print ('test pass: element found by partial link text')
except Exception as e:
    print ("Exception found", format(e))  

注：partial_link_text与link_text链接定位类似，partial link text就是选择这个元素的link text中一部分字段。

例如：百度首页带链接文字为：“把百度设为主页”。

为了更好的验证是否找到了“把百度添加到首页”这个元素，我在这个地方添加了一个click(),运行代码，可以看到确实点击了这个元素，代表找到了这个元素。

选择partial link text的时候，需要选择一个比较唯一的字段，来区分这个元素。

4.利用class name定位元素

 try:
     driver.find_element_by_class_name("s_ipt")
     print ('test pass: element found by class name')
 except Exception as e:
     print ("Exception found", format(e))  

5.利用name定位元素

try:
    driver.find_element_by_name("wd")  # 这里百度搜索输入框有name = 'wd'这个节点信息
    print ('test pass: element found by name value')
except Exception as e:
    print ("Exception found", format(e)) 

6.利用css定位元素

参考：http://blog.csdn.net/u011541946/article/details/68927139

7.清楚文本信息

 # coding=utf-8
 from selenium import webdriver
 driver = webdriver.Firefox()
 driver.maximize_window ()
 driver.get("https://www.baidu.com")
 driver.find_element_by_id("kw").send_keys("Selenium")
 try:
     driver.find_element_by_id("kw").clear()
     driver.find_element_by_id ("kw1").send_keys ("Selenium")
     print ('test pass: clean successful')
 except Exception as e:
     print ("Exception found", format(e))
 driver.quit()

8.基本页面操作方法

driver.refresh() # 刷新方法 refresh  

driver.back()  # 从百度新闻后退到百度首页

driver.forward() # 百度首页前进到百度新闻

print(driver.capabilities['version'])  # 打印浏览器version的值

print (driver.current_url)  # current_url 方法可以得到当前页面的URL

print (driver.title)  # title方法可以获取当前页面的标题显示的字段

在浏览器中新开一个界面：

 # coding=utf-8
 import time
 from selenium import webdriver
 from selenium.webdriver.common.keys import Keys

 driver = webdriver.Firefox ()
 driver.maximize_window ()
 # driver.implicitly_wait (6)

 driver.get ("http://www.baidu.com/")
 time.sleep (1)
 ele = driver.find_element_by_tag_name ('body').send_keys (Keys.CONTROL + 't')  # 触发ctrl + t 在浏览器中新开一个tab
 time.sleep (1)

8.点击单选按钮-Radio Button

勾选一个单选按钮，就是调用元素方法click()

我们利用for语句遍历这两个单选按钮，依次点击他们。

 from selenium import webdriver
 import time
 driver = webdriver.Firefox ()
 driver.maximize_window ()
 driver.get ('http://news.baidu.com')
 # driver.implicitly_wait (8)
 print(driver.find_elements_by_xpath ("//*/input[@type='radio']"))
 for i in driver.find_elements_by_xpath ("//*/input[@type='radio']"):
     i.click ()
     time.sleep (5)

9.获取元素页面上的文字

 # coding=utf-8
 import time
 from selenium import webdriver
 from selenium.webdriver.common.keys import Keys

 driver = webdriver.Firefox ()
 driver.maximize_window ()
 # driver.implicitly_wait (6)

 driver.get ("http://www.baidu.com/")
 time.sleep (1)
 driver.find_element_by_xpath ("//*[@id='u1']/a[7]").click ()
 time.sleep (1)

 driver.find_element_by_xpath ("//*[@id='TANGRAM__PSP_10__submit']").click ()

 # 断言方法二，本文重点介绍方法
 error_mes = driver.find_element_by_xpath ("//*[@id='TANGRAM__PSP_10__error']").text
 print(error_mes)
 try:
     assert error_mes == u'请您填写手机/邮箱/用户名'
     print ('Test pass.')
 except Exception as e:
     print ("Test fail.", format (e))

注：不同版本的火狐浏览器，定位的元素会有所不同

10.验证控件是否选中

 # coding=utf-8
 import time
 from selenium import webdriver

 driver = webdriver.Firefox ()
 driver.maximize_window ()
 # driver.implicitly_wait (6)
 driver.get ("http://news.baidu.com/")
 time.sleep (1)aluse
 try:
     # driver.find_element_by_xpath ("//*[@id='news']").is_selected ()
     assert driver.find_element_by_xpath ("//*[@id='newstitle']").is_selected () == True
     print ('Test Pass.')
 except Exception as e:
     print ('Test fail', format (e))

总结：

元素方法is_selected()返回是是布尔值（True  /Faluse），用来判断单选或者多选控件是否被选中，或者下拉选择菜单是否选择一个默认的option，都可以通过这个方法去判断。

11.获取页面元素的大小

 # coding=utf-8
 import time
 from selenium import webdriver

 driver = webdriver.Firefox ()
 driver.maximize_window ()
 # driver.implicitly_wait (6)
 driver.get ("http://baidu.com/")
 time.sleep (1)
 Size = driver.find_element_by_id("su")
 print(Size.size)

12.多窗口之间的切换

场景：在页面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.Firefox ()
 driver.maximize_window ()
 driver.get ('http://news.baidu.com')
 time.sleep (3)

 driver.find_element_by_xpath ("//*[@id='pane-news']/div/ul/li[1]/strong/a").click ()
 time.sleep (3)
 print (driver.current_window_handle)  # 输出当前窗口句柄
 handles = driver.window_handles  # 获取当前全部窗口句柄集合
 print (handles)  # 输出句柄集合

 for handle in handles:  # 切换窗口
     print(handle)
     if handle != driver.current_window_handle:
         print(handle)
         print ('switch to second window', handle)
         driver.close ()  # 关闭第一个窗口
         driver.switch_to.window (handle)  # 切换到第二个窗口
         print(handle)

注：通过driver.current_window_handle获取的当前窗口句柄，无论打开多少个窗口，只要不使用driver.switch_to.window（）就是第一次打开的窗口。

使用driver.window_handles获取的窗口为当前打开的所有窗口。

只有driver.switch_to.window（）才能切换到对应的窗口，获取窗口句柄。

 # coding=utf-8
 import time
 from selenium import webdriver

 # driver = webdriver.Chrome()
 driver = webdriver.Firefox()
 # driver.maximize_window ()
 driver.get ('http://news.baidu.com')
 time.sleep (1)

 news_link = driver.find_element_by_xpath ("//*[@id='pane-news']/div/ul/li[1]/strong/a")
 page1_title_string = news_link.text  # 得到页面A新闻标题
 print(page1_title_string)
 news_link.click ()  # 点击新闻链接
 time.sleep (1)
 handles = driver.window_handles

 for handle in handles:  # 切换窗口（切换到搜狗）
     if handle != driver.current_window_handle:
         print( 'switch to second window', handle)
         driver.close ()  # 关闭第一个窗口
         driver.switch_to.window (handle)  # 切换到第二个窗口
 page2_title_string = driver.find_element_by_xpath ("/html/body/div[2]/div[2]/div[4]/div[2]/div/div[1]").text

 print(page2_title_string)
 # 详情页有一个原标题

 try:
     assert page1_title_string in page2_title_string  # 判断页面B标题是否包含页面A标题
     print ('Test Pass.')
 except Exception as e:
     print ('Test Fail')

注：

1.一般编程的过程中，要提前设置浏览器最大化，不要使用drive.maxisize_window()去设置窗口的大小。否者当浏览器窗口最大时，使用该方法浏览器就会变成最小，导致有些元素通过xpath定位失效。

2网页的title与新闻详情页的title是不一样的概念，前者通过driver.title即可获取，后者通过元素定位获取。

13.处理iframe切换

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

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

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

14.处理alert弹框

 driver.switch_to_alert().accept()  # 点击弹出里面的确定按钮
 #driver.switch_to_alert().dismiss() # 点击弹出上面的X按钮  

15.获取当前页面的全部图片信息

 # 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)
 print(driver.find_elements_by_tag_name ("img"))
 for image in driver.find_elements_by_tag_name ("img"):
     print (image.text)
     print (image.size)
     print (image.tag_name)

运行结果：

运行结果，发现没有图片名称打印出来，说明百度新闻页面，所有图片都没有给出text这个属性，前端的妹子没有写图片的text属性

16.获取页面元素的herf属性

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

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

 # coding=utf-8
 import time
 from selenium import webdriver

 driver = webdriver.Firefox()
 driver.maximize_window()
 # driver.implicitly_wait(6)
 driver.get("https://www.baidu.com")
 time.sleep(1)
 print(driver.find_elements_by_xpath("//*[@href]"))
 for link in driver.find_elements_by_xpath("//*[@href]"):
     print(link)
     print(link.get_attribute('href'))
 driver.quit()

17.如何截图并保存

 # coding=utf-8
 import time
 from selenium import webdriver

 driver = webdriver.Firefox()
 driver.maximize_window()

 driver.get("https://www.baidu.com")
 time.sleep(1)

 driver.get_screenshot_as_file("‪C:\\Users\\fyr\\Desktop\\新建文本文档 11(2).png")
 time.sleep(4)
 driver.quit()