Selenium之文件上传、下载

文件上传

文件上传功能的话，一般都是需要点击上传按钮，然后打开本地windwos窗口，从窗口中选择文件，然后进行上传，但是windwos的弹窗不属于浏览器页面的元素，通过WebDriver无法操作windwos控件，那么遇到这种情况的话，文件上传操作该如何去模拟呢？下面让小编带你来了解一下吧。

方法一：通过元素定位方法，找到文件上传控件，然后通过send_keys()方法向其输入一个文件地址来实现文件上传。

driver.find_element_by_id('file_upload').send_keys('C:\\wlrun.log')

效果演示：

 

方法二：使用AutoIt 

AutoIt 目前最新是v3版本，这是一个使用类似BASIC脚本语言的免费软件,它设计用于Windows GUI(图形用户界面)中进行自动化操作。它利用模拟键盘按键，鼠标移动和窗口/控件的组合来实现自动化任务。

AutoIt下载地址        https://www.autoitscript.com/site/autoit/downloads/

下载安装之后，在菜单中会出现如下程序：

 

• AutoIt Windows Info:用于识别Windows控件信息

• Compile Script to.exe:用于将AutoIt生成exe执行文件

• Run Script:用于执行AutoIt脚本

• SciTE Script Editor:用于编写AutoIt脚本

 

具体使用方法如下：

1、打开AutoIt Windows Info工具，用鼠标点击Finder Tool,然后按住鼠标左键，将鼠标拖动到需要识别的控件上。

从显示的结果得知，此控件的Title=“打开”，Class为Edit，Instance=1。我们就是利用控件的这些信息，定位控件，编写脚本。

这里需要关注以下信息：

　a.操作页面的title，用于固定操作的页面（也就是文件选择框的title）。

　b.需要填入的信息，在输入框中填入“上传文件的路径及文件名”（windows操作）

　c.点击“打开”按钮，实现文件上传。

 

2、根据以上所识别的控件信息，利用编辑器SciTE Script Editor，根据AutoIT的语法编写脚本。

    实现文件上传需要的几个方法： 

　　ControlFocus ( "窗口标题", "窗口文本", 控件ID)   

　　　　---->设置输入焦点到指定窗口的某个控件上（即：控件ID“文件名”输入框的id）

　　WinWait ( "窗口标题" [, "窗口文本" [, 超时时间]] )  

　　　　---->暂停脚本的执行直至指定窗口存在（出现）为止

　　ControlSetText ( "窗口标题", "窗口文本", 控件ID, "新文本" )   

　　　　---->修改指定控件的文本（即：控件ID“文件名”输入框的id）

　　Sleep ( 延迟 )   

　　　　---->使脚本暂停指定时间段

　　ControlClick ( "窗口标题", "窗口文本", 控件ID [, 按钮] [, 点击次数]] )   

　　　　---->向指定控件发送鼠标点击命令（即：控件ID“打开”按钮的id）

　　其中，title即AutoIt Window Info识别出的Title字段，controlID即AutoIt Window Info识别出的Class和Instance的拼接，如上图拼接后的结果应为：Button1

 

 

那么，最终文件上传的脚本如下：

ControlFocus("打开","","Edit1");识别windows窗口WinWait("[CLASS:#32770]","",10);窗口等待十秒ControlSetText("打开", "", "Edit1", "c:\wlrun.log");想输入框中输入要上传的文件的路径Sleep(2000)ControlClick("打开", "","Button1");;点击打开按钮

接下来这里有个小问题要提醒你，关于OS弹框的title，不同浏览器是不一样的，一般firefox是“文件上传”、chrome叫“打开”、而IE则叫“选择要加载的文件”，对于这个问题，你可以写三个不同脚本，在处理弹框的方法中根据浏览器类型的不同而进行选择，或者每次去获取所有类型弹框，再或者通过参数传入该弹框的名称。具体怎么去应用就要看个人喜好。

 

AutoIT脚本编写完成后，保存后才可以通过菜单栏“Tools”-->"Go" 运行一下脚本。注意，运行的时候，上传窗口需处于打开状态。

效果演示：

调试了文件可以正常上传后，将脚本保存为uploadfile.au3 ,存在在桌面。

3、打开Compile Script to.exe，将脚本转换成exe程序，给python调用

 

python脚本调用：​​​​​​​

#点击文件上传按钮driver.find_element_by_name('divfile').click()#调用uploadfile.exe上传文件os.system('‪C:\\uploadfile.exe')

这里只是简单的介绍一下如何通过AutoIt工具进行文件上传操作，脚本的话还有优化的空间，感兴趣的朋友可以自行深入研究，比如：

• 脚本转换之后生成的exe如何支持动态传入文件路径

• 批量上传文件时如何处理

文件下载

文件下载方法的话，小编目前没有亲自尝试过，从网上找了段代码copy下来，有兴趣的自己试一下是否可以用。思路的话就是先更改浏览器的设置，比如设置默认下载路径，控制是否显示开始下载，以及下载下来的文件类型等。​​​​​​​

from selenium.webdriver.chrome.options import Options#实例化一个Optionschrome_options = Options()
#用于定义下载不弹窗和默认下载地址（默认下载地址还要再后面的commands里启动，默认是不开启的）prefs = {"download.default_directory": "E:\download","download.prompt_for_download": False,}chrome_options.add_experimental_option("prefs", prefs)
#无头模式（就是不打开浏览器）chrome_options.add_argument("--headless")browser = webdriver.Chrome(chrome_options=chrome_options)
#老外的样例browser.command_executor._commands["send_command"] = ("POST", '/session/$sessionId/chromium/send_command')params = {'cmd': 'Page.setDownloadBehavior', 'params': {'behavior': 'allow', 'downloadPath': "E:\download"}}command_result = browser.execute("send_command", params)

另一种方式实现下载的话，就是借助AutoIt控件，操作方式与文件上传类似，这里就不再举例子了。

Selenium系列操作回顾：

Selenium工具的各个组件以及演变历史你都了解吗

Selenium之页面元素定位

Selenium之操作浏览器、元素等待、窗体切换和弹窗处理