1 # coding=utf-8 2 from selenium import webdriver 3 driver = webdriver.Firefox() 4 driver.maximize_window () 5 driver.get("https://www.baidu.com") 6 try: 7 driver.find_element_by_id("kwf") 8 print("id is find") 9 except Exception as e: 10 print("exception find ",format(e)) 11 driver.quit()
运行结果:
注:该断言判断是否获取对应的元素,再自动化脚本执行时,可以定位出现的问题
2.利用link text定位元素
1 try: 2 driver.find_element_by_link_text("新闻") 3 print (‘test pass: element found by link text‘) 4 except Exception as e: 5 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定位元素
1 try: 2 driver.find_element_by_class_name("s_ipt") 3 print (‘test pass: element found by class name‘) 4 except Exception as e: 5 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.清楚文本信息
1 # coding=utf-8 2 from selenium import webdriver 3 driver = webdriver.Firefox() 4 driver.maximize_window () 5 driver.get("https://www.baidu.com") 6 driver.find_element_by_id("kw").send_keys("Selenium") 7 try: 8 driver.find_element_by_id("kw").clear() 9 driver.find_element_by_id ("kw1").send_keys ("Selenium") 10 print (‘test pass: clean successful‘) 11 except Exception as e: 12 print ("Exception found", format(e)) 13 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方法可以获取当前页面的标题显示的字段
在浏览器中新开一个界面:
1 # coding=utf-8 2 import time 3 from selenium import webdriver 4 from selenium.webdriver.common.keys import Keys 5 6 driver = webdriver.Firefox () 7 driver.maximize_window () 8 # driver.implicitly_wait (6) 9 10 driver.get ("http://www.baidu.com/") 11 time.sleep (1) 12 ele = driver.find_element_by_tag_name (‘body‘).send_keys (Keys.CONTROL + ‘t‘) # 触发ctrl + t 在浏览器中新开一个tab 13 time.sleep (1)
8.点击单选按钮-Radio Button
勾选一个单选按钮,就是调用元素方法click()
我们利用for语句遍历这两个单选按钮,依次点击他们。
1 from selenium import webdriver 2 import time 3 driver = webdriver.Firefox () 4 driver.maximize_window () 5 driver.get (‘http://news.baidu.com‘) 6 # driver.implicitly_wait (8) 7 print(driver.find_elements_by_xpath ("//*/input[@type=‘radio‘]")) 8 for i in driver.find_elements_by_xpath ("//*/input[@type=‘radio‘]"): 9 i.click () 10 time.sleep (5)
9.获取元素页面上的文字
1 # coding=utf-8 2 import time 3 from selenium import webdriver 4 from selenium.webdriver.common.keys import Keys 5 6 driver = webdriver.Firefox () 7 driver.maximize_window () 8 # driver.implicitly_wait (6) 9 10 driver.get ("http://www.baidu.com/") 11 time.sleep (1) 12 driver.find_element_by_xpath ("//*[@id=‘u1‘]/a[7]").click () 13 time.sleep (1) 14 15 driver.find_element_by_xpath ("//*[@id=‘TANGRAM__PSP_10__submit‘]").click () 16 17 18 # 断言方法二,本文重点介绍方法 19 error_mes = driver.find_element_by_xpath ("//*[@id=‘TANGRAM__PSP_10__error‘]").text 20 print(error_mes) 21 try: 22 assert error_mes == u‘请您填写手机/邮箱/用户名‘ 23 print (‘Test pass.‘) 24 except Exception as e: 25 print ("Test fail.", format (e))
注:不同版本的火狐浏览器,定位的元素会有所不同
10.验证控件是否选中
1 # coding=utf-8 2 import time 3 from selenium import webdriver 4 5 driver = webdriver.Firefox () 6 driver.maximize_window () 7 # driver.implicitly_wait (6) 8 driver.get ("http://news.baidu.com/") 9 time.sleep (1)aluse 10 try: 11 # driver.find_element_by_xpath ("//*[@id=‘news‘]").is_selected () 12 assert driver.find_element_by_xpath ("//*[@id=‘newstitle‘]").is_selected () == True 13 print (‘Test Pass.‘) 14 except Exception as e: 15 print (‘Test fail‘, format (e))
总结:
元素方法is_selected()返回是是布尔值(True /Faluse),用来判断单选或者多选控件是否被选中,或者下拉选择菜单是否选择一个默认的option,都可以通过这个方法去判断。
11.获取页面元素的大小
1 # coding=utf-8 2 import time 3 from selenium import webdriver 4 5 driver = webdriver.Firefox () 6 driver.maximize_window () 7 # driver.implicitly_wait (6) 8 driver.get ("http://baidu.com/") 9 time.sleep (1) 10 Size = driver.find_element_by_id("su") 11 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两个新闻标题进行对比
先看看窗口切换的脚本代码:
1 # coding=utf-8 2 import time 3 from selenium import webdriver 4 5 driver = webdriver.Firefox () 6 driver.maximize_window () 7 driver.get (‘http://news.baidu.com‘) 8 time.sleep (3) 9 10 driver.find_element_by_xpath ("//*[@id=‘pane-news‘]/div/ul/li[1]/strong/a").click () 11 time.sleep (3) 12 print (driver.current_window_handle) # 输出当前窗口句柄 13 handles = driver.window_handles # 获取当前全部窗口句柄集合 14 print (handles) # 输出句柄集合 15 16 for handle in handles: # 切换窗口 17 print(handle) 18 if handle != driver.current_window_handle: 19 print(handle) 20 print (‘switch to second window‘, handle) 21 driver.close () # 关闭第一个窗口 22 driver.switch_to.window (handle) # 切换到第二个窗口 23 print(handle)
注:通过driver.current_window_handle获取的当前窗口句柄,无论打开多少个窗口,只要不使用driver.switch_to.window()就是第一次打开的窗口。
使用driver.window_handles获取的窗口为当前打开的所有窗口。
只有driver.switch_to.window()才能切换到对应的窗口,获取窗口句柄。
1 # coding=utf-8 2 import time 3 from selenium import webdriver 4 5 # driver = webdriver.Chrome() 6 driver = webdriver.Firefox() 7 # driver.maximize_window () 8 driver.get (‘http://news.baidu.com‘) 9 time.sleep (1) 10 11 news_link = driver.find_element_by_xpath ("//*[@id=‘pane-news‘]/div/ul/li[1]/strong/a") 12 page1_title_string = news_link.text # 得到页面A新闻标题 13 print(page1_title_string) 14 news_link.click () # 点击新闻链接 15 time.sleep (1) 16 handles = driver.window_handles 17 18 for handle in handles: # 切换窗口(切换到搜狗) 19 if handle != driver.current_window_handle: 20 print( ‘switch to second window‘, handle) 21 driver.close () # 关闭第一个窗口 22 driver.switch_to.window (handle) # 切换到第二个窗口 23 page2_title_string = driver.find_element_by_xpath ("/html/body/div[2]/div[2]/div[4]/div[2]/div/div[1]").text 24 25 print(page2_title_string) 26 # 详情页有一个原标题 27 28 try: 29 assert page1_title_string in page2_title_string # 判断页面B标题是否包含页面A标题 30 print (‘Test Pass.‘) 31 except Exception as e: 32 print (‘Test Fail‘)
注:
1.一般编程的过程中,要提前设置浏览器最大化,不要使用drive.maxisize_window()去设置窗口的大小。否者当浏览器窗口最大时,使用该方法浏览器就会变成最小,导致有些元素通过xpath定位失效。
2网页的title与新闻详情页的title是不一样的概念,前者通过driver.title即可获取,后者通过元素定位获取。
13.处理iframe切换
问题:发现元素定位没问题,在测试回放的过程,发现就是找不到元素报错。
由于没有找到合适的iframe网站,这里不好用代码举例,简单文字加图片来介绍。
1 driver.switch_to.frame("iframe1") 2 # 操作目标元素,这个目标元素在 iframe1里面,这里就是百度文本输入框输入文字 3 driver.switch_to.default_content()
14.处理alert弹框
1 driver.switch_to_alert().accept() # 点击弹出里面的确定按钮 2 #driver.switch_to_alert().dismiss() # 点击弹出上面的X按钮
15.获取当前页面的全部图片信息
1 # coding=utf-8 2 import time 3 from selenium import webdriver 4 5 driver = webdriver.Chrome () 6 driver.maximize_window () 7 driver.implicitly_wait (6) 8 driver.get ("http://news.baidu.com") 9 time.sleep (1) 10 print(driver.find_elements_by_tag_name ("img")) 11 for image in driver.find_elements_by_tag_name ("img"): 12 print (image.text) 13 print (image.size) 14 print (image.tag_name)
运行结果:
运行结果,发现没有图片名称打印出来,说明百度新闻页面,所有图片都没有给出text这个属性,前端的妹子没有写图片的text属性
16.获取页面元素的herf属性
通过Selenium获取页面元素的某一个属性。一个元素可能有多个属性,例如 class, id, name, text, href, vale等等。这里我们举例一个爬虫中经常需要处理的链接问题:找出当前页面所有的超链接。
已百度首页为例,打印所有包含href的元素的链接
1 # coding=utf-8 2 import time 3 from selenium import webdriver 4 5 driver = webdriver.Firefox() 6 driver.maximize_window() 7 # driver.implicitly_wait(6) 8 driver.get("https://www.baidu.com") 9 time.sleep(1) 10 print(driver.find_elements_by_xpath("//*[@href]")) 11 for link in driver.find_elements_by_xpath("//*[@href]"): 12 print(link) 13 print(link.get_attribute(‘href‘)) 14 driver.quit()
17.如何截图并保存
1 # coding=utf-8 2 import time 3 from selenium import webdriver 4 5 driver = webdriver.Firefox() 6 driver.maximize_window() 7 8 driver.get("https://www.baidu.com") 9 time.sleep(1) 10 11 driver.get_screenshot_as_file("?C:\\Users\\fyr\\Desktop\\新建文本文档 11(2).png") 12 time.sleep(4) 13 driver.quit()