Python selenium实现网页截图

代码如下，通过 set_window_size方法可以设置浏览器大小

from selenium import webdriver
from time import sleep
from selenium.webdriver.chrome.options import Options

 

# 无界面模式
def ChromeDriverNOBrowser():
   chrome_options = Options()
   chrome_options.add_argument('--headless')
   chrome_options.add_argument('--disable-gpu')
   driverChrome = webdriver.Chrome(chrome_options=chrome_options)
   return driverChrome

driver=ChromeDriverNOBrowser()
driver.set_window_size(1920,1080)
driver.get("网址")
sleep(2)
#网页截图并保存
driver.get_screenshot_as_file(r'D:\test\test.png')
print("over!")