FastAPI(52)- Response Cookies 响应设置 Cookies

前言

有两种实现场景

  • 路径操作函数声明 Response 参数来设置 Cookie
  • 在函数内,通过 return Response 来设置 Cookie

路径操作函数声明 Response 参数来设置 Cookie

from fastapi import FastAPI, Response

app = FastAPI()


@app.get("/item")
# 路径操作函数声明一个 Response 类型的参数
async def get_item(response: Response):
    response.set_cookie("test_token", "tokenABC")
    return {
    "name": "设置Cookie"}
  • 然后可以像往常一样返回需要的任何对象(字典、数据库模型等)
  • 如果声明了一个 response_model,它仍将用于过滤和转换返回的对象
  • FastAPI 将使用该临时响
  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
FastAPIcookies 是使用 Python 的 `http.cookies` 模块来处理的。要实现 cookies 保持登录的功能,需要将用户的身份信息存储在 cookies 中,并在每个请求中读取 cookies 中的信息来验证用户身份。 以下是一个使用 FastAPI 实现 cookies 保持登录的示例: ```python from fastapi import FastAPI, Request, Response from fastapi.responses import HTMLResponse from fastapi.templating import Jinja2Templates from http import cookies app = FastAPI() templates = Jinja2Templates(directory="templates") # 假设这是一个数据库,存储了用户的身份信息 users_db = { "alice": "password1", "bob": "password2", } # 登录页面 @app.get("/") async def login_form(request: Request): return templates.TemplateResponse("login.html", {"request": request}) # 处理登录请求 @app.post("/login") async def login(request: Request, response: Response, username: str, password: str): # 验证用户身份信息 if username in users_db and users_db[username] == password: # 将用户信息存储在 cookies 中 c = cookies.SimpleCookie() c["username"] = username c["username"]["max-age"] = 3600 response.headers["Set-Cookie"] = c.output(header='') # 登录成功,跳转到受保护的页面 return templates.TemplateResponse("protected.html", {"request": request, "username": username}) else: # 登录失败,返回错误信息 return templates.TemplateResponse("login.html", {"request": request, "error": "Invalid username or password"}) # 受保护的页面,只有登录后才能访问 @app.get("/protected", response_class=HTMLResponse) async def protected(request: Request): # 从 cookies 中读取用户信息 username = request.cookies.get("username") if not username: # 用户未登录,跳转到登录页面 return templates.TemplateResponse("login.html", {"request": request}) else: # 用户已登录,显示受保护的页面 return templates.TemplateResponse("protected.html", {"request": request, "username": username}) ``` 在上面的代码中,首先定义了一个 `users_db` 字典,用于存储用户的身份信息。登录页面使用 `login.html` 模板,显示登录表单;处理登录请求的函数使用 `post` 方法,验证用户身份信息,并将用户信息存储在 cookies 中;受保护的页面使用 `protected.html` 模板,只有登录后才能访问。 在处理登录请求时,使用 `http.cookies` 模块创建一个 `SimpleCookie` 对象,将用户名存储在 cookies 中,并设置了 `max-age` 属性为 3600 秒,表示 cookies 会在 1 小时后过期。然后将 cookies 添加到响应头中,以便浏览器可以将 cookies 保存到本地。 在受保护的页面中,使用 `request.cookies.get("username")` 方法从 cookies 中读取用户名,如果不存在,则表示用户未登录,跳转到登录页面;如果存在,则表示用户已登录,显示受保护的页面。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

小菠萝测试笔记

来支持下测试小锅锅

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值