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 #調用chrome瀏覽器
driver = webdriver.Chrome(chromedriver)
driver.get("http://xxxx.com") #該處爲具體網址
driver.refresh() #刷新頁面
driver.maximize_window() #瀏覽器最大化
#獲取全屏圖片，並截取驗證碼圖片的位置
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)
#打開保存的驗證碼圖片
image = Image.open("a.png")
#圖片轉換成字符
vcode = pytesseract.image_to_string(image)
print(vcode)
#填充用戶名 密碼 驗證碼
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)
#點擊登錄
driver.find_element_by_id("loginBtn").click()
