selenium自动化实战-基于python语言（二： 编写脚本）

上一篇文章说到显示等待和隐式等待语句，我们继续学习下面的命令方法。

8.  定位一组元素

这里书上是自己写了一个页面代码，通过访问本地这个页面来举例。但我觉得找一个现有的页面自己琢磨更有意思，而且复杂的页面也会遇到复杂的问题。比如我根据163邮箱登录页面的 “十天内免登录” 复选框写了如下的代码：

 1 from selenium import webdriver
 2 import time
 3 
 4 driver = webdriver.Firefox()
 5 driver.get('http://mail.163.com/')
 6 driver.implicitly_wait(20) #这里加一个隐式等待命令，等待页面加载，不然可能会报错
 7 driver.switch_to_frame('x-URS-iframe') #163邮箱登录页面的复选框在一个框架内，所以需要定位到该框架内
 8 
 9 #选择页面上所有的tag name 为input的元素
10 inputs = driver.find_elements_by_tag_name('input') #一定要注意定位一组元素需要是 elements, 而不是element
11 
12 
13 #通过for循环找出type为checkbox的元素并且单击勾选
14 for i in inputs:
15     if i.get_attribute('type') == 'checkbox':
16         i.click()
17         time.sleep(1)
18         
19 print(len(inputs)) #这里结果打印出来是9，所以在这个框架内复选框的个数应该是9，但是比较直观的我们只能看到“十天内免登录”
20 
21 #driver.quit()

View Code

上面这段代码会勾选 “十天内免登录” 复选框，其实如果单纯实现这个行为有更简单的方法，但是这里只是为了理解定位一组元素的方法。下面介绍 pop() 方法：

获取一组元素中最后一个	pop() 或者 pop(-1)	例如： driver.find_element_by_id('xx').pop().click()
获取一组元素中的第一个	pop(0)
获取一组元素中的第二个	pop(1)

 

 

 

 

9. 多表单切换

• 嵌套的iframe有 id 或 name: switch_to_frame('') 里面填写 id 或者 name。 这里书里写的是： switch_to.frame()，这两个方法我都试过了，全都是正确的。但还是建议用规范的写法：switch_to.frame()
• 嵌套的iframe 没有 id 或 name:

1 .......
2 
3 xf = driver. find_element_by_xpath('//*[@class="if"]')
4 
5 driver.switch_to_frame(xf)
6 
7 driver.switch_to.parent_frame() # 默认跳出离它最近的switch_to.frame() 方法
8 
9 driver.switch_to.default_content() # 跳回最外层的页面

View Code

 

10. 多窗口切换

获得当前窗口句柄	current_window_handle	例如： search_window = driver.current_window_handle
返回所有打开的窗口的句柄到当前会话	window_handles	例如： all_handles = driver.window_handles
切换到相应的窗口	switch_to.window()	例如： driver.switch_to.window(search_window)

 

 

 

 

11. 警告框处理： switch_to_alert() 方法定位到alert/confirm/prompt，然后使用下面的方法进行操作

返回 alert/confirm/prompt 中的文字信息	text
接受现有警告框	accept()	举例：switch_to_alert().accept()
解散现有警告框	dismiss()
发送文本至警告框	send_keys(keysToSend)

 

 

 

 

 

12. 上传文件

• send_keys('')： 和模拟键盘输入操作是一样的，value替换成 文件的本地目录
• AutoIt 实现： 不过多解释，有兴趣可以自己学习一下

 

13. 下载文件: 一个脚本解释一切

1 from selenium import webdriverimport os
2 fp = webdriver.FirefoxProfile()
3 fp.set_preference("browser.download.folderList", 2) #浏览器下载路径，0代表默认路径，2代表指定目录fp.set_preference("browser.download.manager.showWhenStarting",False) #是否显示开始fp.set_preference("browser.download.dir", os.getcwd()) #指定下载目录为当前目录fp.set_preference("browser.helperApps.neverAsk.saveToDisk","application/octet-stream") #下载文件的类型
4 driver = webdriver.Firefox(firefox_profile=fp)driver.get("https://pypi.python.org/pypi/selenium")driver.implicitly_wait(30) driver.find_element_by_partial_link_text("selenium-3.4.3.tar.gz").click()

View Code

 

14. 操作cookie

1 from selenium import webdriver
2 driver = webdriver.Firefox()driver.get("http://youdao.com")
3 cookie = driver.get_cookies()
4 print(cookie) #打印当前页面cookie信息
5 driver.add_cookie({'name': 'key-aaa', 'value': 'value-bbb'})#添加cookie信息
6 for cookie in driver.get_cookies():    print("%s -> %s" %(cookie['name'], cookie['value']))#再次打印cookie的name和value信息
7 driver.quit()

View Code

运行上面的脚本，查看运行结果就能理解

 

15. 调用JavaScript

• 通过 JavaScript 设置浏览器滚动条位置

1 from selenium import webdriverfrom time import sleep
2 driver = webdriver.Firefox()driver.get("http://www.baidu.com")driver.implicitly_wait(20)
3 driver.set_window_size(500,500)
4 driver.find_element_by_id("kw").send_keys("selenium")driver.find_element_by_id("su").click()sleep(2)
5 js="window.scrollTo(100,450);" #通过JavaScript设置浏览器窗口的滚动条位置driver.execute_script(js) #执行js脚本sleep(3)
6 driver.quit()

View Code

• 向页面中的 textarea 文本框输入内容

1 text = "input text"js = "var sum=document.getElementById('id'); sum.value='" + text + "';"driver.execute_script(js)

View Code

 

16. 处理 HTML5 的视频播放

1 from selenium import webdriverfrom time import sleep
2 driver = webdriver.Firefox()driver.get("http://www.miaopai.com/show/NvFBW3o2ZprzmFd8Xsq96RnB5Ax6nFSi.html") #秒拍上的一个视频driver.implicitly_wait(20)
3 video = driver.find_element_by_xpath("/html/body/div[7]/div[1]/div[1]/div[2]/div/div/video") #定位视频位置
4 url = driver.execute_script("return arguments[0].currentSrc;", video) #返回视频的urlprint(url) #打印视频url
5 print("start")driver.execute_script("return arguments[0].play()", video) #开始播放视频
6 sleep(15) #播放15秒
7 print("stop")driver.execute_script("arguments[0].pause()", video) #停止播放视频
8 driver.quit()

View Code

 

17. 窗口截图

driver.get_screenshot_as_file("D:\\pictures\\test.jpg")

 

18. 关闭窗口

• quit(): 退出相关驱动程序和关闭所有窗口
• close(): 关闭当前窗口

 

19.  验证码的处理

• 去掉验证码： 开发人员在测试环境注释掉验证码的相关代码即可
• 设置万能验证码： 设置一个万能验证码
• 验证码识别技术： Python-tesseract 能够读取任何常规图片
• 记录cookie： 直接添加自己的用户名密码信息到 cookie 中，再次访问网站，服务器直接调取cookie登录

转载于:https://www.cnblogs.com/nothingc/p/7111917.html