python使用FastAPI获取请求头信息的两种方法

http请求头

HTTP 请求头是 HTTP 请求的重要组成部分，它包含了关于请求的各种元信息，以键值对的形式存在，为服务器处理请求提供额外信息。下面从作用、常见请求头、示例等方面进行简要解释。

1. 作用：它可以告知服务器关于客户端的一些信息，比如客户端类型、支持的内容类型、缓存策略等，让服务器能根据这些信息来正确处理请求，返回合适的响应。例如，通过请求头中的User - Agent，服务器能识别客户端是浏览器、手机应用还是其他工具，从而返回适配的内容。

2. 常见请求头

   • User-Agent：包含发出请求的客户端的相关信息，如浏览器类型、版本、操作系统等。例如Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36 ，就表明这是来自 Windows 10 系统、使用 Chrome 浏览器的请求。
   • Content-Type：用于说明请求体的媒体类型，常见的有application/json 表示请求体是 JSON 格式数据，application/x-www-form-urlencoded表示表单数据 。
   • Accept：告诉服务器客户端能够接受的响应内容类型，如Accept: text/html表示客户端期望接收 HTML 格式的响应。

1. 示例：在一个简单的登录请求中，可能会有Content-Type: application/json，表明请求体中是 JSON 格式的数据，里面可能包含用户名和密码等信息；User-Agent表明客户端的相关信息，服务器根据这些信息来进行相应的处理，如验证登录信息、返回合适格式的响应。

在 FastAPI 中，获取 HTTP 请求头可以帮助开发者更好地理解请求的来源和特性，从而实现更灵活和个性化的业务逻辑，比如根据User - Agent进行不同的页面渲染，根据Content - Type正确解析请求体等。

在 FastAPI 中，有多种方式可以获取请求头信息，下面为你详细介绍两种常见的方法，并结合示例代码进行说明。

方法一：使用 Request 对象

Request 对象是 FastAPI 中代表 HTTP 请求的对象，通过它可以访问请求的各个部分，包括请求头。你可以在路由处理函数中注入 Request 对象，然后使用 headers 属性来获取请求头信息。

示例代码

from fastapi import FastAPI, Request

app = FastAPI()

@app.get("/get_headers_with_request")
async def get_headers_with_request(request: Request):
    # 获取所有请求头信息
    all_headers = dict(request.headers)
    # 获取特定请求头信息，例如 'User-Agent'
    user_agent = request.headers.get('User-Agent')

    return {
        "all_headers": all_headers,
        "User-Agent": user_agent
    }

代码解释

• 导入必要的模块：导入 FastAPI 和 Request。

• 创建 FastAPI 应用实例：app = FastAPI()。

• 定义路由处理函数

1. • 在函数参数中注入 Request 对象。
   • 使用 request.headers 获取所有请求头信息，并将其转换为字典形式。
   • 使用 request.headers.get('User - Agent') 获取特定的请求头信息。

2. 返回结果：将所有请求头信息和特定请求头信息以 JSON 格式返回。

方法二：使用参数依赖注入

FastAPI 支持直接在路由处理函数的参数中声明请求头参数，通过指定参数名和类型，FastAPI 会自动从请求头中提取相应的值。

示例代码

from fastapi import FastAPI, Header

app = FastAPI()

@app.get("/get_headers_with_header")
async def get_headers_with_header(user_agent: str = Header(None), custom_header: str = Header(None)):
    return {
        "User-Agent": user_agent,
        "Custom-Header": custom_header
    }

代码解释

1. 导入必要的模块：导入 FastAPI 和 Header。
2. 创建 FastAPI 应用实例：app = FastAPI()。
3. 定义路由处理函数
   • 在函数参数中使用 Header 依赖注入请求头参数。
   • user_agent: str = Header(None) 表示从请求头中提取 User - Agent 字段的值，并将其赋值给 user_agent 变量。如果请求头中不存在该字段，则默认值为 None。
   • 同样，custom_header: str = Header(None) 用于提取自定义请求头 Custom - Header 的值。
4. 返回结果：将提取的请求头信息以 JSON 格式返回。

运行示例

将上述代码保存为 main.py，然后在终端中运行以下命令启动 FastAPI 应用：

uvicorn main:app --reload

启动成功后，你可以使用工具（如 curl 或 Postman）向对应的路由发送请求，查看返回的请求头信息。例如，使用 curl 发送请求：

curl -H "User-Agent: MyCustomUserAgent" -H "Custom-Header: MyCustomValue" http://127.0.0.1:8000/get_headers_with_header

这样你就可以看到请求头信息被正确提取并返回。

最后总结

在 FastAPI 开发中，获取请求头信息至关重要。Request对象方法提供全面灵活的操作，可对所有请求头进行整体处理和分析，适用于需要全面了解请求头情况的场景。参数依赖注入法则简洁直观，代码更易读，适合明确知道需要获取哪些特定请求头的情况。开发者可依据具体需求选择合适方法，通过合理运用这两种方式，能高效处理 HTTP 请求头信息，提升 FastAPI 应用的开发效率和功能性。

以上就是python使用FastAPI获取请求头信息的两种方法的详细内容，更多关于python FastAPI获取信息的资料请关注脚本之家其它相关文章！

最新评论