上一篇文章说到显示等待和隐式等待语句，我们继续学习下面的命令方法。
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()
上面这段代码会勾选 “十天内免登录” 复选框，其实如果单纯实现这个行为有更简单的方法，但是这里只是为了理解定位一组元素的方法。下面介绍 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() # 跳回最外层的页面
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()
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()
运行上面的脚本，查看运行结果就能理解
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()
- 向页面中的 textarea 文本框输入内容
1 text = "input text"js = "var sum=document.getElementById('id'); sum.value='" + text + "';"driver.execute_script(js)
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()
17. 窗口截图
driver.get_screenshot_as_file("D:\\pictures\\test.jpg")
18. 关闭窗口
- quit(): 退出相关驱动程序和关闭所有窗口
- close(): 关闭当前窗口
19. 验证码的处理
- 去掉验证码： 开发人员在测试环境注释掉验证码的相关代码即可
- 设置万能验证码： 设置一个万能验证码
- 验证码识别技术： Python-tesseract 能够读取任何常规图片
- 记录cookie： 直接添加自己的用户名密码信息到 cookie 中，再次访问网站，服务器直接调取cookie登录