是否有一个 Selenium 包装器来查找具有给定选择器类型作为输入的元素（By.css_selector、By.xpath、By.class ...）？

Question

Python Selenium WebDriver 中提供了以下方法来查找多个元素。 这些方法将返回元素列表。

find_elements_by_name
find_elements_by_xpath
find_elements_by_link_text
find_elements_by_partial_link_text
find_elements_by_tag_name
find_elements_by_class_name
find_elements_by_css_selector

我需要在那些 find_elements_by_xxxx 之上使用包装器（一个函数），以便找到具有给定选择器类型的给定元素的任何元素。

Answer 1

为什么不写一个以By作为参数的函数呢？

def find_elements_by(driver, by, locator):
    return driver.find_elemements(by, locator)

然后，你可以这样称呼它：

xpath_result = find_elements_by(driver, By.XPATH, "//input")

class_result = find_elements_by(driver, By.CLASS_NAME, "className")

css_result = find_elements_by(driver, By.CSS_SELECTOR, "css.selector")

您可以将其更改为静态，因此您只需使用driver.find_elements_by(By.CSS_SELECTOR, "css.selector")

Answer 2

大多数 WebDriver 实现都有一个find_element方法，它接受一个By值和一个选择器值，这有效地完成了 Christine 所写的。

driver = ...
driver.find_element(By.ID, "element_id")
driver.find_element(By.CSS_SELECTOR, "css.selector")

python 的 ChromeDriver 中的实现如下所示：

def find_element(self, by=By.ID, value=None):
    """
    Find an element given a By strategy and locator. Prefer the find_element_by_* methods when
    possible.

    :Usage:
        element = driver.find_element(By.ID, 'foo')

    :rtype: WebElement
    """
    if self.w3c:
        if by == By.ID:
            by = By.CSS_SELECTOR
            value = '[id="%s"]' % value
        elif by == By.TAG_NAME:
            by = By.CSS_SELECTOR
        elif by == By.CLASS_NAME:
            by = By.CSS_SELECTOR
            value = ".%s" % value
        elif by == By.NAME:
            by = By.CSS_SELECTOR
            value = '[name="%s"]' % value
    return self.execute(Command.FIND_ELEMENT, {
        'using': by,
        'value': value})['value']