FastAPI（51）- 自定义响应之 StreamingResponse、FileResponse-阿里云开发者社区

FastAPI（51）- 自定义响应之 StreamingResponse、FileResponse

2022-05-15 1089

版权

本文内容由阿里云实名注册用户自发贡献，版权归原作者所有，阿里云开发者社区不拥有其著作权，亦不承担相应法律责任。具体规则请查看《阿里云开发者社区用户服务协议》和《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容，填写侵权投诉表单进行举报，一经查实，本社区将立刻删除涉嫌侵权内容。

简介： FastAPI（51）- 自定义响应之 StreamingResponse、FileResponse

StreamingResponse

作用

采用异步生成器或普通生成器（generator）/迭代器（iterator）流式传输响应数据

实际代码

from fastapi import FastAPI
from fastapi.responses import StreamingResponse
file_path = "test.mp4"
app = FastAPI()
@app.get("/")
def main():
    # 这是生成器函数。它是一个“生成器函数”，因为它里面包含了 yield 语句
    def iterfile():
        # 通过使用 with 块，确保在生成器函数完成后关闭类文件对象
        with open(file_path, "rb") as file_like:
            # yield from 告诉函数迭代名为 file_like 的东西
            # 对于迭代的每个部分，yield 的内容作为来自这个生成器函数
            yield from file_like
    return StreamingResponse(iterfile(), media_type="video/mp4")

如果有一个类文件对象（例如 open() 返回的对象），可以创建一个生成器函数来迭代该类文件对象
这样，不必首先在内存中读取所有内容，可以将该生成器函数传递给 StreamingResponse，然后返回它
这包括许多与云存储、视频处理等交互的库

请求结果

返回了视频啦！

源码

FileResponse

作用

异步流式传输文件作为响应，重点一定是异步的

实际代码

from fastapi import FastAPI
from fastapi.responses import FileResponse
file_path = "test.mp4"
app = FastAPI()
@app.get("/file", response_class=FileResponse)
async def main():
    return file_path