FastAPI（53）- Response Headers 响应设置 Headers
前言
- 前面讲过如何获取 Headers：https://www.cnblogs.com/poloyy/p/15316932.html
- 也顺带提了下如何设置 Response Header，还是比较简单的
有两种实现场景
- 路径操作函数声明 Response 参数来设置 Header
- 在函数内，通过 return Response 来设置 Header
路径操作函数声明 Response 参数来设置 Header
from fastapi import FastAPI, Response
app = FastAPI()
@app.get("/item")
# 路径操作函数声明一个 Response 类型的参数
async def get_item(response: Response):
response.headers["x-token"] = "token_!@#%#$^$%&"
return {"name": "设置 headers"}
- 然后可以像往常一样返回需要的任何对象（字典、数据库模型等）
- 如果声明了一个 response_model，它仍将用于过滤和转换返回的对象
- FastAPI 将使用该临时响应来提取 headers（还有 cookies、status_code），并将它们放入包含返回值的最终响应中，由任何 response_model 过滤
- 还可以在依赖项中声明 Response 参数，并在其中设置 headers、cookies
请求结果
声明 response_model 和使用依赖项的栗子
from fastapi import FastAPI, Response, Depends
from pydantic import BaseModel
app = FastAPI()
class Item(BaseModel):
id: str
name: str
async def depnds_header(response: Response):
response.headers["x-depends-token"] = "************"
@app.get("/item2", dependencies=[Depends(depnds_header)], response_model=Item)
async def get_item():
return {"id": "12345", "name": "测试 dependencies和response_model", "age": 24}
请求结果
通过 return Response 来设置 Header
from fastapi import FastAPI, Response
from fastapi.responses import JSONResponse
app = FastAPI()
@app.get("/items")
async def get_item():
response = JSONResponse(content={"name": "JSONResponse"})
response.headers["x-auth-token"] = "XXX_TOKEN"
return response
请求结果
headers 的源码
它是 Starlette 库的 Response 类里面的方法哦，加了 @property 变成属性，就可以访问私有属性 _headers 了