seleniumbase 4.24.11__py3-none-any.whl → 4.33.15__py3-none-any.whl

seleniumbase/core/sb_driver.py

@@ -1,14 +1,21 @@
 """Add new methods to extend the driver"""
+from contextlib import suppress
 from selenium.webdriver.remote.webelement import WebElement
+from seleniumbase.config import settings
 from seleniumbase.fixtures import js_utils
 from seleniumbase.fixtures import page_actions
 from seleniumbase.fixtures import page_utils
+from seleniumbase.fixtures import shared_utils
 
 
 class DriverMethods():
     def __init__(self, driver):
         self.driver = driver
 
+    def __is_cdp_swap_needed(self):
+        """If the driver is disconnected, use a CDP method when available."""
+        return shared_utils.is_cdp_swap_needed(self.driver)
+
     def find_element(self, by=None, value=None):
         if not value:
             value = by
@@ -36,20 +43,29 @@ class DriverMethods():
             selector, by = page_utils.swap_selector_and_by_if_reversed(
                 selector, by
             )
-        try:
+        with suppress(Exception):
             return self.driver.default_find_element(by=by, value=selector)
-        except Exception:
-            pass
         raise Exception('No such Element: {%s} (by="%s")!' % (selector, by))
 
     def get_attribute(self, selector, attribute, by="css selector"):
         element = self.locator(selector, by=by)
         return element.get_attribute(attribute)
 
+    def get_current_url(self):
+        if self.__is_cdp_swap_needed():
+            current_url = self.driver.cdp.get_current_url()
+        else:
+            current_url = self.driver.current_url
+        return current_url
+
     def get_page_source(self):
+        if self.__is_cdp_swap_needed():
+            return self.driver.cdp.get_page_source()
         return self.driver.page_source
 
     def get_title(self):
+        if self.__is_cdp_swap_needed():
+            return self.driver.cdp.get_title()
         return self.driver.title
 
     def open_url(self, *args, **kwargs):
@@ -94,6 +110,16 @@ class DriverMethods():
     def assert_exact_text(self, *args, **kwargs):
         page_actions.assert_exact_text(self.driver, *args, **kwargs)
 
+    def assert_non_empty_text(self, *args, **kwargs):
+        return page_actions.assert_non_empty_text(
+            self.driver, *args, **kwargs
+        )
+
+    def assert_text_not_visible(self, *args, **kwargs):
+        return page_actions.assert_text_not_visible(
+            self.driver, *args, **kwargs
+        )
+
     def wait_for_element(self, *args, **kwargs):
         return page_actions.wait_for_element(self.driver, *args, **kwargs)
 
@@ -112,6 +138,16 @@ class DriverMethods():
     def wait_for_exact_text(self, *args, **kwargs):
         return page_actions.wait_for_exact_text(self.driver, *args, **kwargs)
 
