4.15 操作cookie · Python / Linux / Mysql / 軟件測試等等

[TOC] ## **4.15 操作 Cookie** 有時我們需要驗證瀏覽器中的 Cookie 是否正確，因為基于真實的 Cookie 是無法通過白盒測試和集成測試的。WebDriver 提供了操作 Cookie 的相關方法，可以讀取、添加和刪除 Cookie。 WebDriver 操作 Cookie 的方法如下。 * get_cookies()：獲得所有 Cookie。 * get_cookie(name)：返回字典中 key 為“name”的 Cookie。 * add_cookie(cookie_dict)：添加 Cookie。 * delete_cookie(name,optionsString)：刪除名為 OpenString 的 Cookie。 * delete_all_cookies()：刪除所有 Cookie。下面通過 get_cookies()獲取當前瀏覽器的所有 Cookie。 ``` from selenium import webdriver driver = webdriver.Chrome() driver.get("https://www.baidu.com") # 獲得所有cookie信息并打印 cookie = driver.get_cookies() print(cookie) ``` 執行結果如下。 ``` [{'domain': '.baidu.com', 'expiry': 1635838659, 'httpOnly': False, 'name': 'BA_HECTOR', 'path': '/', 'secure': False, 'value': 'asa02h2l2ha52k458f1go1n5j0r'}, {'domain': '.baidu.com', 'expiry': 1667371059, 'httpOnly': False, 'name': 'BAIDUID_BFESS', 'path': '/', 'sameSite': 'None', 'secure': True, 'value': '7E94D5A577F495FB11F3BDB10ED26D3B:FG=1'}, {'domain': '.baidu.com', 'httpOnly': False, 'name': 'H_PS_PSSID', 'path': '/', 'secure': False, 'value': '34836_34884_34448_34067_34712_34584_34518_34709_34916_34812_26350_34827_34868'}, {'domain': '.baidu.com', 'expiry': 1667371058, 'httpOnly': False, 'name': 'BAIDUID', 'path': '/', 'secure': False, 'value': '7E94D5A577F495FB11F3BDB10ED26D3B:FG=1'}, {'domain': '.baidu.com', 'expiry': 3783318705, 'httpOnly': False, 'name': 'BIDUPSID', 'path': '/', 'secure': False, 'value': '7E94D5A577F495FB591BC48C7FCE223E'}, {'domain': '.baidu.com', 'expiry': 3783318705, 'httpOnly': False, 'name': 'PSTM', 'path': '/', 'secure': False, 'value': '1635835060'}, {'domain': 'www.baidu.com', 'expiry': 1636699059, 'httpOnly': False, 'name': 'BD_UPN', 'path': '/', 'secure': False, 'value': '12314753'}, {'domain': 'www.baidu.com', 'httpOnly': False, 'name': 'BD_HOME', 'path': '/', 'secure': False, 'value': '1'}] ``` 從執行結果可以看出，Cookie 中的數據是以字典形式存放的。知道了 Cookie 中數據的存放形式后，即可按照這種形式向瀏覽器中添加 Cookie。 ``` # 添加cookie信息 driver.add_cookie({'name': 'key-aaaaaa', 'value': 'value-bbbbbb'}) # 遍歷指定的cookies for cookie in driver.get_cookies(): print("%s -> %s" % (cookie['name'], cookie['value'])) ``` 執行結果如下： ``` BA_HECTOR -> asa02h2l2ha52k458f1go1n5j0r BAIDUID_BFESS -> 7E94D5A577F495FB11F3BDB10ED26D3B:FG=1 H_PS_PSSID -> 34836_34884_34448_34067_34712_34584_34518_34709_34916_34812_26350_34827_34868 BAIDUID -> 7E94D5A577F495FB11F3BDB10ED26D3B:FG=1 key-aaaaaa -> value-bbbbbb BIDUPSID -> 7E94D5A577F495FB591BC48C7FCE223E PSTM -> 1635835060 BD_UPN -> 12314753 BD_HOME -> 1 ``` 從執行結果可以看出，最后一條 Cookie 是在腳本執行過程中通過 add_cookie()方法添加的。通過遍歷得到所有的 Cookie，從而找到字典中 key 為“name”和“value”的 Cookie值。 delete_cookie() 和 delete_all_cookies() 方法的使用也很簡單，前者通過 name 刪除一個指定的 Cookie，后者直接刪除瀏覽器中的所有 Cookies。