本文主要总结使用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)+'_'+d