selenium 验证码识别

前言：做web端ui自动化，往往会遇到登录需要验证码这个问题，很多时候解决方法是要求前端在非正式环境设置万能验证码或者剔除验证码。但这样不利于测试正常流程，存在漏测或者万能验证码代码发布上线的风险。

　　　所以可以通过ocr、cookic等方法去验证的登录，以下主要说明通过ocr去解决的方法：

注：ddddocr 比较适用于数字这种简单验证码，对于向复杂字符、需要点击验证码内容选择、滑动图片验证这些难度高的，可参考这位大佬：https://wenanzhe.com/

driver = webdriver.Chrome()
# 通过ddddocr 识别 数字验证码
def web_login():
    driver.maximize_window()
    driver.get("https://xxxx.com")
    time.sleep(5)
    # 识别完成 输入账户密码/验证码
    driver.find_element_by_name('user_name').send_keys('username')

    driver.find_element_by_name('password').send_keys('password')

    driver.find_element_by_name('captcha').send_keys(captcha_ocr())
    time.sleep(5)

    driver.find_element_by_xpath(element.login.value).click()
    time.sleep(2)
    try:
        # TODO:因为ocr识别不是100%准确，所以可以根据项目来决定重试次数，使用循环来判断更优；
        if iselement(element.login.value) is not False:
            log.log_info('第一次输入验证码不正确')
            captcha_input()
            if iselement(element.login.value) is not False:
                log.log_info('第二次输入验证码不正确')
                captcha_input()
                if iselement(element.login.value) is not False:
                    log.log_info('第三次输入验证码不正确')
                    captcha_input()
    except:
        print('验证码正确，登录成功')


# 当验证码出错，删除已输入的验证码，重新识别输入
def captcha_input():
    # 因为ocr识别成不是4位数字，造成没有刷新验证码，需要手动刷新一次
    driver.find_element_by_xpath(element.captcha_img.value).click()
    # 删除已输入的，重新识别
    rc = driver.find_element_by_name('captcha')
    ActionChains(driver).double_click(rc).perform()
    rc.send_keys(Keys.BACK_SPACE)
    time.sleep(1)
    rc.send_keys(captcha_ocr())
    time.sleep(1)
    driver.find_element_by_xpath(element.login.value).click()


# 进行验证码截图，并通过ocr识别验证码信息
def captcha_ocr():
    driver.find_element_by_xpath(element.captcha_img.value)\
        .screenshot('captcha.jpg')
    time.sleep(3)

    # 验证码识别
    ocr = ddddocr.DdddOcr()
    with open('captcha.jpg', 'rb') as f:
        img_bytes = f.read()
    res = ocr.classification(img_bytes)
    return res


#  判断元素是否存在，不存在抛出异常捕获
def iselement(xpath):
    try:
        driver.find_element_by_xpath(xpath)
    except NoSuchElementException:
        return False