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

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

8. 定位一组元素

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

1 fromselenium import webdriver2 import time3

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, 而不是element11

12

13 #通过for循环找出type为checkbox的元素并且单击勾选14 for i ininputs: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 importwebdriverimport os2 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 importwebdriver2 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 importsleep2 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 importsleep2 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登录