如何通过 FastAPI 从服务器下载大文件？

Answer 1

如果您发现yield from f 在using StreamingResponse with file-like objects 时相当慢，您可以创建一个生成器，在其中使用指定的块大小以块的形式读取文件；因此，加快了进程。可以在下面找到示例。

请注意，StreamingResponse 可以采用 async 生成器或普通生成器/迭代器来流式传输响应正文。如果您使用不支持async/await 的标准open() 方法，则必须使用普通def 声明生成器函数。无论如何，FastAPI/Starlette 仍将异步工作，因为它会检查您传递的生成器是否是异步的（如 source code 所示），如果不是，它将在单独的线程中运行生成器，使用 @987654323 @，然后等待。

您可以在响应中设置Content-Disposition 标头（如this answer 中所述，以及here 和here）以指示内容是否应在浏览器中显示inline（如果您是流，例如，.mp4 视频、.mp3 音频文件等），或作为attachment 下载并保存在本地（使用指定的filename）。

至于media_type（也称为MIME 类型），有两种主要的MIME 类型（参见Common MIME types）：

• text/plain 是文本文件的默认值。文本文件应该是人类可读的，并且不能包含二进制数据。
• application/octet-stream 是所有其他情况的默认值。一个未知文件类型应使用此类型.

对于带有.tar 扩展名的文件，如您的问题所示，您还可以使用不同的亚型来自octet-stream，即x-tar。有关常见 MIME 类型的列表，请参阅上面的链接文档。

选项 1 - 使用普通发电机

from fastapi import FastAPI
from fastapi.responses import StreamingResponse

CHUNK_SIZE = 1024 * 1024  # = 1MB - adjust the chunk size as desired
some_file_path = 'large_file.tar'
app = FastAPI()

@app.get('/')
def main():
    def iterfile():
        with open(some_file_path, 'rb') as f:
            while chunk := f.read(CHUNK_SIZE):
                yield chunk

    headers = {'Content-Disposition': 'attachment; filename="large_file.tar"'}
    return StreamingResponse(iterfile(), headers=headers, media_type='application/x-tar')

选项 2 - 使用 async 生成器和 aiofiles

from fastapi import FastAPI
from fastapi.responses import StreamingResponse
import aiofiles

CHUNK_SIZE = 1024 * 1024  # = 1MB - adjust the chunk size as desired
some_file_path = 'large_file.tar'
app = FastAPI()

@app.get('/')
async def main():
    async def iterfile():
       async with aiofiles.open(some_file_path, 'rb') as f:
            while chunk := await f.read(CHUNK_SIZE):
                yield chunk

    headers = {'Content-Disposition': 'attachment; filename="large_file.tar"'}
    return StreamingResponse(iterfile(), headers=headers, media_type='application/x-tar')