seleniumbase 4.41.3__py3-none-any.whl → 4.45.10__py3-none-any.whl

seleniumbase/core/sb_driver.py

@@ -12,6 +12,10 @@ from seleniumbase.fixtures import shared_utils
 class DriverMethods(WebDriver):
     def __init__(self, driver):
         self.driver = driver
+        if hasattr(driver, "session_id"):
+            self.session_id = driver.session_id
+        if hasattr(driver, "command_executor"):
+            self.command_executor = driver.command_executor
 
     def __is_cdp_swap_needed(self):
         """If the driver is disconnected, use a CDP method when available."""
@@ -37,6 +41,36 @@ class DriverMethods(WebDriver):
             value, by = page_utils.swap_selector_and_by_if_reversed(value, by)
         return self.driver.default_find_elements(by=by, value=value)
 
+    def add_cookie(self, *args, **kwargs):
+        page_actions._reconnect_if_disconnected(self.driver)
+        self.driver.default_add_cookie(*args, **kwargs)
+
+    def get_cookie(self, *args, **kwargs):
+        page_actions._reconnect_if_disconnected(self.driver)
+        self.driver.default_get_cookie(*args, **kwargs)
+
+    def delete_cookie(self, *args, **kwargs):
+        page_actions._reconnect_if_disconnected(self.driver)
+        self.driver.default_delete_cookie(*args, **kwargs)
+
+    def back(self):
+        if self.__is_cdp_swap_needed():
+            self.driver.cdp.go_back()
+            return
+        self.driver.default_back()
+
+    def forward(self):
+        if self.__is_cdp_swap_needed():
+            self.driver.cdp.go_forward()
+            return
+        self.driver.default_forward()
+
+    def refresh(self, *args, **kwargs):
+        if self.__is_cdp_swap_needed():
+            self.driver.cdp.refresh(*args, **kwargs)
+            return
+        self.driver.default_refresh()
+
     def locator(self, selector, by=None):
         if not by:
             by = "css selector"
@@ -137,6 +171,16 @@ class DriverMethods(WebDriver):
     def wait_for_element_present(self, *args, **kwargs):
         return page_actions.wait_for_selector(self.driver, *args, **kwargs)
 
+    def wait_for_element_absent(self, *args, **kwargs):
+        return page_actions.wait_for_element_absent(
+            self.driver, *args, **kwargs
+        )
+
+    def wait_for_element_not_visible(self, *args, **kwargs):
+        return page_actions.wait_for_element_not_visible(
+            self.driver, *args, **kwargs
+        )
+
     def wait_for_selector(self, *args, **kwargs):
         return page_actions.wait_for_selector(self.driver, *args, **kwargs)
 
@@ -278,11 +322,8 @@ class DriverMethods(WebDriver):
                 selector = kwargs["selector"]
             else:
                 selector = args[0]
-            if ":contains(" not in selector:
-                self.driver.cdp.highlight(selector)
-                return
-            else:
-                self.driver.connect()
+            self.driver.cdp.highlight(selector)
+            return
         if "scroll" in kwargs:
             kwargs.pop("scroll")
         w_args = kwargs.copy()

seleniumbase/core/session_helper.py

@@ -3,10 +3,8 @@ from seleniumbase import config as sb_config
 
 def end_reused_class_session_as_needed():
     if (
-        hasattr(sb_config, "reuse_class_session")
-        and sb_config.reuse_class_session
-        and hasattr(sb_config, "shared_driver")
-        and sb_config.shared_driver
+        getattr(sb_config, "reuse_class_session", None)
+        and getattr(sb_config, "shared_driver", None)
     ):
         if (
             hasattr(sb_config.shared_driver, "service")

seleniumbase/core/tour_helper.py

@@ -1134,10 +1134,9 @@ def export_tour(tour_steps, name=None, filename="my_tour.js", url=None):
             os.makedirs(exported_tours_folder)
         except Exception:
             pass
-    import codecs
 
     file_path = exported_tours_folder + "/" + filename
-    out_file = codecs.open(file_path, "w+", encoding="utf-8")
+    out_file = open(file_path, mode="w+", encoding="utf-8")
     out_file.writelines(instructions)
     out_file.close()
     print("\n>>> [%s] was saved!\n" % file_path)