selenium获取cookies并持久化登陆

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)

​ 又过了一坑。

鬼刀镇楼