1. 程式人生 > >基於Python的Selenium自動化(3)— 實現驗證碼擷取並識別

基於Python的Selenium自動化(3)— 實現驗證碼擷取並識別

這些天實在忙的冒煙,一大堆的專案堆在一起,沒日沒夜的加班。加上有些懶惰,學習進度一直沒有太多進展。這篇文章主要介紹前段時間抽空實現的一個功能,希望有需要用到可以得到一點啟發。

基於UI層的自動化,有一些坑在裡面,幾乎幾個每個人都會遇到的,其中之一就是註冊或登入時需要驗證碼。由於驗證碼時動態生成的。每次訪問都會不同,這樣的話,使用指令碼去登入會是一件麻煩的事。
目前有的一些解決方案:
1、針對公司內部的專案有兩個方法,
(1)設定一個萬能驗證碼,只要每次填寫這個驗證碼就可以驗證通過
(2)將手機號設定為白名單,只要輸入特定的手機號,則不校驗驗證碼
2、針對外部專案則可使用下面的方法
(3)擷取驗證碼部分並使用圖片識別技術識別
本文主要介紹(3)的實現過程。僅供參考
步驟可分為擷取驗證碼+識別驗證碼
一、擷取驗證碼
selenium webdriver 的api中直接提供了截圖的方法.但是是全屏,要實現對元素的擷取,則需要繞一繞彎
1、獲取到元素的大小、元素的座標
2、擷取整屏
3、根據元素的座標和大小,定位要剪裁的區域
4、使用影象庫對元素區域進行剪裁
python程式碼實現:

        driver.get_screenshot_as_file('a.jpg')
        location = driver.find_element_by_id('validate-img').location
        size = driver.find_element_by_id('validate-img').size
        left = location['x']
        top =  location['y']
        right = location['x'] + size['width']
        bottom = location['y'
] + size['height'] a = Image.open("a.jpg") im = a.crop((left,top,right,bottom)) im.save('a.jpg') time.sleep(1)

程式碼沒有註釋部分,解釋起來就是上面的四個步驟

二、識別驗證碼
本文不設計如何實現驗證碼,這部分的任務我選用的是第三方的API來實現的。(如有需要,可以搜尋“超級鷹驗證碼識別”)畢竟術業有專攻,如果要做OCR識別的同學,本文沒有任何參考價值,就此略過