8、打造个性化 REST API 响应与项目架构优化

打造个性化 REST API 响应与项目架构优化

在开发 REST API 时，定制响应是一项至关重要的技能，它能够让我们根据不同的业务需求返回合适的信息。同时，合理地组织项目结构也能提高代码的可维护性和可扩展性。下面将详细介绍如何定制响应以及如何使用多个路由器来构建更大的项目。

定制响应

在 HTTP 响应中，除了响应体和状态码，还有其他一些重要的部分，如自定义头部和 Cookie。FastAPI 允许我们在路径操作逻辑中动态地设置这些内容。

1. 设置自定义头部

要设置自定义头部，我们可以将 Response 对象作为路径操作函数的参数注入。以下是一个示例代码：

from fastapi import FastAPI, Response
app = FastAPI()

@app.get("/")
async def custom_header(response: Response):
    response.headers["Custom-Header"] = "Custom-Header-Value"
    return {"hello": "world"}

在这个例子中，我们通过 response.headers 字典设置了一个自定义头部。需要注意的是，我们不需要返回 Response 对象，FastAPI 会自动处理响应的构建，包括我们设置的头部。

2. 设置 Cookie

当需要在浏览器中