Python requests上传文件详解

一、介绍

Python requests库提供了一个简单而优雅的方式来使用HTTP。其中一个使用场景就是上传文件。在本文中，我们将介绍如何使用requests上传文件，以及如何处理上传的文件。

二、基本用法

使用requests上传文件非常简单。最基本的方法是使用POST请求，并将文件作为请求的一部分提交到服务器。

import requests

url = 'http://httpbin.org/post' # 测试网站，返回POST请求数据
files = {'file': open('test.txt', 'rb')} # 文件名为test.txt

r = requests.post(url, files=files)

print(r.text) # 查看返回的POST请求数据

在上述代码中，我们使用‘open’函数读取文件，并将其作为文件。’files’参数是一个字典，键为提交时的参数名，值为文件对象。

三、上传多个文件

上传多个文件也很简单，只需要在‘files’参数中添加多个参数即可。

import requests

url = 'http://httpbin.org/post'
files = {'file1': open('test1.txt', 'rb'),
         'file2': open('test2.txt', 'rb')} # 多个文件

r = requests.post(url, files=files)

print(r.text)

四、使用其他表单数据

除了上传文件，我们还可以将其他参数作为表单数据提交到服务器。可以通过在‘data’参数中传递参数来实现。例如：

import requests

url = 'http://httpbin.org/post'
files = {'file': open('test.txt', 'rb')}
data = {'foo': 'bar'} # 其他表单数据

r = requests.post(url, data=data, files=files)

print(r.text)

五、自定义文件名和文件类型

在默认情况下，requests会将上传的文件名和文件类型设置为原始文件名和文件类型。我们可以通过传递一个元组来自定义文件名和文件类型:

import requests

url = 'http://httpbin.org/post'
files = {'file': ('test.txt', open('test.txt', 'rb'), 'text/plain')}
r = requests.post(url, files=files)

print(r.text)

在上述代码中，我们使用一个元组来定义文件名和文件类型。第一个参数是文件名，第二个参数是文件对象，第三个参数是文件类型。

六、批量上传

在某些情况下，我们需要一次上传多个文件，可以使用列表和循环来实现批量上传。例如：

import requests

url = 'http://httpbin.org/post'
files = {'file': [('test1.txt', open('test1.txt', 'rb')), 
                  ('test2.txt', open('test2.txt', 'rb'))]}
# 文件名为test1.txt和test2.txt

r = requests.post(url, files=files)

print(r.text)

在上述代码中，我们将文件名和文件对象作为元组添加到列表中，然后将该列表作为‘files’参数的值传递给requests.post方法。

七、处理上传的文件

有时候，我们需要在上传文件后对上传的文件进行处理。例如，我们可能需要将文件保存到本地文件系统或上传到另一个服务器。

import requests

url = 'http://httpbin.org/post'
files = {'file': open('test.txt', 'rb')}

r = requests.post(url, files=files)

with open('response.txt', 'wb') as f:
    f.write(response.content) # 将响应内容写入本地文件

print('文件已保存到本地')

在上述代码中，我们使用‘open’函数打开一个文件‘response.txt’，然后将响应内容写入该文件。注意，在使用requests.post方法上传文件时，返回的响应包含了服务器返回的数据，因此我们需要使用‘content’属性来获取响应内容。

八、处理chunked编码的响应

有些服务器使用chunked编码来返回响应，这种情况下，requests必须解码响应内容才能使其可读。可以通过设置‘stream’参数来请求一个具有流式传输支持的响应，然后使用‘iter_content’方法来迭代响应内容。例如：

import requests

url = 'http://httpbin.org/post'
files = {'file': open('test.txt', 'rb')}

r = requests.post(url, files=files, stream=True)

with open('response.txt', 'wb') as f:
    for chunk in r.iter_content(chunk_size=1024):
        f.write(chunk)

print('文件已保存到本地')

在上述代码中，我们首先设置‘stream’参数为True，以请求一个具有流式传输支持的响应。然后使用‘iter_content’方法来迭代响应内容，并将内容分块存储到文件中。请注意，‘iter_content’方法的第一个参数‘chunk_size’用于定义一次性迭代多少字节。在这个例子中，我们将一次迭代的字节数设置为1024。

九、结语

本文介绍了如何使用Python requests库上传文件。我们了解了如何上传单个文件、多个文件、自定义文件名和文件类型、处理上传的文件和chunked编码的响应。使用requests上传文件非常简单，但是在处理上传的文件和响应的过程中可能会遇到一些挑战。