Selenium+Python+Webdriver:保存截圖到指定文件夾

保存圖片到指定文件夾：

from selenium import webdriver
from pathlib import Path
from time import sleep
driver = webdriver.Chrome()
driver.get("http://www.baidu.com")
driver.find_element_by_id('kw').send_keys('selenium')
driver.find_element_by_id('su').click()
sleep(5)
# 截取當前窗口，並指定截圖圖片的保存位置

i=1
scrpath="C:\\Users\\whu024\\Desktop\\AmyPython"  #指定的保存目錄
capturename = '\\'+str(i) + '.png'  #自定義命名截圖
wholepath=scrpath+capturename
print (wholepath)
if Path(scrpath).is_dir():  #判斷文件夾路徑是否已經存在
    pass   
else:
    Path(scrpath).mkdir()   #如果不存在，創建文件夾
while Path(wholepath).exists():   #判斷文件是否已經存在，也可使用is_file()判斷
    i+=1
    capturename = '\\'+str(i) + '.png'
    wholepath = scrpath+capturename
driver.get_screenshot_as_file(wholepath) #不能接受Path類的值，只能是字符串，否則無法截圖
sleep(5)
driver.quit()

保存圖片到Excel:

import xlsxwriter
book = xlsxwriter.Workbook('C:\AmyPersonal\AmyMoKuai\pict.xlsx')
sheet = book.add_worksheet('demo')
sheet.insert_image('A1','0.jpg')
book.close()

 (pip install XlsxWriter)