上传文件

文件上传操作也比较常见功能之一,上传功能操作 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 为内容的类型。

原文地址:https://www.cnblogs.com/zhang-jun-jie/p/9313815.html