51Testing软件测试论坛

 找回密码
 (注-册)加入51Testing

QQ登录

只需一步,快速开始

微信登录,快人一步

查看: 279|回复: 0

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

[复制链接]
  • TA的每日心情
    无聊
    昨天 09:12
  • 签到天数: 918 天

    连续签到: 3 天

    [LV.10]测试总司令

    发表于 2023-3-14 09:57:32 | 显示全部楼层 |阅读模式
    说在前面
      验证码登录的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', '[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
      这个库将重试这件事基本做透了。
      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
      import  arrow
      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
      import  arrow
      @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 下一条

    小黑屋|手机版|Archiver|51Testing软件测试网 ( 沪ICP备05003035号 关于我们

    GMT+8, 2024-3-28 19:20 , Processed in 0.065504 second(s), 24 queries .

    Powered by Discuz! X3.2

    © 2001-2024 Comsenz Inc.

    快速回复 返回顶部 返回列表