Selenium 屏幕截图运行缓慢（Python）

Question

带有 selenium 和 phantomjs 的脚本可以检查大约 20 个动态页面并在有更改时警告我，无需屏幕截图部分即可快速运行，但是当我想要获取页面的屏幕截图时，大约需要 1-2 分钟来警告我并获取截屏。有没有更好更快的方法用 python 截取页面的特定部分？

这是我用于截图的代码。

from selenium import webdriver
from PIL import Image

fox = webdriver.Firefox()
fox.get('http://stackoverflow.com/')

# now that we have the preliminary stuff out of the way time to get that image :D
element = fox.find_element_by_id('hlogo') # find part of the page you want image of
location = element.location
size = element.size
fox.save_screenshot('screenshot.png') # saves screenshot of entire page
fox.quit()

im = Image.open('screenshot.png') # uses PIL library to open image in memory

left = location['x']
top = location['y']
right = location['x'] + size['width']
bottom = location['y'] + size['height']


im = im.crop((left, top, right, bottom)) # defines crop points
im.save('screenshot.png') # saves new cropped image

已解决：

问题不在于 selenium 模块，无论是屏幕截图。这是 关于 phantomjs，在我开始使用 chromedriver 之后，它非常快而且更多 有效。

解决方案更新：

phantomjs 的问题是禁用图像。当我使用 --load-images=no 我面临内存泄漏问题，脚本变得非常慢，没有它 没问题。

Answer 1

您可以通过在内存中裁剪屏幕截图而不先将其保存到文件来节省一些时间：

import StringIO
from selenium import webdriver
from PIL import Image

driver = webdriver.Firefox()
driver.get('http://stackoverflow.com')
element = driver.find_element_by_id('hlogo')

crop_points = driver.execute_script("""
    var r = arguments[0].getBoundingClientRect();
    return [r.left, r.top, r.left + r.width, r.top + r.height];
    """, element)

with Image.open(StringIO.StringIO(driver.get_screenshot_as_png())) as img :
    with img.crop(crop_points) as imgsub :
        imgsub.save(logo.png', 'PNG')