Webdriver ждет появления одного из нескольких элементов

Есть ли способ заставить webDriverWait ждать появления одного из нескольких элементов и действовать соответственно на основе того, какой элемент появляется?

В настоящий момент я выполняю webDriverWait внутри цикла try, и если возникает исключение тайм-аута, я запускаю альтернативный код, ожидающий появления другого элемента. Это кажется неуклюжим. Есть ли способ лучше? Вот мой (неуклюжий) код:

try:
    self.waitForElement("//a[contains(text(), '%s')]" % mime)
    do stuff ....
except TimeoutException:
    self.waitForElement("//li[contains(text(), 'That file already exists')]")
    do other stuff ...

Он включает в себя ожидание целых 10 секунд, прежде чем он увидит сообщение о том, что файл уже существует в системе.

Функция waitForElement просто выполняет несколько вызовов webDriverWait:

def waitForElement(self, xPathLocator, untilElementAppears=True):
    self.log.debug("Waiting for element located by:\n%s\nwhen untilElementAppears is set to %s" % (xPathLocator,untilElementAppears))
    if untilElementAppears:
        if xPathLocator.startswith("//title"):
            WebDriverWait(self.driver, 10).until(lambda driver : self.driver.find_element_by_xpath(xPathLocator))
        else:
            WebDriverWait(self.driver, 10).until(lambda driver : self.driver.find_element_by_xpath(xPathLocator).is_displayed())
    else:   
        WebDriverWait(self.driver, 10).until(lambda driver : len(self.driver.find_elements_by_xpath(xPathLocator))==0)

У кого-нибудь есть предложения, чтобы сделать это более эффективным способом?

Ответ 1

Создайте функцию, которая берет карту идентификаторов для запросов xpath и возвращает идентификатор, который был сопоставлен.

def wait_for_one(self, elements):
    self.waitForElement("|".join(elements.values())
    for (key, value) in elements.iteritems():
        try:
            self.driver.find_element_by_xpath(value)
        except NoSuchElementException:
            pass
        else:
            return key

def othermethod(self):

    found = self.wait_for_one({
        "mime": "//a[contains(text(), '%s')]",
        "exists_error": "//li[contains(text(), 'That file already exists')]"
    })

    if found == 'mime':
        do stuff ...
    elif found == 'exists_error':
        do other stuff ...

Ответ 2

Что-то вроде этого:

def wait_for_one(self, xpath0, xpath1):
    self.waitForElement("%s|%s" % (xpath0, xpath1))
    return int(self.selenium.is_element_present(xpath1))