from pyvirtualdisplay import Display
from selenium import webdriver

# 啟動虛擬屏幕
display = Display(visible=0, size=(800, 600))
display.start()

# 創(chuàng)建瀏覽器實例并訪問頁面
browser = webdriver.Chrome()
browser.get('https://www.google.com')

# 截圖并保存
browser.save_screenshot('screenshot.png')

# 關閉瀏覽器和虛擬屏幕
browser.quit()
display.stop()

在這個例子中，我們先啟動虛擬屏幕，然后創(chuàng)建了一個Chrome瀏覽器實例，并訪問了谷歌首頁。接著我們使用save_screenshot方法對頁面進行截圖，并保存到本地文件。最后我們關閉了瀏覽器和虛擬屏幕。

請注意，虛擬屏幕的大小應該與瀏覽器窗口大小相同，否則可能會導致截圖不完整。在代碼中，我們指定了虛擬屏幕的大小為800x600，你可以根據(jù)實際情況進行修改。

Q&A

任務太多不是要啟動很多虛擬屏幕，這樣是不是很耗費內存

是的，多開虛擬屏幕會占用大量內存。

Xvfb

你可以考慮使用Xvfb（虛擬X11窗口系統(tǒng)）來模擬屏幕，這樣截圖的時候就不會彈出頁面了。以下是基于Python的Xvfb截圖代碼示例：

import os
import time
from selenium import webdriver
from pyvirtualdisplay import Display
from PIL import Image

# 設置虛擬屏幕分辨率
display = Display(visible=0, size=(1920, 1080))
display.start()

# 啟動瀏覽器
browser = webdriver.Chrome()

# 打開網(wǎng)頁
browser.get('https://www.baidu.com')

# 等待頁面加載完成
time.sleep(5)

# 截圖
browser.save_screenshot('screenshot.png')

# 退出瀏覽器
browser.quit()

# 關閉虛擬屏幕
display.stop()

# 打開截圖
Image.open('screenshot.png').show()

這個示例中，我們使用了pyvirtualdisplay庫來創(chuàng)建虛擬屏幕，Chrome瀏覽器以及Selenium來打開網(wǎng)頁和截圖，最后使用Pillow庫打開截圖。注意，這種方式需要在Linux或者macOS環(huán)境下運行。

到此這篇關于python實現(xiàn)selenium截圖的兩種方法的文章就介紹到這了,更多相關python selenium截圖內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: