python+selenium实现自动截图保存

本文主要总结使用python+selenium实现自动截图保存的功能。

一、背景

为了代替人工检查上传的几百个视频用户是否能正常观看，于是结合网上的资料研究了下通过python调用文件url后自动截屏保存，这样就不必一个个打开去看了。

本文以实现 打开斗鱼直播并截图 为例进行说明。下图的url是网上随便找的几个直播地址，可根据需要替换。

 

二、技术方法

python+selenium+pandas

三、代码实现

import pandas as pd
import time
from selenium import webdriver

def get_screen_shot(path):

   # 读取excel表格数据
   data = pd.read_excel(path, header=0)
   for i in range(len(data)):
      driver = webdriver.Chrome()
      driver.maximize_window()
      driver.get(data.iloc[i, 1])
      time.sleep(5)
      driver.save_screenshot('.\\images\\'+str(i)+'_'+driver.title+'.png')
      driver.quit()

if __name__ == '__main__':
   path = 'E:\\learning\\screenshot\\tvList.xlsx'
   get_screen_shot(path)

四、效果

五、问题记录

问题1：python 调用 selenium 库使用chrome时候报错：selenium.common.exceptions.WebDriverException: Message: 'chromedriver' executable needs to be in PATH.

解决方法：

1.打开chrome 地址栏输入 “chrome://version/”来查看chrome版本

2.http://chromedriver.storage.googleapis.com/index.html 下载对应版本的chromedriver

3.解压到python的安装路径下即可（我的python安装路径：D:\Python\Python37）