FastAPI学习-10. 路由管理APIRouter

上海-悠悠

发布于 2023-09-21 08:12:38

4100

发布于 2023-09-21 08:12:38

文章被收录于专栏：从零开始学自动化测试从零开始学自动化测试

前言

在 Flask 中，我们一般用蓝图 Blueprint 来处理多个模块的视图，在fastapi 中也有类似的功能通过APIRouter 来管理。

路由管理 APIRouter

如果你正在开发一个应用程序或 Web API，很少会将所有的内容都放在一个文件中。 FastAPI 提供了一个方便的工具，可以在保持所有灵活性的同时构建你的应用程序（如果你学过 Flask，那这将相当于 Flask 的 Blueprints）。

假设你的文件结构如下：

├── app
│   ├── __init__.py
│   ├── main.py
│   └── routers
│   │   ├── __init__.py
│   │   ├── items.py
│   │   └── users.py

app 目录包含了所有内容。并且它有一个空文件 app/__init__.py，它包含一个 app/main.py 文件。 routers 目录下有 items.py 和 users.py 2个文件。

假设专门用于处理用户逻辑的文件是位于 /app/routers/users.py 的子模块

from fastapi import APIRouter

router = APIRouter()

@router.get("/users/", tags=["users"])
async def read_users():
    return [{"username": "Rick"}, {"username": "Morty"}]

@router.get("/users/me", tags=["users"])
async def read_user_me():
    return {"username": "fakecurrentuser"}

@router.get("/users/{username}", tags=["users"])
async def read_user(username: str):
    return {"username": username}

假设你在位于 app/routers/items.py 的模块中还有专门用于处理应用程序中「项目」的端点。这和 app/routers/users.py 的结构完全相同。我们知道此模块中的所有路径操作都有相同的：

路径 prefix：路径前缀 /items。
tags：（仅有一个 items 标签）。
responses: 定义响应状态码
dependencies：依赖项。

因此，我们可以将其添加到 APIRouter 中，而不是将其添加到每个路径操作中。

from fastapi import APIRouter, Depends, HTTPException

router = APIRouter(
    prefix="/items",
    tags=["items"],
    responses={404: {"description": "Not found"}},
)

fake_items_db = {"plumbus": {"name": "Plumbus"}, "gun": {"name": "Portal Gun"}}

@router.get("/")
async def read_items():
    return fake_items_db

@router.get("/{item_id}")
async def read_item(item_id: str):
    if item_id not in fake_items_db:
        raise HTTPException(status_code=404, detail="Item not found")
    return {"name": fake_items_db[item_id]["name"], "item_id": item_id}

@router.put(
    "/{item_id}",
    tags=["custom"],
    responses={403: {"description": "Operation forbidden"}},
)
async def update_item(item_id: str):
    if item_id != "plumbus":
        raise HTTPException(
            status_code=403, detail="You can only update the item: plumbus"
        )
    return {"item_id": item_id, "name": "The great Plumbus"}

FastAPI 主体

现在，让我们来看看位于 app/main.py 的模块。在这里你导入并使用 FastAPI 类。

from fastapi import Depends, FastAPI
from .routers import items, users

app = FastAPI()

app.include_router(users.router)
app.include_router(items.router)

@app.get("/")
async def root():
    return {"message": "Hello Bigger Applications!"}

本文参与腾讯云自媒体分享计划，分享自微信公众号。

原始发表：2023-09-12 14:51，如有侵权请联系 cloudcommunity@tencent.com 删除

app