FastAPI 如何检查 FastAPI 中是否设置了cookie

在本文中，我们将介绍在 FastAPI 中如何检查是否设置了cookie，并提供一些示例说明。

什么是Cookie？

Cookie是一个存储在用户计算机中的小文件，用于存储网站的一些用户信息。通过HTTP协议，服务器可以将Cookie发送给用户的浏览器，并在浏览器中进行存储。当用户再次访问相同网站时，浏览器会将该Cookie发送给服务器，以便服务器可以读取其中的信息。

在FastAPI中，我们可以使用Request类的cookies属性来检查是否设置了Cookie。

如何检查Cookie是否设置？

要检查FastAPI中是否设置了Cookie，我们需要使用Request类的cookies属性，并根据需要检查特定的Cookie键是否存在。

以下是一段示例代码，展示了如何使用FastAPI检查Cookie是否设置：

from fastapi import FastAPI, Request

app = FastAPI()

@app.get("/check-cookie")
def check_cookie(request: Request):
    if "my_cookie" in request.cookies:
        return {"message": "Cookie is set"}
    else:
        return {"message": "Cookie is not set"}

上述代码定义了一个GET请求的路由/check-cookie，当访问该路由时，它将检查名为my_cookie的Cookie是否设置。如果设置了该Cookie，则返回JSON响应{"message": "Cookie is set"}，否则返回{"message": "Cookie is not set"}。

在上述示例中，我们首先导入了FastAPI和Request类。然后，我们定义了一个名为check_cookie的路由，该路由接受一个Request对象作为参数。在路由处理函数中，我们使用"my_cookie" in request.cookies的语法来检查是否设置了名为my_cookie的Cookie。

如何设置Cookie以进行测试？

为了测试上述的代码，我们可以使用Response类的set_cookie方法来设置Cookie。在下面的示例中，我们使用FastAPI的TestClient来模拟请求和测试：

from fastapi.testclient import TestClient

client = TestClient(app)

def test_check_cookie():
    response = client.get("/check-cookie")
    assert response.status_code == 200
    assert response.json() == {"message": "Cookie is not set"}

    response = client.get("/set-cookie")
    assert response.status_code == 200
    assert response.json() == {"message": "Cookie is set"}

    response = client.get("/check-cookie")
    assert response.status_code == 200
    assert response.json() == {"message": "Cookie is set"}

上述代码中，我们先发送一个GET请求到/check-cookie路由，来检查是否设置了Cookie。由于我们还没有设置Cookie，所以预期的响应应该是{"message": "Cookie is not set"}。

然后，我们发送一个GET请求到/set-cookie路由，来设置名为my_cookie的Cookie。设置成功后，预期的响应应该是{"message": "Cookie is set"}。

最后，我们再次发送一个GET请求到/check-cookie路由，来确认是否设置了Cookie。由于我们已经设置了my_cookie，所以预期的响应应该是{"message": "Cookie is set"}。