我想使用 Selenium python 获取完整网页的屏幕截图。为此，我使用了 this answer .
但现在我想分割屏幕截图并将生成的屏幕截图的高度限制为 30 000px。例如，如果网页高度为 40 000 像素，我想要第一个 30 000 像素的屏幕截图，然后是一个 10 000 像素的屏幕截图。
解决方案不应该是保存整页然后裁剪图像，因为我需要它来处理很长的网页。事实上，不可能获得高度为 120 000 像素的屏幕截图，否则会出现此类错误:
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()
有人可以帮我吗？
谢谢，
米哈
请您参考如下方法:
您可以使用 pillow 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 设置您想要的窗口的高度和宽度，这将使您在屏幕截图中获得相同的大小
