怎么在Python3中利用Selenium获取session和token
本篇文章给大家分享的是有关怎么在Python3中利用Selenium获取session和token,小编觉得挺实用的,因此分享给大家学习,希望大家阅读完这篇文章后可以有所收获,话不多说,跟着小编一起来看看吧。
环境搭建
下载驱动
下载驱动文件,放到后续python文件同级目录下或加入到环境变量中即可。其中Chrome驱动要注意Driver版本与浏览器版本对应要求。
Firefox驱动下载地址:https://github.com/mozilla/geckodriver/releases
Chrome驱动下载地址:http://chromedriver.chromium.org/downloads
Safari驱动下载地址:https://webkit.org/blog/6900/webdriver-support-in-safari-10/
Edge驱动下载地址:https://developer.microsoft.com/en-us/microsoft-edge/tools/webdriver/
2.2 安装selenium
pip install selenium
三、程序实现
3.1 系统状况
手动登录系统后,查看存储状况如下图所示,sessionid和token分别存储在sessionStorage的sessionId和token两个变量中
3.2 实现代码
import json import time from selenium import webdriver from selenium.webdriver.support.wait import WebDriverWait from selenium.webdriver.support import expected_conditions as EC from selenium.webdriver.common.by import By class GetSessionAndToken(): def __init__(self): # 启用无头模式,可选 browser_options = webdriver.FirefoxOptions() browser_options.add_argument('--headless') browser_options.add_argument('--disable-gpu') self.browser = webdriver.Firefox(firefox_options=browser_options) # self.browser = webdriver.Chrome() # 登录系统,具体到自己系统时需要自行修改 def login_system(self): # 登录用户名密码,改成目标系统用户名密码 username = "admin" password = "123456" # 登录页面url,改成目标系统登录页面 url = "https://10.10.6.93/#login" self.browser.get(url) # 显性等待,直到用户名控件加载出来才进行下一步 WebDriverWait(self.browser,20,0.5).until(EC.presence_of_element_located((By.ID,"txtUserName"))) # 填写用户名 self.browser.find_element_by_id("txtUserName").send_keys(username) # 填写密码 self.browser.find_element_by_id("txtPassword").send_keys(password) # 点击登录 self.browser.find_element_by_id("btnLogin").click() # 强制等待5秒,待session和token都成功返回并存到浏览器中 # restful隐性等待不太好用?self.browser.implicitly_wait(5) time.sleep(5) # 获取sessionid def get_sessionid(self): # 是要从localStorage中获取还是要从sessionStorage中获取,具体看目标系统存到哪个中 # window.sessionStorage和直接写sessionStorage是等效的 # 一定要使用return,不然获取到的一直是None # get的Item不一定就叫sessionId,得具体看目标系统把sessionid存到哪个变量中 sessionid = self.browser.execute_script('return sessionStorage.getItem("sessionId");') # 另外sessionid一般都直接通过返回Set-Cookies头设置到Cookie中,所以也可以从Cookie读取 # 获取浏览器所有Set-Cookie,返回对象是字典列表 # cookies = self.browser.get_cookies() # 获取单项Cookie,是不是叫sessionId取决于系统存成什么变量,单项Cookie是字典 # cookie = self.browser.get_cookie("sessionId") # cookie = cookie["value"] # print(f"{cookies}") return sessionid # 获取token def get_token(self): # 是要从localStorage中获取还是要从sessionStorage中获取,具体看目标系统存到哪个中 # window.sessionStorage和直接写sessionStorage是等效的 # 一定要使用return,不然获取到的一直是None # get的Item不一定就叫token,得具体看目标系统把token存到哪个变量中 token = self.browser.execute_script('return sessionStorage.getItem("token");') # print(f"{token}") return token def __del__(self): # 退出程序时关闭浏览器 self.browser.close() if __name__ == "__main__": obj = GetSessionAndToken() obj.login_system() sessionid = obj.get_sessionid() token = obj.get_token() print(f"sessionid为: {sessionid}\n" f"token为: {token}")
3.3 实现效果
以上就是怎么在Python3中利用Selenium获取session和token,小编相信有部分知识点可能是我们日常工作会见到或用到的。希望你能通过这篇文章学到更多知识。更多详情敬请关注行业资讯频道。