How Can I Save and Load Cookies in Selenium Python for Session Management?

Storing and Retrieving Cookies in Python Selenium

In web automation scenarios, it becomes essential to handle cookies effectively. Saving and loading cookies allow us to maintain user sessions and interact with web applications as a logged-in user. This article explains how to utilize Selenium WebDriver in Python to save cookies to a text file and load them later during session resumption.

Saving Cookies to a Text File

To store all cookies generated during a Selenium WebDriver session, you can employ the pickle module. Here's an example:

import pickle
import selenium.webdriver

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

This code saves the current cookies in a Python object using the pickle module and stores it in a text file named "cookies.txt."

Loading Cookies from the Text File

To retrieve the saved cookies and add them to a subsequent Selenium WebDriver session, you can use the following steps:

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)

In this code, we load the saved cookies from the text file, and for each cookie, we add it to the Selenium WebDriver instance using the add_cookie method. This enables us to resume the session with the same cookies as the previous session.

The above is the detailed content of How Can I Save and Load Cookies in Selenium Python for Session Management?. For more information, please follow other related articles on the PHP Chinese website!