接口自动化，sso单点登陆不支持通过接口返回token，其他请求又需要token。解决方式，使用UI自动化的selenium框架登陆获取token

import time
from selenium import  webdriver
from cfg import LoginUrl,usename,password
class GetToken():
    def __init__(self):
        self.driver=webdriver.Chrome()
    def loginUI(self):
        self.driver.get(LoginUrl)
        self.driver.maximize_window()
        self.driver.implicitly_wait(10)
        # 登录输入用户名，密码
        self.driver.find_element_by_id(‘LoginForm_username‘).send_keys(usename)
        self.driver.find_element_by_id(‘LoginForm_password‘).send_keys(password)
        self.driver.find_element_by_id(‘loginSubmit‘).click()
        # 强制等待5秒，待session和token都成功返回并存到浏览器中
        time.sleep(5)

    # 获取token
    def get_token(self):
        # 是要从localStorage中获取还是要从sessionStorage中获取，具体看目标系统存到哪个中
        # window.sessionStorage和直接写sessionStorage是等效的
        # 一定要使用return，不然获取到的一直是None
        # get的Item不一定就叫token，得具体看目标系统把token存到哪个变量中
        token = self.driver.execute_script(‘return localStorage.getItem("token");‘)
        # print(‘token:‘,token)
        return token

    #关闭浏览器
    def close_browser(self):
        self.driver.quit()

if __name__ == ‘__main__‘:
    gt = GetToken()
    gt.loginUI()
    ret_token=gt.get_token()
    print(‘获取到的token:‘,ret_token)
    gt.close_browser()