25

我将 Selenium 2 / WebDriver 与 Python API 一起使用，如下所示：

from selenium.webdriver.support import expected_conditions as EC

# code that causes an ajax query to be run

WebDriverWait(driver, 10).until( EC.presence_of_element_located( \
    (By.CSS_SELECTOR, "div.some_result")));

我想等待返回结果（）或“未找到”字符串。那可能吗？有点儿：div.some_result

WebDriverWait(driver, 10).until( \
    EC.presence_of_element_located( \
         (By.CSS_SELECTOR, "div.some_result")) \
    or 
    EC.presence_of_element_located( \
         (By.CSS_SELECTOR, "div.no_result")) \
);

我意识到我可以使用 CSS 选择器 ( div.no_result, div.some_result) 来做到这一点，但是有没有办法使用 Selenium 预期条件方法来做到这一点？

4

3 回答 3

25

我是这样做的：

class AnyEc:
    """ Use with WebDriverWait to combine expected_conditions
        in an OR.
    """
    def __init__(self, *args):
        self.ecs = args
    def __call__(self, driver):
        for fn in self.ecs:
            try:
                res = fn(driver)
                if res:
                    return True
                    # Or return res if you need the element found
            except:
                pass

然后称它为...

from selenium.webdriver.support import expected_conditions as EC
# ...
WebDriverWait(driver, 10).until( AnyEc(
    EC.presence_of_element_located(
         (By.CSS_SELECTOR, "div.some_result")),
    EC.presence_of_element_located(
         (By.CSS_SELECTOR, "div.no_result")) ))

显然，同样实现一个AllEc类也是微不足道的。

NB。try:块很奇怪。我很困惑，因为有些 EC 返回 true/false，而其他 EC 会抛出NoSuchElementExceptionFalse。异常被 WebDriverWait 捕获，所以我的 AnyEc 事情产生了奇怪的结果，因为第一个抛出异常意味着 AnyEc 没有进行下一个测试。

于 2013-05-09T14:32:59.867 回答
9

古老的问题，但是，

WedDriverWait在一个独立于 selenium 的示例中，考虑如何工作：

def is_even(n):
    return n % 2 == 0

x = 10

WebDriverWait(x, 5).until(is_even)

这将等待最多 5 秒is_even(x)才能返回True

现在，WebDriverWait(7, 5).until(is_even)将需要 5 秒，他们会引发 TimeoutException

事实证明，您可以返回任何非 Falsy 值并捕获它：

def return_if_even(n):
    if n % 2 == 0:
        return n
    else:
        return False

x = 10
y = WebDriverWait(x, 5).until(return_if_even)
print(y) # >> 10

现在考虑如何EC工作的方法：

print(By.CSS_SELECTOR) # first note this is only a string
>> 'css selector'

cond = EC.presence_of_element_located( ('css selector', 'div.some_result') )
# this is only a function(*ish), and you can call it right away:

cond(driver)
# if element is in page, returns the element, raise an exception otherwise

您可能想尝试以下方法：

def presence_of_any_element_located(parent, *selectors):
    ecs = []
    for selector in selectors:
        ecs.append(
            EC.presence_of_element_located( ('css selector', selector) )
        )

     # Execute the 'EC' functions agains 'parent'
     ecs = [ec(parent) for ec in ecs]

     return any(ecs)

如果在中找不到时EC.presence_of_element_located返回，这将起作用，但它会引发异常，一个易于理解的解决方法是：Falseselectorparent

def element_in_parent(parent, selector):
    matches = parent.find_elements_by_css_selector(selector)
    if len(matches) == 0:
        return False
    else:
        return matches

def any_element_in_parent(parent, *selectors):
    for selector in selectors:
        matches = element_in_parent(parent, selector)
        # if there is a match, return right away
        if matches:
            return matches
    # If list was exhausted
    return False

# let's try 
any_element_in_parent(driver, 'div.some_result', 'div.no_result')
# if found in driver, will return matches, else, return False

# For convenience, let's make a version wich takes a tuple containing the arguments (either one works):
cond = lambda args: any_element_in_parent(*args)
cond( (driver, 'div.some_result', 'div.no_result') )
# exactly same result as above

# At last, wait up until 5 seconds for it 
WebDriverWait((driver, 'div.some_result', 'div.no_result'), 5).until(cond)

我的目标是解释，artfulrobot已经给出了一个片段，用于实际EC方法的一般使用，请注意

class A(object):
    def __init__(...): pass
    def __call__(...): pass

只是定义函数的一种更灵活的方式（实际上是“类函数”，但在这种情况下无关紧要）

于 2017-01-29T06:15:52.483 回答
1

不完全通过 EC，但确实达到了相同的结果 - 有奖金。
仍然使用WebDriverWait's方法，但在 lambda 表达式until()中传递纯方法：find_elements_*()

WebDriverWait(driver, 10).until(lambda driver: driver.find_elements_by_id("id1") or \
                                               driver.find_elements_by_css_selector("#id2"))[0]

这些find_elements_*()方法返回所有匹配元素的列表，如果没有，则返回一个空元素 - 这是一个布尔值 false。因此，如果第一个调用没有找到任何东西，则评估第二个；重复直到他们中的任何一个找到匹配项，或者时间用完。

奖励 - 当它们返回值[0]时，最后的索引实际上会返回匹配的元素 - 如果你对它有任何用处，在后续调用中。

于 2020-05-19T16:25:24.260 回答