requests- 处理 multipartform-data 类型的上传接口

multipart/form-data类型的上传接口

我们在做接口测试工作时，会遇到含有文件上传的接口。

抓到的上传接口的信息大概是以下这样子：

请求体的content-type为：
multipart/form-data; boundary=---------------------------(一堆数字)

请求体当中，除了有要上传的文件以外，还有其它的数据字段。

在下面的请求体截图当中，含有以下字段：

id: 文件的id

name: 文件的名字

type: 文件的类型。如果是图片则是image/png。若是图中的文本文件，则为application/octet-stream

lastModifiedDate：上一次修改时间。截图中为当前时间点

size: 文件的大小

file: 有文件名、文件类型、文件内容3项内容。

这样的接口，用requests如何来发送呢？

requests的requests_toolbelt模块处理multipart/form-data

1、安装requests_toolbelt

pip命令： pip install requests_toolbelt

2、导入requests_toolbelt下的MultipartEncoder类

from requests_toolbelt import MultipartEncoder

3、MultipartEncoder类应用

3.1 处理请求体

MultipartEncoder类，有一个fields参数，用来传递请求数据以及文件。

在实例化MultipartEncoder类，传递fields

fields：字典类型。将请求数据放在此参数中

key为下图中的name的值。下图中的key分别为：id，name，type，lastModifiedDate，size，file

对于key为file的，它的值为一个元组：(文件名称，文件流对象(binary模式打开文件)，文件类型)。

示例："file":("chromedriver.log", open(file_path,"rb"), "application/octet-stream")})

上传文本文件的代码(实例化MultipartEncoder)：

filesize = os.path.getsize(file_path)  # 获取文件大小
m = MultipartEncoder(fields={"id":"lemon66",
                             "name":"chromedriver.log",
                             "type":"application/octet-stream",
                             "lastModifiedDate":"Mon Nov 30 2020 20:21:44 GMT+0800 (中国标准时间)",
                             "size":str(filesize),
                             "file":("chromedriver.log",open(file_path,"rb"),"application/octet-stream")})

上传图片的代码(实例化MultipartEncoder)：

# 上传文件为图片：图片类型不一样。
file_path = 'D:\\BaiduNetdiskDownload\\lemon_log_1.png'
filesize = os.path.getsize(file_path) # 获取文件大小

m = MultipartEncoder(fields={"id":"lemon66",
                         "name":"xiaojian.png",
                         "type":"image/png",
                         "lastModifiedDate":"Mon Nov 30 2020 20:21:44 GMT+0800 (中国标准时间)",
                         "size":str(filesize),
                         "file":("xiaojian.png",open(file_path,"rb"),"image/png")})

3.2 处理请求头的content-type

content-type要包含boundary。MultipartEncoder类定义了属性：content_type。

我们在代码当中，直接使用它的content-type来设置即可。

m = MultipartEncoder(fields={"id":"lemon66",
                         "name":"chromedriver.log",
                         "type":"application/octet-stream",
                         "lastModifiedDate":"Mon Nov 30 2020 20:21:44 GMT+0800 (中国标准时间)",
                         "size":str(filesize),
                         "file":("chromedriver.log",open(file_path,"rb"),"application/octet-stream")})
# content-type
# m.content_type

# 发送上传文件的请求：
resp = s.post(upload_url,data=m,headers={'Content-Type': m.content_type})

以上就是完整的处理过程啦，可以完美上传文件了。

测试派 Give testers a home.