from selenium import webdriver
import time
from PIL import Image
base_url = '***********'
browser = webdriver.Chrome()
browser.maximize_window()
browser.implicitly_wait(10)
browser.get(base_url)
# (1)登錄頁(yè)面截圖
browser.save_screenshot("D:/pic.png")#可以修改保存地址
# (2)基操
browser.find_element_by_name("username").send_keys("gxx")
browser.find_element_by_name("password").send_keys("123456")
time.sleep(2)
# (3)獲取圖片驗(yàn)證碼坐標(biāo)
code_ele = browser.find_element_by_xpath("http://*[@id='app']/div/div[1]/form/div[3]/div/div/div[2]/div/img")
print("驗(yàn)證碼的坐標(biāo)為：", code_element.location)#控制臺(tái)查看{'x': 1086, 'y': 368}
print("驗(yàn)證碼的大小為：", code_element.size)# 圖片大小{'height': 40, 'width': 110}
# (4)圖片4個(gè)點(diǎn)的坐標(biāo)位置
left = code_ele.location['x']#x點(diǎn)的坐標(biāo)
top = code_ele.location['y']#y點(diǎn)的坐標(biāo)
right = code_ele.size['width']+left#上面右邊點(diǎn)的坐標(biāo)
down = code_ele.size['height']+top#下面右邊點(diǎn)的坐標(biāo)
image = Image.open('D:/pic.png')
# (4)將圖片驗(yàn)證碼截取
code_image = image.crop((left, top, right, height))
code_image.save('D:/pic1.png')#截取的驗(yàn)證碼圖片保存為新的文件

#百度ocr提供了模板，我們直接復(fù)制就ok
#下載通用文字識(shí)別的python sdk,一定要放在你寫的代碼的文件夾下面
from aip import AipOcr 
from os import path
def baiduOCR(picfile,outfile):#picfile:圖片文件名 outfile:輸出文件
  filename = path.basename(picfile)#圖片名稱
  #百度提供
  """ 你的 APPID AK SK """
  APP_ID = ''#這是你產(chǎn)品服務(wù)的appid
  API_KEY = ''#這是你產(chǎn)品服務(wù)的appkey
  SECRET_KEY = ''#這是你產(chǎn)品服務(wù)的secretkey
  client = AipOcr(APP_ID, API_KEY, SECRET_KEY)
  i = open(picfile, 'rb')
  img = i.read()

  print("正在識(shí)別圖片：\t" + filename)
  """ 調(diào)用通用文字識(shí)別（高精度版） """
  message = client.basicAccurate(img)
  print("識(shí)別成功！")
  i.close()
  with open(outfile, 'a+') as fo:#這邊是寫進(jìn).txt文件
    fo.writelines("*" * 60 + '\n')#搞點(diǎn)花里胡哨的做區(qū)分
    fo.writelines("識(shí)別圖片：\t" + filename + "\n" * 2)
    fo.writelines("文本內(nèi)容：\n")
    # 輸出文本內(nèi)容
    for text in message.get('words_result'):#識(shí)別的內(nèi)容 
      fo.writelines(text.get('words') + '\n')
    fo.writelines('\n' * 2)
  print("文本導(dǎo)出成功！")
  print()

if __name__ == '__main__' :
  outfile = 'D:/export1.txt'#保存的文件
  baiduOCR('D:/pic1.png', outfile)
  print('圖片文本提取結(jié)束！文本輸出結(jié)果位于 %s 文件中。' % outfile)