使用无头Chrome驱动程序拍摄屏幕截图

Question

我正在尝试使用Selenium和无头Chrome驱动程序在网站上拍摄图像的屏幕快照。此过程涉及两个步骤。首先，我确定了网站上需要单击的区域（它是一个全屏按钮，可以显示完整图像）。第二部分是截图。当我在无头模式下运行时，一切工作都很好。但是，当我切换到无头模式时，我会截取网站完全错误的部分的屏幕快照，好像脚本忽略了单击全屏按钮的部分一样。以下是我脚本的相关部分。

    element = WebDriverWait(driver, 
    20).until(EC.visibility_of_element_located((By.CSS_SELECTOR, '.fullscreen')))
    driver.execute_script("return 
    arguments[0].scrollIntoView(true);", element)

    element.click()

    element = driver.find_element_by_id('id')

    png = driver.get_screenshot_as_png()
    driver.quit()