博客

把find_element方法封装，循环查找，返回element，如果异常，等待2秒，打印异常信息，如果10次循环结束仍未找到元素，就在向上抛出未找到元素。

直接上代码喽！

import time
def find_element(driver,idOrXpath,n=10):
    for i in xrange(n):
        try:
            if idOrXpath.startswith('/'):
                ret = driver.find_element_by_xpath(idOrXpath)
            else:
                ret = driver.find_element_by_id(idOrXpath)
            return ret
        except Exception as err:
            print err.message
            time.sleep(2)
    raise Exception('element %s could not be found'%idOrXpath)