Selenium中使用Cookies绕过登录

提示：扫一扫查出行
【扫一扫了解最新限行尾号】
复制提示

在使用selenium测试后台时常常每个流程都需要走登录流程，这样自然比较浪费时间。如果遇到登录需要输入验证码等情况，就可能出师未捷身先死。

在Web应用中，登录状态通常是通过Cookie中对应的session id验证的。也是就是说，我们只要携带上登录后的Cookies，浏览器就会自动识别我们为登录状态。由于Selenium每次启动的浏览器是一个隔离的环境，不能直接使用本地已经保存的Cookies值，我们需要通过driver的add_cookie()方法手动添加Cookies,来绕过登录。

手工抓取Cookie并绕过登录

如图，先手工登录后，通过开发者工具-网络面板，查看任意请求中的Cookie值，经尝试分析，ECSCP_ID的值为session id，我们只需要添加此条Cookie即可。

Cookie的存储格式，如下图所示，每一条包含name,value,path,secure等字段

WebDriver相关的Cookies操作

get_cookies(self): 获取当前会话中当前域名所有cookies
get_cookie(self, name): 获取当前会话中当前域名指定name对应的cookie值
delete_cookie(self, name): 删除指定cookie
delete_all_cookies(self): 删除所有cookie
add_cookie(self, cookie_dict): 添加cookie

添加Cookie时需要逐条添加，cookie_dict示例：

driver.add_cookie({'name' : 'foo', 'value' : 'bar'})
driver.add_cookie({'name' : 'foo', 'value' : 'bar', 'path' : '/'})
driver.add_cookie({'name' : 'foo', 'value' : 'bar', 'path' : '/', 'secure':True})"

示例代码：

from selenium import webdriver from time import sleepdriver = webdriver.Chrome()driver.get('http://39.104.14.232/ecshop/wwwroot/admin/privilege.php?act=login') driver.add_cookie(dict(name='ECSCP_ID',value='7406d5a6f0c5bdfbebcc29b4e41cea9731d9e9f0')) driver.get('http://39.104.14.232/ecshop/wwwroot/admin/index.php')sleep(3) driver.quit()

注意：必须先打开一个页面（同域名页面）才能设置cookie，否则会报UnableToSetCookieException

使用Requests配合Selenium绕过登录

手工抓取的Cookie有一定的有效期，我们可以通过发送登录接口，从响应中拿到Cookies再放入Selenium中使用。

示例代码：

import requests from selenium import webdriver from time import sleepdata = dict(username='****', password='****', act='signin') res = requests.post('http://39.104.14.232/ecshop/wwwroot/admin/privilege.php', data=data, allow_redirects=False)esscp_id = res.cookies.get('ECSCP_ID')driver = webdriver.Chrome() driver.get('http://39.104.14.232/ecshop/wwwroot/admin/privilege.php?act=login')driver.add_cookie(dict(name='ECSCP_ID',value=esscp_id)) driver.get('http://39.104.14.232/ecshop/wwwroot/admin/index.php')sleep(3) driver.quit()

同样，如果有多条Cookies,需要从请求响应中逐个提取，在Selenium中逐个添加。