Python中requests上传大文件

最新推荐文章于 2024-03-14 14:51:29 发布

aduocd

最新推荐文章于 2024-03-14 14:51:29 发布

阅读量2.3k

点赞数

分类专栏： Python 自动化测试文章标签： python requests 上传文件 memoryerror

本文链接：https://blog.csdn.net/aduocd/article/details/124966226

版权

自动化测试同时被 2 个专栏收录

27 篇文章 2 订阅

订阅专栏

Python

11 篇文章 0 订阅

订阅专栏

1.背景

2.上传大文件

3.参考资料

1.背景

在使用requests上传文件时，尽管后台接口上已经做了分片处理，但不知道为何，python脚本还是报了memoryerror。

2.上传大文件

网上也有不少文件上传的代码。这里稍微有所差异的是，前端使用了分片操作，所以请求体中需要传一些参数；此外，文件的二进制流也在请求体中。描述过于复杂，代码更加直接~~

import os
from requests_toolbelt import *


file_path = "D:\\mytest\\myfile.rar"
file_name = "myfile.rar"        # 这里为了简化就直接给出文件名了，也可以从文件路径中获取
file_stats = os.stat(file_path)
file_size = file_stats.st_size

with open(file_path, mode='rb') as f:
    file_rb = f.read()

def upload_file(file_name, file_size, file_rb):

    url = "http://localhost:9090/abc"    

    item = {
        "name": file_name,
        "chunkNumber": "1",
        "chunkSize": "204800000",
        "fileSize": str(file_size),
        "file": (file_name, file_rb, "application/x-zip-compressed")    # 这里application/x-zip-compressed是从请求接口的header是中看到的
    }
    
    temp = MultipartEncoder(data)
    data = MultipartEncoderMonitor(temp)

    headers = {
        "Content-Type": temp.content_type
    }

    response = request.post(url=url, data=data, headers=headers)
    return response