文件上传操作也比较常见功能之一,上传功能操作 webdriver 并没有提供对应的方法,关键上传文件的思路。
上传过程一般要打开一个系统的 window 窗口,从窗口选择本地文件添加。所以,一般会卡在如何操作本地 window 窗口。其实,上传本地文件没我们想的那么复杂;只要定位上传按钮,通 send_keys 添加本地文件路径就可以了。绝对路径和相对路径都可以,关键是上传的文件存在。下面通地例子演示操作过程
upload_file.html
<html> <head> <meta http-equiv="content-type" content="text/html;charset=utf-8" /> <title>upload_file</title> <script type="text/javascript" async="" src="http://ajax.googleapis.com/ajax/libs/jquery/1.9.1/jquery.min.js"></script> <link href="http://netdna.bootstrapcdn.com/twitter-bootstrap/2.3.2/css/bootstrap-combined.min.css" rel="stylesheet" /> <script type="text/javascript"> </script> </head> <body> <div class="row-fluid"> <div class="span6 well"> <h3>upload_file</h3> <input type="file" name="file" /> </div> </div> </body>
<script src="http://netdna.bootstrapcdn.com/twitter-bootstrap/2.3.2/js/bootstrap.min.js"></script> </html>
通过浏览器打开
使用脚本实现上传文件
#coding=utf-8 from selenium import webdriver import os,time driver = webdriver.Firefox() #打开上传文件页面 file_path = 'file:///' + os.path.abspath('upload_file.html') driver.get(file_path) #定位上传按钮,添加本地文件 driver.find_element_by_name("file").send_keys('D:\selenium_use_caseupload _file.txt') time.sleep(2) driver.quit()
从上面例子可以看到,send_keys()方法除可以输入内容外,也可以跟一个本地的文件路径。从而达到上传文件的目的
文件上传成功的效果如图
下载文件
webdriver 允许我们设置默认的文件下载路径。也就是说文件会自动下载并且存在设置的那个目录中。要想下载文件,首选要先确定你所要下载的文件的类型。要识别自动文件的下载类型可以使用 curl
curl 是利用 URL 语法在命令行方式下工作的开源文件传输工具
Content-Type ,内容类型,一般是指网页中存在的 Content-Type ,用于定义网络文件的类型和网页的编码,决定浏览器将以什么形式、什么编码读取这个文件。
另一种方法是使用 requests 模块来查找内容类型。Requests 是一个 Python 的 HTTP 客户端库,默认下载的 python 环境包不包含这个类库,需要另外安装。使用方法如下
import requests print requests.head(’http://www.python.org’).headers[’content-type’]
一旦确定了内容的类型,就可以用它来设置 Firefox 的默认配置文件,具体实例如下:
#coding=utf-8 import os from selenium import webdriver fp = webdriver.FirefoxProfile() fp.set_preference("browser.download.folderList",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") browser = webdriver.Firefox(firefox_profile=fp) browser.get("http://pypi.python.org/pypi/selenium") browser.find_element_by_partial_link_text("selenium-2").click()
browser.download.dir 用于指定你所下载文件的目录。
os.getcwd() 该函数不需要传递参数,用于返回当前的目录。
application/octet-stream 为内容的类型。