FASTAPI系列 14-使用JSONResponse 返回JSON内容
文章目录
- FASTAPI系列 14-使用JSONResponse 返回JSON内容
- 前言
- 一、默认返回的JSON格式
- 二、JSONResponse 自定义返回
- 三、自定义返回 headers 和 media_type
- 总结
前言
当你创建一个 FastAPI 接口时,可以正常返回以下任意一种数据:dict,list,Pydantic 模型,数据库模型等等。FastAPI 默认会使用 jsonable_encoder 将这些类型的返回值转换成 JSON 格式,默认情况下会以content-type: application/json 格式返回。
在有些情况下,我们需要在路径操作中直接返回Response对象,这样我们能有更多的操作灵活性,比如自定义头headers 信息、自定义Cookie信息等
提示:以下是本篇文章正文内容,下面案例可供参考
一、默认返回的JSON格式
定义一个字典类型,然后fastpai会默认转化成json然后返回
from fastapi import FastAPI
import uvicorn
app = FastAPI()
@app.get('/users')
def users():
user = {
"user_name": "Teacher Li",
"email": "Teacher_Li@qq.com"
}
return user
当使用get请求时,返回的格式:
HTTP/1.1 200 OK
date: Tue, 25 Mar 2023 10:40:41 GMT
server: uvicorn
content-length: 36
content-type: application/json
{"user_name": "Teacher Li","email": "Teacher_Li@qq.com"}
二、JSONResponse 自定义返回
可以使用 from starlette.responses import JSONResponse 定制返回内容,包含响应状态码,响应headers 和 响应body;
JSONResponse 继承自 Response 类,部分源码如下:
class JSONResponse(Response):
media_type = "application/json"
def __init__(
self,
content: typing.Any,
status_code: int = 200,
headers: typing.Optional[typing.Dict[str, str]] = None,
media_type: typing.Optional[str] = None,
background: typing.Optional[BackgroundTask] = None,
) -> None:
super().__init__(content, status_code, headers, media_type, background)
JSONResponse可传参数:
- content: 响应body内容,str 或者 bytes.
- status_code: 响应状态码,int类型,默认200.
- headers: 响应头部,dict类型.
- media_type:media type. 例如"text/html".
- background:后台任务
自定义 JSONResponse 响应, status_code 可以自定义状态码,FastAPI 会自动包含 Content-Length,以及Content-Type,charset等头信息。
from fastapi import FastAPI, status
from fastapi.responses import JSONResponse
import uvicorn
app = FastAPI()
@app.get('/users')
def users():
user = {"user_name": "Teacher Li","email": "Teacher_Li@qq.com"}
return JSONResponse(
content=user, status_code=status.HTTP_200_OK
)
当使用get请求时,返回的格式:
HTTP/1.1 200 OK
date: Tue, 25 Mar 2023 10:40:41 GMT
server: uvicorn
content-length: 36
content-type: application/json
{"user_name": "Teacher Li","email": "Teacher_Li@qq.com"}
三、自定义返回 headers 和 media_type
响应头部添加 headers 内容和设置 media_type 响应 body 媒体类型
@app.get('/resp/users')
async def users():
user = {"user_name": "Teacher Li", "email": "Teacher_Li@qq.com"}
return JSONResponse(
content=user,
status_code=status.HTTP_201_CREATED,
headers={"x-token": "abcdefghijklmnop"},
media_type="text/html"
)
当使用get请求时,返回的格式:
HTTP/1.1 201 Created
date: Tue, 25 Mar 2023 10:40:41 GMT
server: uvicorn
x-token: abcdefghijklmnop
content-length: 36
content-type: text/html
{"user_name": "Teacher Li","email": "Teacher_Li@qq.com"}
总结
jsonable_encoder 是 FastAPI(一个用于构建高性能 Web 应用的现代、异步 Python 框架)中的一个实用函数,其主要功能是对 Pydantic 模型或任何其他可迭代对象进行编码,使其转换为可以安全地序列化为 JSON 的格式。
该函数会遍历对象的所有属性,并对其中的复杂数据类型如 datetime、UUID 等进行特殊处理,确保它们能够被 JSON 序列化引擎识别和正确处理。例如,datetime 对象会被转换为 ISO8601 格式的字符串,而 UUID 对象则通常会转换为字符串形式。
使用 jsonable_encoder 可以确保你在将数据返回给前端或者存入数据库时,不会因为数据类型的复杂性而出现问题。这极大地方便了开发者处理 API 返回的数据结构,提高了开发效率。
总结来说,jsonable_encoder 在 FastAPI 中的作用是提供一种便捷的方法,将符合 Pydantic 模型或其他特定类型的数据结构转换成适合 JSON 序列化的格式。