### Python Selenium 实现避免重复登录的最佳实践 在使用 Python 和 Selenium 时，可以通过保存和加载会话信息或 Cookie 来避免重复登录。以下是实现这一功能的详细方法[^3]。 #### 方法一：通过保存和加载 Cookie 1. **首次登录并保存 Cookie** 在第一次访问目标网站时，完成手动或自动登录操作后，将当前浏览器的 Cookie 保存到文件中。例如： ```python import pickle from selenium import webdriver driver = webdriver.Chrome() driver.get("https://example.com/login") # 使用用户名和密码登录（假设已登录） username_input = driver.find_element('name', "username") password_input = driver.find_element('name', "password") username_input.send_keys("your_username") password_input.send_keys("your_password") password_input.send_keys(Keys.RETURN) # 登录成功后保存 Cookie 到文件 with open("cookies.pkl", "wb") as file: pickle.dump(driver.get_cookies(), file) ``` 2. **后续访问时加载 Cookie** 在后续访问时，先加载保存的 Cookie 文件，并将其应用到浏览器会话中。 ```python import pickle from selenium import webdriver driver = webdriver.Chrome() driver.get("https://example.com/login") # 需要先访问目标域名 # 加载 Cookie with open("cookies.pkl", "rb") as file: cookies = pickle.load(file) for cookie in cookies: driver.add_cookie(cookie) # 再次访问目标页面以应用 Cookie driver.get("https://example.com/xxx") ``` #### 方法二：通过 Session 保持会话状态 Selenium 还可以通过保持会话状态来避免重复登录。如果会话未被关闭，则可以在任何时候重新连接到浏览器实例[^1]。 ```python from selenium import webdriver # 第一次启动浏览器并登录 driver = webdriver.Chrome() driver.get("https://petstore.octoperf.com/actions/Catalog.action") # 手动或自动登录后，保持会话状态 # 后续脚本可以直接连接到现有会话 driver.get("https://petstore.octoperf.com/actions/Catalog.action") driver.find_element('link text', 'My Account').click() ``` #### 注意事项 - 确保在加载 Cookie 前访问目标域名，否则可能会遇到 `InvalidCookieDomainException` 错误[^3]。 - 如果目标网站的会话超时或 Cookie 失效，则需要重新登录并保存新的 Cookie。 - 驱动程序版本需与浏览器版本匹配，否则可能导致运行失败[^2]。 ### 示例代码 以下是一个完整的示例，展示如何通过保存和加载 Cookie 避免重复登录： ```python import pickle from selenium import webdriver from selenium.webdriver.common.keys import Keys # 第一次登录并保存 Cookie def save_cookies(): driver = webdriver.Chrome() driver.get("https://example.com/login") username_input = driver.find_element('name', "username") password_input = driver.find_element('name', "password") username_input.send_keys("your_username") password_input.send_keys("your_password") password_input.send_keys(Keys.RETURN) # 登录成功后保存 Cookie with open("cookies.pkl", "wb") as file: pickle.dump(driver.get_cookies(), file) driver.quit() # 加载 Cookie 并访问目标页面 def load_cookies(): driver = webdriver.Chrome() driver.get("https://example.com/login") # 必须先访问目标域名 with open("cookies.pkl", "rb") as file: cookies = pickle.load(file) for cookie in cookies: driver.add_cookie(cookie) # 再次访问目标页面以应用 Cookie driver.get("https://example.com/xxx") return driver # 调用示例 # save_cookies() # 第一次登录时调用 driver = load_cookies() # 后续访问时调用 ```