Check existence of two elements using Python, Selenium

Question

I am trying to check the existence of two elements(A,B) on a website.What I need is to click on the element A if it exists if not go ahead and look for B and click on it if it exists. Below is part of my sample code.

try:
        abc= WebDriverWait(driver, 20).until(EC.visibility_of_element_located((By.CSS_SELECTOR, "#rso > div > div > div:nth-child(1) > div > div > h3 > a")))
    except NoSuchElementException:
        continue
    except TimeoutException:
        continue
    else:
        element.click()


try:
        element = WebDriverWait(driver, 20).until(EC.visibility_of_element_located((By.CSS_SELECTOR, "#rso > div > div > div:nth-child(1) > div > div > h3 > a")))
    except NoSuchElementException:
        continue
    except TimeoutException:
        continue
    else:
        abc.click()
        time.sleep(randint(1, 15))


print('Process completed successfully')

The issue that I face is that the code only looks for the first element and doesn't go ahead into the next try. Any advice would be great.

Andersson · Accepted Answer · 2018-09-06 13:36:54Z

1

Try below to click first element and in case it's not found - click second:

try:
    WebDriverWait(driver, 20).until(EC.element_to_be_clickable((By.CSS_SELECTOR, "first.element"))).click()
except TimeoutException:
    WebDriverWait(driver, 20).until(EC.element_to_be_clickable((By.CSS_SELECTOR, "second.element"))).click()

P.S. Replace "first.element", "second.element" with real CSS selectors

If both elements might not be present on page:

try:
    WebDriverWait(driver, 20).until(EC.element_to_be_clickable((By.CSS_SELECTOR, "first.element"))).click()
except TimeoutException:
    try:
        WebDriverWait(driver, 20).until(EC.element_to_be_clickable((By.CSS_SELECTOR, "second.element"))).click()
    except TimeoutException:
        print("Both elements not found")

answered Sep 6, 2018 at 13:36

Andersson

52.8k18 gold badges83 silver badges132 bronze badges

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

1 Comment

Shashi Shankar Singh Over a year ago

Thanks a ton Andersson :)

user21945972 · Accepted Answer · 2023-06-22 03:57:00Z

0

wait = WebDriverWait(driver, 30)
try:
    elements = wait.until(
        EC.presence_of_all_elements_located(
            (By.CSS_SELECTOR, 'first.element'
                              'second.element')
        )
    )
    if elements[0].tag_name == 'first.element':
        print("login fail")
    elif elements[0].tag_name == 'second.element':
        print("login")
except:
    print("Both elements not found")

answered Jun 22, 2023 at 3:57

user21945972

133 bronze badges

1 Comment

Community Over a year ago

As it’s currently written, your answer is unclear. Please edit to add additional details that will help others understand how this addresses the question asked. You can find more information on how to write good answers in the help center.

Collectives™ on Stack Overflow

Check existence of two elements using Python, Selenium

2 Answers 2

1 Comment

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

1 Comment

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related