seleniumbase 4.39.6a2__py3-none-any.whl → 4.40.0__py3-none-any.whl

seleniumbase/core/sb_cdp.py

@@ -2014,6 +2014,150 @@ class CDPMethods():
             % (selector, timeout, plural)
         )
 
+    def wait_for_any_of_elements_visible(self, *args, **kwargs):
+        """Waits for at least one of the elements to be visible.
+        Returns the first element that is found.
+        The input is a list of elements. (Should be CSS selectors)
+        Optional kwargs include: "timeout" (used by all selectors).
+        Raises an exception if no elements are visible by the timeout.
+        Examples:
+            sb.cdp.wait_for_any_of_elements_visible("h1", "h2", "h3")
+            OR
+            sb.cdp.wait_for_any_of_elements_visible(["h1", "h2", "h3"]) """
+        selectors = []
+        timeout = None
+        for kwarg in kwargs:
+            if kwarg == "timeout":
+                timeout = kwargs["timeout"]
+            elif kwarg == "by":
+                pass  # Autodetected
+            elif kwarg == "selector" or kwarg == "selectors":
+                selector = kwargs[kwarg]
+                if isinstance(selector, str):
+                    selectors.append(selector)
+                elif isinstance(selector, list):
+                    selectors_list = selector
+                    for selector in selectors_list:
+                        if isinstance(selector, str):
+                            selectors.append(selector)
+            else:
+                raise Exception('Unknown kwarg: "%s"!' % kwarg)
+        if not timeout:
+            timeout = settings.SMALL_TIMEOUT
+        for arg in args:
+            if isinstance(arg, list):
+                for selector in arg:
+                    if isinstance(selector, str):
+                        selectors.append(selector)
+            elif isinstance(arg, str):
+                selectors.append(arg)
+        if not selectors:
+            raise Exception("The selectors list was empty!")
+        start_ms = time.time() * 1000.0
+        stop_ms = start_ms + (timeout * 1000.0)
+        any_present = False
+        for i in range(int(timeout * 10)):
+            for selector in selectors:
+                if self.is_element_visible(selector):
+                    return self.select(selector)
+                if self.is_element_present(selector):
+                    any_present = True
+            now_ms = time.time() * 1000.0
+            if now_ms >= stop_ms:
+                break
+            time.sleep(0.1)
+        plural = "s"
+        if timeout == 1:
+            plural = ""
+        if not any_present:
+            # None of the elements exist in the HTML
+            raise Exception(
+                "None of the elements {%s} were present after %s second%s!" % (
+                    str(selectors),
+                    timeout,
+                    plural,
+                )
+            )
+        raise Exception(
+            "None of the elements %s were visible after %s second%s!" % (
+                str(selectors),
+                timeout,
+                plural,
+            )
+        )
+
+    def wait_for_any_of_elements_present(self, *args, **kwargs):
+        """Waits for at least one of the elements to be present.
+        Visibility not required, but element must be in the DOM.
+        Returns the first element that is found.
+        The input is a list of elements. (Should be CSS selectors)
+        Optional kwargs include: "timeout" (used by all selectors).
+        Raises an exception if no elements are present by the timeout.
+        Examples:
+            self.wait_for_any_of_elements_present("style", "script")
+            OR
+            self.wait_for_any_of_elements_present(["style", "script"]) """
+        selectors = []
+        timeout = None
+        for kwarg in kwargs:
+            if kwarg == "timeout":
+                timeout = kwargs["timeout"]
+            elif kwarg == "by":
+                pass  # Autodetected
+            elif kwarg == "selector" or kwarg == "selectors":
+                selector = kwargs[kwarg]
+                if isinstance(selector, str):
+                    selectors.append(selector)
+                elif isinstance(selector, list):
+                    selectors_list = selector
+                    for selector in selectors_list:
+                        if isinstance(selector, str):
+                            selectors.append(selector)
+            else:
+                raise Exception('Unknown kwarg: "%s"!' % kwarg)
+        if not timeout:
+            timeout = settings.SMALL_TIMEOUT
+        for arg in args:
+            if isinstance(arg, list):
+                for selector in arg:
+                    if isinstance(selector, str):
+                        selectors.append(selector)
+            elif isinstance(arg, str):
+                selectors.append(arg)
+        if not selectors:
+            raise Exception("The selectors list was empty!")
+        start_ms = time.time() * 1000.0
+        stop_ms = start_ms + (timeout * 1000.0)
+        for i in range(int(timeout * 10)):
+            for selector in selectors:
+                if self.is_element_present(selector):
+                    return self.select(selector)
+            now_ms = time.time() * 1000.0
+            if now_ms >= stop_ms:
+                break
+            time.sleep(0.1)
+        plural = "s"
+        if timeout == 1:
+            plural = ""
+        # None of the elements exist in the HTML
+        raise Exception(
+            "None of the elements %s were present after %s second%s!" % (
+                str(selectors),
+                timeout,
+                plural,
+            )
+        )
+
+    def assert_any_of_elements_visible(self, *args, **kwargs):
+        """Like wait_for_any_of_elements_visible(), but returns nothing."""
+        self.wait_for_any_of_elements_visible(*args, **kwargs)
+        return True
+
+    def assert_any_of_elements_present(self, *args, **kwargs):
+        """Like wait_for_any_of_elements_present(), but returns nothing."""
+        self.wait_for_any_of_elements_present(*args, **kwargs)
+        return True
+
     def assert_element(self, selector, timeout=None):
         """Same as assert_element_visible()"""
         self.assert_element_visible(selector, timeout=timeout)

