autoIT 自动化上传/下载文件图文详解【python selenium】
情景:
在用selenium进行web页面自动化时,时不时会遇到上传附件的情况,常见的情况就是一个上传按钮,点击后弹出windows窗口,选择文件后上传,如下图1所示
图1
这种情况超出了selenium的能力范围,需要借助其他工具来实现,这里介绍AutoIt这个工具:
1.下载并安装
进入AutoIt官网,选择AutoIt -> Downloads,如图2:
图2
然后选择Full Installation版本,里面包含了所需的所有工具,如图3:
图3
下载后直接安装即可。
2.探测控件
(1).首先从“开始”菜单里找到"Au3Info_x64.exe" 打开,如图4:
图4
(2).打开需要上传的页面,点击上传,弹出上传窗口,如上面图1所示
(3).在AutoIt上面,用鼠标按住Finder Tool不放(鼠标图标变成一个轮子的样子),移动到上传窗口的文本输入框中,然后松开,如图5
图5
松开后查看探测结果,如图6:
图6
主要关注图6中红色标注的信息:
窗口标题,即Title,图中是“文件上传”
窗口的class,图中是32770
控件的类型以及编号,图中类型是Edit,编号(Instance)是1,表明是窗口中第一个文本输入框
类似地,重复上面的一步,获取“打开”按钮的信息,图如7:
图7
由于是同一个窗口,因此标题和class都和上面的一样,不同的是,这里的"打开"按钮的类型是Button,编号为1(如果是“取消”按钮,编号就是2)。获取到这些必须信息后,就可以开始编写脚本了。
3.编写脚本
从开始菜单中打开SciTE Script Editor,编写如图8的内容:
图8
脚本内容如下,分号开头的是注释:
; 等待5秒钟,让上传窗口出现
WinWait("[CLASS:#32770]","",5)
;把输入焦点定位到上传输入文本框中,类型为Edit,编号为1,也就是上面获取到内容
ControlFocus("文件上传", "","Edit1")
;在文件名那里,输入需要上传的文件绝对路径
ControlSetText("文件上传", "", "Edit1", 'D:\files\file_name')
;等待上传时间,单位是毫秒 1秒 = 1000 毫秒,文件大的话需要设置长点
Sleep(5000)
;点击"打开"按钮,也就是上传,完成整个上传过程
ControlClick("文件上传", "","Button1");
保存为.au3脚本,然后在浏览器中点击上传,使上传窗口出现,如图1所示,然后在SciTe Scripts Editor里面选Tools -> Go (或直接按F5) 即可看到自动上传效果。
4.转换为exe文件
在开始菜单中打开"Compile sctrip to .exe x64"(32位的机器选x86),然后选择要装换的.au3文件和输出的exe文件路径和名称,最后点击下面的"Convert"即可,如图9:
图9
5.使用python调用
使用配Python调用是,首先用python控制selenium,打开上传窗口(图1),然后用
import os
os.system(r'C:\Users\dell\Desktop\no_argument.exe') #这里是保存的exe文件,根据自己的实际情况修改
然后跑python脚本,即可完成文件上传。
6.参数化
上面的过程虽然可以上传文件,但是路径、文件名都是写死在脚本里面的,如果要上传其它路径、其它文件,就要要重新修改脚本,编译成exe文件;如果上传多个文件,每个文件都要写个脚本然后转成exe
那岂不是很蛋疼? 参数化可以解决这个问题。
控件探测过程还和原来一样,脚本参数化只需要修改脚本,如图10:
图10
脚本内容如下,注意黄色背景的内容:
; 等待5秒钟,让上传窗口出现
WinWait("[CLASS:#32770]","",5)
;把输入焦点定位到上传输入文本框中
ControlFocus("文件上传", "","Edit1")
;判断是否有参数
IF $CmdLine[0] > 0 Then ;有参数
$file = $CmdLine[1]
;Else ;无参数,传递默认的文件
;$file = 'D:\work\1.png'
EndIf
;在文件名那里,使用参数
ControlSetText("文件上传", "", "Edit1", $file)
;等待上传时间,单位是毫秒 1秒 = 1000 毫秒
Sleep(5000)
;点击"打开"按钮,也就是上传
ControlClick("文件上传", "","Button1");
同样保存为.au3文件,然后打开上传窗口,按F5运行,成功后转换为exe文件
注意:
参数化之后,调用时需要传递参数,再用os.system(r'C:\Users\dell\Desktop\upload.exe 参数') 这种绝对路径调用,并不能生效,解决办法是把生成的exe文件,放到PATH里面,然后不带路径,直接调用
os.system('upload.exe upload_filename') 即可(过程中会有命令提示符窗口一闪而过,请无视)。
举一反三:如果连窗口id,文本输入框都变化,该怎么办?如果上传的文件比较到,需要等待较长时间怎么办?
提示:控件标识,比如类型、编号,等待时间,都可以参数化。下提供一个大体上通用(不敢说所有情况都适用)的脚本以及调用函数:
7.进一步参数化
把上面参数化的脚本修改为如图11的内容:
图11
其中第3~8行,分别对应的是:
windowTitle是上传窗口的标题,对应图6上面的Title,“文件上传” (这个跟浏览器有关,firefox可能叫做“上传文件”,chrome则可能是“打开”)
windowID是上传窗口的标识ID,对应图6上面的class,"#32770" (这个ID好像基本都是一样的,不过不是很肯定)
windowText是上传窗口的文本输入框,对应图6上面的class和instance,这里把他们合并到一起了
windowButton是上传窗口的“打开”按钮,这里把class和instance也合并到一起
uploadTime就是上传时等待的时间,如果上传的文件比较大或者网速比较慢,这个时间需要设置长一点,单位是秒
把这个脚本转换为exe文件,比如upload_file.exe,然后放入PATH中,
使用下面代码调用:
os.system('%s "%s" "%s" "%s" "%s" %d %s' % ('upload_file.exe','文件上传','32770','Edit1','Button1',2,r'D:\testfile\thefile'))
#注意上面这个写法中,除了uploadTime需要传递为整数之外,其他参数都要传递为字符串
下面提供一个现成的函数,有兴趣的话可以试试:
(别问我为什么只贴图不贴内容。 纸上得来终觉浅,绝知此事要躬行。手动敲一遍比看十遍学得都快!怎么样 !
说明:
第一个参数dirver是浏览器对象,第二个就是我们刚编译成的exe文件,第三个是要上传的文件所在的目录,第四个是要上传的文件,如果有多个,需要放在列表中,比如['file1','file2']
考虑到windows上大部分上传窗口都是一样的,后面几个采用了默认参数
需要注意的是,第17、18行获取上传按钮并点击,需要修改成具体的情况,而且上传一个文件后,之前获取到的upload_button这个元素句柄失效,必须重新用find_element获取一次
下面是调用的示例(ff是生成的浏览器对象,我用是firefox,叫做ff):
upload_files(ff,'upload_file.exe',r'D:\Work\测试图片','php.jpg') #上传单个文件
upload_files(ff,'upload_file.exe',r'D:\Work\测试图片','redhat.jpg')
upload_files(ff,'upload_file.exe',r'D:\Work\测试图片',['redhat.jpg','ubuntu.jpg','php.jpg','linux.png']) #上传多个文件
upload_files(ff,'upload_file.exe',r'D:\tmp','data.zip',uploadTime=30) #zip文件比较大,设置上传时间为30秒
upload_files(ff,'upload_file.exe',r'D:\tmp',['Git使用.docx','debian.jpg','vim.jpg'],uploadTime=1) #jpg文件比较小,设置上传时间1秒
当然,如果你的上传窗口跟这个完全不同,最完整的情况就是:
upload_files(ff,'upload_file.exe',r'D:\tmp',['Git使用.docx','debian.jpg','vim.jpg'],windowTitle="打开",windowID="12345",windowText="Edit100",windowButton="Btn100",uploadTime=1) #jpg文件比较小,设置上传时间1秒
不过你要是提供了所有的参数,大可以这样:
upload_files(ff,'upload_file.exe',r'D:\tmp',['Git使用.docx','debian.jpg','vim.jpg'],"打开","12345","Edit100","Btn100",1)
如果不知道为什么的话,补补Python去吧。
至于下载,过程和上传相反,稍后补充。