如何使用Python和Selenium高效保存和加载Cookie？

使用 Python 和 Selenium 保存和加载 Cookie

保存和检索 Cookie 可以显着增强您的 Web 应用程序测试能力，使您能够维护状态并模拟用户行为。然而，尽管它很重要，Selenium WebDriver 的文档提供的关于“getCookies”功能的见解很有限。让我们深入研究使用 Python 的 pickle 模块来简化 cookie 管理过程的综合解决方案。

将 Cookie 保存到文件

为了保存您当前的 cookie，我们将利用Python pickle 模块的强大功能：

import pickle<br>import selenium.webdriver<p>driver = selenium.webdriver.Firefox()<br>driver.get("http://www.google.com")<br>pickle.dump(driver.get_cookies(), open("cookies.pkl", "wb"))<br></p>

此代码片段将创建一个名为“cookies.pkl”的新文件，并在其中存储当前 cookie 的序列化版本。

恢复 Cookie

要检索和恢复您的 cookie，该过程同样简单：

导入 pickle<br>导入selenium.webdriver</p>
<p>driver = selenium.webdriver.Firefox()<br>driver.get("http://www.google.com")<br>cookies = pickle.load(open("cookies) .pkl", "rb"))<br>用于 cookie cookies:</p>
<div class="code" style="position:relative; padding:0px; margin:0px;"><pre class="brush:php;toolbar:false">driver.add_cookie(cookie)