检查pag上是否存在h1或h2

Question

我有这个密码：

driver = webdriver.Chrome()
driver.implicitly_wait(60)
driver.get(link)

driver.add_cookie({'name' : 'access_token', 'value' : token})
driver.refresh()

time.sleep(5)
driver.find_element_by_name('next').click()
time.sleep(5)

如何检查页上是否存在或标记元素？ 我这样试过：

falha = driver.find_element_by_tag_name('h1')
sucesso = driver.find_element_by_tag_name('h2')

if falha.text == "Por segurança":
    print ("Retorno: " + element.text + "\nToken: " + token)

if sucesso.text == "Pronto":
    print("Retorno: " + element.text + "\nToken: " + token)

driver.close()

但是我的代码很简单，如果其中一个元素不存在。你知道吗

Answer 1

您可以为此使用driver.find_elements_by_tag_name()方法，它将返回匹配标记元素的列表。你知道吗

然后您可以使用它的大小来确定它是否存在。请尝试以下代码：

elementsh1 = driver.find_elements_by_tag_name('h1')
elementsh2 = driver.find_elements_by_tag_name('h2')

if len(elementsh1) > 0 :
    print("=> h1 is present...")
    print(elementsh1[0].text)
else:
    print("=> h1 tags are not there...")

if len(elementsh2) > 0 :
    print("=> h2 is present...")
    print(elementsh2[0].text)
else:
    print("=> h2 tags are not there...")

我希望这有帮助。。。你知道吗