Scraping - 如何在 python 中使用 selenium 查找 webelement 的标签

Question

例如，如果我有一个 webelement，我想检查它是否包含 h2 标签。这可能吗？如果是，那么如何

from selenium import webdriver
import re
import pandas as pd
from bs4 import BeautifulSoup
chrome_path = r"C:\Users\ajite\Desktop\web scraping\chromedriver.exe"
driver = webdriver.Chrome(chrome_path)
driver.get('....')

header = driver.find_element_by_xpath('/html/body/div[5]/div[2]/div/ol/li[1]/div/div/div[1]/h2')

if header.contains('h2'):
    print("successful")
else:
    print("unsuccesful")

header 变量包含 h2 标记，但我不能使用 header.contains，因为它是 web 元素而不是文本

错误

Traceback (most recent call last):
  File "test2.py", line 11, in <module>
    if header.contains('h2'):
AttributeError: 'WebElement' object has no attribute 'contains'

Answer 1

如果你想检查 webelement 是否包含`标签，你可以使用下面的代码块：

header = driver.find_element_by_xpath('/html/body/div[5]/div[2]/div/ol/li[1]/div/div/div[1]/h2')
if 'h2' in header.tag_name:
    print("successful")
else:
    print("unsuccesful")

Answer 2

您可以使用tag_name 方法获取webelement 的标签名称，如下所示。

from selenium import webdriver
chrome_path = r"path"
driver = webdriver.Chrome(chrome_path)
driver.get("http://www.google.co.in")
print ("Launching chrome")
element = driver.find_element_by_xpath('//*[@title="Search"]')
print("element.tag_name "+element.tag_name)
if element.tag_name == "input":
    print("successful")
else:
    print("unsuccesful")
driver.quit()