Python流式下载文件

转载请注明出处：http://blog.youkuaiyun.com/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工程中去。如果你想要完整的代码，请点击这里。

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