seleniumbase/fixtures/base_case.py

@@ -4894,7 +4894,7 @@ class BaseCase(unittest.TestCase):
                 self.driver.connect()
             current_url = self.get_current_url()
             if not current_url.startswith(("about", "data", "chrome")):
-                self.get_new_driver(undetectable=True)
+                self.open("about:blank")
             self.driver.uc_open_with_cdp_mode(url, **kwargs)
         else:
             self.get_new_driver(undetectable=True)
@@ -9227,6 +9227,162 @@ class BaseCase(unittest.TestCase):
             original_selector=original_selector,
         )
 
+    def wait_for_any_of_elements_visible(self, *args, **kwargs):
+        """Waits for at least one of the elements to be visible.
+        Returns the first element that is found.
+        The input is a list of elements. (Should be CSS selectors or XPath)
+        Optional kwargs include: "timeout" (used by all selectors).
+        Raises an exception if no elements are visible by the timeout.
+        Allows flexible inputs (Eg. Multiple args or a list of args)
+        Examples:
+            self.wait_for_any_of_elements_visible("h1", "h2", "h3")
+            OR
+            self.wait_for_any_of_elements_visible(["h1", "h2", "h3"]) """
+        self.__check_scope()
+        selectors = []
+        timeout = None
+        for kwarg in kwargs:
+            if kwarg == "timeout":
+                timeout = kwargs["timeout"]
+            elif kwarg == "by":
+                pass  # Autodetected
+            elif kwarg == "selector" or kwarg == "selectors":
+                selector = kwargs[kwarg]
+                if isinstance(selector, str):
+                    selectors.append(selector)
+                elif isinstance(selector, list):
+                    selectors_list = selector
+                    for selector in selectors_list:
+                        if isinstance(selector, str):
+                            selectors.append(selector)
+            else:
+                raise Exception('Unknown kwarg: "%s"!' % kwarg)
+        if not timeout:
+            timeout = settings.LARGE_TIMEOUT
+        if self.timeout_multiplier and timeout == settings.LARGE_TIMEOUT:
+            timeout = self.__get_new_timeout(timeout)
+        for arg in args:
+            if isinstance(arg, list):
+                for selector in arg:
+                    if isinstance(selector, str):
+                        selectors.append(selector)
+            elif isinstance(arg, str):
+                selectors.append(arg)
+        if not selectors:
+            raise Exception("The selectors list was empty!")
+        original_selectors = selectors
+        updated_selectors = []
+        for selector in selectors:
+            by = "css selector"
+            if page_utils.is_xpath_selector(selector):
+                by = "xpath"
+            selector, by = self.__recalculate_selector(selector, by)
+            updated_selectors.append(selector)
+        selectors = updated_selectors
+        if self.__is_cdp_swap_needed():
+            return self.cdp.wait_for_any_of_elements_visible(
+                selectors, timeout=timeout
+            )
+        return page_actions.wait_for_any_of_elements_visible(
+            self.driver,
+            selectors,
+            timeout=timeout,
+            original_selectors=original_selectors,
+        )
+
+    def wait_for_any_of_elements_present(self, *args, **kwargs):
+        """Waits for at least one of the elements to be present.
+        Visibility not required, but element must be in the DOM.
+        Returns the first element that is found.
+        The input is a list of elements. (Should be CSS selectors or XPath)
+        Optional kwargs include: "timeout" (used by all selectors).
+        Raises an exception if no elements are present by the timeout.
+        Allows flexible inputs (Eg. Multiple args or a list of args)
+        Examples:
+            self.wait_for_any_of_elements_present("style", "script")
+            OR
+            self.wait_for_any_of_elements_present(["style", "script"]) """
+        self.__check_scope()
+        selectors = []
+        timeout = None
+        for kwarg in kwargs:
+            if kwarg == "timeout":
+                timeout = kwargs["timeout"]
+            elif kwarg == "by":
+                pass  # Autodetected
+            elif kwarg == "selector" or kwarg == "selectors":
+                selector = kwargs[kwarg]
+                if isinstance(selector, str):
+                    selectors.append(selector)
+                elif isinstance(selector, list):
+                    selectors_list = selector
+                    for selector in selectors_list:
+                        if isinstance(selector, str):
+                            selectors.append(selector)
+            else:
+                raise Exception('Unknown kwarg: "%s"!' % kwarg)
+        if not timeout:
+            timeout = settings.LARGE_TIMEOUT
+        if self.timeout_multiplier and timeout == settings.LARGE_TIMEOUT:
+            timeout = self.__get_new_timeout(timeout)
+        for arg in args:
+            if isinstance(arg, list):
+                for selector in arg:
+                    if isinstance(selector, str):
+                        selectors.append(selector)
+            elif isinstance(arg, str):
+                selectors.append(arg)
+        if not selectors:
+            raise Exception("The selectors list was empty!")
+        original_selectors = selectors
+        updated_selectors = []
+        for selector in selectors:
+            by = "css selector"
+            if page_utils.is_xpath_selector(selector):
+                by = "xpath"
+            selector, by = self.__recalculate_selector(selector, by)
+            updated_selectors.append(selector)
+        selectors = updated_selectors
+        if self.__is_cdp_swap_needed():
+            return self.cdp.wait_for_any_of_elements_present(
+                selectors, timeout=timeout
+            )
+        return page_actions.wait_for_any_of_elements_present(
+            self.driver,
+            selectors,
+            timeout=timeout,
+            original_selectors=original_selectors,
+        )
+
+    def assert_any_of_elements_visible(self, *args, **kwargs):
+        """Similar to wait_for_any_of_elements_visible(), but returns nothing.
+        As above, raises an exception if none of the set elements are visible.
+        Returns True if successful. Default timeout = SMALL_TIMEOUT.
+        Allows flexible inputs (Eg. Multiple args or a list of args)
+        Examples:
+            self.assert_any_of_elements_visible("h1", "h2", "h3")
+            OR
+            self.assert_any_of_elements_visible(["h1", "h2", "h3"]) """
+        if "timeout" not in kwargs:
+            kwargs["timeout"] = settings.SMALL_TIMEOUT
+        self.wait_for_any_of_elements_visible(*args, **kwargs)
+        return True
+
+    def assert_any_of_elements_present(self, *args, **kwargs):
+        """Similar to wait_for_any_of_elements_present(), but returns nothing.
+        As above, raises an exception if none of the given elements are found.
+        Visibility is not required, but element must exist in the DOM.
+        Returns True if successful. Default timeout = SMALL_TIMEOUT.
+        Allows flexible inputs (Eg. Multiple args or a list of args)
+        Examples:
+            self.assert_any_of_elements_present("h1", "h2", "h3")
+            OR
+            self.assert_any_of_elements_present(["h1", "h2", "h3"]) """
+        if "timeout" not in kwargs:
+            kwargs["timeout"] = settings.SMALL_TIMEOUT
+        self.wait_for_any_of_elements_present(*args, **kwargs)
+        return True
+
     def select_all(self, selector, by="css selector", limit=0):
         return self.find_elements(selector, by=by, limit=limit)
 
