Python + Selenium（五）- web文件上传及下载

短命女 2023-07-03 04:59 22阅读 0赞

### 1. 图片及文件的上传 ###

#### 1.1 input 标签的上传 ####

以慕课网的个人设置上传头像为例：  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM0NjU5Nzc3_size_16_color_FFFFFF_t_70]  
该页面的上传头像其实就是一个 **type=‘file’ 的 input 标签卡**，这种情况，上传头像只需要调用 webdriver 的 sendkey 方法即可：

driver.find_element_by_id('upload').send_keys('D:\\photo.png')

由于该元素 .hide 的 css 使其隐藏了，所以首先应该将其变为可见的标签卡，再传入文件信息，就成功上传了头像：

js = "document.getElementById('upload').removeAttribute('class');"
    driver.execute_script(js)
    # 再传入文件位置及属性
    driver.find_element_by_id('upload').send_keys('D:\\photo.png')

#### 1.2 使用 pywinauto 对非 input 标签进行上传 ####

pywinauto的下载使用pip 进行：`pip install pywinauto`

from pywinauto.keyboard import send_keys
    # 点击上传图片按钮，会弹出选择文件框
    # driver.find_element_by_xpath("//button[@id='test1']").click()
    # 点击多图片上传按钮
    driver.find_element_by_xpath("//button[@id='test2']").click()
    time.sleep(2)
    # 注意：多图片上传需要再额外输入双引号将各张图片隔开
    send_keys(r'"D:\22.png"')
    send_keys(r'"D:\11.png"')
    send_keys(r'"D:\33.png"')
    # 输入回车，相当于点击确定按钮
    send_keys("{VK_RETURN}")

注意：该模块只能够在window系统下进行使用，如果你是linux或者mac系统，就需要使用到下面的这个跨平台模块pyautogui。

#### 1.3 使用pyautogui对非 input 标签的上传 ####

优点：能够跨平台使用  
缺点：只能够选择一个文件，且路径中不能够使用中文  
使用方法：通过模拟真人敲击键盘，来输入对应的路径信息。因为电脑上可能会安装输入法软件，所以输入完成后需要敲击两次回车键。

import pyautogui
    pyautogui.write(r"D:\22.png")
    pyautogui.press('enter',2)

#### 1.4 使用pykeyboard对非 input 标签的上传 ####

在该种情况下，只能够使用 PyKeyBoard 模拟键盘事件来上传文件。  
以谷歌浏览器的导入本地书签为例：  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM0NjU5Nzc3_size_16_color_FFFFFF_t_70 1]  
点击“选择文件”之后就会弹出我们熟悉的上传文件窗口：

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM0NjU5Nzc3_size_16_color_FFFFFF_t_70 2]  
而此时我们不能够通过获得网页元素对其进行相关操作。此时就只能够借助 PyKeyBoard 模拟键盘事件来传入文件信息并上传。  
PyKeyBoard 使用需要先安装其插件，在我的[这篇博客][Link 1]中有详细说明。  
安装完成 PyKeyBoard 后就可以开始使用了，我们从点击“选择文件”按钮弹出文件框开始，写出其简单的操作代码：

# 导包
    from pykeyboard import PyKeyboard
    
    # 实例化
    pykey = PyKeyboard()
    # 点击 ‘shift’ 按钮， 使输入法转化为英文输入法
    pykey.tap_key(pykey.shift_key)
    # 传入文件路径及名称
    pykey.type_string(filename)
    # 点击 enter 按钮，确定上传
    pykey.tap_key(pykey.enter_key)

### 2. 文件的下载 ###

通常网页的文件下载步骤是：

1.  点击下载按钮跳转到下载页面
2.  浏览器弹窗 -> 选择文件存放的路径
3.  点击确定后浏览器开始下载
4.  下载完成后，在指定的路径下能够查看到刚刚下载的这个文件

那么我们要解决的就是第2-3步的自动执行。  
在 selenium 中，每个浏览器都有一些默认设置，例如：默认下载路径、是否弹窗等。这些默认设置就能够有效的解决我们的上个问题。

Chrome 的相关设置：

options = webdriver.ChromeOptions()
    # 设置文件默认下载存放路径 和 禁止弹出下载窗口
    prefs = {'download.default_directory' : 'F:\\Download\\','profile.default_content_setting.popups':0}
    options.add_experimental_option('prefs',prefs)
    driver = webdriver.Chrome(chrome_options=options)
    
    driver.get('https://www.imooc.com/mobile/app')
    driver.find_element_by_link_text('Android 下载').click()

注：这行代码 `driver = webdriver.Chrome(chrome_options=options)` 运行时如果报错“**DeprecationWarning: use options instead of chrome\_options**”，请将 **代码中的 chrome\_options 换为 options**。

Firefox 的相关设置：

profile = webdriver.FirefoxProfile()
    profile.set_preference('browser.download.dir','F:\\Download\\')
    profile.set_preference('browser.download.folderList',2)
    profile.set_preference('browser.helperApps.neverAsk.saveToDisk','application/zip')
    driver = webdriver.Firefox(firefox_profile=profile)
    
    driver.get('https://www.imooc.com/mobile/app')
    driver.find_element_by_link_text('Android 下载').click()

其中：

browser.download.folderList

设置成 0 代表下载到浏览器默认下载路径， 设置成 2 则可以保存到指定目录。

browser.download.dir

用于指定所下载文件的目录。 os.getcwd() 函数不需要传递参数， 用于返回当前的目录。

browser.helperApps.neverAsk.saveToDisk

指定要下载页面的 **Content-type** 值， **“application/zip”** 为文件的类型。下载的文件不同，这里的类型也会有所不一样。如果不清楚你下载的文件什么类型，请用Fiddler抓包查看。

最后总结一下：  
如果你对此文有任何疑问，如果你也需要接口项目实战，如果你对软件测试、接口测试、自动化测试、面试经验交流感兴趣欢迎加入：软件测试技术群：**593462778**，群里的免费资料都是笔者十多年测试生涯的精华。还有同行大神一起交流技术哦。

作者：暗潮汹涌  
原创不易，欢迎转载，但未经作者同意请保留此段声明，并在文章页面明显位置给出原文链接。

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM0NjU5Nzc3_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20200202194024230.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM0NjU5Nzc3,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM0NjU5Nzc3_size_16_color_FFFFFF_t_70 1]: https://img-blog.csdnimg.cn/20200202195415690.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM0NjU5Nzc3,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM0NjU5Nzc3_size_16_color_FFFFFF_t_70 2]: https://img-blog.csdnimg.cn/20200202195507771.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM0NjU5Nzc3,size_16,color_FFFFFF,t_70
[Link 1]: https://blog.csdn.net/qq_34659777/article/details/103807535