Selenium之文件上传、下载

向量数据库大模型数据中台

文件上传

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

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

  
driver.find\_element\_by\_id('file\_upload').send\_keys('C:\\wlrun.log')

效果演示:

picture.image

方法二:使用AutoIt

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

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

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

picture.image

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

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

  • Run Script:用于执行AutoIt脚本

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

具体使用方法如下:

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

picture.image

从显示的结果得知,此控件的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

picture.image

那么,最终文件上传的脚本如下:

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

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

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

效果演示:

picture.image

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

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

picture.image

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  
#实例化一个Options  
chrome_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之操作浏览器、元素等待、窗体切换和弹窗处理

0
0
0
0
关于作者

文章

0

获赞

0

收藏

0

相关资源
云原生可观测性技术的落地实践
云原生技术和理念在近几年成为了备受关注的话题。应用通过云原生改造,变得更动态、弹性,可以更好地利用云的弹性能力。但是动态、弹性的环境也给应用以及基础设施的观测带来了更大的挑战。本次分享主要介绍了云原生社区中可观测性相关的技术和工具,以及如何使用这些工具来完成对云原生环境的观测。
相关产品
评论
未登录
看完啦,登录分享一下感受吧~
暂无评论