Python, Selenium - handle multiple wait-conditions

Question

I have a file called webdriver.py which implements methods from selenium.webdriver library. Had a wait-function that were handling most of the cases I need:

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)

Where func is a selector.

The problem is that sometimes the DOM element would be invisible, causing exception and test failure. So I would like to extend wait_for to also wait for the element to become visible.

Something like

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 is a selenium.driver.expected_conditions

This would not work of course - either until().until() syntax is not supported.. or something else happens, like EC doesn't exist.

Any ideas?

Mesut GUNES · Accepted Answer · 2015-11-11 15:15:21Z

1

You can use EC.presence_of_element_located for waiting an element in DOM to visible, no need for second .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)

answered Nov 11, 2015 at 15:15

Mesut GUNES

7,4612 gold badges36 silver badges52 bronze badges

Sign up to request clarification or add additional context in comments.

4 Comments

0leg Over a year ago

Thank you for help, I also used "from selenium.webdriver.support import expected_conditions as EC" to fix EC problem.

0leg Over a year ago

I also think it should be: "By.ByCssSelector" according to seleniumhq.github.io/selenium/docs/api/java/org/openqa/selenium/…

Mesut GUNES Over a year ago

@h3d0 it should be By.CSS_SELECTOR for python check selenium-python.readthedocs.org/…

0leg Over a year ago

Indeed,By.CSS_SELECTOR it is.

Collectives™ on Stack Overflow

Python, Selenium - handle multiple wait-conditions

1 Answer 1

4 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

4 Comments

Your Answer

Sign up or log in

Post as a guest

Related