Как ограничить высоту скриншота в Селен Python? - PullRequest
Новая
       11

Как ограничить высоту скриншота в Селен Python?

2 голосов
/ 27 апреля 2019

Я хотел бы получить скриншот полной веб-страницы с Selenium Python.Для этого я использовал этот ответ .

Но теперь я хотел бы разделить снимок экрана и ограничить высоту полученного снимка экрана до 30 000 пикселей.Например, если высота веб-страницы составляет 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()

Может кто-нибудь помочь мне здесь, пожалуйста?

Спасибо,

Мюльха

1 Ответ

0 голосов
/ 27 апреля 2019

Вы можете обрезать сохраненный скриншот, используя подушку 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, чтобы установить высоту и ширину окна, которое вы хотите, что даст вам тот же размер на скриншоте

...