Python，Selenium - 处理多个等待条件

Question

我有一个名为webdriver.py的文件，它实现了selenium.webdriver库中的方法。有同时处理最多的，我需要的情况下，等待功能：

def wait_for(self, func, target=None, timeout=None, **kwargs): 
    timeout = timeout or self.timeout 
    try: 
     return WebDriverWait(self, timeout).until(func) 
    except TimeoutException: 
     if not target: 
      raise WebDriverException('Wait for: "%s" failed!' % inspect.getsource(func).strip()) 
     raise NoSuchElementException(target) 

凡func是一个选择。

的问题是，有时DOM元素将是不可见的，从而引起异常，并检测失败。所以我想延长wait_for也等待元素变得可见。

喜欢的东西

def wait_for(self, func, target=None, timeout=None, **kwargs): 
    timeout = timeout or self.timeout 
    try: 
     return WebDriverWait(self, timeout).until(EC.element_to_be_clickable(func)).until(func) 
    except TimeoutException: 
     if not target: 
      raise WebDriverException('Wait for: "%s" failed!' % inspect.getsource(func).strip()) 
     raise NoSuchElementException(target) 

EC是selenium.driver.expected_conditions

这当然不是工作 - 不支持任何until().until()语法..什么事情发生，就像EC不存在。

任何想法？

Answer 1

您可以使用EC.presence_of_element_located等待DOM中的元素可见，不需要第二个.until。

def wait_for(self, func, target=None, timeout=self.timeout): 
    try: 
     WebDriverWait(self, timeout).until(EC.presence_of_element_located((By.CSS_SELECTOR, func))) 
    except TimeoutException: 
     if not target: 
      raise WebDriverException('Wait for: "%s" failed!' % inspect.getsource(func).strip()) 
     raise NoSuchElementException(target)