如何在SeleniumBase中正确遍历多个元素？现有写法是否合规？

场景说明 #

给定HTML结构如下：

<div style="background-color:#FFF4A3;">
  <h2>London</h2>
  <p>London is the capital city of England.</p>
  <p>London has over 9 million inhabitants.</p>
</div>

<div style="background-color:#FFC0C7;">
  <h2>Oslo</h2>
  <p>Oslo is the capital city of Norway.</p>
  <p>Oslo has over 700,000 inhabitants.</p>
</div>

<div style="background-color:#D9EEE1;">
  <h2>Rome</h2>
  <p>Rome is the capital city of Italy.</p>
  <p>Rome has over 4 million inhabitants.</p>
</div>

需求：找到所有<h2>元素，获取它们的父级<div>的文本。场景中存在不含h2的div和无父div的h2，因此无法直接遍历所有div来实现。

用户当前的SeleniumBase代码如下：

from seleniumbase import SB
from selenium.webdriver.common.by import By

with cm as SB(test=True, uc=True):
    sb.get("https://www.w3schools.com/html/tryit.asp?filename=tryhtml_div4")
    # accept choices
    sb.click("div#accept-choices")
    sb.switch_to_frame("iframe")
    for el in sb.find_elements("h2"):
        parent = el.find_element(By.XPATH, "..")
        print(parent.text, "
---")

疑问：sb.find_elements()返回原生Selenium WebElement对象，使用原生方法操作的遍历方式是否正确？若不正确，正确方式是什么？

结论与优化方案 #

1. 原始写法的问题 #

原始写法逻辑上能获取到父元素，但存在两个关键缺陷：

• 未校验父元素是否为<div>：如果某个h2的父元素不是div，会错误获取到其他标签的文本
• 无异常处理：如果某个h2没有父元素（或父元素查找失败），find_element()会抛出NoSuchElementException，直接中断遍历

2. 优化后的原生方法写法 #

添加父元素类型校验和异常捕获，保证遍历稳定执行：

from seleniumbase import SB
from selenium.webdriver.common.by import By
from selenium.common.exceptions import NoSuchElementException

with SB(test=True, uc=True) as sb:
    sb.get("https://www.w3schools.com/html/tryit.asp?filename=tryhtml_div4")
    sb.click("div#accept-choices")
    sb.switch_to_frame("iframe")
    
    for h2_el in sb.find_elements("h2"):
        try:
            # 直接定位父级div，而非泛化的父元素
            parent_div = h2_el.find_element(By.XPATH, "./parent::div")
            print(parent_div.text)
            print("---")
        except NoSuchElementException:
            # 处理无父div的h2场景，可跳过或记录日志
            print(f"跳过无父div的h2：{h2_el.text}")
            continue

3. 更贴合SeleniumBase的写法 #

SeleniumBase提供了更简洁的元素定位方法，可直接通过XPATH一次性定位到符合条件的父div，避免遍历后二次查找：

from seleniumbase import SB

with SB(test=True, uc=True) as sb:
    sb.get("https://www.w3schools.com/html/tryit.asp?filename=tryhtml_div4")
    sb.click("div#accept-choices")
    sb.switch_to_frame("iframe")
    
    # 直接定位所有包含h2的div元素
    target_divs = sb.find_elements("//div[h2]")
    for div in target_divs:
        print(div.text)
        print("---")

这种写法更高效，且天然过滤掉了不含h2的div，同时确保拿到的父元素是div。

内容的提问来源于stack exchange，提问作者robertspierre