def login(): ''' Fonction permettant se récupérer le cookie de session après une connexion via selenium ''' firefoxOptions = FirefoxOptions() #firefoxOptions.addArguments("--window-size=1920,1080") #firefoxOptions.addArguments("--disable-gpu") #firefoxOptions.addArguments("--disable-extensions") #firefoxOptions.addArguments("--proxy-server='direct://'") #firefoxOptions.addArguments("--proxy-bypass-list=*") #firefoxOptions.addArguments("--start-maximized") firefoxOptions.addArguments("--headless") webDriver = FirefoxDriver(firefoxOptions) webDriver.get(URL_LOGIN) timeOut = 3 waiter = WebDriverWait(webDriver, timeOut) waiter.until(ExpectedConditions.visibilityOfElement(By.id("login-form"))) webDriver.findElement(By.name("email")).sendKeys("*****@*****.**") webDriver.findElement(By.name("pass")).sendKeys("123") webDriver.findElement(By.name("login-button")).click() waiter.until(ExpectedConditions.visibilityOfElement(By.id("welcome-page"))) GlobalVariables.setGlobalCustomVar( COOKIE, str(webDriver.manage().getCookieNamed(COOKIE_NAME).getValue()))
def clearAccessTokenFromGlobalVar(): GlobalVariables.setGlobalVar(ACCESS_TOKEN_KEY_NAME, None); GlobalVariables.setGlobalCustomVar(ACCESS_TOKEN_CREATION_TIMESTAMP_KEY_NAME, None); GlobalVariables.setGlobalCustomVar(ACCESS_TOKEN_EXPIRY_IN_SECONDS_KEY_NAME, None);
def setAccessTokenInGlobalVar(accessToken, expiryInSeconds): GlobalVariables.setGlobalVar(ACCESS_TOKEN_KEY_NAME, str(accessToken)); GlobalVariables.setGlobalCustomVar(ACCESS_TOKEN_CREATION_TIMESTAMP_KEY_NAME, time.time()); GlobalVariables.setGlobalCustomVar(ACCESS_TOKEN_EXPIRY_IN_SECONDS_KEY_NAME, expiryInSeconds);
import org.zaproxy.zap.extension.script.ScriptVars as GlobalVariables GlobalVariables.setGlobalCustomVar("COOKIE", None)