Selenium中如何将已定位的WebElement复用为定位器？

Selenium的text_to_be_present_in_element_attribute预期条件（EC）的第一个参数要求传入定位器元组（比如(By.ID, "element-id")），而非已定位完成的WebElement。当直接传入WebElement时，EC内部会尝试调用driver.find_element(*locator)，但WebElement不是可迭代的元组类型，因此触发TypeError。

方案1：复用定位器（推荐，保持单点维护优势） #

改造元素查找逻辑，将定位器信息单独抽离，既保留“一处修改全生效”的优势，又满足EC的参数要求：

首先单独定义定位器：

from selenium.webdriver.common.by import By

def get_city_filter_locator():
    # 根据实际DOM结构调整定位方式，示例用ID定位
    return (By.ID, "city-filter-input")

修改原元素查找方法，基于定位器获取元素：

def get_city_filter_field(driver: WebDriver) -> WebElement:
    return driver.find_element(*get_city_filter_locator())

最后在add_city_filter方法中传入定位器调用EC：

def add_city_filter(driver: WebDriver, city: str) -> None:
    city_filter: WebElement

    if not is_search_filter_box_visible(driver):
        click_show_filters_button(driver)

    city_filter = get_city_filter_field(driver)
    WebDriverWait(driver, 10).until(EC.element_to_be_clickable(city_filter)).click()
    city_filter.send_keys(city)
    Utility.wait_for_network_inactivity(driver)

    # 传入定位器而非WebElement
    WebDriverWait(driver, 20).until(
        EC.text_to_be_present_in_element_attribute(get_city_filter_locator(), "value", "")
    )

    if not has_applied_filter(driver, Filters.FilterOptions.City):
        raise Exception("Filter was not added")

方案2：自定义预期条件（适用于无法修改原方法的场景） #

如果无法调整原有元素查找方法，可以自定义一个接受WebElement的预期条件，绕过定位器要求：

from selenium.webdriver.support.expected_conditions import ExpectedCondition
from selenium.webdriver.remote.webelement import WebElement

class TextPresentInElementAttribute(ExpectedCondition):
    def __init__(self, element: WebElement, attribute: str, text: str):
        self.element = element
        self.attribute = attribute
        self.text = text

    def __call__(self, driver):
        attr_value = self.element.get_attribute(self.attribute)
        # 处理属性值为None的情况，判断目标文本是否在属性值中
        return self.text in attr_value if attr_value else False

在代码中使用自定义EC：

def add_city_filter(driver: WebDriver, city: str) -> None:
    # ... 原有代码 ...

    WebDriverWait(driver, 20).until(
        TextPresentInElementAttribute(city_filter, "value", "")
    )

    # ... 原有代码 ...

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