82 lines
2.4 KiB
Python
82 lines
2.4 KiB
Python
from fastapi.responses import Response
|
||
from fastapi.requests import Request
|
||
import logging
|
||
|
||
|
||
class CookieUtil:
|
||
@staticmethod
|
||
def set_cookie(res: Response, key: str, value: str, max_age: int = 3600, path: str = "/", secure: bool = False,
|
||
httponly: bool = True):
|
||
"""
|
||
设置cookie
|
||
|
||
:param res: FastAPI的Response对象
|
||
:param key: cookie的键
|
||
:param value: cookie的值
|
||
:param max_age: cookie的有效时间（秒），默认1小时（3600秒）
|
||
:param path: cookie的路径，默认为根路径("/")
|
||
:param secure: 是否仅通过HTTPS传输，默认False
|
||
:param httponly: 是否仅通过HTTP请求访问，默认True
|
||
"""
|
||
res.set_cookie(
|
||
key=key,
|
||
value=value,
|
||
httponly=httponly,
|
||
secure=secure,
|
||
max_age=max_age,
|
||
path=path
|
||
)
|
||
logging.info(f"设置 cookie: {key}={value}")
|
||
|
||
@staticmethod
|
||
def get_cookie(req: Request, key: str) -> str:
|
||
"""
|
||
从请求中获取cookie的值
|
||
|
||
:param req: FastAPI的Request对象
|
||
:param key: cookie的键
|
||
:return: cookie的值，如果未找到则返回None
|
||
"""
|
||
token_value = req.cookies.get(key)
|
||
logging.info(f"从 cookie 中获取的 {key}: {token_value}")
|
||
return token_value
|
||
|
||
@staticmethod
|
||
def remove_cookie(res: Response, key: str, path: str = "/"):
|
||
"""
|
||
移除cookie
|
||
|
||
:param res: FastAPI的Response对象
|
||
:param key: cookie的键
|
||
:param path: cookie的路径，默认为根路径("/")
|
||
"""
|
||
res.delete_cookie(key, path=path)
|
||
logging.info(f"移除 cookie: {key}")
|
||
|
||
|
||
# 示例使用
|
||
if __name__ == "__main__":
|
||
# 创建一个示例响应对象
|
||
response = Response(content="示例响应")
|
||
|
||
# 设置一个cookie
|
||
CookieUtil.set_cookie(response, key="auth_token", value="your_token_value")
|
||
|
||
# 创建一个示例请求对象（通常从FastAPI请求中获取）
|
||
request = Request(scope={
|
||
"type": "http",
|
||
"headers": [
|
||
(b"cookie", b"auth_token=your_token_value")
|
||
]
|
||
})
|
||
|
||
# 获取cookie
|
||
token = CookieUtil.get_cookie(request, key="auth_token")
|
||
print(token)
|
||
|
||
# 移除cookie
|
||
CookieUtil.remove_cookie(response, key="auth_token")
|
||
|
||
# 打印示例响应头
|
||
print(response.headers)
|