@@ -9698,6 +9854,7 @@ class BaseCase(unittest.TestCase):
         The input is a list of elements.
         Optional kwargs include "by" and "timeout" (used by all selectors).
         Raises an exception if any of the elements are not visible.
+        Allows flexible inputs (Eg. Multiple args or a list of args)
         Examples:
             self.assert_elements_present("head", "style", "script", "body")
             OR
@@ -9711,7 +9868,7 @@ class BaseCase(unittest.TestCase):
                 timeout = kwargs["timeout"]
             elif kwarg == "by":
                 by = kwargs["by"]
-            elif kwarg == "selector":
+            elif kwarg == "selector" or kwarg == "selectors":
                 selector = kwargs["selector"]
                 if isinstance(selector, str):
                     selectors.append(selector)
@@ -9804,6 +9961,7 @@ class BaseCase(unittest.TestCase):
         The input is a list of elements.
         Optional kwargs include "by" and "timeout" (used by all selectors).
         Raises an exception if any of the elements are not visible.
+        Allows flexible inputs (Eg. Multiple args or a list of args)
         Examples:
             self.assert_elements("h1", "h2", "h3")
             OR
@@ -9817,7 +9975,7 @@ class BaseCase(unittest.TestCase):
                 timeout = kwargs["timeout"]
             elif kwarg == "by":
                 by = kwargs["by"]
