现场打脸：如何使用Selenium批量上传文件？

发表于 更新于 分类于 阅读次数： 本文字数： 663 阅读时长 ≈ 2 分钟

我们知道，Selenium里面，当我们获得一个 element 对象的时候，如果它是一个输入框，那么我们可以使用.send_keys()方法，模拟键盘按键，发送特定的字符串到输入框中，例如：

1
2
input_box = driver.find_element_by_xpath('//input[@class="xxx"]')
input_box.send_keys('账号 xxx')

但如果要说.send_keys()可以上传文件，你可能会非常吃惊吧。今天有个读者在问我的时候，我也非常吃惊，觉得这怎么可能：

结果我到 Selenium 的文档里面一看，发现send_keys()竟然真的可以上传文件: 8.5. How to upload files into file inputs ?

为了验证这个说法，我们使用 Flask 手写一个支持上传功能的简陋网站。网站代码如下：

网站运行效果如下图所示：

点击“选择文件”按钮，在弹出的对话框里面选中一个文件，然后点击“Upload”按钮，就会把文件上传到代码里面的uploads文件夹中，如下图所示：

现在我们在 Selenium 里面进行测试：

1
2
3
4
5
6
7
8
from selenium.webdriver import Chrome

driver = Chrome('./chromedriver')
driver.get('http://127.0.0.1:5000')
file_input = driver.find_element_by_xpath('//input[@type="file"]')
file_input.send_keys('/Users/kingname/test_send_keys/target/x.txt')
submit = driver.find_element_by_xpath('//input[@type="submit"]')
submit.click()

经过测试，发现确实可以正常上传文件。如下图所示：

这样一来，既然 .send_keys()能够正常工作，那么就可以反向推测出，浏览器上传文件的原理，选择文件的对话框实际上提供给浏览器的仅仅是一个文件路径。当我们点击了上传按钮以后，浏览器会根据这个路径去读硬盘，找到这个文件然后上传。由于文件路径本质上就是一个字符串，所以用.send_keys()本质上就是直接替代了选择文件对话框生成的文件路径，直接把这个路径上传给了文件输入表单。

那么如何一次性上传多个文件呢？

只要网站支持同时上传多个文件，那么我们可以把多个文件的路径拼接到一个长字符串中，路径与路径之间使用换行符\n来进行分割。

假设在文件夹/Users/kingname/test_send_keys/target里面有多个文件，如下图所示：

我们需要一次性全部上传。那么，可以使用换行符把每一个文件的路径拼接起来：

代码可以写为：

1
2
3
4
5
6
7
8
9
10
11
12
13
import os
from selenium.webdriver import Chrome
folder = '/Users/kingname/test_send_keys/target'
file_name_list = os.listdir(folder)
path_list = [os.path.join(folder, x) for x in file_name_list]
path_split_by_newline = '\n'.join(path_list)

driver = Chrome('./chromedriver')
driver.get('http://127.0.0.1:5000')
file_input = driver.find_element_by_xpath('//input[@type="file"]')
file_input.send_keys(path_split_by_newline)
submit = driver.find_element_by_xpath('//input[@type="submit"]')
submit.click()

运行效果如下图所示：

成功上传多个文件。