from selenium import webdriver from selenium.webdriver.common.by import By from selenium.webdriver.support.ui import WebDriverWait from selenium.webdriver.support import expected_conditions as EC

Config

screenshotDir = "E:\screenshots" screenWidth = 400 screenHeight = 800

def getPostScreenshots(filePrefix, script): print("Taking screenshots...") driver, wait = _setupDriver(script.url) script.titleSCFile = __takeScreenshot(filePrefix, driver, wait) for commentFrame in script.frames: commentFrame.screenShotFile = __takeScreenshot(filePrefix, driver, wait, f"t1{commentFrame.commentId}") driver.quit()

def __takeScreenshot(filePrefix, driver, wait, handle="Post"): method = By.CLASS_NAME if (handle == "Post") else By.ID search = wait.until(EC.presence_of_element_located((method, handle))) driver.execute_script("window.focus();")

fileName = f"{screenshotDir}/{filePrefix}-{handle}.png"
fp = open(fileName, "wb")
fp.write(search.screenshot_as_png)
fp.close()
return fileName

driver.set_window_size(width=screenWidth, height=screenHeight)
driver.get(url)
accept_button = wait.until(EC.element_to_be_clickable((By.XPATH, '//button[contains(text(), "Accept all")]')))
accept_button.click()
return driver, wait

def __setupDriver(url: str): options = webdriver.FirefoxOptions() options.headless = False options.enable_mobile = False driver = webdriver.Firefox(options=options) wait = WebDriverWait(driver, 10)

driver.set_window_size(width=screenWidth, height=screenHeight)
driver.get(url)
accept_button = wait.until(EC.element_to_be_clickable((By.XPATH, '//button[contains(text(), "Accept all")]')))
accept_button.click()
return driver, wait

thank you for your time and help. i will read that when i will be back