selenium结合tenacity的retry实现验证码失败重试
说在前面验证码登录的demo后续可以单独讲解，VIP学员对这部分应该都是掌握的，此处不再赘述。
本文假设了一个场景：
·你通过OCR识别的验证码是有一定的错误几率的
· 本文是通过识别后的验证码去加一个随机字符，如果取到的是''空字符则可能成功，否则必然不会成功
所涉及的python库：
· selenium
· ddddocr
· tenacity
上代码
细节详见注释：
from selenium import webdriver
from time import sleep
from tenacity import TryAgain, retry, wait_random
def get_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:
return False
driver = webdriver.Chrome()
driver.implicitly_wait(5)
driver.get('http://114.116.2.138: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存在的最大时间
def code_login():
'''
这个函数是用来反复验证码登录的
'''
# 验证码元素
ele_code = driver.find_element('css selector', '>img')
import ddddocr
ocr = ddddocr.DdddOcr()
code_text = ocr.classification(ele_code.screenshot_as_png)
# 当失败的时候尤其要注意: 清空已输入的内容
driver.find_element('css selector', 'input').clear()
test_data = ['1','a','','2','b']
from random import choice
choice_data = choice(test_data)
# 输入识别后的数据+随机字符
driver.find_element('css selector', 'input').send_keys(code_text + choice_data)
# 点击登录
driver.find_element('css selector', ">strong").click()
# 注意！ 你可以去操作网页，点击登录如果失败会弹出提示 "抱歉，验证码填写错误"
hint_locator = 'css selector', '.pc_inner>i'
if get_element(hint_locator): # 如果出现这个元素，就...
# 点击下验证码，刷新一下
driver.find_element('css selector', '>img').click()
# 抛出异常，重跑
raise TryAgain
code_login()
聊聊tenacity
这个库将重试这件事基本做透了。
1、 无条件重试
你疯了
from tenacity import retry
@retry
def retry_without_anything():
print('retry...')
raise Exception# 不加这个可不会重试
retry_without_anything()
2、重试指定次数后停止
from tenacity import retry, stop_after_attempt
@retry(stop=stop_after_attempt(3))
def retry_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, in iter
raise retry_exc from fut.exception()
tenacity.RetryError: RetryError[<Future at 0x16628b45460 state=finished raised Exception>]
别担心，你可能在最后一次的时候就不抛出异常了。
3、过一定时间后停止重试
from tenacity import retry, stop_after_delay
importarrow
from time import sleep
@retry(stop=stop_after_delay(10))
def retry_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)))
def retry_multi_conditions():
print("10秒后或者5次重试后停止重试")
raise Exception
5、重试间隔
之前的重试是无缝衔接的，你可以让重试之间有延迟。
@retry(wait=wait_fixed(2))
def retry_wait_time1():
print(arrow.now().format('YYYY-MM-DD HH:mm:ss'))
print("每次重试前等待2秒")
raise Exception
retry_wait_time1()
当然上面的仍然是个无限重试，你可以组合前面的停止。
@retry(wait=wait_fixed(2),stop=stop_after_attempt(3))
重试等待间隔可以设定一个区间(最大最小值)
from tenacity import retry, stop_after_attempt, wait_random
importarrow
@retry(wait=wait_random(min=1,max=4),stop=stop_after_attempt(3))
def retry_wait_time1():
print(arrow.now().format('YYYY-MM-DD HH:mm:ss'))
print("每次重试前等待1~4秒区间")
raise Exception
retry_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))
def retry_if_exception():
print('retry...')
print(10/0) # 现在触发的就是ZeroDivisionError，如果把此处改为 print(a)，则遇到的是NameError，那就不会重试
raise Exception
retry_if_exception()
引发 TryAgain 异常随时显式重试。比如这样：
@retry
def try_if_condition():
result = 23
if result == 23:
raise TryAgain
上面的demo中就用到了这个。官网还有很多的例子，相对高级一些，如有兴趣可以自行前往或者搜索之。
页:
[1]
