selenium获取cookies并持久化登陆
需求背景:
这几天需要写一个接口,用来批量上传数据,最开始考虑的是 UI 自动化,然后选值的时候自动化难以判别,最终选择 接口 自动化。
然后操蛋的找了两天没找着该环境的登录接口,只能选择最笨的办法,selenium 模拟登陆,获取到cookie然后为 requests 所用。
我的思路是这样,听起来好像没啥难点,但还是卡了大晚上。
主要遇到的问题是,我 selenium 模拟登陆是正常的,但到了requests cookie就是不对,左检查右检查,终于找到,犯了一个致命的错误:
那我正常登陆后跳转到的是 index 页面,我从这里取cookies,那我取到的是 index 的cookies,虽然理论上他是一样的,但我也想不明白为啥就是不对
最后的解决方案也很简单,在selenium 中在进行一次跳转,跳转到目标页面,然后再获取 cookies进行持久化操作,很幸运出来了。
逻辑代码大概如下:
options = webdriver.ChromeOptions()
# 忽略SSL验证
options.add_argument('--ignore-certificate-errors') # 忽略证书错误
options.add_argument('headless') # 隐藏窗口
#options.add_argument("--disable-gpu") # 规避BUG
driver = webdriver.Chrome(chrome_options=options)
# driver.implicitly_wait(30)
driver.get(u'https://tfssc.ynby.cn/CAS_Server/login?service=https://tfssc.ynby.cn/portal-web/login/casSuccess##')
driver.delete_all_cookies()
driver.find_element_by_id('username').send_keys('user')
driver.find_element_by_id('password').send_keys('pwd')
driver.find_element_by_name('submit').click()
driver.get('https:xxxxx')# 获取请求头
agent = driver.execute_script("return navigator.userAgent")
# 创建一个requests.session对象
session = requests.Session()
print(agent,type(agent))
# 获取登录cookies
saveCookies = driver.get_cookies()
c = saveCookies[0]['value']
print(saveCookies)
# 将cookies设置到session中
for cookie in saveCookies:
session.cookies.set(cookie['name'],cookie['value'])
print(session.cookies.values())
# cookies_value = saveCookies['name']
# 目标地址
url = "https:xxxxx"
driver.quit()
payload={'bizCategoryId': '12874',
'orgIds': ''}
headers = {
'Accept': 'application/json, text/javascript, */*; q=0.01',
'User-Agent': agent ,
'Content-Type':'application/x-www-form-urlencoded; charset=UTF-8',
#'Cookie': 'JSESSIONID='+c,
'Host': 'tfssc.ynby.cn'
}
print(headers)
response = session.request('POST', url, headers=headers, data=payload)
又过了一坑。
鬼刀镇楼