python+selenium 实现验证码由base64转为jpg/png格式，并识别验证码-Toy模板网

这篇具有很好参考价值的文章主要介绍了python+selenium 实现验证码由base64转为jpg/png格式，并识别验证码。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

在使用python+selenium实现自动化登录时，遇到了关于验证码的一些问题，主要是验证码格式的转换和识别。我这里的验证码都是用的base64格式，我的思路是先将base64格式转化为jpg或者png格式，然后用ddddocr模型来识别图片内容。

#base64转jpg
import base64
from selenium import webdriver
from selenium.webdriver.common.by import By
#使用xpath定位到验证码所在的位置
el = driver.find_element_by_xpath('//*[@id="app"]/div/form/div[3]/div/div[2]/img')
data = el.get_attribute("src")
print(len(data))
#去掉base64的'data:image/png;base64,'
a = data.replace('data:image/png;base64,', '')
print(len(a))
missing_padding = 4 - len(a) % 4
if missing_padding:
    a += '=' * missing_padding
print(a, file=open('base64.txt', 'w'))
with open('base64.txt', 'r') as f_obj:
    b = base64.b64decode(f_obj.read())
#注意base64的数据前头部分，如果是data:image/png,则转换成jpg格式
    file = open('1.jpg', 'wb')
    file.write(b)
print('图片已下载')
code = driver.find_element_by_xpath('//*[@id="app"]/div/form/div[3]/div/div[1]/input')  # 验证码输入框位置
ocr = ddddocr.DdddOcr()
time.sleep(3)
try:
    with open('1.jpg', 'rb') as fp:
        image = fp.read()
    catch = ocr.classification(image)  # 验证码返回给catch
    print(catch)
    #将验证码内容写入输入框
    code.send_keys(catch)
except Exception as e:
    print(e)
print('successful')

总结：文章来源地址https://www.toymoban.com/news/detail-547797.html