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 sleep

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='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 sleep


data = 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中逐个添加。

用户名、密码可微信联系lockingfree获取

猜你喜欢

转载自www.cnblogs.com/superhin/p/11481803.html