有时,我们要为整个应用添加依赖项。通过与定义FastAPI 学习之路(二十五)路径操作装饰器依赖项
类似的方式,可以把依赖项添加至整个 FastAPI 应用。
那么我们看下,如何去实现,比如我们全局都需要校验token。我们去看下,我们应该如何实现代码。
from fastapi import FastAPI, Depends, Header, HTTPException
from typing import Optional
fake_db_items = [{"city": "beijing"}, {"city": "shanghai"}, {"city": "guangzhou"}]
def verify_token(token: str = Header(...)):
if token != "mrli_token":
raise HTTPException(status_code=400, detail="token is invalid")
app = FastAPI(dependencies=[Depends(verify_token)])
@app.get("/items")
def read_items():
return fake_db_items
@app.get("/users")
def read_users():
return "this is test return user"
if __name__ == '__main__':
import uvicorn
uvicorn.run("main:app", reload=True, debug=True)
那么我们看下,接口是否都需要token。
①:测试 /items 不带token
然后我们增加token
②测试下 /users 接口不带token
带token的 /users