Update CDP Mode

Author: mdmintz

examples/cdp_mode/ReadMe.md

@@ -368,6 +368,9 @@ sb.cdp.get_element_attribute(selector, attribute)
 sb.cdp.get_element_html(selector)
 sb.cdp.set_locale(locale)
 sb.cdp.set_attributes(selector, attribute, value)
+sb.cdp.gui_press_key(key)
+sb.cdp.gui_press_keys(keys)
+sb.cdp.gui_write(text)
 sb.cdp.gui_click_x_y(x, y)
 sb.cdp.gui_click_element(selector)
 sb.cdp.internalize_links()

seleniumbase/core/browser_launcher.py

@@ -620,6 +620,9 @@ def uc_open_with_cdp_mode(driver, url=None):
     cdp.reset_window_size = CDPM.reset_window_size
     cdp.set_locale = CDPM.set_locale
     cdp.set_attributes = CDPM.set_attributes
+    cdp.gui_press_key = CDPM.gui_press_key
+    cdp.gui_press_keys = CDPM.gui_press_keys
+    cdp.gui_write = CDPM.gui_write
     cdp.gui_click_x_y = CDPM.gui_click_x_y
     cdp.gui_click_element = CDPM.gui_click_element
     cdp.internalize_links = CDPM.internalize_links
@@ -721,6 +724,9 @@ def uc_click(
     timeout=settings.SMALL_TIMEOUT,
     reconnect_time=None,
 ):
+    if __is_cdp_swap_needed(driver):
+        driver.cdp.click(selector)
+        return
     with suppress(Exception):
         rct = float(by)  # Add shortcut: driver.uc_click(selector, RCT)
         if not reconnect_time:

seleniumbase/core/sb_cdp.py

@@ -942,6 +942,45 @@ def __get_configured_pyautogui(self, pyautogui_copy):
                 )
         return pyautogui_copy
 
+    def gui_press_key(self, key):
+        self.__install_pyautogui_if_missing()
+        import pyautogui
+        pyautogui = self.__get_configured_pyautogui(pyautogui)
+        gui_lock = fasteners.InterProcessLock(
+            constants.MultiBrowser.PYAUTOGUILOCK
+        )
+        with gui_lock:
+            pyautogui.press(key)
+            time.sleep(0.0375)
+        self.__slow_mode_pause_if_set()
+        self.loop.run_until_complete(self.page.wait())
+
+    def gui_press_keys(self, keys):
+        self.__install_pyautogui_if_missing()
+        import pyautogui
+        pyautogui = self.__get_configured_pyautogui(pyautogui)
+        gui_lock = fasteners.InterProcessLock(
+            constants.MultiBrowser.PYAUTOGUILOCK
+        )
+        with gui_lock:
+            for key in keys:
+                pyautogui.press(key)
+                time.sleep(0.0375)
+        self.__slow_mode_pause_if_set()
+        self.loop.run_until_complete(self.page.wait())
+
+    def gui_write(self, text):
+        self.__install_pyautogui_if_missing()
+        import pyautogui
+        pyautogui = self.__get_configured_pyautogui(pyautogui)
+        gui_lock = fasteners.InterProcessLock(
+            constants.MultiBrowser.PYAUTOGUILOCK
+        )
+        with gui_lock:
+            pyautogui.write(text)
+        self.__slow_mode_pause_if_set()
+        self.loop.run_until_complete(self.page.wait())
+
     def __gui_click_x_y(self, x, y, timeframe=0.25, uc_lock=False):
         self.__install_pyautogui_if_missing()
         import pyautogui

seleniumbase/plugins/driver_manager.py

@@ -550,11 +550,7 @@ def Driver(
         or uc_sub
     ):
         undetectable = True
-    if (
-        (undetectable or undetected or uc)
-        and (uc_subprocess is None)
-        and (uc_sub is None)
-    ):
+    if undetectable or undetected or uc:
         uc_subprocess = True  # Use UC as a subprocess by default.
     elif (
         "--undetectable" in sys_argv

seleniumbase/plugins/sb_manager.py

@@ -608,11 +608,7 @@ def SB(
         or uc_sub
     ):
         undetectable = True
-    if (
-        (undetectable or undetected or uc)
-        and (uc_subprocess is None)
-        and (uc_sub is None)
-    ):
+    if undetectable or undetected or uc:
         uc_subprocess = True  # Use UC as a subprocess by default.
     elif (
         "--undetectable" in sys_argv

seleniumbase/undetected/cdp_driver/connection.py

@@ -18,6 +18,7 @@
     TypeVar,
 )
 import websockets
+from websockets.protocol import State
 from . import cdp_util as util
 import mycdp as cdp
 import mycdp.network
@@ -261,7 +262,7 @@ async def aopen(self, **kw):
         """
         Opens the websocket connection. Shouldn't be called manually by users.
         """
-        if not self.websocket or self.websocket.closed:
+        if not self.websocket or self.websocket.state is State.CLOSED:
             try:
                 self.websocket = await websockets.connect(
                     self.websocket_url,
@@ -288,7 +289,7 @@ async def aclose(self):
         """
         Closes the websocket connection. Shouldn't be called manually by users.
         """
-        if self.websocket and not self.websocket.closed:
+        if self.websocket and self.websocket.state is not State.CLOSED:
             if self.listener and self.listener.running:
                 self.listener.cancel()
                 self.enabled_domains.clear()
@@ -393,7 +394,7 @@ async def send(
             when multiple calls to connection.send() are made.
         """
         await self.aopen()
-        if not self.websocket or self.closed:
+        if not self.websocket or self.websocket.state is State.CLOSED:
             return
         if self._owner:
             browser = self._owner