如何限制FastAPI请求头中的内容类型

FastAPI是一个基于Python的现代、快速（高性能）的Web框架，用于构建API。它支持异步请求处理和类型注解，同时具有简单易用的API设计和自动生成文档的功能。

要限制FastAPI请求头中的内容类型，可以使用FastAPI提供的Content-Type装饰器。该装饰器用于声明接受的请求内容类型，并可以通过参数指定允许的内容类型。

以下是限制FastAPI请求头中内容类型的步骤：

导入FastAPI和Depends模块：

from fastapi import FastAPI, Depends

创建一个FastAPI应用实例：

app = FastAPI()

定义一个路由处理函数，并使用Content-Type装饰器限制请求头中的内容类型。可以使用media_type参数指定允许的内容类型，多个内容类型可以使用逗号分隔。

@app.post("/endpoint")
async def endpoint(content_type: str = Depends(Content-Type(["application/json"])):
    # 处理请求
    return {"message": "Success"}

在上述代码中，我们定义了一个POST请求的路由处理函数endpoint，并使用Content-Type装饰器限制请求头中的内容类型为application/json。如果请求头中的内容类型不符合要求，FastAPI将返回415 Unsupported Media Type错误。

需要注意的是，Content-Type装饰器是FastAPI提供的一个自定义装饰器，用于限制请求头中的内容类型。它接受一个参数，即允许的内容类型列表。

这是一个完整的示例，演示了如何限制FastAPI请求头中的内容类型为application/json：

from fastapi import FastAPI, Depends
from fastapi.exceptions import HTTPException
from fastapi.param_functions import Header
from fastapi.routing import APIRouter
from starlette.status import HTTP_415_UNSUPPORTED_MEDIA_TYPE

app = FastAPI()

@app.post("/endpoint")
async def endpoint(content_type: str = Header(...)):
    if content_type != "application/json":
        raise HTTPException(
            status_code=HTTP_415_UNSUPPORTED_MEDIA_TYPE,
            detail="Unsupported media type",
        )
    # 处理请求
    return {"message": "Success"}

在上述示例中，我们使用Header参数来获取请求头中的Content-Type，并进行判断。如果内容类型不是application/json，则抛出HTTPException异常，返回415 Unsupported Media Type错误。

这样，我们就成功限制了FastAPI请求头中的内容类型为application/json。

腾讯云相关产品和产品介绍链接地址：