Python接口自动化之文件上传/下载接口怎么实现
今天小编给大家分享一下Python接口自动化之文件上传/下载接口怎么实现的相关知识点，内容详细，逻辑清晰，相信大部分人都还太了解这方面的知识，所以分享这篇文章给大家参考一下，希望大家阅读完这篇文章后有所收获，下面我们一起来了解一下吧。
〇、前言
文件上传/下载接口与普通接口类似，但是有细微的区别。
如果需要发送文件到服务器，例如：上传文档、图片、视频等，就需要发送二进制数据，上传文件一般使用的都是 Content-Type: multipart/form-data 数据类型，可以发送文件，也可以发送相关的消息体数据。
反之，文件下载就是将二进制格式的响应内容存储到本地，并根据需要下载的文件的格式来写文件名，例如：F:/合同文件.pdf。
一、文件上传接口
1. 接口文档
Request URL： /createbyfile
Request Method： POST
Content-Type： multipart/form-data
| 名称 | 类型 | 是否必须 | 描述 |
|---|---|---|---|
| file | File | 是 | 文档文件 |
| title | String | 是 | 文档名称 |
| fileType | String | 是 | 文件类型：doc, docx, txt, pdf, png, gif, jpg, jpeg, tiff, html, rtf, xls, txt |
2. 代码实现
（1）实现步骤：
构造文件数据，通过open函数以二进制方式打开文件
文件上传接口参数与普通post请求一样，需要写成Key和Value模式，Key为参数名称file（也是组件的name属性），Value为一个元组（与普通接口不同的地方）
"file": ( "", # 元组第一个值为文件名称，没有则取None open(r"F:\pdf_file.pdf", "rb"), # 若第一个值非None，则取文件open打开的二进制流，否则直接写文件路径，如"F:\pdf_file.pdf" "pdf" # 文件类型)
"file": ( None, "F:\pdf_file.pdf")
构造其他数据
{ "title": "接口发起的文档", "fileType": "pdf"}发送请求，将文件数据以 files 参数传入，其他消息体数据通过 data 、json 、 headers 、 cookies 等传入
req = { "url": "127.0.0.1/v2/document/createbyfile", "method": "POST", "headers": {}, "files": {"file": ("", open(r"F:\pdf_file.pdf", "rb"), "pdf")}, "data": { "title": "接口发起的文档", "fileType": "pdf" } }（2）完整代码
base_api.py
import requestsclass BaseApi: @staticmethod def requests_http(req): # ** 解包 result = requests.request(**req) return result
api/createbyfile.py
# -*- coding:utf-8 -*-# 作者：IT小学生蔡坨坨# 时间：2022/3/12 21:04# 功能：根据文件类型创建合同文档from base_api import BaseApiclass Createbyfile: def createbyfile(self): req = { "url": "127.0.0.1/createbyfile", "method": "POST", "headers": {}, "files": {"file": ("", open(r"F:\pdf_file.pdf", "rb"), "pdf")}, "data": { "title": "接口发起的文档", "fileType": "pdf" } } res = BaseApi().requests_http(req) assert res.status_code == 200 res_json = res.json() return res_json["result"]["documentId"]if __name__ == '__main__': Createbyfile().createbyfile()二、文件下载接口
1. 接口文档
Request URL：/download
Request Method：GET
| 名称 | 类型 | 是否必须 | 描述 |
|---|---|---|---|
| contractId | Long | ID | ID |
| downloadItems | String[] | 否 | 下载可选项，NORMAL(正文)，ATTACHMENT(附件) |
| needCompressForOneFile | Boolean | 是，默认单文件也压缩 | 当下载的文件仅一份时，是否压缩 |
2. 代码实现
# -*- coding:utf-8 -*-# 作者：IT小学生蔡坨坨# 时间：2022/4/5 2:56# 功能：下载合同from base_api import BaseApiclass Download: def download(self): req = { "url": "127.0.0.1/download", "method": "GET", "headers": {}, "params": { "contractId": 2947403075747869536, "downloadItems": ["NORMAL"], "needCompressForOneFile": False }, } res = BaseApi().requests_http(req).content # 注意".content"获取返回内容 # with open("F:/response.zip", "wb") as f: with open("F:/response.pdf", "wb") as f: f.write(res) return resif __name__ == '__main__': Download().download()以上就是"Python接口自动化之文件上传/下载接口怎么实现"这篇文章的所有内容，感谢各位的阅读！相信大家阅读完这篇文章都有很大的收获，小编每天都会为大家更新不同的知识，如果还想学习更多的知识，请关注行业资讯频道。
