使用Python Behave BDD时无法找到表格元素并更新时长

代码中的核心问题 #

• Driver对象引用错误：代码里直接用driver，但初始化的驱动实例是context.driver，会导致找不到驱动对象报错。
• 文本匹配精度不足：element.text可能因元素包含空格、换行或隐藏文本导致匹配失败，需用更精准的文本获取方式。
• 全局定位元素风险：查找duration-column时用全局定位，可能定位到其他任务的时长列，而非目标任务对应的列。
• 输入框清空不彻底：仅用一次BACKSPACE无法确保清空原有内容，应使用专门的清空方法。
• 缺少显式等待：未等待元素加载完成就执行操作，容易触发元素未找到的异常。

修正后的代码 #

@then('I get the data of elements with class "gantt_tree_content"')
def step_update_end_date(context):
    from selenium.webdriver.support.ui import WebDriverWait
    from selenium.webdriver.support import expected_conditions as EC
    from selenium.webdriver.common.keys import Keys

    element_list = context.driver.find_elements(By.CLASS_NAME, "gantt_tree_content")
    task_exists = False
    target_task_text = "Task by SK-1"

    for element in element_list:
        # 用textContent获取精准文本，去除首尾空格避免匹配偏差
        current_text = element.get_attribute("textContent").strip()
        if current_text == target_task_text:
            task_exists = True
            # 基于当前任务元素定位对应行的时长列（需根据实际DOM结构调整）
            parent_row = element.find_element(By.XPATH, "./ancestor::tr")
            column = parent_row.find_element(By.CLASS_NAME, "duration-column")
            
            # 等待时长列可点击
            WebDriverWait(context.driver, 10).until(EC.element_to_be_clickable(column))
            column.click()

            # 等待时长输入框加载完成
            edit = WebDriverWait(context.driver, 10).until(
                EC.presence_of_element_located((By.XPATH, "//input[@name='duration']"))
            )
            
            # 清空输入框后输入新时长
            edit.clear()
            edit.send_keys("6")
            edit.send_keys(Keys.ENTER)
            break

额外优化建议 #

• 适配DOM结构：如果任务行的DOM层级不同，需调整parent_row的定位XPath，确保定位到目标任务对应的时长列。
• 添加异常处理：加入try-except块捕获元素未找到、点击失败等异常，方便排查问题。
• 增加调试日志：在关键步骤添加日志输出（如匹配到任务、成功点击时长列），便于调试流程。

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