一、背景說明

之前寫了一款簡單的api模糊測試工具，之前系統(tǒng)可以使用http Base認證現(xiàn)在改成session形式并加上了token。

最簡單的改造方法，是自己先在瀏覽器手動登錄，然后提取出session和token（系統(tǒng)token在整個會話期間可重復使用）填到模糊測試工具中即可。但這種非全自動化的方式不到萬不得已不想用。

最直接的方法，最使用requests按登錄流程依次發(fā)包登錄即可。但其中的難點是密碼是加密提交的，詢問開發(fā)人員說是DES加密；DES加密不難，但是用不同的語言編寫的加密算法與別人的結(jié)果完全一致那就比較費工夫。

最后的方法，那就是使用selenium登錄避開密碼的構(gòu)造，這種方法的關(guān)鍵點在于：selenium是基于圖形界面操作的，沒有直接的辦法能獲取request和response的數(shù)據(jù)包，在這種情況下如何獲取session和token。

自己動web前端的存儲并沒有很深入了解，一直想的是如何獲取selenium request和response的數(shù)據(jù)包從中提取session和token，直到看到這篇文章才想起前端的變量（尤其是restful中的全局變量）會放存localStorage和sessionStorage中，從中提取session和token即可。

二、環(huán)境搭建

2.1 下載驅(qū)動

下載驅(qū)動文件，放到后續(xù)python文件同級目錄下或加入到環(huán)境變量中即可。其中Chrome驅(qū)動要注意Driver版本與瀏覽器版本對應要求。

Firefox驅(qū)動下載地址：https://github.com/mozilla/geckodriver/releases

Chrome驅(qū)動下載地址：http://chromedriver.chromium.org/downloads

Safari驅(qū)動下載地址：https://webkit.org/blog/6900/webdriver-support-in-safari-10/

Edge驅(qū)動下載地址：https://developer.microsoft.com/en-us/microsoft-edge/tools/webdriver/

2.2 安裝selenium

pip install selenium

三、程序?qū)崿F(xiàn)

3.1 系統(tǒng)狀況

手動登錄系統(tǒng)后，查看存儲狀況如下圖所示，sessionid和token分別存儲在sessionStorage的sessionId和token兩個變量中

3.2 實現(xiàn)代碼

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()

  # 登錄系統(tǒng)，具體到自己系統(tǒng)時需要自行修改
  def login_system(self):
    # 登錄用戶名密碼，改成目標系統(tǒng)用戶名密碼
    username = "admin"
    password = "123456"
    # 登錄頁面url，改成目標系統(tǒng)登錄頁面
    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中獲取，具體看目標系統(tǒng)存到哪個中
    # window.sessionStorage和直接寫sessionStorage是等效的
    # 一定要使用return，不然獲取到的一直是None
    # get的Item不一定就叫sessionId，得具體看目標系統(tǒng)把sessionid存到哪個變量中
    sessionid = self.browser.execute_script('return sessionStorage.getItem("sessionId");')

    # 另外sessionid一般都直接通過返回Set-Cookies頭設(shè)置到Cookie中，所以也可以從Cookie讀取
    # 獲取瀏覽器所有Set-Cookie，返回對象是字典列表
    # cookies = self.browser.get_cookies()
    # 獲取單項Cookie，是不是叫sessionId取決于系統(tǒng)存成什么變量，單項Cookie是字典
    # cookie = self.browser.get_cookie("sessionId")
    # cookie = cookie["value"]
    # print(f"{cookies}")
    return sessionid

  # 獲取token
  def get_token(self):
    # 是要從localStorage中獲取還是要從sessionStorage中獲取，具體看目標系統(tǒng)存到哪個中
    # window.sessionStorage和直接寫sessionStorage是等效的
    # 一定要使用return，不然獲取到的一直是None
    # get的Item不一定就叫token，得具體看目標系統(tǒng)把token存到哪個變量中
    token = self.browser.execute_script('return sessionStorage.getItem("token");')
    # print(f"{token}")
    return token

  def __del__(self):
    # 退出程序時關(guān)閉瀏覽器
    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 實現(xiàn)效果

更多關(guān)于Python獲取session和token方法請查看下面的相關(guān)鏈接

最新評論