最近在准备一个公开课,主题就是利用不同的语言和不同的工具去实现文件的上传和下载。
在利用Jmeter去实现功能的时候,以及利用loadrunner去写脚本的时候,都很顺利,没有任何问题,当我尝试用Python去解决这个问题的时候,花了一些时间。
这也让我在学习和尝试中,找到了很多乐趣,下面给大家分享下,如何去实现操作。
前提:
1:有一个上传接口,地址如下:http://xx.xx.xx.xx//upload/stream(公司的服务,地址不便外发~有兴趣的同志可以来看我们的公开课!)
2:上传接口的参数如下所示:
{"parentId":"","fileCategory":"personal","fileSize":179,"fileName":"summer_text_0920.txt","uoType":1}
其中有两个参数需要跟大家解释一下:filesize:指的是文件的字节大小。 filename:指的是你上传之后保存的文件名~记得不要搞错了后缀哟。其他的参数可以忽略,在做自己公司接口的时候,就按照自己公司的接口文档去做即可。
开始操作:
1:安装requests模块,安装命令:pip install requests
2:发送请求,代码如下所示:
header={"ct":"eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9"} files = {'file':open('D:\\test_data\\summer_test_data_05.txt','rb')}#此处是重点!我们操作文件上传的时候,把目标文件以open打开,然后存储到变量file里面存到一个字典里面 upload_data={"parentId":"","fileCategory":"personal","fileSize":179,"fileName":"summer_text_0920.txt","uoType":1} upload_res=requests.post(upload_url,upload_data,files=files,headers=header)##此处是重点!我们操作文件上传的时候,接口请求参数直接存到upload_data变量里面,在请求的时候,直接作为数据传递过去
查看结果:
执行完代码后,看到我们上传成功,效果如下:
大家如果有上传文件的接口,赶紧去试试吧!
学习了代码后,有些事情就会显得轻松多啦!
补充:python实现接口上传文件的两种方法
文件上传:上传图片的类型是file,这里没有用到头部信息
import requests def sendImg(img_path, img_name, img_type='image/jpeg'): """ :param img_path:图片的路径 :param img_name:图片的名称 :param img_type:图片的类型,这里写的是image/jpeg,也可以是png/jpg """ url = 'https://www.xxxxxxxxxx.com' # 自己想要请求的接口地址 with open(img_path + img_name, "rb")as f_abs:# 以2进制方式打开图片 body = { # 有些上传图片时可能会有其他字段,比如图片的时间什么的,这个根据自己的需要 'camera_code': (None, "摄像头1"), 'image_face': (img_name, f_abs, img_type) # 图片的名称、图片的绝对路径、图片的类型(就是后缀) "time":(None, "2019-01-01 10:00:00") } # 上传图片的时候,不使用data和json,用files response = requests.post(url=url, files=body).json return response if __name__=='__main__': # 上传图片 res = sendImg(img_path, img_name) # 调用sendImg方法 print(res) **如果上传图片是数组时,value直接写图片路径就可以**
文件上传:上传的类型是file,用到头部信息
# "Content-Type": "multipart/form-data; boundary=76a22e30da2bb7790828887966871012" from urllib3 import encode_multipart_formdata import requests def sendFile(filename, file_path): """ :param filename:文件的名称 :param file_path:文件的绝对路径 """ url = "https://www.xxxxxxx.com" # 请求的接口地址 with open(file_path, mode="r", encoding="utf8")as f: # 打开文件 file = { "file": (filename, f.read()),# 引号的file是接口的字段,后面的是文件的名称、文件的内容 "key": "value", # 如果接口中有其他字段也可以加上 } encode_data = encode_multipart_formdata(file) file_data = encode_data[0] # b'--c0c46a5929c2ce4c935c9cff85bf11d4\r\nContent-Disposition: form-data; name="file"; filename="1.txt"\r\nContent-Type: text/plain\r\n\r\n...........--c0c46a5929c2ce4c935c9cff85bf11d4--\r\n headers_from_data = { "Content-Type": encode_data[1], "Authorization": token } # token是登陆后给的值,如果你的接口中头部不需要上传字段,就不用写,只要前面的就可以 # 'Content-Type': 'multipart/form-data; boundary=c0c46a5929c2ce4c935c9cff85bf11d4',这里上传文件用的是form-data,不能用json response = requests.post(url=url, headers=headers_from_data, data=file_data).json() return response if __name__=='__main__': # 上传文件 res = sendFile(filename, file_path) # 调用sendFile方法 print(res)
以上为个人经验,希望能给大家一个参考,也希望大家多多支持。如有错误或未考虑完全的地方,望不吝赐教。