Python中selenium实现文件上传所有方法整理总结
文件上传是所有UI自动化测试都要面对的一个头疼问题,今天博主在这里给大家分享下自己处理文件上传的经验,希望能够帮助到广大被文件上传坑住的seleniumer。 首先,我们要区分出上传按钮的种类,大体上可以分为两种,一种是input框,另外一种就比较复杂,通过js、flash等实现,标签非input 我们分别对这两种进行分析: 1.input标签 众所周知,input标签是可以直接send_keys的,这里也不例外,来看代码示例: 代码: # -*- 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层面的操作去处理呗,到这里我们基本找到了问题的处理方法。 大体上有以下几种解决方案:
目前我只知道以上四种办法,有其他方法的请留言告诉我,让我学习一下。 我们依次看一下: 1. autoIT 关于autoIT上传以及参数化的方法我已经在另一篇博文中讲过了,请参见selenium之 autoit命令行参数 。这里不再赘述。 2.win32gui 废话不多说,上代码先: 代码: # -*- coding: utf-8 -*- from selenium import webdriver import win32gui import win32con import time dr = webdriver.Firefox() dr.get('http://sahitest.com/demo/php/fileUpload.htm') upload = dr.find_element_by_id('file') upload.click() time.sleep(1) # win32gui 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',None) # 确定按钮Button win32gui.SendMessage(Edit,win32con.WM_SETTEXT,None,'d:baidu.py') # 往输入框输入绝对地址 win32gui.SendMessage(dialog,win32con.WM_COMMAND,1,button) # 按button print upload.get_attribute('value') dr.quit() 结果: baidu.py 在这里你需要一个非常重要的小工具:Spy++,百度一下有很多,当然你也可以用autoIT自带的工具,不过没有这个好用,建议去下一个吧。 而且,你得安装pywin32的库,你可以到这里找到对应你Python版本的库,注意32位还是64位一定要和你安装的Python版本对应。 安装完成之后在【开始菜单Python的文件夹】里看到PyWin32的文档【Python for Windows Documentation】,你能从中找到对应的方法API。 简单介绍几个用到的: win32gui.FindWindow(lpClassName=None,lpWindowName=None):
win32gui.FindWindowEx(hwndParent=0,hwndChildAfter=0,lpszClass=None,lpszWindow=None)
win32gui.SendMessage(hWnd,Msg,wParam,lParam)
至于win32api模块以及其他的方法,这里不进行更多描述,想要了解的自行百度或看pywin32文档。 3.SendKeys 首先要安装SendKeys库,可以用pip安装 pip install SendKeys 代码示例: # -*- coding: utf-8 -*- from selenium import webdriver import win32gui import win32con import time dr = webdriver.Firefox() dr.get('http://sahitest.com/demo/php/fileUpload.htm') upload = dr.find_element_by_id('file') upload.click() time.sleep(1) # SendKeys SendKeys.SendKeys('D:baidu.py') # 发送文件地址 SendKeys.SendKeys("{ENTER}") # 发送回车键 print upload.get_attribute('value') dr.quit() 结果: baidu.py 通过SendKeys库可以直接向焦点里输入信息,不过要注意在打开窗口是略微加一点等待时间,否则容易第一个字母send不进去(或者你可以在地址之前加一个无用字符),不过我觉得这种方法很不稳定,不推荐。 4.keybd_event win32api提供了一个keybd_event()方法模拟按键,不过此方法比较麻烦,也不稳定,所以很不推荐,下面给出部分代码示例,如果想要研究,自己百度去学习吧。 ... # 先找一个input框,输入想要上传的文件的地址,剪切到剪贴板 video.send_keys('C:UsersAdministratorPictures 4b20919fc78baf41fc993fd8ee2c5c9.jpg') video.send_keys(Keys.CONTROL,'a') # selenium的send_keys(ctrl+a) video.send_keys(Keys.CONTROL,'x') # (ctrl+x) driver.find_element_by_id('uploadImage').click() # 点击上传按钮,打开上传框 # 粘贴(ctrl + v) win32api.keybd_event(17,0) # 按下按键 ctrl win32api.keybd_event(86,0) # 按下按键 v win32api.keybd_event(86,win32con.KEYEVENTF_KEYUP,0) # 升起按键 v win32api.keybd_event(17,0) # 升起按键 ctrl time.sleep(1) # 回车(enter) win32api.keybd_event(13,0) # 按下按键 enter win32api.keybd_event(13,0) # 升起按键 enter ... 是不是很麻烦,当然,你甚至可以用按键把整个路径输入进去,不过,我想没人愿意这么做的。而且在此过程中你不能随意移动鼠标,不能使用剪贴板,太不稳定了,所以非常不建议你用这种办法。。 3.多文件上传 接下来还有一种情况值得我们考虑,那就是多文件上传。如何上传多个文件,当然我们还是往输入框里输入文件路径,所以唯一要搞清楚的就是多文件上传时,文件路径是怎么写的。 我来告诉你吧,多文件上传就是在文件路径框里用引号括起单个路径,然后用逗号隔开多个路径,就是这么简单,例如: 但需要注意的是:只有多个文件在同一路径下,才能这样用,否则是会失败的(下面的写法是不可以的): 接下里找一个例子试试: 代码: # -*- coding: utf-8 -*- from selenium import webdriver import win32gui import win32con import time dr = webdriver.Firefox() dr.get('http://www.sucaijiayuan.com/api/demo.php?url=/demo/20150128-1') dr.switch_to.frame('iframe') # 一定要注意frame dr.find_element_by_class_name('filePicker').click() time.sleep(1) dialog = win32gui.FindWindow('#32770',None) ComboBoxEx32 = win32gui.FindWindowEx(dialog,None) ComboBox = win32gui.FindWindowEx(ComboBoxEx32,None) button = win32gui.FindWindowEx(dialog,None) # 跟上面示例的代码是一样的,只是这里传入的参数不同,如果愿意可以写一个上传函数把上传功能封装起来 win32gui.SendMessage(Edit,'"d:baidu.py" "d:upload.py" "d:1.html"') win32gui.SendMessage(dialog,button) print dr.find_element_by_id('status_info').text dr.quit() 结果: 选中3张文件,共1.17KB。 可见,多文件上传并没有那么复杂,也很简单,唯一的区别就是输入的参数不同而已。autoIT也可以实现,有兴趣可以自己试试。 而且我们可以发现一点,就是上面的这个窗口的代码跟之前示例中的基本是一样,说明我们可以把上传的部分抽出来,写一个函数,这样每次要上传,直接去调用函数,传入参数即可。 以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持编程小技巧。 (编辑:李大同) 【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容! |