在 Python 中，`requests.Session` 类允许用户在多个请求之间保持某些参数，例如 cookies、headers 等，并且可以显著提升发起多个请求时的性能。封装 `requests.Session` 类可以实现更高效的 HTTP 请求管理，尤其适用于需要频繁访问相同服务的场景。以下是一个封装 `requests.Session` 的示例。 ### 封装 Session 类的基本结构 ```python import requests class HttpSessionManager: def __init__(self, base_url=None, headers=None, timeout=10): self.session = requests.Session() self.base_url = base_url self.timeout = timeout # 设置默认请求头 if headers: self.session.headers.update(headers) def get(self, url, params=None): full_url = self._construct_url(url) try: response = self.session.get(full_url, params=params, timeout=self.timeout) response.raise_for_status() return response.json() except requests.exceptions.RequestException as e: print(f"GET request failed: {e}") return None def post(self, url, data=None, json=None): full_url = self._construct_url(url) try: response = self.session.post(full_url, data=data, json=json, timeout=self.timeout) response.raise_for_status() return response.json() except requests.exceptions.RequestException as e: print(f"POST request failed: {e}") return None def put(self, url, data=None): full_url = self._construct_url(url) try: response = self.session.put(full_url, data=data, timeout=self.timeout) response.raise_for_status() return response.json() except requests.exceptions.RequestException as e: print(f"PUT request failed: {e}") return None def delete(self, url): full_url = self._construct_url(url) try: response = self.session.delete(full_url, timeout=self.timeout) response.raise_for_status() return response.json() except requests.exceptions.RequestException as e: print(f"DELETE request failed: {e}") return None def _construct_url(self, url): if self.base_url: return f"{self.base_url}/{url}" return url def close(self): self.session.close() ``` ### 使用封装的 Session 类 ```python # 示例：使用封装的 HttpSessionManager if __name__ == "__main__": base_url = "https://httpbin.org" headers = { "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/111.0.0.0 Safari/537.36" } http_client = HttpSessionManager(base_url=base_url, headers=headers) # GET 请求 get_response = http_client.get("get", params={"test": "test01"}) print("GET Response:", get_response) # POST 请求 post_response = http_client.post("post", data={"key": "value"}) print("POST Response:", post_response) # 关闭会话 http_client.close() ``` ### 优势与优化 1. **保持连接**：通过 `requests.Session`，可以复用底层的 TCP 连接，减少连接建立的开销，从而提升性能。 2. **统一管理请求头**：可以在初始化时设置默认的请求头，避免每次请求都需要重复设置。 3. **异常处理**：统一处理请求异常，确保程序的健壮性。 4. **超时机制**：为所有请求设置统一的超时时间，避免请求长时间阻塞。 5. **URL 构造**：支持设置基础 URL，简化每次请求的 URL 构造过程。 ### 扩展功能 - **代理支持**：可以在 `Session` 中添加代理支持，以应对反爬虫机制[^3]。 - **日志记录**：可以集成日志模块，记录每次请求的详细信息。 - **身份验证**：支持在 `Session` 中添加身份验证逻辑，例如 OAuth 或 Token 认证。 - **重试机制**：可以结合 `retrying` 或 `tenacity` 库实现请求失败的自动重试。 ###