In this Python Selenium article we want to learn about Working with Cookies in Python Selenium, In web development and testing, cookies play an important role in storing and retrieving information about a user browsing activities. Python Selenium is a powerful tool for browser automation, and it offers excellent support for handling cookies.
To work the examples of this tutorial you need some requirements, first you should have installed Python in your system, then we need to install Python Selenium and you can use pip for that like this.
|
1 |
pip install selenium |
First of all we need to import our required module from Python Selenium.
|
1 |
from selenium import webdriver |
After that we need to launch a browser instance. In this example, we are going to use Google Chrome. instantiate a webdriver.Chrome object, which represents the browser window.
|
1 |
driver = webdriver.Chrome() |
To demonstrate cookie handling, let’s open Google.com and search for a term. we are going to use the get() method of the webdriver object to navigate to the desired URL.
|
1 |
driver.get("https://www.google.com") |
To retrieve the cookies from the current browser session, we can use the get_cookies() method. it returns a list of dictionaries, where each dictionary represents a cookie.
|
1 2 |
cookies = driver.get_cookies() print(cookies) |
For adding a new cookie, we use add_cookie() method, which accepts a dictionary containing the cookie details. In this example, let’s add a custom cookie with the name example_cookie and value 12345.
|
1 2 |
new_cookie = {'name': 'example_cookie', 'value': '12345'} driver.add_cookie(new_cookie) |
After adding the new cookie, we can verify its presence by retrieving the cookies again and checking if the added cookie is present.
|
1 2 3 4 5 6 7 8 |
cookies = driver.get_cookies() for cookie in cookies: if cookie['name'] == 'example_cookie': print("Added cookie found") break else: print("Added cookie not found") |
If you want to remove a specific cookie, you can use delete_cookie() method and provide the cookie name.
|
1 |
driver.delete_cookie('example_cookie') |
After that you have finished working with the browser, remember to close it using the close() method.
|
1 |
driver.close() |
This is the complete code for this article
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
from selenium import webdriver # Launch the browser driver = webdriver.Chrome() # Open Google.com driver.get("https://www.google.com") # Obtain the current cookies cookies = driver.get_cookies() print(cookies) # Add a new cookie new_cookie = {'name': 'example_cookie', 'value': '12345'} driver.add_cookie(new_cookie) # Verify the added cookie cookies = driver.get_cookies() for cookie in cookies: if cookie['name'] == 'example_cookie': print("Added cookie found!") break else: print("Added cookie not found!") # Delete the cookie driver.delete_cookie('example_cookie') # Close the browser driver.close() |
Run the code and this will be the result
