python自动化--设置超时等待

适用情况：需要循环查询一个接口，直到返回预期内结果时再进行下一步

# !/usr/bin/python3
# -*- coding:utf-8 -*-
import time
from func_timeout import func_set_timeout
from api.test_api import test_api
@func_set_timeout #注解器设置超时时间为10s
def wait_success(uuid):
    while True: #反复查询直到返回预期结果
        res = test_api.test(uuid)
        if res.status_code == 200:
            res_status = res.json()['data']
            if 'success' not in str(res_status):
                return True
            else:
                time.sleep(1)
        else:
            return False

testcase中调用：

  from common.wait import wait_success
  class Test:
  	def setup_class(self):
        self.uuid = 'uuid'
        if wait_success(self.uuid):
            LOG.debug('初始成功，可以进行测试了')
        else:
            LOG.error('初始化失败')