這些天實在忙的冒煙,一大堆的項目堆在一起,沒日沒夜的加班。加上有些懶惰,學習進度一直沒有太多進展。這篇文章主要介紹前段時間抽空實現的一個功能,希望有需要用到可以得到一點啟發。
基於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識別的同學,本文沒有任何參考價值,就此略過