Add the "--proxy=STRING" option for manager formats

Author: mdmintz

seleniumbase/plugins/driver_manager.py

@@ -251,6 +251,16 @@ def Driver(
             is_mobile = False
     test_id = "direct_driver"
     proxy_string = proxy
+    if proxy_string is None and "--proxy" in arg_join:
+        if "--proxy=" in arg_join:
+            proxy_string = arg_join.split("--proxy=")[1].split(" ")[0]
+        elif "--proxy " in arg_join:
+            proxy_string = arg_join.split("--proxy ")[1].split(" ")[0]
+        if proxy_string:
+            if proxy_string.startswith('"') and proxy_string.endswith('"'):
+                proxy_string = proxy_string[1:-1]
+            elif proxy_string.startswith("'") and proxy_string.endswith("'"):
+                proxy_string = proxy_string[1:-1]
     user_agent = agent
     recorder_mode = False
     if recorder_ext:

seleniumbase/plugins/sb_manager.py

@@ -312,6 +312,16 @@ def SB(
     if is_mobile:
         sb_config.mobile_emulator = True
     proxy_string = proxy
+    if proxy_string is None and "--proxy" in arg_join:
+        if "--proxy=" in arg_join:
+            proxy_string = arg_join.split("--proxy=")[1].split(" ")[0]
+        elif "--proxy " in arg_join:
+            proxy_string = arg_join.split("--proxy ")[1].split(" ")[0]
+        if proxy_string:
+            if proxy_string.startswith('"') and proxy_string.endswith('"'):
+                proxy_string = proxy_string[1:-1]
+            elif proxy_string.startswith("'") and proxy_string.endswith("'"):
+                proxy_string = proxy_string[1:-1]
     user_agent = agent
     recorder_mode = False
     if recorder_ext: