如何用循环实现Selenium中按ID批量选择按钮的自动化操作

嘿，我来帮你把重复的点击代码改成更简洁的循环写法，这样不仅减少冗余代码，后续维护也更方便！

方法一：通过ID列表循环（最直接适配你的需求） #

你可以把需要点击的所有按钮ID放进一个列表，然后遍历这个列表逐个执行点击操作，替换掉原来重复的三行代码：

def test_new(self):
    driver = self.driver
    driver.get(self.base_url + "/en-US/docs/Web/HTML/Element/input/radio")
    # 存储所有需要点击的按钮ID
    target_button_ids = ["contactChoice1", "contactChoice2", "contactChoice3"]
    
    # 循环遍历每个ID，执行点击
    for btn_id in target_button_ids:
        driver.find_element(By.ID, btn_id).click()

优化：添加异常处理（避免单个元素找不到导致流程中断） #

如果担心某个按钮可能不存在，你可以在循环里加入异常捕获，确保其他按钮的点击不受影响：

def test_new(self):
    driver = self.driver
    driver.get(self.base_url + "/en-US/docs/Web/HTML/Element/input/radio")
    target_button_ids = ["contactChoice1", "contactChoice2", "contactChoice3"]
    
    for btn_id in target_button_ids:
        try:
            driver.find_element(By.ID, btn_id).click()
        except NoSuchElementException:
            print(f"警告：ID为 {btn_id} 的按钮未找到，已跳过")
            continue

更灵活的方法：通过共同属性定位一组元素 #

观察目标页面的单选框，它们都有相同的name="contact"属性，你可以直接定位这一组元素，然后循环点击，这样就算后续新增同组的单选框，代码也不用修改：

def test_new(self):
    driver = self.driver
    driver.get(self.base_url + "/en-US/docs/Web/HTML/Element/input/radio")
    # 定位所有name为contact的单选框
    contact_radios = driver.find_elements(By.NAME, "contact")
    
    # 遍历每个单选框并点击
    for radio in contact_radios:
        radio.click()
        # 可选：如果页面加载慢，可添加短暂等待避免操作过快
        time.sleep(0.5)

另外提一句：你代码里用的find_element_by_id是Selenium的旧语法，现在官方更推荐使用find_element(By.ID, btn_id)这种写法，刚好你已经导入了By类，直接用就好啦。

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