FastAPI学习-21.response 参数-设置响应Cookies

发布时间 2023-09-17 11:07:40作者: 上海-悠悠

前言

可以在 路径函数 中定义一个类型为 Response的参数,这样你就可以在这个临时响应对象中设置cookie了。

response 参数

设置cookies

from fastapi import FastAPI, Response

app = FastAPI()


@app.post("/cookie-and-object/")
def create_cookie(response: Response):
    response.set_cookie(key="fakesession", value="fake-cookie-session-value")
    return {"message": "Come to the dark side, we have cookies"}

而且你还可以根据你的需要响应不同的对象,比如常用的 dict,数据库model等。

如果你定义了 response_model,程序会自动根据response_model来过滤和转换你响应的对象。

FastAPI 会使用这个 临时 响应对象去装在这些cookies信息 (同样还有headers和状态码等信息), 最终会将这些信息和通过response_model转化过的数据合并到最终的响应里。

你也可以在depend中定义Response参数,并设置cookie和header。

直接响应 Response

你还可以在直接响应Response时直接创建cookies。

你可以参考Return a Response Directly来创建response

然后设置Cookies,并返回:

from fastapi import FastAPI
from fastapi.responses import JSONResponse

app = FastAPI()


@app.post("/cookie/")
def create_cookie():
    content = {"message": "Come to the dark side, we have cookies"}
    response = JSONResponse(content=content)
    response.set_cookie(key="fakesession", value="fake-cookie-session-value")
    return response

需要注意,如果你直接反馈一个response对象,而不是使用Response入参,FastAPI则会直接反馈你封装的response对象。
所以你需要确保你响应数据类型的正确性,如:你可以使用JSONResponse来兼容JSON的场景。
同时,你也应当仅反馈通过response_model过滤过的数据。