Add UC Mode customization options

Author: mdmintz

seleniumbase/core/browser_launcher.py

@@ -413,12 +413,14 @@ def uc_open_with_tab(driver, url):
     return None
 
 
-def uc_open_with_reconnect(driver, url):
-    """Open a url, then reconnect with UC before switching to the window."""
+def uc_open_with_reconnect(driver, url, reconnect_time=None):
+    """Open a url, disconnect chromedriver, wait, and reconnect."""
+    if not reconnect_time:
+        reconnect_time = constants.UC.RECONNECT_TIME
     if (url.startswith("http:") or url.startswith("https:")):
         driver.execute_script('window.open("%s","_blank");' % url)
-        driver.reconnect(2.65)
         driver.close()
+        driver.reconnect(reconnect_time)
         driver.switch_to.window(driver.window_handles[-1])
     else:
         driver.default_get(url)  # The original one
@@ -3479,7 +3481,9 @@ def get_local_driver(
                         lambda url: uc_open_with_tab(driver, url)
                     )
                     driver.uc_open_with_reconnect = (
-                        lambda url: uc_open_with_reconnect(driver, url)
+                        lambda *args, **kwargs: uc_open_with_reconnect(
+                            driver, *args, **kwargs
+                        )
                     )
                     if mobile_emulator:
                         uc_metrics = {}

seleniumbase/fixtures/constants.py

@@ -359,6 +359,10 @@ class Mobile:
     )
 
 
+class UC:
+    RECONNECT_TIME = 2.15  # Seconds
+
+
 class ValidBrowsers:
     valid_browsers = [
         "chrome",

seleniumbase/undetected/__init__.py

@@ -217,7 +217,7 @@ def __init__(
             options.binary_location = (
                 browser_executable_path or find_chrome_executable()
             )
-        self._delay = 2.05
+        self._delay = constants.UC.RECONNECT_TIME
         self.user_data_dir = user_data_dir
         self.keep_user_data_dir = keep_user_data_dir
         if suppress_welcome:
@@ -491,10 +491,7 @@ def __enter__(self):
         return self
 
     def __exit__(self, exc_type, exc_val, exc_tb):
-        self.service.stop()
-        time.sleep(self._delay)
-        self.service.start()
-        self.start_session()
+        self.reconnect(timeout=self._delay)
 
     def __hash__(self):
         return hash(self.options.debugger_address)