【测试】Selenium操作Cookie

news2024/12/22 22:36:32

1.操作Cookie

操作Cookie无非就是读取和删除Cookie，下面这些是主要方法

from selenium.webdriver import Chrome, ChromeOptions
from selenium.webdriver.chrome.service import Service

options = ChromeOptions()
options.add_argument("--ignore-certificate-errors")
s = Service(executable_path=r"E:\webdriver\chromedriver.exe")
driver = Chrome(service=s, options=options)
driver.maximize_window()
driver.get("https://www.baidu.com/")
driver.get_cookies() # 获取所有的cookie
driver.get_cookie(name) # 获取指定的 cookie
driver.delete_cookie(name) # 删除指定的 cookie
driver.delete_all_cookies() # 删除所有的 cookie
driver.add_cookie(cookie_dict) # 添加一个 cookie

2.设置单个Cookie

通过add_cookie()方法添加Cookie，参数是一个字典，一个Cookie字典的键有name、value、domain、httpOnly、path、sameSite、secure等，但是至少，必须要有name和value这两个键，比如说我们设置一下百度的Cookie

首先在浏览器开发者工具找到百度的关键Cookie，BDUSS，把它的name和value复制下来
在这里插入图片描述粘贴带Cookie字典里，如下

driver = Chrome(service=s, options=options)
driver.maximize_window()
driver.get("https://www.baidu.com/")
driver.add_cookie({"name": "BDUSS", "value": "5MbEN....AAAAAAAAAAAAAAAAAAJbLdWSWy3VkVn"})
driver.refresh()

注意，在调用add_cookie()方法之前，至少要先打开一个网页，不然会报错的，添加好Cookie之后再refresh一下，这样就处于登录状态了

3.保存和读取Cookie

上面是通过一个关键的BDUSS Cookie让我们处于登录状态，既然已经处于登录状态了，我们可以趁机把该页面的全部Cookie都保存为一个JSON文件，下次可以读取这个JSON文件

driver.get("https://www.baidu.com/")

# 保存Cookie
# cookies_str = json.dumps(driver.get_cookies())
# with open("baidu_cookies.json", "w", encoding="utf-8") as f:
#     f.write(cookies_str)

# 读取之前保存的Cookie
with open("baidu_cookies.json", "r", encoding="utf-8") as f:
    cookies_list = json.loads(f.read())
    for cookie in cookies_list:
        driver.add_cookie(cookie)

driver.refresh()
time.sleep(1)

4.加载用户文件夹

直接通过selenium打开浏览器，相当于打开了一个无痕模式，它不会加载任何Cookie或其他平时使用Chrome的一些设置。如果你嫌每次都手动设置Cookie很麻烦，希望它能像你平时打开浏览器一样使用自动保存的Cookie，我们可以通过加载本地Chrome浏览器的配置文件夹的方式打开Chrome

那如何知道自己的配置文件夹的路径呢？你可以在浏览器地址栏打开chrome://version/，看到的那个“个人资料路径”就是了
Chrome浏览器个人资料路径但是注意，我们在使用的时候，只用到"User Data"这一级就可以了，后面的删掉，例如我这里就是C:\Users\admin\AppData\Local\Google\Chrome\User Data

options = ChromeOptions()
options.add_argument("--no-sandbox")
options.add_argument(r'--user-data-dir=C:\Users\admin\AppData\Local\Google\Chrome\User Data')
s = Service(executable_path=r"E:\webdriver\chromedriver.exe")
driver = Chrome(service=s, options=options)
driver.get("https://www.baidu.com")
print("cookies:", json.dumps(driver.get_cookies()))