def wait_element(element_type, text, wait_time=10): wait = WebDriverWait(d, wait_time) if wait_time != 10: wait._timeout = wait_time element = wait.until(EC.presence_of_element_located((element_type, text))) return element