flask.Response 无法在 Web 浏览器中开始下载

Question

在 Flask（微型 Web 框架）中，我们有一个视图：

@app.route('/download/<id>/<resolution>/<extension>/')
def download_by_id(id, resolution=None, extension=None):
    stream = youtube.stream_url(id, resolution, extension)
    binary = requests.get(stream['url'], stream=True)
    return flask.Response(
        binary,
        headers={'Content-Disposition': 'attachment; '
                                        'filename=' + stream['filename']})

在模板中，我们有一个链接为<a href="/download/adkdsk457jds/240p/mp4/">Download 240p Video</a>，当点击它时，它应该开始下载该视频。

问题是：

在一些没有安装 IDM 等下载管理器的浏览器中运行良好。但是 IDM 无法下载它。 IDM 只是挂在http://example.com/download/adkdsk457jds/240p/mp4/

Firefox 自己的下载管理器也是如此。 Firefox 只是下载一个普通的 .html 页面，而不是实际的视频。

但是，当没有安装 IDM 或其他下载管理器时，视频会在 Chrome 中成功下载。

请帮助和建议为什么它不起作用。我需要更改代码中的某些内容吗？

Answer 1

在标题中添加'Content-Type': 'application/octet-stream'

Answer 2

您没有包含任何响应信息，包括内容类型；您需要复制有关原始响应的更多信息，以传达您返回的响应类型。否则使用默认值（由 HTTP 标准或 Flask 规定）。

具体来说，您至少要跨内容类型、长度和传输编码进行复制：

headers={
    'Content-Disposition': 'attachment; filename=' + stream['filename']
}
for header in ('content-type', 'content-length', 'transfer-encoding'):
    if header in binary.headers:
        headers[header] = binary.headers[header]
return flask.Response(binary.raw, headers=headers)

我正在使用response.raw 底层原始文件对象；这也应该有效，但还有一个额外的好处是保留了 YouTube 应用的任何压缩。

一些下载管理器可能会尝试使用HTTP range request 来获取并行下载，即使服务器并未宣传它支持此类请求。您可能应该用 406 Not Acceptable 响应（在不受支持时请求字节范围是 Accept-* 违规）。您需要记录下载管理器发送的标头以确定是否是这种情况。