selenium之文件上传

文件上传是所有UI自动化测试都要面对的一个头疼问题，今天博主在这里给大家分享下自己处理文件上传的经验，希望能够帮助到广大被文件上传坑住的seleniumer。

首先，我们要区分出上传按钮的种类，大体上可以分为两种，一种是input框，另外一种就比较复杂，通过js、flash等实现，标签非input

我们分别对这两种进行分析：

1.input标签

众所周知，input标签是可以直接send_keys的，这里也不例外，来看代码示例：

示例网址:http://www.sahitest.com/demo/php/fileUpload.htm

代码：

# -*- coding: utf-8 -*-
from selenium import webdriver

driver = webdriver.Firefox()
driver.get('http://sahitest.com/demo/php/fileUpload.htm')
upload = driver.find_element_by_id('file')
upload.send_keys('d:\\baidu.py')  # send_keys
print upload.get_attribute('value')  # check value

driver.quit()

结果：

baidu.py

很明显，对于input上传，直接send_keys是最简单的解决方案。

2.非input型上传

接下来难度要升级了，对于那些不是input框实现的上传怎么办，这种上传千奇百怪，有用a标签的，有用div的，有用button的，有用object的，我们没有办法通过直接在网页上处理掉这些上传，唯一的办法就是打开OS弹框，去处理弹框。

问题又来了，OS弹框涉及的层面已经不是selenium能解决的了，怎么办？很简单，用OS层面的操作去处理呗，到这里我们基本找到了问题的处理方法。

大体上有以下几种解决方案：

autoIT，借助外力，我们去调用其生成的au3或exe文件。
Python pywin32库，识别对话框句柄，进而操作（推荐！）
SendKeys库
keybd_event，跟3类似，不过是模拟按键，ctrl+a，ctrl+c， ctrl+v…

目前我只知道以上四种办法，有其他方法的请留言告诉我，让我学习一下。

我们依次看一下：

1. autoIT

关于autoIT上传以及参数化的方法我已经在另一篇博文中讲过了，请参见selenium之 autoit命令行参数。这里不再赘述。

2.win32gui

你得先安装pywin32的库，你可以到这里找到对应你Python版本的库，注意32位还是64位一定要和你安装的Python版本对应。或者使用以下方式安装：pip install pypiwin32

示例网址:http://www.sahitest.com/demo/php/fileUpload.htm

代码：

 # -*- coding: utf- -*-
 from selenium import webdriver
 import win32gui
 import win32con
 import time

 dr = webdriver.Firefox()
 dr.get('http://www.sahitest.com/demo/php/fileUpload.htm')
 upload = dr.find_element_by_id('file')
 upload.click()
 time.sleep()

 # win32gui
 # 一级顶层窗口 - 第二个参数Title会因为浏览器不同，而会不同。只需要修改第二个值就好。
 dialog = win32gui.FindWindow('#32770', u'打开')
 # 二级窗口，固定写法
 ComboBoxEx32 = win32gui.FindWindowEx(dialog, , 'ComboBoxEx32', None)
 # 三级窗口，固定写法
 ComboBox = win32gui.FindWindowEx(ComboBoxEx32, , 'ComboBox', None)
 # 四级窗口 - 文件路径输入区域，固定写法
 Edit = win32gui.FindWindowEx(ComboBox, , 'Edit', None)  # 找到输入框Edit对象的句柄
 button = win32gui.FindWindowEx(dialog, , 'Button', "打开(&O)")  # 找到打开按钮
 win32gui.SendMessage(Edit, win32con.WM_SETTEXT, None, 'd:\\test.html')  # 输入文件路径
 win32gui.SendMessage(dialog, win32con.WM_COMMAND, , button)  # 点击打开按钮

结果：

baidu.py

在这里你需要一个非常重要的小工具：Spy++，百度一下有很多，当然你也可以用autoIT自带的工具，不过没有这个好用，建议去下一个吧。

使用方法：

拖动风扇图标到上传窗口，title是打开，class是#32770，对应代码第15行

拖动风扇图标到文件输入框，text是空，class是Edit，对应代码第21行

拖动风扇图标到打开按钮，text是打开(&O)，class是Button，对应代码第22行

安装完成之后在【开始菜单Python的文件夹】里看到PyWin32的文档【Python for Windows Documentation】，你能从中找到对应的方法API。

简单介绍几个用到的：

win32gui.FindWindow(lpClassName=None, lpWindowName=None):

自顶层窗口开始寻找匹配条件的窗口，并返回这个窗口的句柄。
lpClassName：类名，在Spy++里能够看到
lpWindowName：窗口名，标题栏上能看到的名字
代码示例里我们用来寻找上传窗口，你可以只用其中的一个，用classname定位容易被其他东西干扰，用windowname定位不稳定，不同的上传对话框可能window_name不同，怎么定位取决于你的情况。

win32gui.FindWindowEx(hwndParent=0, hwndChildAfter=0, lpszClass=None, lpszWindow=None)

搜索类名和窗体名匹配的窗体，并返回这个窗体的句柄。找不到就返回0。
hwndParent：若不为0，则搜索句柄为hwndParent窗体的子窗体。
hwndChildAfter：若不为0，则按照z-index的顺序从hwndChildAfter向后开始搜索子窗体，否则从第一个子窗体开始搜索。
lpClassName：字符型，是窗体的类名，这个可以在Spy++里找到。
lpWindowName：字符型，是窗口名，也就是标题栏上你能看见的那个标题。
代码示例里我们用来层层寻找输入框和寻找确定按钮

win32gui.SendMessage(hWnd, Msg, wParam, lParam)

hWnd：整型，接收消息的窗体句柄
Msg：整型，要发送的消息，这些消息都是windows预先定义好的，可以参见系统定义消息（System-Defined Messages）
wParam：整型，消息的wParam参数
lParam：整型，消息的lParam参数
代码示例里我们用来向输入框输入文件地址以及点击确定按钮

至于win32api模块以及其他的方法，这里不进行更多描述，想要了解的自行百度或看pywin32文档。