Question

我想获取一个使用Selenium python的完整网页的屏幕截图。为此，我使用了this answer。

但是现在我想对屏幕截图进行划分，并将结果屏幕截图的高度限制为30 000px。例如，如果网页高度为40000像素，则我需要第一个30000像素的屏幕截图，然后是10000像素的屏幕截图。

解决方案不应是保存整个页面然后裁剪图像，因为我需要它才能处理很长的网页。实际上，不可能获得高度为120000px的屏幕截图，否则会出现这种错误：

    raise exception_class(message, screen, stacktrace)
selenium.common.exceptions.WebDriverException: Message: [Exception... "Failure"  nsresult: "0x80004005 (NS_ERROR_FAILURE)"  location: "JS frame :: chrome://marionette/content/capture.js :: capture.canvas :: line 154"  data: no]

我试过了，但是根本没用：

from selenium import webdriver
from selenium.webdriver.firefox.options import Options
from time import sleep

def save_screenshot(d, path):
    max_size = 30000
    original_size = d.get_window_size()
    required_width = d.execute_script('return document.body.parentNode.scrollWidth')+74
    required_height = d.execute_script('return document.body.parentNode.scrollHeight')+74
    if required_height <= max_size:
        d.set_window_size(required_width, required_height)
        d.find_element_by_tag_name('body').screenshot(path)  # avoids scrollbar
    else :
        for i in range(0,int(required_height/max_size)):
            d.set_window_position(0,max_size*i)
            d.set_window_size(required_width, max_size)
            d.find_element_by_tag_name('body').screenshot(path.replace(".png",str(i+1)+".png"))
        d.set_window_position(0,max_size*int(required_height/max_size))
        d.set_window_size(required_width, required_height%max_size)
        d.find_element_by_tag_name('body').screenshot(path.replace(".png",str(int(required_height/max_size)+1)+".png"))
        d.set_window_position(0,0)
    d.set_window_size(original_size['width'], original_size['height'])

if __name__ == '__main__':
    options = Options()
    options.headless = True
    driver = webdriver.Firefox(options=options,executable_path=r"C:/Users/path/geckodriver.exe")
    driver.get("http://www.lwb.com/")
    sleep(3)
    save_screenshot(driver,"C:/Users/path/test.png")
    driver.close()

有人可以在这里帮助我吗？

谢谢

Mylha

Answer 1

您可以使用枕头pip install pillow剪裁保存的屏幕截图

def crop_image(img, img_limit):
    img_width, img_height = img.size
    crop_dim = (0, 0, img_width, img_limit) # left top right bottom
    cropped_img = img.crop(crop_dim)
    return cropped_img

完成save_screenshot(driver, "path/to/img")之后，请执行以下操作：

from PIL import Image

img_limit = 30000 # your image size limit
img = Image.open("path/to/img")
img = crop_image(img, img_limit)
img.save("path/to/img")

如果您不想在操作图像之前保存图像，可以使用get_screenshot_as_png，它将返回二进制数据而不是保存它：

from PIL import Image
from io import BytesIO

img_limit = 30000 # your image size limit
img_binary = driver.get_screenshot_as_png()
img = Image.open(BytesIO(img_binary))
img = crop_image(img, img_limit)
img.save("path/to/save")

完成从内存中删除二进制数据后，请确保执行del img和del img_binary

为了拍摄整个页面的一个屏幕截图，请执行以下操作：

from selenium import webdriver

DRIVER_PATH = "path/to/chrome/driver"
URL = "site.url"

options = webdriver.ChromeOptions()
options.add_argument("headless")
driver  =  webdriver.Chrome(executable_path = DRIVER_PATH, chrome_options = options)

# setting a long window height to take one full screenshot
driver.set_window_size(1920, 90000) # width , height
driver.get(URL)
driver.maximize_window()

img_binary = driver.get_screenshot_as_png()

PS：如果使用此方法拍摄屏幕截图，则不需要pillow。只需使用set_window_size设置所需窗口的高度和宽度，即可在屏幕截图中获得相同的大小

如何限制Selenium Python中的屏幕截图高度？

1 个答案: