seleniumbase 4.19.1__py3-none-any.whl → 4.20.0__py3-none-any.whl

seleniumbase/__version__.py

@@ -1,2 +1,2 @@
 # seleniumbase package
-__version__ = "4.19.1"
+__version__ = "4.20.0"

seleniumbase/behave/behave_sb.py

@@ -111,6 +111,8 @@ from seleniumbase.core import session_helper
 from seleniumbase.fixtures import constants
 from seleniumbase.fixtures import shared_utils
 
+is_linux = shared_utils.is_linux()
+is_windows = shared_utils.is_windows()
 sb_config.__base_class = None
 
 
@@ -829,10 +831,10 @@ def get_configured_sb(context):
             '\n  (Your browser choice was: "%s")\n' % sb.browser
         )
     # The Xvfb virtual display server is for Linux OS Only.
-    if sb.xvfb and not shared_utils.is_linux():
+    if sb.xvfb and not is_linux:
         sb.xvfb = False
     if (
-        shared_utils.is_linux()
+        is_linux
         and not sb.headed
         and not sb.headless
         and not sb.headless2
@@ -1020,7 +1022,7 @@ def dashboard_pre_processing():
     filename = None
     feature_name = None
     scenario_name = None
-    if shared_utils.is_windows():
+    if is_windows:
         output = output.decode("latin1")
     else:
         output = output.decode("utf-8")
@@ -1129,8 +1131,11 @@ def behave_dashboard_prepare():
         stars = "*" * star_len
         c1 = ""
         cr = ""
-        if not shared_utils.is_linux():
-            colorama.init(autoreset=True)
+        if not is_linux:
+            if is_windows and hasattr(colorama, "just_fix_windows_console"):
+                colorama.just_fix_windows_console()
+            else:
+                colorama.init(autoreset=True)
             c1 = colorama.Fore.BLUE + colorama.Back.LIGHTCYAN_EX
             cr = colorama.Style.RESET_ALL
         print("Dashboard: %s%s%s\n%s" % (c1, dash_path, cr, stars))
@@ -1144,7 +1149,7 @@ def _perform_behave_unconfigure_():
         if sb_config.shared_driver:
             try:
                 if (
-                    not shared_utils.is_windows()
+                    not is_windows
                     or sb_config.browser == "ie"
                     or sb_config.shared_driver.service.process
                 ):
@@ -1221,7 +1226,7 @@ def do_final_driver_cleanup_as_needed():
     try:
         if hasattr(sb_config, "last_driver") and sb_config.last_driver:
             if (
-                not shared_utils.is_windows()
+                not is_windows
                 or sb_config.browser == "ie"
                 or sb_config.last_driver.service.process
             ):
@@ -1245,8 +1250,11 @@ def _perform_behave_terminal_summary_():
     equals = "=" * (equals_len + 2)
     c2 = ""
     cr = ""
-    if not shared_utils.is_linux():
-        colorama.init(autoreset=True)
+    if not is_linux:
+        if is_windows and hasattr(colorama, "just_fix_windows_console"):
+            colorama.just_fix_windows_console()
+        else:
+            colorama.init(autoreset=True)
         c2 = colorama.Fore.MAGENTA + colorama.Back.LIGHTYELLOW_EX
         cr = colorama.Style.RESET_ALL
     if sb_config.dashboard:

seleniumbase/console_scripts/logo_helper.py

@@ -3,6 +3,7 @@
     http://www.patorjk.com/software/taag/#p=display&f=Slant&t=SeleniumBase """
 
 import colorama
+import sys
 
 r"""
    ______     __           _                  ____
@@ -14,7 +15,13 @@ r"""
 
 
 def get_seleniumbase_logo():
-    colorama.init(autoreset=True)
+    if (
+        "win32" in sys.platform
+        and hasattr(colorama, "just_fix_windows_console")
+    ):
+        colorama.just_fix_windows_console()
+    else:
+        colorama.init(autoreset=True)
     c1 = colorama.Fore.BLUE + colorama.Back.LIGHTCYAN_EX
     c2 = colorama.Fore.BLUE + colorama.Back.LIGHTGREEN_EX
     c3 = colorama.Back.CYAN

seleniumbase/console_scripts/run.py

@@ -41,8 +41,12 @@ import sys
 import time
 from seleniumbase.config import settings
 from seleniumbase.fixtures import constants
+from seleniumbase.fixtures import shared_utils
 
-colorama.init(autoreset=True)
+if shared_utils.is_windows() and hasattr(colorama, "just_fix_windows_console"):
+    colorama.just_fix_windows_console()
+else:
+    colorama.init(autoreset=True)
 
 
 def show_usage():
@@ -1330,7 +1334,6 @@ def main():
         show_detailed_help()
     else:
         show_usage()
-        colorama.init(autoreset=True)
         c5 = colorama.Fore.RED + colorama.Back.LIGHTYELLOW_EX
         c7 = colorama.Fore.BLACK + colorama.Back.MAGENTA
         cr = colorama.Style.RESET_ALL

seleniumbase/console_scripts/sb_behave_gui.py

@@ -38,7 +38,13 @@ def set_colors(use_colors):
     c6 = ""
     cr = ""
     if use_colors:
-        colorama.init(autoreset=True)
+        if (
+            "win32" in sys.platform
+            and hasattr(colorama, "just_fix_windows_console")
+        ):
+            colorama.just_fix_windows_console()
+        else:
+            colorama.init(autoreset=True)
         c0 = colorama.Fore.BLUE + colorama.Back.LIGHTCYAN_EX
         c1 = colorama.Fore.BLUE + colorama.Back.LIGHTGREEN_EX
         c2 = colorama.Fore.RED + colorama.Back.LIGHTYELLOW_EX

seleniumbase/console_scripts/sb_caseplans.py

@@ -42,7 +42,13 @@ def set_colors(use_colors):
     c5 = ""
     cr = ""
     if use_colors:
-        colorama.init(autoreset=True)
+        if (
+            "win32" in sys.platform
+            and hasattr(colorama, "just_fix_windows_console")
+        ):
+            colorama.just_fix_windows_console()
+        else:
+            colorama.init(autoreset=True)
         c0 = colorama.Fore.BLUE + colorama.Back.LIGHTCYAN_EX
         c1 = colorama.Fore.BLUE + colorama.Back.LIGHTGREEN_EX
         c2 = colorama.Fore.RED + colorama.Back.LIGHTYELLOW_EX

seleniumbase/console_scripts/sb_commander.py

@@ -42,7 +42,13 @@ def set_colors(use_colors):
     c5 = ""
     cr = ""
     if use_colors:
-        colorama.init(autoreset=True)
+        if (
+            "win32" in sys.platform
+            and hasattr(colorama, "just_fix_windows_console")
+        ):
+            colorama.just_fix_windows_console()
+        else:
+            colorama.init(autoreset=True)
         c0 = colorama.Fore.BLUE + colorama.Back.LIGHTCYAN_EX
         c1 = colorama.Fore.BLUE + colorama.Back.LIGHTGREEN_EX
         c2 = colorama.Fore.RED + colorama.Back.LIGHTYELLOW_EX

seleniumbase/console_scripts/sb_install.py

@@ -293,7 +293,10 @@ def main(override=None, intel_for_uc=None, force_uc=None):
     use_version = ""
     new_file = ""
     f_name = ""
-    colorama.init(autoreset=True)
+    if IS_WINDOWS and hasattr(colorama, "just_fix_windows_console"):
+        colorama.just_fix_windows_console()
+    else:
+        colorama.init(autoreset=True)
     c1 = colorama.Fore.BLUE + colorama.Back.LIGHTCYAN_EX
     c2 = colorama.Fore.BLUE + colorama.Back.LIGHTGREEN_EX
     c3 = colorama.Fore.BLUE + colorama.Back.LIGHTYELLOW_EX

seleniumbase/console_scripts/sb_mkchart.py

@@ -62,7 +62,13 @@ def main():
     c7 = ""
     cr = ""
     if "linux" not in sys.platform:
-        colorama.init(autoreset=True)
+        if (
+            "win32" in sys.platform
+            and hasattr(colorama, "just_fix_windows_console")
+        ):
+            colorama.just_fix_windows_console()
+        else:
+            colorama.init(autoreset=True)
         c1 = colorama.Fore.BLUE + colorama.Back.LIGHTCYAN_EX
         c5 = colorama.Fore.RED + colorama.Back.LIGHTYELLOW_EX
         c7 = colorama.Fore.BLACK + colorama.Back.MAGENTA

seleniumbase/console_scripts/sb_mkdir.py

@@ -54,7 +54,13 @@ def main():
     c7 = ""
     cr = ""
     if "linux" not in sys.platform:
-        colorama.init(autoreset=True)
+        if (
+            "win32" in sys.platform
+            and hasattr(colorama, "just_fix_windows_console")
+        ):
+            colorama.just_fix_windows_console()
+        else:
+            colorama.init(autoreset=True)
         c1 = colorama.Fore.BLUE + colorama.Back.LIGHTCYAN_EX
         c5 = colorama.Fore.RED + colorama.Back.LIGHTYELLOW_EX
         c7 = colorama.Fore.BLACK + colorama.Back.MAGENTA

seleniumbase/console_scripts/sb_mkfile.py

@@ -92,7 +92,13 @@ def main():
     c7 = ""
     cr = ""
     if "linux" not in sys.platform:
-        colorama.init(autoreset=True)
+        if (
+            "win32" in sys.platform
+            and hasattr(colorama, "just_fix_windows_console")
+        ):
+            colorama.just_fix_windows_console()
+        else:
+            colorama.init(autoreset=True)
         c1 = colorama.Fore.BLUE + colorama.Back.LIGHTCYAN_EX
         c5 = colorama.Fore.RED + colorama.Back.LIGHTYELLOW_EX
         c7 = colorama.Fore.BLACK + colorama.Back.MAGENTA

seleniumbase/console_scripts/sb_mkpres.py

@@ -62,7 +62,13 @@ def main():
     c7 = ""
     cr = ""
     if "linux" not in sys.platform:
-        colorama.init(autoreset=True)
+        if (
+            "win32" in sys.platform
+            and hasattr(colorama, "just_fix_windows_console")
+        ):
+            colorama.just_fix_windows_console()
+        else:
+            colorama.init(autoreset=True)
         c1 = colorama.Fore.BLUE + colorama.Back.LIGHTCYAN_EX
         c5 = colorama.Fore.RED + colorama.Back.LIGHTYELLOW_EX
         c7 = colorama.Fore.BLACK + colorama.Back.MAGENTA

seleniumbase/console_scripts/sb_mkrec.py

@@ -71,7 +71,13 @@ def set_colors(use_colors):
     c7 = ""
     cr = ""
     if use_colors:
-        colorama.init(autoreset=True)
+        if (
+            "win32" in sys.platform
+            and hasattr(colorama, "just_fix_windows_console")
+        ):
+            colorama.just_fix_windows_console()
+        else:
+            colorama.init(autoreset=True)
         c0 = colorama.Fore.BLUE + colorama.Back.LIGHTCYAN_EX
         c1 = colorama.Fore.RED + colorama.Back.LIGHTYELLOW_EX
         c2 = colorama.Fore.LIGHTRED_EX + colorama.Back.LIGHTYELLOW_EX

seleniumbase/console_scripts/sb_print.py

@@ -61,7 +61,13 @@ def get_width(line):
 
 
 def main():
-    colorama.init(autoreset=True)
+    if (
+        "win32" in sys.platform
+        and hasattr(colorama, "just_fix_windows_console")
+    ):
+        colorama.just_fix_windows_console()
+    else:
+        colorama.init(autoreset=True)
     c5 = colorama.Fore.RED + colorama.Back.LIGHTYELLOW_EX
     c7 = colorama.Fore.BLACK + colorama.Back.MAGENTA
     cr = colorama.Style.RESET_ALL

seleniumbase/console_scripts/sb_recorder.py

@@ -42,7 +42,13 @@ def set_colors(use_colors):
     c4 = ""
     cr = ""
     if use_colors:
-        colorama.init(autoreset=True)
+        if (
+            "win32" in sys.platform
+            and hasattr(colorama, "just_fix_windows_console")
+        ):
+            colorama.just_fix_windows_console()
+        else:
+            colorama.init(autoreset=True)
         c0 = colorama.Fore.BLUE + colorama.Back.LIGHTCYAN_EX
         c1 = colorama.Fore.BLUE + colorama.Back.LIGHTGREEN_EX
         c2 = colorama.Fore.RED + colorama.Back.LIGHTYELLOW_EX

seleniumbase/core/browser_launcher.py

@@ -116,6 +116,9 @@ def extend_driver(driver):
     page = types.SimpleNamespace()
     page.open = DM.open_url
     page.click = DM.click
+    page.click_link = DM.click_link
+    page.click_if_visible = DM.click_if_visible
+    page.click_active_element = DM.click_active_element
     page.send_keys = DM.send_keys
     page.press_keys = DM.press_keys
     page.type = DM.update_text
@@ -134,10 +137,13 @@ def extend_driver(driver):
     page.is_element_visible = DM.is_element_visible
     page.is_text_visible = DM.is_text_visible
     page.is_exact_text_visible = DM.is_exact_text_visible
+    page.is_attribute_present = DM.is_attribute_present
     page.get_text = DM.get_text
     page.find_element = DM.find_element
     page.find_elements = DM.find_elements
     page.locator = DM.locator
+    page.get_page_source = DM.get_page_source
+    page.get_title = DM.get_title
     driver.page = page
     js = types.SimpleNamespace()
     js.js_click = DM.js_click
@@ -150,6 +156,8 @@ def extend_driver(driver):
     driver.open = DM.open_url
     driver.click = DM.click
     driver.click_link = DM.click_link
+    driver.click_if_visible = DM.click_if_visible
+    driver.click_active_element = DM.click_active_element
     driver.send_keys = DM.send_keys
     driver.press_keys = DM.press_keys
     driver.type = DM.update_text
@@ -168,6 +176,7 @@ def extend_driver(driver):
     driver.is_element_visible = DM.is_element_visible
     driver.is_text_visible = DM.is_text_visible
     driver.is_exact_text_visible = DM.is_exact_text_visible
+    driver.is_attribute_present = DM.is_attribute_present
     driver.get_text = DM.get_text
     driver.js_click = DM.js_click
     driver.get_active_element_css = DM.get_active_element_css
@@ -176,6 +185,10 @@ def extend_driver(driver):
     driver.get_user_agent = DM.get_user_agent
     driver.highlight = DM.highlight
     driver.sleep = time.sleep
+    driver.get_page_source = DM.get_page_source
+    driver.get_title = DM.get_title
+    if hasattr(driver, "proxy"):
+        driver.set_wire_proxy = DM.set_wire_proxy
     return driver
 
 
@@ -306,7 +319,15 @@ def has_cf(text):
     return False
 
 
-def uc_special_open_if_cf(driver, url, proxy_string=None):
+def uc_special_open_if_cf(
+    driver,
+    url,
+    proxy_string=None,
+    mobile_emulator=None,
+    device_width=None,
+    device_height=None,
+    device_pixel_ratio=None,
+):
     if (
         url.startswith("http:") or url.startswith("https:")
     ):
@@ -332,6 +353,36 @@ def uc_special_open_if_cf(driver, url, proxy_string=None):
                 driver.close()
                 driver.switch_to.window(driver.window_handles[-1])
                 time.sleep(0.02)
+                if mobile_emulator:
+                    uc_metrics = {}
+                    if (
+                        type(device_width) is int
+                        and type(device_height) is int
+                        and type(device_pixel_ratio) is int
+                    ):
+                        uc_metrics["width"] = device_width
+                        uc_metrics["height"] = device_height
+                        uc_metrics["pixelRatio"] = device_pixel_ratio
+                    else:
+                        uc_metrics["width"] = constants.Mobile.WIDTH
+                        uc_metrics["height"] = constants.Mobile.HEIGHT
+                        uc_metrics["pixelRatio"] = constants.Mobile.RATIO
+                    set_device_metrics_override = dict(
+                        {
+                            "width": uc_metrics["width"],
+                            "height": uc_metrics["height"],
+                            "deviceScaleFactor": uc_metrics["pixelRatio"],
+                            "mobile": True
+                        }
+                    )
+                    try:
+                        driver.execute_cdp_cmd(
+                            'Emulation.setDeviceMetricsOverride',
+                            set_device_metrics_override
+                        )
+                    except Exception:
+                        pass
+                    time.sleep(0.03)
         else:
             driver.default_get(url)  # The original one
     else:
@@ -726,7 +777,7 @@ def _set_chrome_options(
             "excludeSwitches",
             ["enable-automation", "enable-logging", "enable-blink-features"],
         )
-    if mobile_emulator:
+    if mobile_emulator and not is_using_uc(undetectable, browser_name):
         emulator_settings = {}
         device_metrics = {}
         if (
@@ -738,9 +789,9 @@ def _set_chrome_options(
             device_metrics["height"] = device_height
             device_metrics["pixelRatio"] = device_pixel_ratio
         else:
-            device_metrics["width"] = 360
-            device_metrics["height"] = 640
-            device_metrics["pixelRatio"] = 2
+            device_metrics["width"] = constants.Mobile.WIDTH
+            device_metrics["height"] = constants.Mobile.HEIGHT
+            device_metrics["pixelRatio"] = constants.Mobile.RATIO
         emulator_settings["deviceMetrics"] = device_metrics
         if user_agent:
             emulator_settings["userAgent"] = user_agent
@@ -911,7 +962,8 @@ def _set_chrome_options(
             debug_port = service_utils.free_port()
         chrome_options.add_argument("--remote-debugging-port=%s" % debug_port)
     if swiftshader:
-        chrome_options.add_argument("--use-gl=swiftshader")
+        chrome_options.add_argument("--use-gl=angle")
+        chrome_options.add_argument("--use-angle=swiftshader-webgl")
     elif not is_using_uc(undetectable, browser_name):
         chrome_options.add_argument("--disable-gpu")
     if not IS_LINUX and is_using_uc(undetectable, browser_name):
@@ -1265,8 +1317,8 @@ def get_driver(
     if (uc_cdp_events or uc_subprocess) and not undetectable:
         undetectable = True
     if is_using_uc(undetectable, browser_name) and mobile_emulator:
-        mobile_emulator = False
-        user_agent = None
+        if not user_agent:
+            user_agent = constants.Mobile.AGENT
     if page_load_strategy and page_load_strategy.lower() == "none":
         settings.PAGE_LOAD_STRATEGY = "none"
     proxy_auth = False
@@ -2345,7 +2397,7 @@ def get_local_driver(
         elif headless:
             if "--headless" not in edge_options.arguments:
                 edge_options.add_argument("--headless")
-        if mobile_emulator:
+        if mobile_emulator and not is_using_uc(undetectable, browser_name):
             emulator_settings = {}
             device_metrics = {}
             if (
@@ -2357,9 +2409,9 @@ def get_local_driver(
                 device_metrics["height"] = device_height
                 device_metrics["pixelRatio"] = device_pixel_ratio
             else:
-                device_metrics["width"] = 360
-                device_metrics["height"] = 640
-                device_metrics["pixelRatio"] = 2
+                device_metrics["width"] = constants.Mobile.WIDTH
+                device_metrics["height"] = constants.Mobile.HEIGHT
+                device_metrics["pixelRatio"] = constants.Mobile.RATIO
             emulator_settings["deviceMetrics"] = device_metrics
             if user_agent:
                 emulator_settings["userAgent"] = user_agent
@@ -2493,7 +2545,8 @@ def get_local_driver(
                 free_port = service_utils.free_port()
             edge_options.add_argument("--remote-debugging-port=%s" % free_port)
         if swiftshader:
-            edge_options.add_argument("--use-gl=swiftshader")
+            edge_options.add_argument("--use-gl=angle")
+            edge_options.add_argument("--use-angle=swiftshader-webgl")
         else:
             edge_options.add_argument("--disable-gpu")
         if IS_LINUX:
@@ -3401,7 +3454,13 @@ def get_local_driver(
                 driver.default_get = driver.get  # Save copy of original
                 if uc_activated:
                     driver.get = lambda url: uc_special_open_if_cf(
-                        driver, url, proxy_string
+                        driver,
+                        url,
+                        proxy_string,
+                        mobile_emulator,
+                        device_width,
+                        device_height,
+                        device_pixel_ratio,
                     )
                     driver.uc_open = lambda url: uc_open(driver, url)
                     driver.uc_open_with_tab = (
@@ -3410,6 +3469,35 @@ def get_local_driver(
                     driver.uc_open_with_reconnect = (
                         lambda url: uc_open_with_reconnect(driver, url)
                     )
+                    if mobile_emulator:
+                        uc_metrics = {}
+                        if (
+                            type(device_width) is int
+                            and type(device_height) is int
+                            and type(device_pixel_ratio) is int
+                        ):
+                            uc_metrics["width"] = device_width
+                            uc_metrics["height"] = device_height
+                            uc_metrics["pixelRatio"] = device_pixel_ratio
+                        else:
+                            uc_metrics["width"] = constants.Mobile.WIDTH
+                            uc_metrics["height"] = constants.Mobile.HEIGHT
+                            uc_metrics["pixelRatio"] = constants.Mobile.RATIO
+                        set_device_metrics_override = dict(
+                            {
+                                "width": uc_metrics["width"],
+                                "height": uc_metrics["height"],
+                                "deviceScaleFactor": uc_metrics["pixelRatio"],
+                                "mobile": True
+                            }
+                        )
+                        try:
+                            driver.execute_cdp_cmd(
+                                'Emulation.setDeviceMetricsOverride',
+                                set_device_metrics_override
+                            )
+                        except Exception:
+                            pass
                 return extend_driver(driver)
             else:  # Running headless on Linux (and not using --uc)
                 try:

seleniumbase/core/sb_driver.py

@@ -41,6 +41,12 @@ class DriverMethods():
             pass
         raise Exception('No such Element: {%s} (by="%s")!' % (selector, by))
 
+    def get_page_source(self):
+        return self.driver.page_source
+
+    def get_title(self):
+        return self.driver.title
+
     def open_url(self, *args, **kwargs):
         page_actions.open_url(self.driver, *args, **kwargs)
 
@@ -50,6 +56,12 @@ class DriverMethods():
     def click_link(self, *args, **kwargs):
         page_actions.click_link(self.driver, *args, **kwargs)
 
+    def click_if_visible(self, *args, **kwargs):
+        page_actions.click_if_visible(self.driver, *args, **kwargs)
+
+    def click_active_element(self, *args, **kwargs):
+        page_actions.click_active_element(self.driver, *args, **kwargs)
+
     def send_keys(self, *args, **kwargs):
         page_actions.send_keys(self.driver, *args, **kwargs)
 
@@ -114,6 +126,9 @@ class DriverMethods():
     def js_click(self, *args, **kwargs):
         return page_actions.js_click(self.driver, *args, **kwargs)
 
+    def is_attribute_present(self, *args, **kwargs):
+        return page_actions.has_attribute(self.driver, *args, **kwargs)
+
     def get_active_element_css(self, *args, **kwargs):
         return js_utils.get_active_element_css(self.driver, *args, **kwargs)
 
@@ -128,3 +143,26 @@ class DriverMethods():
 
     def highlight(self, *args, **kwargs):
         js_utils.highlight(self.driver, *args, **kwargs)
+
+    def set_wire_proxy(self, string):
+        """Set a proxy server for selenium-wire mode ("--wire")
+        Examples:  (ONLY avilable if using selenium-wire mode!)
+        driver.set_wire_proxy("SERVER:PORT")
+        driver.set_wire_proxy("socks5://SERVER:PORT")
+        driver.set_wire_proxy("USERNAME:PASSWORD@SERVER:PORT")
+        """
+        the_http = "http"
+        the_https = "https"
+        if string.startswith("socks4://"):
+            the_http = "socks4"
+            the_https = "socks4"
+        elif string.startswith("socks5://"):
+            the_http = "socks5"
+            the_https = "socks5"
+        string = string.split("//")[-1]
+        if hasattr(self.driver, "proxy"):
+            self.driver.proxy = {
+                "http": "%s://%s" % (the_http, string),
+                "https": "%s://%s" % (the_https, string),
+                "no_proxy": "localhost,127.0.0.1",
+            }