Python + Selenium(二十三)使用百度文字識(shí)別API識(shí)別驗(yàn)證碼

上一節(jié)我們講到驗(yàn)證碼较锡,通過圖像識(shí)別的方式可以識(shí)別簡單的驗(yàn)證碼提澎,所以識(shí)別對(duì)我們的意義不是特別大。

不過我單獨(dú)開這一節(jié)來講念链,也算是擴(kuò)充一些知識(shí)面,同時(shí)也可以練習(xí)一下關(guān)于 requests 的一些用法。對(duì)接口測試的理解也有很大的好處掂墓。

首先我們需要注冊百度開發(fā)者平臺(tái)谦纱,并且認(rèn)證。看這里

注冊好以后君编,創(chuàng)建應(yīng)用并勾選文字識(shí)別服務(wù)(其實(shí)都可以勾起來玩一玩)跨嘉,文字識(shí)別服務(wù)分了幾種,有完全免費(fèi)的“通用文字識(shí)別”吃嘿,每天有 5W 次的訪問許可祠乃,但是識(shí)別準(zhǔn)確性較低。

還有一種是“通用文字識(shí)別(高精度版)”識(shí)別率更高兑燥,每天有 500 次免費(fèi)調(diào)用亮瓷,超出收費(fèi)。免費(fèi)的也夠用了降瞳。這里用高精度版來演示嘱支。

要使用這類的開放 API 一定要閱讀相關(guān)的接口調(diào)用文檔和示例代碼(如果你沒有興趣閱讀,可以直接使用我貼出的代碼挣饥,不過我建議你依然應(yīng)該好好閱讀一下)除师。好在 Python 是很流行的語言,所以基本上都有關(guān)于 Python 的調(diào)用示例扔枫。不過很多時(shí)候用的都是 urllib 系列的庫而非 requests汛聚。

通用文字識(shí)別過程:

  1. 使用創(chuàng)建應(yīng)用分配的 APP_ID、SECRET_KEY 獲取 access_token
  2. 打開本地圖片或者提供網(wǎng)絡(luò)圖片的地址
  3. 如果是本地圖片短荐,需要對(duì)圖片的二進(jìn)制數(shù)據(jù)進(jìn)行 base64 編碼
  4. 對(duì) base64 編碼后的數(shù)據(jù)倚舀,需要進(jìn)行 urlencode 編碼,編碼格式為{'image': base64數(shù)據(jù)}
  5. 訪問 API 地址搓侄,以 url 參數(shù)形式傳遞 access_token瞄桨, 以消息體的方式傳遞編碼后的圖片數(shù)據(jù)
  6. 獲取識(shí)別后的圖片文字

我使用 requests 對(duì)其調(diào)用示例代碼改造了一下:

import requests
from urllib.parse import urlencode
import base64


APP_ID = ''  # 填寫你的appid
SECRET_KEY = ''  # 填寫你的密鑰

TOKEN_URL = 'https://aip.baidubce.com/oauth/2.0/token'

OCR_URL = 'https://aip.baidubce.com/rest/2.0/ocr/v1/general_basic'

ACCURATE_URL = 'https://aip.baidubce.com/rest/2.0/ocr/v1/accurate_basic'

def fetch_token():
    '''
    獲取 API 調(diào)用需要的 access_token
    :params:
    :return: 返回 access_token 字符串
    '''
    data = {
        'grant_type': 'client_credentials',
        'client_id': APP_ID,
        'client_secret': SECRET_KEY
    }
    r = requests.post(TOKEN_URL, data=data)
    if 'access_token' in r.json():
        return r.json().get('access_token')
    else:
        print('請(qǐng)檢查獲取access_token的URL, APP_ID, SECRET_KEY!')

def ocr_b64(b64):
    '''
    傳入base64編碼格式的圖片數(shù)據(jù),識(shí)別圖片中的文字
    :params: base64編碼格式的圖片數(shù)據(jù)
    :return: 返回識(shí)別后的文字字符串
    '''
    access_token = fetch_token() # 獲取 token
    if access_token:
        payload = {'access_token': access_token}    # token通過url參數(shù)傳遞
        data = urlencode({'image': b64})  # 對(duì)傳入的base64編碼格式的圖片數(shù)據(jù)進(jìn)行urlencode編碼
        r = requests.post(ACCURATE_URL, data=data, params=payload)
        r.encoding = 'utf-8'
        if 'words_result' in r.json():
            return r.json().get('words_result')[0].get('words') # 這里可以直接返回 r.json() 在后續(xù)代碼取值
        else: 
            print('圖片識(shí)別失敗讶踪,請(qǐng)根據(jù)返回編碼檢查錯(cuò)誤芯侥,https://ai.baidu.com/docs#/OCR-API-ErrorCode/top')
            print(r.text)

def ocr_file(image):
    '''
    傳入圖片路徑,識(shí)別圖片中的文字
    :params: 圖片路徑
    :return: 返回識(shí)別后的文字字符串
    '''
    with open(image, 'rb') as f:
        data = base64.b64encode(f.read())
    return ocr_b64(data)

if __name__ == "__main__":
    print(ocr_file('./2886.png'))

上面代碼實(shí)現(xiàn)了三個(gè)函數(shù):

  • fetch_token() 用于獲取調(diào)用 API 需要的 access_token乳讥,該 token 需要在調(diào)用時(shí)通過參數(shù)傳遞柱查。
  • ocr_b64(b64) 中傳入 base64 編碼格式的數(shù)據(jù),并通過 requests 庫將編碼后的 base64 數(shù)據(jù)傳遞給文字識(shí)別 API云石,并從獲取到的返回結(jié)果中提取識(shí)別后的字符串唉工。
  • ocr_file(image) 中傳入圖片,并通過 requests 庫將圖片傳遞給文字識(shí)別 API汹忠,并從獲取到的返回結(jié)果中提取識(shí)別后的字符串淋硝。

