Python: как получить скриншот всей страницы?

Question

[vlarkanov]

Всем привет.

Вот мой код:

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()

Использованием

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

я пытаюсь добиться того, чтобы делался скрин всей страницы, но получаю только часть, как если бы я использовал

driver.get_screenshot_as_file(filename)

Что я делаю не так? Страницу смотрел, вся нужная информация находится в пределах тега body.