用selenium在python下实现批量网页 截图

需要用到的工具是selenium，

引用下百度的解释：

Selenium [1]  是一个用于Web应用程序测试的工具。Selenium测试直接运行在浏览器中，就像真正的用户在操作一样。支持的浏览器包括IE（7, 8, 9, 10, 11），Mozilla Firefox，Safari，Google Chrome，Opera等。这个工具的主要功能包括：测试与浏览器的兼容性——测试你的应用程序看是否能够很好得工作在不同浏览器和操作系统之上。测试系统功能——创建回归测试检验软件功能和用户需求。支持自动录制动作和自动生成 .Net、Java、Perl等不同语言的测试脚本。

 

demo代码如下，python写的

import os
from selenium import webdriver

c=0
delay=1  #网页打开后停留时间，单位是秒
path=os.getcwd()+'/html_dir'

files=os.listdir(path)
num=len(files)

for c in range(num): 
    fn=str(c)+'.html'  
    tmpurl='file://{path}/{mapfile}'.format(path=path,mapfile=fn)
    browser=webdriver.Chrome()
    #browser.set_window_size(960,540)  #可设置打开网页窗口大小
    browser.get(tmpurl)
    time.sleep(delay)
    browser.save_screenshot('./pic_of_html/'+str(c)+'.png')
    browser.quit()

上面的代码实现的是：加入在html_dir文件夹下有名称为1.html, 2.html , 3.html……等总计num个网页文件；使用Chrome浏览器加载然后截图保存到pic_of_html文件夹下，图片命名为：1.png, 2.png ……

除了安装好selenium外，还需要安装浏览器的驱动，比如我这用的是Chrome浏览器，首先需要确定自己使用的Chrome版本，然后下载对应的Chrome 网页驱动。