你只需要修改上面代碼中的常量APP_IDSECRET_KEY雹熬,這些都是從百度開發(fā)者平臺(tái)中創(chuàng)建應(yīng)用后可以得到。

好谣膳,接下來我們以 showdoc 系統(tǒng)為例竿报,演示如何通過識(shí)別驗(yàn)證碼實(shí)現(xiàn)登錄。你可以選擇 showdoc 的在線版來練習(xí)继谚。

需要以下步驟完成驗(yàn)證碼識(shí)別并登錄:

  1. 打開 showdoc 登錄頁面
  2. 獲取 驗(yàn)證碼 圖片烈菌,這里直接獲取該元素截圖的 base64 編碼
  3. 調(diào)用 ocr_b64() 函數(shù)識(shí)別驗(yàn)證碼
  4. 輸入登錄賬號(hào)、密碼及驗(yàn)證碼
  5. 點(diǎn)擊登錄
  6. 如果驗(yàn)證碼錯(cuò)誤花履,則循環(huán)這個(gè)過程(驗(yàn)證碼識(shí)別錯(cuò)誤芽世,就刷新頁面重來)直到登錄成功為止
from selenium import webdriver
import time
driver = webdriver.Chrome()
driver.get('https://www.showdoc.cc/user/login')

while True:
    driver.find_element_by_css_selector('[placeholder="用戶名/郵箱"]').send_keys('nemo')
    driver.find_element_by_css_selector('[placeholder="密碼"]').send_keys('123456')
    v_code = ocr_b64(driver.find_element_by_class_name('v_code_img').screenshot_as_base64)  # => 識(shí)別驗(yàn)證碼
    driver.find_element_by_css_selector('[placeholder="驗(yàn)證碼"]').send_keys(v_code)
    driver.find_element_by_xpath('//button/*[text()="登錄"]').click()
    time.sleep(0.5)
    try:
        # 如果彈出框出現(xiàn),則說明驗(yàn)證碼錯(cuò)誤 
        driver.find_element_by_css_selector(
            '.el-message-box .el-button--primary').click()
        # 刷新頁面
        driver.refresh()
    except:
        # 如果彈出框不出現(xiàn)诡壁,則說明驗(yàn)證碼正確
        break
time.sleep(5)
driver.quit()

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末济瓢,一起剝皮案震驚了整個(gè)濱河市,隨后出現(xiàn)的幾起案子欢峰,更是在濱河造成了極大的恐慌葬荷,老刑警劉巖,帶你破解...
    沈念sama閱讀 218,284評(píng)論 6 506
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異,居然都是意外死亡磨澡,警方通過查閱死者的電腦和手機(jī),發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,115評(píng)論 3 395
  • 文/潘曉璐 我一進(jìn)店門扒吁,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人室囊,你說我怎么就攤上這事雕崩。” “怎么了融撞?”我有些...
    開封第一講書人閱讀 164,614評(píng)論 0 354
  • 文/不壞的土叔 我叫張陵盼铁,是天一觀的道長。 經(jīng)常有香客問我尝偎,道長饶火,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 58,671評(píng)論 1 293
  • 正文 為了忘掉前任致扯,我火速辦了婚禮肤寝,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘抖僵。我一直安慰自己鲤看,他們只是感情好,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,699評(píng)論 6 392
  • 文/花漫 我一把揭開白布耍群。 她就那樣靜靜地躺著义桂,像睡著了一般找筝。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上慷吊,一...
    開封第一講書人閱讀 51,562評(píng)論 1 305
  • 那天呻征,我揣著相機(jī)與錄音,去河邊找鬼罢浇。 笑死,一個(gè)胖子當(dāng)著我的面吹牛沐祷,可吹牛的內(nèi)容都是我干的嚷闭。 我是一名探鬼主播,決...
    沈念sama閱讀 40,309評(píng)論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼赖临,長吁一口氣:“原來是場噩夢啊……” “哼胞锰!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起兢榨,我...
    開封第一講書人閱讀 39,223評(píng)論 0 276
  • 序言:老撾萬榮一對(duì)情侶失蹤嗅榕,失蹤者是張志新(化名)和其女友劉穎,沒想到半個(gè)月后吵聪,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體凌那,經(jīng)...
    沈念sama閱讀 45,668評(píng)論 1 314
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,859評(píng)論 3 336
  • 正文 我和宋清朗相戀三年吟逝,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了帽蝶。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點(diǎn)故事閱讀 39,981評(píng)論 1 348
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡块攒,死狀恐怖励稳,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情囱井,我是刑警寧澤驹尼,帶...
    沈念sama閱讀 35,705評(píng)論 5 347
  • 正文 年R本政府宣布,位于F島的核電站庞呕,受9級(jí)特大地震影響新翎,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜千扶,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,310評(píng)論 3 330
  • 文/蒙蒙 一料祠、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧澎羞,春花似錦髓绽、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,904評(píng)論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽枫攀。三九已至,卻和暖如春株茶,著一層夾襖步出監(jiān)牢的瞬間来涨,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 33,023評(píng)論 1 270
  • 我被黑心中介騙來泰國打工启盛, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留蹦掐,地道東北人。 一個(gè)月前我還...
    沈念sama閱讀 48,146評(píng)論 3 370
  • 正文 我出身青樓僵闯,卻偏偏與公主長得像卧抗,于是被迫代替她去往敵國和親。 傳聞我的和親對(duì)象是個(gè)殘疾皇子鳖粟,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 44,933評(píng)論 2 355