How to go back to top of loop after finishing?-CodePudding

When visiting some websites with selenium there are times where the page doesn't load correctly. So I want to write a code that checks if the website loaded correctly by searching for a particular button. If it can't find the button it needs to refresh until it finds the button and if the code does find the button it needs to execute the rest of the code.

So for example: Button not there: >>> refresh >>> (checks again if button is not there) Button not there >>> refresh >>> (checks again if button is not there) Button is there >>> rest of code

The loop that I currently have looks like this but after refreshing the loop doesn't restart and runs the else: function. So the question is how do I make a loop that restarts the loop after it refreshes.

while not (driver.find_elements(By.CLASS_NAME, "button")):
    driver.refresh()
else: 
    Rest of code

Help would be much appreciated, thanks in advance

CodePudding user response：

You can have an infinite while loop, and an if condition with find_elements, Please note that find_elements does not return any exception, it returns a list of web elements or either 0.

Code:

while True:
    try:
        if len(driver.find_elements(By.XPATH, "xpath of the button")) >0:
            print("Button is present, since find elements list is non empty")
            # execute the code, may be click on the button or whatever it is.
            #make sure to exit from infinite loop as well
            break
        else:
            driver.refresh()
            #may be put some delay here to let button available to click again.
    except:
        print("There was some problem trying to find the button element, tearing it down")
        break