+    def wait_for_non_empty_text(self, *args, **kwargs):
+        return page_actions.wait_for_non_empty_text(
+            self.driver, *args, **kwargs
+        )
+
+    def wait_for_text_not_visible(self, *args, **kwargs):
+        return page_actions.wait_for_text_not_visible(
+            self.driver, *args, **kwargs
+        )
+
     def wait_for_and_accept_alert(self, *args, **kwargs):
         return page_actions.wait_for_and_accept_alert(
             self.driver, *args, **kwargs
@@ -134,28 +170,111 @@ class DriverMethods():
     def is_exact_text_visible(self, *args, **kwargs):
         return page_actions.is_exact_text_visible(self.driver, *args, **kwargs)
 
-    def get_text(self, *args, **kwargs):
-        return page_actions.get_text(self.driver, *args, **kwargs)
+    def is_attribute_present(self, *args, **kwargs):
+        return page_actions.has_attribute(self.driver, *args, **kwargs)
+
+    def is_non_empty_text_visible(self, *args, **kwargs):
+        return page_actions.is_non_empty_text_visible(
+            self.driver, *args, **kwargs
+        )
+
+    def is_valid_url(self, url):
+        """Return True if the url is a valid url."""
+        return page_utils.is_valid_url(url)
+
+    def is_alert_present(self):
+        try:
+            self.driver.switch_to.alert
+            return True
+        except Exception:
+            return False
+
+    def is_online(self):
+        if self.__is_cdp_swap_needed():
+            return self.driver.cdp.evaluate("navigator.onLine;")
+        return self.driver.execute_script("return navigator.onLine;")
+
+    def is_connected(self):
+        """
+        Return True if WebDriver is connected to the browser.
+        Note that the stealthy CDP-Driver isn't a WebDriver.
+        In CDP Mode, the CDP-Driver controls the web browser.
+        The CDP-Driver can be connected while WebDriver isn't.
+        """
+        if shared_utils.is_windows():
+            return (
+                not hasattr(self.driver, "_is_connected")
+                or self.driver._is_connected
+            )
+        try:
+            self.driver.window_handles
+            return True
+        except Exception:
+            return False
+
+    def is_uc_mode_active(self):
+        """Return True if the driver is using UC Mode. False otherwise."""
+        return (
+            hasattr(self.driver, "_is_using_uc")
+            and self.driver._is_using_uc
+        )
+
+    def is_cdp_mode_active(self):
+        """CDP Mode is a special mode within UC Mode. Activated separately.
+        Return True if CDP Mode is loaded in the driver. False otherwise."""
+        return (
+            hasattr(self.driver, "_is_using_cdp")
+            and self.driver._is_using_cdp
+        )
 
     def js_click(self, *args, **kwargs):
         return page_actions.js_click(self.driver, *args, **kwargs)
 
-    def is_attribute_present(self, *args, **kwargs):
-        return page_actions.has_attribute(self.driver, *args, **kwargs)
+    def get_text(self, *args, **kwargs):
+        return page_actions.get_text(self.driver, *args, **kwargs)
 
     def get_active_element_css(self, *args, **kwargs):
+        if self.__is_cdp_swap_needed():
+            return self.driver.cdp.get_active_element_css()
         return js_utils.get_active_element_css(self.driver, *args, **kwargs)
 
     def get_locale_code(self, *args, **kwargs):
+        if self.__is_cdp_swap_needed():
+            return self.driver.cdp.get_locale_code()
         return js_utils.get_locale_code(self.driver, *args, **kwargs)
 
+    def get_screen_rect(self, *args, **kwargs):
+        if self.__is_cdp_swap_needed():
+            return self.driver.cdp.get_screen_rect()
+        return js_utils.get_screen_rect(self.driver, *args, **kwargs)
+
     def get_origin(self, *args, **kwargs):
+        if self.__is_cdp_swap_needed():
+            return self.driver.cdp.get_origin()
         return js_utils.get_origin(self.driver, *args, **kwargs)
 
     def get_user_agent(self, *args, **kwargs):
+        if self.__is_cdp_swap_needed():
+            return self.driver.cdp.get_user_agent()
         return js_utils.get_user_agent(self.driver, *args, **kwargs)
 
+    def get_cookie_string(self, *args, **kwargs):
+        if self.__is_cdp_swap_needed():
+            return self.driver.cdp.get_cookie_string()
+        return js_utils.get_cookie_string(self.driver, *args, **kwargs)
+
     def highlight(self, *args, **kwargs):
+        if self.__is_cdp_swap_needed():
+            selector = None
+            if "selector" in kwargs:
+                selector = kwargs["selector"]
+            else:
+                selector = args[0]
+            if ":contains(" not in selector:
+                self.driver.cdp.highlight(selector)
+                return
+            else:
+                self.driver.connect()
         if "scroll" in kwargs:
             kwargs.pop("scroll")
         w_args = kwargs.copy()
@@ -182,13 +301,48 @@ class DriverMethods():
         if self.is_element_visible(selector, by=by):
             self.highlight(selector, by=by, loops=loops, scroll=scroll)
 
-    def switch_to_frame(self, frame):
+    def switch_to_default_window(self):
+        self.driver.switch_to.window(self.driver.window_handles[0])
+
+    def switch_to_newest_window(self):
+        self.driver.switch_to.window(self.driver.window_handles[-1])
+
+    def open_new_window(self, switch_to=True):
+        if switch_to:
+            try:
+                self.driver.switch_to.new_window("tab")
+            except Exception:
+                self.driver.execute_script("window.open('');")
+                self.switch_to_newest_window()
+        else:
+            self.driver.execute_script("window.open('');")
+
+    def open_new_tab(self, switch_to=True):
+        self.open_new_window(switch_to=switch_to)
+
+    def switch_to_window(self, *args, **kwargs):
+        page_actions.switch_to_window(self.driver, *args, **kwargs)
+
+    def switch_to_tab(self, *args, **kwargs):
+        self.switch_to_window(*args, **kwargs)
+
+    def switch_to_frame(self, frame="iframe"):
         if isinstance(frame, WebElement):
             self.driver.switch_to.frame(frame)
         else:
             iframe = self.locator(frame)
             self.driver.switch_to.frame(iframe)
 
+    def reset_window_size(self):
+        if self.__is_cdp_swap_needed():
+            self.driver.cdp.reset_window_size()
+            return
+        x = settings.WINDOW_START_X
+        y = settings.WINDOW_START_Y
+        width = settings.CHROME_START_WIDTH
+        height = settings.CHROME_START_HEIGHT
+        self.driver.set_window_rect(x, y, width, height)
+
     def set_wire_proxy(self, string):
         """Set a proxy server for selenium-wire mode ("--wire")
         Examples:  (ONLY avilable if using selenium-wire mode!)

seleniumbase/core/settings_parser.py

@@ -71,6 +71,8 @@ def set_settings(settings_file):
             settings.LARGE_TIMEOUT = override_settings[key]
         elif key == "EXTREME_TIMEOUT":
             settings.EXTREME_TIMEOUT = override_settings[key]
+        elif key == "PAGE_LOAD_TIMEOUT":
+            settings.PAGE_LOAD_TIMEOUT = override_settings[key]
         elif key == "ARCHIVE_EXISTING_LOGS":
             settings.ARCHIVE_EXISTING_LOGS = override_settings[key]
         elif key == "ARCHIVE_EXISTING_DOWNLOADS":
@@ -113,6 +115,10 @@ def set_settings(settings_file):
             settings.RAISE_INVALID_PROXY_STRING_EXCEPTION = override_settings[
                 key
             ]
+        elif key == "WINDOW_START_X":
+            settings.WINDOW_START_X = override_settings[key]
+        elif key == "WINDOW_START_Y":
+            settings.WINDOW_START_Y = override_settings[key]
         elif key == "CHROME_START_WIDTH":
             settings.CHROME_START_WIDTH = override_settings[key]
         elif key == "CHROME_START_HEIGHT":

seleniumbase/core/style_sheet.py

@@ -1,4 +1,5 @@
 import sys
+import textwrap
 from seleniumbase.fixtures import constants
 
 
@@ -116,6 +117,7 @@ def get_report_style():
         }
         </style>"""
     )
