Python: how to get a screenshot of the entire page?

Question

linux

Hello.
Here is my code:

from selenium import webdriver
from selenium.webdriver.chrome.options import Options

chrome_options = Options()
chrome_options.add_argument("--headless")
driver = webdriver.Chrome(options=chrome_options)
driver.get(URL)
filename = "screenshots/" + filename_prefix + "-" + get_timestamp() + ".png"
print("Saving " + filename)
element = driver.find_element_by_tag_name('body')
element.screenshot(filename)
driver.quit()

Using

element = driver.find_element_by_tag_name('body')
element.screenshot(filename)

i'm trying to get the whole page to be screened, but i get only part of it, as if i use
driver.get_screenshot_as_file(filename)
what i'm doing wrong? I looked at the page, all the necessary information is within the body tag.

Answer 1

driver.set_window_size to match the height or even more of your body