在 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中文網其他相關文章!