一般在做selenium时会有,回放快慢的需求。
实现思路:
1.一般写selenium会自定义findelement函数,来实现查找元素。
2.在查找函数上加个睡眠时间的装饰器,函数执行完等待若干秒
3.同理可以举一返三的使用,装饰器,可以实现很多的功能。
def replay(retime):
"""
设置回放时间,装饰器
:param retime: 回放时间,毫秒
:return: 无
"""
def _wrapper(func):
def wrapper(*args,**kwargs):
ret=func(*args,**kwargs)
time.sleep(float(retime)/1000)
return ret
return wrapper
return _wrapper
找元素,函数:
@replay(500) #等待500毫秒
def find_element(self,*loc):
"""
在指定时间内,查找元素;否则抛出异常
:param loc: 定位器
:return: 元素 或 抛出异常
"""
TimeOut = 20
try:
self.driver.implicitly_wait(TimeOut) #智能等待;超时设置
element = self.driver.find_element(*loc) #如果element没有找到,到此处会开始等待
if self.isDisplayTimeOut(element,TimeOut):
self.hightlight(element) #高亮显示
self.driver.implicitly_wait(0) # 恢复超时设置
return element
else:
raise ElementNotVisibleException #抛出异常,给except捕获
except (
NoSuchElementException,
ElementNotVisibleException
) as ex:
self.getImage
raise ex
else:
self.getImage