Python流式下载文件

admin管理员组

文章数量:1794759

Python流式下载文件

转载请注明出处：blog.csdn/jinixin/article/details/79053741

之前写了使用WebUploader上传大文件的文章，既然有上传文件的需求，下载文件也是免不了的。在此使用Flask举例，为了便于移植，不使用Flask框架集成的send_file方法。

普通下载

直接上代码：

@app.route('/file/download/<filename>', methods=['GET']) def file_download(filename): with open('./upload/%s' % filename, 'rb') as target_file: # 读取文件内容 data = target_file.read() response = Response(data, content_type='application/octet-stream') # 响应指明类型，写入内容 return response

在响应中指定了内容类型为字节流，浏览器收到后便会直接下载内容。

流式下载

虽然上面的代码可以顺利下载，但是如果文件很大，直接打开文件读取内容会大幅占用内存，影响服务器性能。所以最好能流式返回，一点一点读取文件内容。提到流式，在Python中我想到了迭代器和yield关键字。

修改后的代码：

@app.route('/file/download/<filename>', methods=['GET']) def file_download(filename): def send_chunk(): # 流式读取 store_path = './upload/%s' % filename with open(store_path, 'rb') as target_file: while True: chunk = target_file.read(20 * 1024 * 1024) # 每次读取20M if not chunk: break yield chunk return Response(send_chunk(), content_type='application/octet-stream')

运行时，打开活动监视器，未出现大幅占用内存的情况。

最后，关于流式下载的demo我一同上传到了upload_demo工程中去。如果你想要完整的代码，请点击这里。

对流式的理解还不很清晰，文中如有不妥之处，还望大家包容和支持，感谢。

本文标签： 流式文件Python