通过网址自动网页截图（Selenium实现）

  在自动化测试或者想要批量获取网页首页图片时，就需要一个脚本实现自动化网页截图，我们可以通过Selenium实现这个功能。

1. 安装Selenium

pip install selenium

2. 功能代码

from selenium import webdriver
import time

def web_screenshot(url, filepath):
    option = webdriver.ChromeOptions()
    option.add_argument('headless')  # 设置option
    
    driver = webdriver.Chrome()  # 初始化一个谷歌浏览器实例
    driver.maximize_window()  # 窗口最大化
    
    driver.get(url)
    time.sleep(1)  # 设置等待时间，确保页面完全加载
    driver.get_screenshot_as_file(filepath)
    driver.quit()

3. 效果展示

url = 'https://www.baidu.com'
filepath = 'test.png'
web_screenshot(url, filepath)

网页截图效果示例：