使用“OR”组合多个硒等待？

Question

我的selenium代码通过等待网站标题更改来检查完成的子例程，该子例程运行良好。代码如下所示：

waitUntilDone = WebDriverWait(session, 15).until(EC.title_contains(somestring))

但是，这有时可能会失败，因为站点的登录页面在手动网站访问后发生变化。服务器会记住您离开的位置。这迫使我检查另一个条件（网站标题="notestring2）。

以下是我到目前为止所想到的（据我所知也有效）：

    try:
        waitUntilDone = WebDriverWait(session, 15).until(EC.title_contains(somestring)) # the old condition
    except:
        try:
            waitUntilDone = WebDriverWait(session, 15).until(EC.title_contains(somestring2)) # the new other condition which is also valid
        except:
            print "oh crap" # we should never reach this point

这些条件中的任何一个始终为真。我不知道你是哪一个。

有没有什么方法可以在这些等待中包含一个“或”，或者让try/⃣块看起来更好？

Answer 1

看起来硒会让你通过创建自己的类来做到这一点。查看此处的文档：http://selenium-python.readthedocs.io/waits.html

下面是针对您的情况的快速示例。请注意，关键是在类中有一个名为 __call__ 的方法，用于定义所需的检查。Selenium将每500毫秒调用该函数，直到它返回True或一些非空值。

class title_is_either(object):

  def __init__(self, locator, string1, string2):
    self.locator = locator
    self.string1 = string1
    self.string2 = string2

  def __call__(self, driver):
    element = driver.find_element(*self.locator)   # Finding the referenced element
    title = element.text
    if self.string1 in title or self.string2 in title
        return element
    else:
        return False

# Wait until an element with id='ID-of-title' contains text from one of your two strings
somestring = "Title 1"
somestring2 = "Title 2"

wait = WebDriverWait(driver, 10)
element = wait.until(title_is_either((By.ID, 'ID-of-title'), somestring, somestring2))