python – Selenium webdriver screenshot is different when using ‘–headless’

While taking a screenshot ideally you need to induce WebDriverWait for the visibility_of_element_located() of some static and visible element e.g. some <h1> / <h2> element on the desired page and then take the screenshot as follows:

WebDriverWait(driver, 20).until(EC.visibility_of_element_located((By.CSS_SELECTOR, "css_visible_element")))
driver.save_screenshot("Federico Albrieu.png")

Note : You have to add the following imports :

from import WebDriverWait
from import By
from import expected_conditions as EC

Read more here: Source link