大致思路:用Pandas读取EXCEL中的数据,将链接存放入列表中,再用Selenium打开并截图,保存到data文件夹中。
文件结构
EXCEL文件中的内容
代码
from selenium import webdriver
import time
import pandas as pd
df = pd.read_excel("urls.xlsx", header=None)
urls = df[0].tolist()
for i in range(len(urls)):
driver = webdriver.Chrome()
driver.get(urls[i])
time.sleep(2)
try:
driver.get_screenshot_as_file("data/{}.png".format(i+1))
except BaseException as msg:
print(msg)
driver.quit()
运行效果
截取图片