FastAPI传输响应文件 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('') for number in range(minimum, maximum + 1): yield '%d' % number await asyncio.sleep(0.5) yield('')#定义接口,主要部分 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
【FastAPI传输响应文件】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类其实就是那里面的.
这样我们就实现了一个文件的传输操作.
以上是这篇文章的全部,错误的地方恳请指正.
希望能和大家一起学习.
最后,都看到这了,赞赏一下呗!(^ ~ ^||).