python使用selenium实现网页截图

近期需求：网站首页截图功能，需要将8万个网站全部截取首页图片，并保存下来。

其中chromedriver需要自行下载；

直接上代码！

from selenium import webdriver

# 获取网页截图
def road_situation(url):
    # 初始化一个谷歌浏览器实例
    chrome_driver = 'C:\chromedriver.exe'
    driver = webdriver.Chrome(executable_path=chrome_driver)
    driver.maximize_window()
    driver.get(url)
    if 'http://' in url:
        name = url.replace('http://', '')
    else:
        name = url.replace('https://', '')
    driver.get_screenshot_as_file("D:\\python\\2020-12-23【网页截图】\\cover\\%s.png" % name)
    driver.quit()if __name__ == "__main__":
    f = open('./link.txt', 'r', encoding='utf-8')
    data = f.read().split("\n")
    for link in data:
        print(link)
        road_situation(link)
    print('end')