python接口自动化7-post文件上传

前言

文件上传在我们软件是不可少的，最多的使用是体现在我们后台，当然我们前台也会有。但是了解过怎样上传文件吗？这篇我们以禅道文档-创建文档，上传文件为例。

post请求中的：Content-Type：multipart/form-data  这种类型便是上传文件。

一、环境安装、抓包分析

1、pip install requests_toolbelt，post请求 multipart/form-data  类型

C:\Users\Administrator>pip install requests_toolbelt
Requirement already satisfied: requests_toolbelt in d:\path_python\lib\site-packages (0.9.1)
Requirement already satisfied: requests<3.0.0,>=2.0.1 in d:\path_python\lib\site-packages (from requests_toolbelt) (2.20.1)
Requirement already satisfied: chardet<3.1.0,>=3.0.2 in d:\path_python\lib\site-packages (from requests<3.0.0,>=2.0.1->requests_toolbelt) (3.0.4)
Requirement already satisfied: certifi>=2017.4.17 in d:\path_python\lib\site-packages (from requests<3.0.0,>=2.0.1->requests_toolbelt) (2019.6.16)
Requirement already satisfied: idna<2.8,>=2.5 in d:\path_python\lib\site-packages (from requests<3.0.0,>=2.0.1->requests_toolbelt) (2.7)
Requirement already satisfied: urllib3<1.25,>=1.21.1 in d:\path_python\lib\site-packages (from requests<3.0.0,>=2.0.1->requests_toolbelt) (1.24.3)

2、先抓下登录接口，因为先登录成功后才能上传文件。写上登录代码，让其能正常登录。

import requests

s = requests.session()
login_url = 'http://127.0.0.1/zentao/user-login.html'       # 登录url
data = 'account=admin&password=e10adc3949ba59abbe56e057f20f883e&keepLogin%5B%5D=on&referer=http%3A%2F%2F127.0.0.1%2Fzentao%2Fdoc-browse-1-byModule-0-id_desc-doc.html'
login_r = s.post(login_url, params=data)                         # 传 params 参数

r1 = s.get('http://127.0.0.1/zentao/doc-browse-1-byModule-0-id_desc-doc.html')  # 登录访问的 html
print(r1.content.decode('utf-8'))

 2、操作上传文件（文件我就上传一个桌面的：test.png），fiddler 切换至WebForms就看得比较清楚了。

二、写python代码请求

1、将抓包的接口写下，顺序：调试登录、上传、检查点。

2、也需导入模块from requests_toolbelt import MultipartEncoder

3、MultipartEncoder 传的参数注意，特别是图片参数，这里list里面元组：fields=[(name,value), (name2, value)]

4、禅道文件上传代码参考如下：

from requests_toolbelt import MultipartEncoder
import requests

# 登录
s = requests.session()
login_url = 'http://127.0.0.1/zentao/user-login.html'
data = 'account=admin&password=e10adc3949ba59abbe56e057f20f883e&keepLogin%5B%5D=on&referer=http%3A%2F%2F127.0.0.1%2Fzentao%2Fdoc-browse-1-byModule-0-id_desc-doc.html'
login_r = s.post(login_url, params=data)  # 传 params 参数

# 断言是否登录成功
r1 = s.get('http://127.0.0.1/zentao/doc-browse-1-byModule-0-id_desc-doc.html')
if '产品主库' in r1.content.decode('utf-8'):
    print('登录成功')
else:print('登录失败')

# 文件上传保存
# 传参数，name 对应 value，注意图片路径的填写。为空的一些参数不影响的去前提下可以删除。
s_url = 'http://127.0.0.1/zentao/doc-create-1-0.html'
body = MultipartEncoder(
    fields=[
        ('lib', '1'),
        ('uid', '5dec436e32b85'),
        ('module', '0'),
        ('title', '我的一个title'),
        ('type', 'text'),
        ('contentType', 'html'),
        ('files[]', ('test.png', open('C:\\Users\\Administrator\\Desktop\\test.png', 'rb'), 'image/png'))
    ])
r2 = s.post(s_url, data=body, headers={'Content-Type': body.content_type})      # Content-Type 自动获取

# 检查是否成功上传图片
res = s.get('http://127.0.0.1/zentao/doc-browse-1.html')
if '我的一个title' in res.content.decode('utf-8'):
    print('已上传成功')
else:print('上传失败')

 5、python控制台说“已上传成功”，但是我们首次执行最好还是肉眼看一下有没有问题。（下图显然没有问题）

但说到这里，不得不说这只是一次的测试请求，但是以后迭代，无数次回归咋办呢？这个问题和我们手工操作是一样的，好比如你想title每次都一样，但是只能存在一条数据。

那么就可以①逻辑删除：操作删除按钮、物理删除：连接数据库执行删除语句。欢迎来QQ交流群：482713805