Disclaimer: Although there is a similar question already out there, none of the answers provided worked for a headless browser. Therefore, I have decided to create a more detailed version addressing this specific issue (the original question I referred to can be found here: Take screenshot of full page with Selenium Python with chromedriver)
Greetings to all.
I've come across a seemingly simple yet challenging problem. I am trying to capture a screenshot of a NON-HEADLESS BROWSER on a display with dimensions of 1920x1080 (this will be relevant later), capturing the entire webpage rather than just the visible portion.
Here's what I've attempted:
import os
import time
from selenium import webdriver
chrome_options = webdriver.ChromeOptions()
chrome_options.add_argument('--headless')
chrome_options.add_argument("--start-maximized")
chromedriver = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'chromedriver.exe')
chrome = webdriver.Chrome(chromedriver, options=chrome_options)
url = 'https://stackoverflow.com/'
chrome.get(url)
time.sleep(2)
total_height = chrome.execute_script("return document.body.parentNode.scrollHeight") + 1000
chrome.set_window_size(1920, total_height)
time.sleep(2)
chrome.save_screenshot("screenshot1.png")
chrome.quit()
The above code works perfectly fine with Headless mode. However, if I remove the --headless option, Selenium attempts to resize itself, but because it tries to exceed the 1080 height of the display, it automatically adjusts to 1080, resulting in a screenshot size of 1920x1080. In an ideal scenario, I would want Selenium to go headless only during the moment when it takes the screenshot (which, unfortunately as per my knowledge, is not feasible).
Other commonly used methods that do not work when the browser is not headless:
el = driver.find_element_by_tag_name('body')
el.screenshot(path)
original_size = driver.get_window_size()
required_width = driver.execute_script('return document.body.parentNode.scrollWidth')
required_height = driver.execute_script('return document.body.parentNode.scrollHeight')
driver.set_window_size(required_width, required_height)
driver.find_element_by_tag_name('body').screenshot(path) # avoids scrollbar
driver.set_window_size(original_size['width'], original_size['height'])
element = chrome.find_element_by_tag_name('body')
element_png = element.screenshot_as_png
with open("test2.png", "wb") as file:
file.write(element_png)
With the headless option
https://i.stack.imgur.com/KV5Qg.png
Without the headless option