+    style = textwrap.dedent(style)
     Saved.report_style = style
     return style
 
@@ -132,6 +134,7 @@ def get_bt_backdrop_style():
             box-shadow: 0 0 0 88422px rgba(0, 0, 0, 0.42);
             pointer-events: auto !important;
         }"""
+    bt_backdrop_style = textwrap.dedent(bt_backdrop_style)
     Saved.bt_backdrop_style = bt_backdrop_style
     return bt_backdrop_style
 
@@ -150,6 +153,7 @@ def get_dt_backdrop_style():
         button.driver-prev-btn.driver-disabled {
             visibility: hidden;
         }"""
+    dt_backdrop_style = textwrap.dedent(dt_backdrop_style)
     Saved.dt_backdrop_style = dt_backdrop_style
     return dt_backdrop_style
 
@@ -167,6 +171,7 @@ def get_messenger_style():
             box-shadow: 2px 2px 9px 4px rgba(32, 142, 120, 0.28),
             2px 2px 9px 4px rgba(200, 240, 80, 0.34) !important;
         }""" % font_family
+    messenger_style = textwrap.dedent(messenger_style)
     Saved.messenger_style = messenger_style
     return messenger_style
 
@@ -181,6 +186,7 @@ def get_sh_style_test():
             scrollTo: true
           }
         });"""
+    sh_style_test = textwrap.dedent(sh_style_test)
     Saved.sh_style_test = sh_style_test
     return sh_style_test
 
@@ -193,6 +199,7 @@ def get_hops_backdrop_style():
         .hopscotch-bubble-container {
             font-size: 110%;
         }"""
+    hops_backdrop_style = textwrap.dedent(hops_backdrop_style)
     Saved.hops_backdrop_style = hops_backdrop_style
     return hops_backdrop_style
 
@@ -227,6 +234,7 @@ def get_introjs_style():
             box-sizing: content-box;
             position: absolute;
         }"""
+    introjs_style = textwrap.dedent(introjs_style)
     Saved.introjs_style = introjs_style
     return introjs_style
 
@@ -261,6 +269,7 @@ def get_sh_backdrop_style():
         body.shepherd-active {
             pointer-events: none !important;
         }"""
+    sh_backdrop_style = textwrap.dedent(sh_backdrop_style)
     Saved.sh_backdrop_style = sh_backdrop_style
     return sh_backdrop_style
 
@@ -387,5 +396,6 @@ def get_pytest_style():
         .desc.active .sort-icon {
             border-top: 8px solid #999;
         }"""
+    pytest_style = textwrap.dedent(pytest_style)
     Saved.pytest_style = pytest_style
     return pytest_style