python - 如何在selenium中以特定分辨率捕获屏幕截图？

Question

我想捕获具有特定分辨率的 Canvas 元素的屏幕截图。我正在使用下面的代码片段来捕获屏幕截图。这可以工作，但总是截取分辨率为 1544px*638px 的屏幕截图。基于什么(浏览器窗口或我的窗口屏幕)进行屏幕截图。我正在使用 Chrome 浏览器。我如何修改下面的代码来截取分辨率为 500 * 325 左右的屏幕截图。

我使用了 set_window_size(500, 325) 并得到了尺寸为 750 * 135 的屏幕截图

def capture_screenshot():
    driver = LiveLibrary.get_webdriver_instance()
    driver.set_window_size(500, 325) 
    canvas_element = driver.find_element_by_xpath("//canvas")
    result = canvas_element.screenshot_as_png
    with open('save.png', 'wb') as f:
        f.write(result)

有人可以帮我解决这个问题吗？谢谢。

Answer 1

您可以使用 Image.resize 调整屏幕截图的大小来自太平船务。喜欢:

from PIL import Image
import io

    ...

    result = canvas_element.screenshot_as_png
    image = Image.open(io.BytesIO(result))
    imageResized = image.resize( (500,325), Image.ANTIALIAS) 
    with open('save.png', 'wb') as f:
        imageResized.save(f , format='PNG')