Function find_element

src/py/reactpy/tests/tooling/select.py:37–67

Find an element by a selector. Parameters: element: The tree to search. selector: A function that returns True if the element matches. first: If True, return the first element found. If False, raise an error if multiple ele

(
    element: VdomJson,
    selector: Selector,
    *,
    first: bool = False,
)

def find_element(
    element: VdomJson,
    selector: Selector,
    *,
    first: bool = False,
) -> tuple[VdomJson, ElementInfo]:
    """Find an element by a selector.

    Parameters:
        element:
            The tree to search.
        selector:
            A function that returns True if the element matches.
        first:
            If True, return the first element found. If False, raise an error if
            multiple elements are found.

    Returns:
        Element info, or None if not found.
    """
    find_iter = find_elements(element, selector)
    found = next(find_iter, None)
    if found is None:
        raise ValueError("Element not found")
    if not first:
        try:
            next(find_iter)
            raise ValueError("Multiple elements found")
        except StopIteration:
            pass
    return found


def find_elements(