Selenium+Python+Webdriver:儲存截圖到指定檔案夾

標籤：視窗   while   png   OLE   pat   存在   圖片   apt   字串   

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()
 

Selenium+Python+Webdriver:儲存到指定檔案夾