如何在 Selenium Python 中保存和加载 Cookie 以进行会话管理？

在 Python Selenium 中存储和检索 Cookie

在 Web 自动化场景中，有效处理 Cookie 变得至关重要。保存和加载 cookie 使我们能够以登录用户的身份维护用户会话并与 Web 应用程序进行交互。本文介绍如何在 Python 中使用 Selenium WebDriver 将 Cookie 保存到文本文件，并在会话恢复期间加载它们。

将 Cookie 保存到文本文件

到存储 Selenium WebDriver 会话期间生成的所有 cookie，您可以使用 pickle 模块。下面是一个示例：

import pickle
import selenium.webdriver

driver = selenium.webdriver.Firefox()
driver.get("http://www.google.com")
pickle.dump(driver.get_cookies(), open("cookies.txt", "wb"))

此代码使用 pickle 模块将当前 cookie 保存在 Python 对象中，并将其存储在名为“cookies.txt”的文本文件中。

从文本文件加载 Cookie

检索保存的 Cookie 并将其添加到后续的 Selenium WebDriver会话中，您可以使用以下步骤：

import pickle
import selenium.webdriver

driver = selenium.webdriver.Firefox()
driver.get("http://www.google.com")
cookies = pickle.load(open("cookies.txt", "rb"))

for cookie in cookies:
    driver.add_cookie(cookie)

在此代码中，我们从文本文件加载保存的 cookie，对于每个 cookie，我们使用 add_cookie 方法将其添加到 Selenium WebDriver 实例。这使我们能够使用与前一个会话相同的 cookie 来恢复会话。

以上是如何在 Selenium Python 中保存和加载 Cookie 以进行会话管理？的详细内容。更多信息请关注PHP中文网其他相关文章！