selenium结合tenacity的retry实现验证码失败重试

说在前面

• 本文假设了一个场景 - 你通过OCR识别的验证码是有一定的错误几率的- 本文是通过识别后的验证码去加一个随机字符，如果取到的是’'空字符则可能成功，否则必然不会成功
• 所涉及的python库 - selenium- ddddocr- tenacity

上代码

• 细节详见注释

from selenium import webdriver
from time import sleep
from tenacity import TryAgain, retry, wait_random

defget_element(locator):'''
    这个函数用来判断是否存在某个元素
    '''try:from selenium.webdriver.support.wait import WebDriverWait
        from selenium.webdriver.support import expected_conditions as EC
        return WebDriverWait(driver,5,0.5).until(EC.visibility_of_element_located(locator))except:returnFalse

driver = webdriver.Chrome()
driver.implicitly_wait(5)
driver.get('http://101.43.47.168:8090/forum.php')# 这个地址你可以访问，是我搭建在云端的一个容器，开源论坛
driver.find_element('css selector','#ls_username').send_keys('admin')
driver.find_element('css selector','#ls_password').send_keys('123456')
driver.find_element('css selector','button.pn.vm').click()@retry(wait=wait_random(min=3,max=5))# 等待一个时间区间，3-5s，注意这个时间的下限建议>hint存在的最大时间defcode_login():'''
    这个函数是用来反复验证码登录的
    '''# 验证码元素
    ele_code = driver.find_element('css selector','[id^=vseccode_cS]>img')import ddddocr
    ocr = ddddocr.DdddOcr()
    code_text = ocr.classification(ele_code.screenshot_as_png)# 当失败的时候尤其要注意: 清空已输入的内容
    driver.find_element('css selector','input[id^=seccodeverify_cS]').clear()
    test_data =['1','a','','2','b']from random import choice
    choice_data = choice(test_data)# 输入识别后的数据+随机字符
    driver.find_element('css selector','input[id^=seccodeverify_cS]').send_keys(code_text + choice_data)# 点击登录
    driver.find_element('css selector',"[name='loginsubmit']>strong").click()# 注意！ 你可以去操作网页，点击登录如果失败会弹出提示 "抱歉，验证码填写错误"
    hint_locator ='css selector','.pc_inner>i'if get_element(hint_locator):# 如果出现这个元素，就...# 点击下验证码，刷新一下
        driver.find_element('css selector','[id^=vseccode_cS]>img').click()# 抛出异常，重跑raise TryAgain
code_login()

聊聊tenacity

https://tenacity.readthedocs.io/en/latest/api.html

https://github.com/jd/tenacity

• 这个库将重试这件事基本做透了

1、 无条件重试

• 你疯了

from tenacity import retry

@retrydefretry_without_anything():print('retry...')raise Exception  # 不加这个可不会重试

retry_without_anything()

2、重试指定次数后停止

from tenacity import retry, stop_after_attempt
@retry(stop=stop_after_attempt(3))defretry_times():print(f'retry... times')raise Exception

retry_times()

• 实际运行的效果是这样

retry... times
retry... times
retry... times
Traceback (most recent call last):
  File "D:\Python39\lib\site-packages\tenacity\__init__.py", line 407,in __call__
    result = fn(*args,**kwargs)
  File "demo_retry.py", line 20,in retry_times
    raise Exception
Exception

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "demo_retry.py", line 23,in<module>
    retry_times()
  File "D:\Python39\lib\site-packages\tenacity\__init__.py", line 324,in wrapped_f
    return self(f,*args,**kw)
  File "D:\Python39\lib\site-packages\tenacity\__init__.py", line 404,in __call__
    do = self.iter(retry_state=retry_state)
  File "D:\Python39\lib\site-packages\tenacity\__init__.py", line 361,initerraise retry_exc from fut.exception()
tenacity.RetryError: RetryError[<Future at 0x16628b45460 state=finished raised Exception>]

• 别担心，你可能在最后一次的时候就不抛出异常了。

3、过一定时间后停止重试

from tenacity import retry, stop_after_delay
import  arrow
from time import sleep
@retry(stop=stop_after_delay(10))defretry_times():print(arrow.now().format('YYYY-MM-DD HH:mm:ss'))
    sleep(1)raise Exception
retry_times()

• 输出像这样

2023-02-21 17:32:01
2023-02-21 17:32:02
2023-02-21 17:32:03
2023-02-21 17:32:04
2023-02-21 17:32:05
2023-02-21 17:32:06
2023-02-21 17:32:07
2023-02-21 17:32:08
2023-02-21 17:32:10
2023-02-21 17:32:11
# 最后抛出异常

4、组合条件

@retry(stop=(stop_after_delay(10)| stop_after_attempt(5)))defretry_multi_conditions():print("10秒后或者5次重试后停止重试")raise Exception

5、重试间隔

• 之前的重试是无缝衔接的
• 你可以让重试之间有延迟@retry(wait=wait_fixed(2))defretry_wait_time1():print(arrow.now().format('YYYY-MM-DD HH:mm:ss'))print("每次重试前等待2秒")raise Exceptionretry_wait_time1()
• 当然上面的仍然是个无限重试
• 你可以组合前面的停止@retry(wait=wait_fixed(2),stop=stop_after_attempt(3))

• 重试等待间隔可以设定一个区间(最大最小值)from tenacity import retry, stop_after_attempt, wait_randomimport arrow@retry(wait=wait_random(min=1,max=4),stop=stop_after_attempt(3))defretry_wait_time1():print(arrow.now().format('YYYY-MM-DD HH:mm:ss'))print("每次重试前等待1~4秒区间")raise Exceptionretry_wait_time1()

6、是否重试！！！

• 这是最重要的了
• 重试的条件一：引发特定或一般异常的重试from tenacity import retry, retry_if_exception_type, stop_after_attempt@retry(retry=retry_if_exception_type(ZeroDivisionError), stop= stop_after_attempt(5))defretry_if_exception():print('retry...')print(10/0)# 现在触发的就是ZeroDivisionError，如果把此处改为 print(a)，则遇到的是NameError，那就不会重试raise Exceptionretry_if_exception()

• 引发 TryAgain 异常随时显式重试
• 比如这样@retrydeftry_if_condition(): result =23if result ==23:raise TryAgain
• 上面的demo中就用到了这个
• 官网还有很多的例子，相对高级一些，如有兴趣可以自行前往或者搜索之