12、FastAPI 中的依赖注入与数据库交互

FastAPI 中的依赖注入与数据库交互

1. FastAPI 中的依赖注入

在开发 RESTful API 时，我们可以创建多个路由器来清晰地划分 API 的不同部分，并将它们连接到主 FastAPI 应用程序。此时，为整个路由器注入依赖是一个不错的选择，这样每个路由都会调用该依赖。有两种方式可以实现：
- 在 APIRouter 类上设置依赖 ：

from fastapi import APIRouter, Depends

# 假设 secret_header 是一个依赖函数
def secret_header():
    pass

router = APIRouter(dependencies=[Depends(secret_header)])

@router.get("/route1")
async def router_route1():
    return {"route": "route1"}

@router.get("/route2")
async def router_route2():
    return {"route": "route2"}

from fastapi import FastAPI
app = FastAPI()
app.include_router(router, prefix="/router")

• 在 include_router 方法上设置依赖 ：