Установить обработчик исключений по умолчанию, если не удается найти элемент в селене? - PullRequest
Новая
       24

Установить обработчик исключений по умолчанию, если не удается найти элемент в селене?

3 голосов
/ 10 февраля 2012

Довольно часто мой сценарий селен будет запускаться, и вдруг он вылетает с ошибкой: 

<class 'selenium.common.exceptions.NoSuchElementException'>
Message: u'Unable to locate element: {"method":"id","selector":"the_element_id"}' 
<traceback object at 0x1017a9638>

И если я запускаю в интерактивном режиме (python -i myseltest.py), если я просто сделаю что-то вроде: 

driver.switch_to_window(driver.window_handles[0])

, а затем снова запустите конкретный find_element_by_id(), это будет успешно.* вызов в случае возникновения исключения?

1 Ответ

3 голосов
/ 10 февраля 2012

<UPDATE>
Во-первых, рассмотрите возможность использования неявного ожидания , потому что эта проблема возникает в основном, когда присутствие элемента инициируется на странице javascript, и вы можете получить задержку в несколько секунд между событием DOMReady и функциями js или выполнением запросов ajax.
</UPDATE>

Это 

from selenium.webdriver import Firefox  
from selenium.webdriver.support.ui import WebDriverWait  

from selenium.common.exceptions import TimeoutException  
from selenium.common.exceptions import NoSuchElementException

class MyFirefox(Firefox):

    RETRIES = 3
    TIMEOUT_SECONDS = 10

    def find_element_by_id(self, id):
        tries = 0
        element = None

        while tries < self.RETRIES:
            try:
                element = WebDriverWait(self, self.TIMEOUT_SECONDS).until(
                    lambda browser: super(MyFirefox, browser).find_element_by_id(id)
                )   
            except TimeoutException:
                tries = tries + 1
                self.switch_to_window(self.window_handles[0])
                continue
            else:
                return element

        raise NoSuchElementException("Element with id=%s was not found." % id)
...