-            elif kwarg == "selector":
+            elif kwarg == "selector" or kwarg == "selectors":
                 selector = kwargs["selector"]
                 if isinstance(selector, str):
                     selectors.append(selector)
@@ -16747,7 +16905,11 @@ class BaseCase(unittest.TestCase):
                     self._last_page_url = "(Error: Unknown URL)"
             if hasattr(self, "is_behave") and self.is_behave and has_exception:
                 if hasattr(sb_config, "pdb_option") and sb_config.pdb_option:
-                    self.__activate_behave_post_mortem_debug_mode()
+                    if (
+                        hasattr(sb_config, "behave_step")
+                        and hasattr(sb_config.behave_step, "exc_traceback")
+                    ):
+                        self.__activate_behave_post_mortem_debug_mode()
             if self._final_debug:
                 self.__activate_debug_mode_in_teardown()
             elif (

seleniumbase/fixtures/page_actions.py

@@ -751,6 +751,155 @@ def wait_for_exact_text_visible(
         return element
 
 
+def wait_for_any_of_elements_visible(
+    driver,
+    selectors,
+    timeout=settings.LARGE_TIMEOUT,
+    original_selectors=[],
+    ignore_test_time_limit=False,
+):
+    """
+    Waits for at least one of the elements in the selector list to be visible.
+    Returns the first element that is found.
+    Raises NoSuchElementException if none of the elements exist in the HTML
+    within the specified timeout.
+    Raises ElementNotVisibleException if the element exists in the HTML,
+    but is not visible (eg. opacity is "0") within the specified timeout.
+    @Params
+    driver - the webdriver object (required)
+    selectors - the list of selectors for identifying page elements (required)
+    timeout - the time to wait for elements in seconds
+    original_selectors - handle pre-converted ":contains(TEXT)" selectors
+    ignore_test_time_limit - ignore test time limit (NOT related to timeout)
+    @Returns
+    A web element object
+    """
+    if not isinstance(selectors, (list, tuple)):
+        raise Exception("`selectors` must be a list or tuple!")
+    if not selectors:
+        raise Exception("`selectors` cannot be an empty list!")
+    _reconnect_if_disconnected(driver)
+    element = None
+    any_present = False
+    start_ms = time.time() * 1000.0
+    stop_ms = start_ms + (timeout * 1000.0)
+    for x in range(int(timeout * 10)):
+        if not ignore_test_time_limit:
+            shared_utils.check_if_time_limit_exceeded()
+        try:
+            for selector in selectors:
+                by = "css selector"
+                if page_utils.is_xpath_selector(selector):
+                    by = "xpath"
+                try:
+                    element = driver.find_element(by=by, value=selector)
+                    any_present = True
+                    if element.is_displayed():
+                        return element
+                    element = None
+                except Exception:
+                    pass
+            raise Exception("Nothing found yet!")
+        except Exception:
+            now_ms = time.time() * 1000.0
+            if now_ms >= stop_ms:
+                break
+            time.sleep(0.1)
+    plural = "s"
+    if timeout == 1:
+        plural = ""
+    if original_selectors:
+        selectors = original_selectors
+    if not element:
+        if not any_present:
+            # None of the elements exist in the HTML
+            message = (
+                "None of the elements {%s} were present after %s second%s!" % (
+                    str(selectors),
+                    timeout,
+                    plural,
+                )
+            )
+            timeout_exception(NoSuchElementException, message)
+        # At least one element exists in the HTML, but none are visible
+        message = "None of the elements %s were visible after %s second%s!" % (
+            str(selectors),
+            timeout,
+            plural,
+        )
+        timeout_exception(ElementNotVisibleException, message)
+    else:
+        return element
+
+
+def wait_for_any_of_elements_present(
+    driver,
+    selectors,
+    timeout=settings.LARGE_TIMEOUT,
+    original_selectors=[],
+    ignore_test_time_limit=False,
+):
+    """
+    Waits for at least one of the elements in the selector list to be present.
+    Visibility not required. (Eg. <head> hidden in the HTML)
+    Returns the first element that is found.
+    Raises NoSuchElementException if none of the elements exist in the HTML
+    within the specified timeout.
+    @Params
+    driver - the webdriver object (required)
+    selectors - the list of selectors for identifying page elements (required)
+    timeout - the time to wait for elements in seconds
+    original_selectors - handle pre-converted ":contains(TEXT)" selectors
+    ignore_test_time_limit - ignore test time limit (NOT related to timeout)
+    @Returns
+    A web element object
+    """
+    if not isinstance(selectors, (list, tuple)):
+        raise Exception("`selectors` must be a list or tuple!")
+    if not selectors:
+        raise Exception("`selectors` cannot be an empty list!")
+    _reconnect_if_disconnected(driver)
+    element = None
+    start_ms = time.time() * 1000.0
+    stop_ms = start_ms + (timeout * 1000.0)
+    for x in range(int(timeout * 10)):
+        if not ignore_test_time_limit:
+            shared_utils.check_if_time_limit_exceeded()
+        try:
+            for selector in selectors:
+                by = "css selector"
+                if page_utils.is_xpath_selector(selector):
+                    by = "xpath"
+                try:
+                    element = driver.find_element(by=by, value=selector)
+                    return element
+                except Exception:
+                    pass
+            raise Exception("Nothing found yet!")
+        except Exception:
+            now_ms = time.time() * 1000.0
+            if now_ms >= stop_ms:
+                break
+            time.sleep(0.1)
+    plural = "s"
+    if timeout == 1:
+        plural = ""
+    if original_selectors:
+        selectors = original_selectors
+    if not element:
+        # None of the elements exist in the HTML
+        message = (
+            "None of the elements %s were present after %s second%s!" % (
+                str(selectors),
+                timeout,
+                plural,
+            )
+        )
+        timeout_exception(NoSuchElementException, message)
+    else:
+        return element
+
+
 def wait_for_attribute(
     driver,
     selector,

seleniumbase/plugins/pytest_plugin.py

@@ -28,6 +28,8 @@ def pytest_addoption(parser):
     --edge  (Shortcut for "--browser=edge".)
     --firefox  (Shortcut for "--browser=firefox".)
     --safari  (Shortcut for "--browser=safari".)
+    --cft  (Shortcut for using `Chrome for Testing`)
+    --chs  (Shortcut for using `Chrome-Headless-Shell`)
     --settings-file=FILE  (Override default SeleniumBase settings.)
     --env=ENV  (Set the test env. Access with "self.env" in tests.)
     --account=STR  (Set account. Access with "self.account" in tests.)

seleniumbase/plugins/selenium_plugin.py

@@ -16,6 +16,8 @@ class SeleniumBrowser(Plugin):
     --edge  (Shortcut for "--browser=edge".)
     --firefox  (Shortcut for "--browser=firefox".)
     --safari  (Shortcut for "--browser=safari".)
+    --cft  (Shortcut for using `Chrome for Testing`)
+    --chs  (Shortcut for using `Chrome-Headless-Shell`)
     --user-data-dir=DIR  (Set the Chrome user data directory to use.)
     --protocol=PROTOCOL  (The Selenium Grid protocol: http|https.)
     --server=SERVER  (The Selenium Grid server/IP used for tests.)
@@ -1221,6 +1223,7 @@ class SeleniumBrowser(Plugin):
             test.test.binary_location = "cft"
         elif self.options.use_chs and not test.test.binary_location:
             test.test.binary_location = "chs"
+        sb_config.binary_location = test.test.binary_location
         if (
             test.test.binary_location
             and test.test.binary_location.lower() == "chs"

{seleniumbase-4.39.6a2.dist-info → seleniumbase-4.40.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: seleniumbase
-Version: 4.39.6a2
+Version: 4.40.0
 Summary: A complete web automation framework for end-to-end testing.
 Home-page: https://github.com/seleniumbase/SeleniumBase
 Author: Michael Mintz
@@ -108,7 +108,7 @@ Requires-Dist: wsproto==1.2.0
 Requires-Dist: websocket-client==1.8.0
 Requires-Dist: selenium==4.27.1; python_version < "3.9"
 Requires-Dist: selenium==4.32.0; python_version >= "3.9" and python_version < "3.10"
-Requires-Dist: selenium==4.33.0; python_version >= "3.10"
+Requires-Dist: selenium==4.34.0; python_version >= "3.10"
 Requires-Dist: cssselect==1.2.0; python_version < "3.9"
 Requires-Dist: cssselect==1.3.0; python_version >= "3.9"
 Requires-Dist: sortedcontainers==2.4.0
@@ -124,7 +124,7 @@ Requires-Dist: pytest-ordering==0.6
 Requires-Dist: pytest-rerunfailures==14.0; python_version < "3.9"
 Requires-Dist: pytest-rerunfailures==15.1; python_version >= "3.9"
 Requires-Dist: pytest-xdist==3.6.1; python_version < "3.9"
-Requires-Dist: pytest-xdist==3.7.0; python_version >= "3.9"
+Requires-Dist: pytest-xdist==3.8.0; python_version >= "3.9"
 Requires-Dist: parameterized==0.9.0
 Requires-Dist: behave==1.2.6
 Requires-Dist: soupsieve==2.7
@@ -140,7 +140,7 @@ Requires-Dist: allure-python-commons>=2.13.5; extra == "allure"
 Requires-Dist: allure-behave>=2.13.5; extra == "allure"
 Provides-Extra: coverage
 Requires-Dist: coverage>=7.6.1; python_version < "3.9" and extra == "coverage"
-Requires-Dist: coverage>=7.9.1; python_version >= "3.9" and extra == "coverage"
+Requires-Dist: coverage>=7.9.2; python_version >= "3.9" and extra == "coverage"
 Requires-Dist: pytest-cov>=5.0.0; python_version < "3.9" and extra == "coverage"
 Requires-Dist: pytest-cov>=6.2.1; python_version >= "3.9" and extra == "coverage"
 Provides-Extra: flake8
@@ -161,12 +161,12 @@ Provides-Extra: pdfminer
 Requires-Dist: pdfminer.six==20250324; python_version < "3.9" and extra == "pdfminer"
 Requires-Dist: pdfminer.six==20250506; python_version >= "3.9" and extra == "pdfminer"
 Requires-Dist: cryptography==39.0.2; python_version < "3.9" and extra == "pdfminer"
-Requires-Dist: cryptography==45.0.4; python_version >= "3.9" and extra == "pdfminer"
+Requires-Dist: cryptography==45.0.5; python_version >= "3.9" and extra == "pdfminer"
 Requires-Dist: cffi==1.17.1; extra == "pdfminer"
 Requires-Dist: pycparser==2.22; extra == "pdfminer"
 Provides-Extra: pillow
 Requires-Dist: Pillow>=10.4.0; python_version < "3.9" and extra == "pillow"
-Requires-Dist: Pillow>=11.2.1; python_version >= "3.9" and extra == "pillow"
+Requires-Dist: Pillow>=11.3.0; python_version >= "3.9" and extra == "pillow"
 Provides-Extra: pip-system-certs
 Requires-Dist: pip-system-certs==4.0; platform_system == "Windows" and extra == "pip-system-certs"
 Provides-Extra: proxy