FastAPI传输响应文件

最近博客项目有一个需求,博客前端通过mavon-editor上传图片,后端将其保存在数据库中,而这就出现了一个为问题,如何通过FastAPI传输这种文件,以及如何对它编码解码.
对这个问题,我们很容易显出一个很烂简单的做法,将图片直接压缩为base64格式,字符串传输存储,但这样的问题在于,传输的数据量大,处理起来也非常麻烦,而我们从fastapi的官网可以看到,fastapi的响应依赖starlette,那么我们可以再跳转到starlette,看到这一段:

StreamingResponse
Takes an async generator or a normal generator/iterator and streams the response body.
Have in mind that file-like objects (like those created by open()) are normal iterators. So, you can return them directly in a StreamingResponse.

大意是:使用一个异步的生成器或普通的生成器/可迭代对象作为这个响应主体.请记住,类文件对象（如open（）创建的对象）是普通的迭代器。所以，你可以直接以流响应的形式返回它们。

这是非常有帮助的一段信息,让我们再看看官方的实例:

from starlette.responses import StreamingResponse     #引入需要的文件
import asyncio

#定义一个异步生成器
async def slow_numbers(minimum, maximum):
    yield('<html><body><ul>')
    for number in range(minimum, maximum + 1):
        yield '<li>%d</li>' % number
        await asyncio.sleep(0.5)
    yield('</ul></body></html>')

#定义接口,主要部分
async def app(scope, receive, send):
    assert scope['type'] == 'http'
    generator = slow_numbers(1, 10)     #获取生成器
    response = StreamingResponse(generator, media_type='text/html')     #关键部分!通过生成器生成对象.
    await response(scope, receive, send)

根据这个,我们就可以进行代码编写了,我的博客项目使用的是sqlite的blob类型字段存储图像,所以我们要借助io库,并最后返回一个StreamingResponse实例.话不多说,上代码:

@app.get("/asset/img/{imgId}")        #定义路径,使用了一个路径参数
async def getImgApi(imgId: int):        #定义函数,注意要加上async,其实不加也可以运行,但访问数据库时变为多线程访问,需要设置check_sme_thread=False
    img = db.get("files", id=imgId)        #获取图像,这里使用了一个自己封装的sqlite3数据库接口,这行代码的意思是,设定img为数据库中从"files"表中获取的id=imgId的数据.
    imgD = img.file        #获取其中的blob字段的值,也就是说imgD是真正的BLOB字段的值.
    buffer = io.BytesIO(imgD)        #创建一个BytesIO,因为之前说过,可以使用like-file objects(和文件具有相同接口的对象)作为参数.
    r = StreamingResponse(buffer)        #创建响应体
    return r        #返回

这里我们在fastapi中可以直接通过这种方式返回一个starlette的response,因为其官网上说:

Return a Response

In fact, you can return any Response or any sub-class of it.
And when you return a Response, FastAPI will pass it directly.
It won't do any data conversion with Pydantic models, it won't convert the contents to any type, etc.
This gives you a lot of flexibility. You can return any data type, override any data declaration or validation, etc.

也就是说,我们可以直接返回STARLETTE的response,因为Response类其实就是那里面的.
这样我们就实现了一个文件的传输操作.

以上是这篇文章的全部,错误的地方恳请指正.
希望能和大家一起学习.
最后,都看到这了,赞赏一下呗!(^ ~ ^||).