Selenium Python：嵌套元素点击与循环执行异常排查

页面存在多组如下结构的元素：

<div class="course-lesson__course-wrapper" data-qa-level="z1">
   <div class="course-lesson__course-title"></div>
   <section class="course-lesson__wrap" data-qa-lesson="trial"></section>
   <section class="course-lesson__wrap" data-qa-lesson="trial"></section>
</div>

多页面采用同一布局，需求是获取所有data-qa-level="z1"容器内的元素，点击其中data-qa-lesson="trial"的元素。但下方编写的代码无法正常运行：

#finds all the elements for z1 - ...etc
listofA1 = driver.find_element(By.CSS_SELECTOR, "div.course-lesson__course-wrapper:nth-child(1)")
for elemen in listofA1:
    #checks for the attribute i need to see if it's clickable
    elementcheck = elemen.getAttribute("data-qa-lesson")
    if elementcheck == "objective":
        elemen.click()

        #do some stuff then go back to main and begin again on the next element
        driver.get(home_link)

1. 元素定位逻辑错误 #

• 原代码用find_element()只能获取单个元素，无法实现遍历，必须使用find_elements()（复数形式）才能拿到所有匹配的容器。
• 原CSS选择器div.course-lesson__course-wrapper:nth-child(1)仅选中第一个容器，不符合“获取所有”的需求，换成div[data-qa-level="z1"]直接定位所有目标容器更精准。

2. 条件判断不匹配 #

需求是点击data-qa-lesson="trial"的元素，但代码里判断的是elementcheck == "objective"，条件完全不匹配，导致永远不会触发点击操作。

3. 遍历对象错误 #

原代码直接遍历单个容器元素listofA1，逻辑错误，应该遍历每个z1容器内的section元素，或者直接定位所有符合条件的section元素。

4. 页面跳转后的元素失效问题 #

执行driver.get(home_link)跳转页面后，之前获取的元素会变成“过期元素”，后续操作会报错，需要重新定位元素。

修正后的代码方案 #

方案一：逐个处理容器内元素（需重新定位）

# 获取所有data-qa-level="z1"的容器
z1_containers = driver.find_elements(By.CSS_SELECTOR, "div[data-qa-level='z1']")

for idx in range(len(z1_containers)):
    # 每次循环重新定位容器，避免页面跳转后元素失效
    current_container = driver.find_elements(By.CSS_SELECTOR, "div[data-qa-level='z1']")[idx]
    # 在当前容器内查找所有data-qa-lesson="trial"的元素
    trial_elements = current_container.find_elements(By.CSS_SELECTOR, "section[data-qa-lesson='trial']")
    
    for elem in trial_elements:
        elem.click()
        
        # 这里执行你需要的操作
        # ...
        
        # 返回主页
        driver.get(home_link)

方案二：先收集所有目标链接再处理（更稳定）

如果点击trial元素是跳转到新页面，可以先收集所有跳转链接，再逐个访问处理，避免元素失效问题：

trial_links = []
# 先收集所有目标元素的跳转链接
z1_containers = driver.find_elements(By.CSS_SELECTOR, "div[data-qa-level='z1']")
for container in z1_containers:
    trial_elements = container.find_elements(By.CSS_SELECTOR, "section[data-qa-lesson='trial']")
    for elem in trial_elements:
        # 若元素是a标签，用get_attribute('href')获取链接；非a标签则根据实际属性调整
        link = elem.get_attribute('href')
        if link:
            trial_links.append(link)

# 逐个访问链接并处理
for link in trial_links:
    driver.get(link)
    # 执行需要的操作
    # ...
    driver.get(home_link)

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