Make improvements to "driver.uc_click(selector)" (UC Mode only)

Author: mdmintz

seleniumbase/core/browser_launcher.py

@@ -431,11 +431,24 @@ def uc_open_with_reconnect(driver, url, reconnect_time=None):
 
 
 def uc_click(
-    driver, selector, by="css selector", timeout=settings.SMALL_TIMEOUT
+    driver,
+    selector,
+    by="css selector",
+    timeout=settings.SMALL_TIMEOUT,
+    reconnect_time=None,
 ):
+    try:
+        rct = float(by)  # Add shortcut: driver.uc_click(selector, RCT)
+        if not reconnect_time:
+            reconnect_time = rct
+        by = "css selector"
+    except Exception:
+        pass
     element = driver.wait_for_element(selector, by=by, timeout=timeout)
     try:
-        element.uc_click()
+        element.uc_click(
+            driver, selector, by=by, reconnect_time=reconnect_time
+        )
     except ElementClickInterceptedException:
         driver.js_click(selector, by=by, timeout=timeout)
 

seleniumbase/undetected/webelement.py

@@ -1,14 +1,25 @@
 import selenium.webdriver.remote.webelement
-from seleniumbase.config import settings
 
 
 class WebElement(selenium.webdriver.remote.webelement.WebElement):
-    def uc_click(self):
-        super().click()
-        if (
-            hasattr(settings, "PAGE_LOAD_STRATEGY")
-            and settings.PAGE_LOAD_STRATEGY == "none"
-        ):
-            pass
+    def uc_click(
+        self,
+        driver=None,
+        selector=None,
+        by=None,
+        reconnect_time=None,
+    ):
+        if driver and selector and by:
+            driver.js_click(selector, by=by, timeout=1)
         else:
+            super().click()
+        if not reconnect_time:
             self._parent.reconnect(0.1)
+        else:
+            self._parent.reconnect(reconnect_time)
+
+    def uc_reconnect(self, reconnect_time=None):
+        if not reconnect_time:
+            self._parent.reconnect(0.1)
+        else:
+            self._parent.reconnect(reconnect_time)