Python使用selenium實(shí)現(xiàn)網(wǎng)頁(yè)用戶名 密碼 驗(yàn)證碼自動(dòng)登錄功能

from selenium import webdriver
from PIL import Image
import pytesseract
import os,time
chromedriver = "D:\Program Files\Anaconda3\selenium\webdriver\chromedriver.exe" #這里寫本地的chromedriver 的所在路徑
os.environ["webdriver.Chrome.driver"] = chromedriver #調(diào)用chrome瀏覽器
driver = webdriver.Chrome(chromedriver)
driver.get("http://xxxx.com") #該處為具體網(wǎng)址
driver.refresh() #刷新頁(yè)面
driver.maximize_window() #瀏覽器最大化
#獲取全屏圖片，并截取驗(yàn)證碼圖片的位置
driver.get_screenshot_as_file('a.png')
location = driver.find_element_by_id('imgValidateCode').location
size = driver.find_element_by_id('imgValidateCode').size
left = location['x']
top = location['y']
right = location['x'] + size['width']
bottom = location['y'] + size['height']
a = Image.open("a.png")
im = a.crop((left,top,right,bottom))
im.save('a.png')
time.sleep(1)
#打開保存的驗(yàn)證碼圖片
image = Image.open("a.png")
#圖片轉(zhuǎn)換成字符
vcode = pytesseract.image_to_string(image)
print(vcode)
#填充用戶名 密碼 驗(yàn)證碼
driver.find_element_by_id("staffCode").send_keys("username")
driver.find_element_by_id("pwd").send_keys("password")
driver.find_element_by_id("validateCode").send_keys(vcode)
#點(diǎn)擊登錄 
driver.find_element_by_id("loginBtn").click()