seleniumbase 4.24.10__py3-none-any.whl → 4.33.15__py3-none-any.whl

sbase/__init__.py

@@ -11,3 +11,4 @@ from seleniumbase import MasterQA  # noqa
 from seleniumbase import page_actions  # noqa
 from seleniumbase import page_utils  # noqa
 from seleniumbase import SB  # noqa
+from seleniumbase import translate  # noqa

sbase/steps.py

@@ -1189,3 +1189,10 @@ def set_attributes(context, selector, attribute, value):
     if attribute.endswith("'") or attribute.endswith('"'):
         attribute = attribute[:-1]
     sb.set_attributes(selector, attribute, value)
+
+
+@step("Activate CDP Mode")
+@step("User activates CDP Mode")
+def activate_cdp_mode(context):
+    sb = context.sb
+    sb.activate_cdp_mode()

seleniumbase/__init__.py

@@ -1,10 +1,8 @@
 import collections
+import os
 import pdb
-try:
-    import pdbp  # (Pdb+) --- Python Debugger Plus
-except Exception:
-    pass
 import sys
+from contextlib import suppress
 from selenium import webdriver
 from seleniumbase.__version__ import __version__
 from seleniumbase.common import decorators  # noqa
@@ -14,7 +12,7 @@ from seleniumbase.core.browser_launcher import get_driver  # noqa
 from seleniumbase.fixtures import js_utils  # noqa
 from seleniumbase.fixtures import page_actions  # noqa
 from seleniumbase.fixtures import page_utils  # noqa
-from seleniumbase.fixtures import shared_utils  # noqa
+from seleniumbase.fixtures import shared_utils
 from seleniumbase.fixtures.base_case import BaseCase  # noqa
 from seleniumbase.masterqa.master_qa import MasterQA  # noqa
 from seleniumbase.plugins.sb_manager import SB  # noqa
@@ -22,6 +20,16 @@ from seleniumbase.plugins.driver_manager import Driver  # noqa
 from seleniumbase.plugins.driver_manager import DriverContext  # noqa
 from seleniumbase import translate  # noqa
 
+with suppress(Exception):
+    import colorama
+
+with suppress(Exception):
+    import pdbp  # (Pdb+) --- Python Debugger Plus
+
+with suppress(Exception):
+    shared_utils.fix_colorama_if_windows()
+    colorama.init(autoreset=True)
+
 if sys.version_info[0] < 3 and "pdbp" in locals():
     # With Python3, "import pdbp" is all you need
     for key in pdbp.__dict__.keys():
@@ -34,11 +42,12 @@ if sys.version_info[0] < 3 and "pdbp" in locals():
         pdb.DefaultConfig.truncate_long_lines = False
         pdb.DefaultConfig.sticky_by_default = True
 colored_traceback.add_hook()
-if sys.version_info >= (3, 7):
-    webdriver.TouchActions = None  # Lifeline for past selenium-wire versions
+os.environ["SE_AVOID_STATS"] = "true"  # Disable Selenium Manager stats
+webdriver.TouchActions = None  # Lifeline for past selenium-wire versions
 if sys.version_info >= (3, 10):
     collections.Callable = collections.abc.Callable  # Lifeline for nosetests
 del collections  # Undo "import collections" / Simplify "dir(seleniumbase)"
+del os  # Undo "import os" / Simplify "dir(seleniumbase)"
 del sys  # Undo "import sys" / Simplify "dir(seleniumbase)"
 del webdriver  # Undo "import webdriver" / Simplify "dir(seleniumbase)"
 

seleniumbase/__version__.py

@@ -1,2 +1,2 @@
 # seleniumbase package
-__version__ = "4.24.10"
+__version__ = "4.33.15"

seleniumbase/behave/behave_sb.py

@@ -42,11 +42,14 @@ behave -D agent="User Agent String" -D demo
 -D binary-location=PATH  (Set path of the Chromium browser binary to use.)
 -D driver-version=VER  (Set the chromedriver or uc_driver version to use.)
 -D sjw  (Skip JS Waits for readyState to be "complete" or Angular to load.)
+-D wfa  (Wait for AngularJS to be done loading after specific web actions.)
 -D pls=PLS  (Set pageLoadStrategy on Chrome: "normal", "eager", or "none".)
--D headless  (Run tests in headless mode. The default arg on Linux OS.)
--D headless2  (Use the new headless mode, which supports extensions.)
+-D headless  (The default headless mode. Linux uses this mode by default.)
+-D headless1  (Use Chrome's old headless mode. Fast, but has limitations.)
+-D headless2  (Use Chrome's new headless mode, which supports extensions.)
 -D headed  (Run tests in headed/GUI mode on Linux OS, where not default.)
 -D xvfb  (Run tests using the Xvfb virtual display server on Linux OS.)
+-D xvfb-metrics=STRING  (Set Xvfb display size on Linux: "Width,Height".)
 -D locale=LOCALE_CODE  (Set the Language Locale Code for the web browser.)
 -D pdb  (Activate Post Mortem Debug Mode if a test fails.)
 -D interval=SECONDS  (The autoplay interval for presentations & tour steps)
@@ -68,7 +71,8 @@ behave -D agent="User Agent String" -D demo
 -D rec-behave  (Same as Recorder Mode, but also generates behave-gherkin.)
 -D rec-sleep  (If the Recorder is enabled, also records self.sleep calls.)
 -D rec-print  (If the Recorder is enabled, prints output after tests end.)
--D disable-js  (Disable JavaScript on Chromium. May break websites!)
+-D disable-cookies  (Disable Cookies on websites. Pages might break!)
+-D disable-js  (Disable JavaScript on websites. Pages might break!)
 -D disable-csp  (Disable the Content Security Policy of websites.)
 -D disable-ws  (Disable Web Security on Chromium-based browsers.)
 -D enable-ws  (Enable Web Security on Chromium-based browsers.)
@@ -89,6 +93,7 @@ behave -D agent="User Agent String" -D demo
 -D rcs | -D reuse-class-session  (Reuse session for tests in class/feature)
 -D crumbs  (Delete all cookies between tests reusing a session.)
 -D disable-beforeunload  (Disable the "beforeunload" event on Chrome.)
+-D window-position=X,Y  (Set the browser's starting window position.)
 -D window-size=WIDTH,HEIGHT  (Set the browser's starting window size.)
 -D maximize  (Start tests with the browser window maximized.)
 -D screenshot  (Save a screenshot at the end of each test.)
@@ -103,6 +108,7 @@ import colorama
 import os
 import re
 import sys
+from contextlib import suppress
 from seleniumbase import config as sb_config
 from seleniumbase.config import settings
 from seleniumbase.core import download_helper
@@ -140,10 +146,12 @@ def get_configured_sb(context):
     sb.browser = "chrome"
     sb.is_behave = True
     sb.headless = False
+    sb.headless1 = False
     sb.headless2 = False
     sb.headless_active = False
     sb.headed = False
     sb.xvfb = False
+    sb.xvfb_metrics = None
     sb.start_page = None
     sb.locale_code = None
     sb.pdb_option = False
@@ -173,6 +181,7 @@ def get_configured_sb(context):
     sb.database_env = "test"
     sb.log_path = constants.Logs.LATEST + os.sep
     sb.archive_logs = False
+    sb.disable_cookies = False
     sb.disable_js = False
     sb.disable_csp = False
     sb.disable_ws = False
@@ -192,6 +201,7 @@ def get_configured_sb(context):
     sb._disable_beforeunload = False
     sb.visual_baseline = False
     sb.use_wire = False
+    sb.window_position = None
     sb.window_size = None
     sb.maximize_option = False
     sb.is_context_manager = False
@@ -290,6 +300,11 @@ def get_configured_sb(context):
             sb.headless = True
             continue
         # Handle: -D headless2
+        if low_key == "headless1":
+            sb.headless1 = True
+            sb.headless = True
+            continue
+        # Handle: -D headless2
         if low_key == "headless2":
             sb.headless2 = True
             continue
@@ -301,6 +316,13 @@ def get_configured_sb(context):
         if low_key == "xvfb":
             sb.xvfb = True
             continue
+        # Handle: -D xvfb-metrics=STR / xvfb_metrics=STR
+        if low_key in ["xvfb-metrics", "xvfb_metrics"]:
+            xvfb_metrics = userdata[key]
+            if xvfb_metrics == "true":
+                xvfb_metrics = sb.xvfb_metrics  # revert to default
+            sb.xvfb_metrics = xvfb_metrics
+            continue
         # Handle: -D start-page=URL / start_page=URL / url=URL
         if low_key in ["start-page", "start_page", "url"]:
             start_page = userdata[key]
@@ -515,16 +537,20 @@ def get_configured_sb(context):
         if low_key in ["archive-logs", "archive_logs"]:
             sb.archive_logs = True
             continue
+        # Handle: -D disable-cookies / disable_cookies
+        if low_key in ["disable-cookies", "disable_cookies"]:
+            sb.disable_cookies = True
+            continue
         # Handle: -D disable-js / disable_js
         if low_key in ["disable-js", "disable_js"]:
             sb.disable_js = True
             continue
-        # Handle: -D disable-csp / disable_csp
-        if low_key in ["disable-csp", "disable_csp"]:
+        # Handle: -D disable-csp / disable_csp / dcsp
+        if low_key in ["disable-csp", "disable_csp", "dcsp"]:
             sb.disable_csp = True
             continue
-        # Handle: -D disable-ws / disable_ws
-        if low_key in ["disable-ws", "disable_ws"]:
+        # Handle: -D disable-ws / disable_ws / dws
+        if low_key in ["disable-ws", "disable_ws", "dws"]:
             sb.disable_ws = True
             continue
         # Handle: -D enable-ws / enable_ws
@@ -588,6 +614,10 @@ def get_configured_sb(context):
         if low_key in ["sjw", "skip-js-waits", "skip_js_waits"]:
             settings.SKIP_JS_WAITS = True
             continue
+        # Handle: -D wfa / wait-for-angularjs / wait_for_angularjs
+        if low_key in ["wfa", "wait-for-angularjs", "wait_for_angularjs"]:
+            settings.WAIT_FOR_ANGULARJS = True
+            continue
         # Handle: -D visual-baseline / visual_baseline
         if low_key in ["visual-baseline", "visual_baseline"]:
             sb.visual_baseline = True
@@ -596,6 +626,13 @@ def get_configured_sb(context):
         if low_key == "wire":
             sb.use_wire = True
             continue
+        # Handle: -D window-position=X,Y / window_position=X,Y
+        if low_key in ["window-position", "window_position"]:
+            window_position = userdata[key]
+            if window_position == "true":
+                window_position = sb.window_position  # revert to default
+            sb.window_position = window_position
+            continue
         # Handle: -D window-size=Width,Height / window_size=Width,Height
         if low_key in ["window-size", "window_size"]:
             window_size = userdata[key]
@@ -840,6 +877,7 @@ def get_configured_sb(context):
     # Recorder Mode can still optimize scripts in "-D headless2" mode.
     if sb.recorder_ext and sb.headless:
         sb.headless = False
+        sb.headless1 = False
         sb.headless2 = True
     if sb.headless2 and sb.browser == "firefox":
         sb.headless2 = False  # Only for Chromium browsers
@@ -862,22 +900,27 @@ def get_configured_sb(context):
         and not sb.headless2
         and not sb.xvfb
     ):
-        print(
-            '(Linux uses "-D headless" by default. '
-            'To override, use "-D headed" / "-D gui". '
-            'For Xvfb mode instead, use "-D xvfb". '
-            "Or you can hide this info by using"
-            '"-D headless" / "-D headless2".)'
-        )
-        sb.headless = True
+        if not sb.undetectable:
+            print(
+                '(Linux uses "-D headless" by default. '
+                'To override, use "-D headed" / "-D gui". '
+                'For Xvfb mode instead, use "-D xvfb". '
+                "Or you can hide this info by using"
+                '"-D headless" / "-D headless2" / "-D uc".)'
+            )
+            sb.headless = True
+        else:
+            sb.xvfb = True
     # Recorder Mode can still optimize scripts in --headless2 mode.
     if sb.recorder_mode and sb.headless:
         sb.headless = False
+        sb.headless1 = False
         sb.headless2 = True
     if not sb.headless and not sb.headless2:
         sb.headed = True
     if sb.browser == "safari" and sb.headless:
         sb.headless = False  # Safari doesn't support headless mode
+        sb.headless1 = False
     if sb.save_screenshot_after_test and sb.no_screenshot_after_test:
         sb.save_screenshot_after_test = False  # "no_screenshot" has priority
     if sb.servername != "localhost":
@@ -886,6 +929,39 @@ def get_configured_sb(context):
         # If the port is "443", the protocol is "https"
         if str(sb.port) == "443":
             sb.protocol = "https"
+    if (
+        (sb.enable_ws is None and sb.disable_ws is None)
+        or (sb.disable_ws is not None and not sb.disable_ws)
+        or (sb.enable_ws is not None and sb.enable_ws)
+    ):
+        sb.enable_ws = True
+        sb.disable_ws = False
+    else:
+        sb.enable_ws = False
+        sb.disable_ws = True
+    if sb.window_position:
+        window_position = sb.window_position
+        if window_position.count(",") != 1:
+            message = (
+                '\n\n  window_position expects an "x,y" string!'
+                '\n  (Your input was: "%s")\n' % window_position
+            )
+            raise Exception(message)
+        window_position = window_position.replace(" ", "")
+        win_x = None
+        win_y = None
+        try:
+            win_x = int(window_position.split(",")[0])
+            win_y = int(window_position.split(",")[1])
+        except Exception:
+            message = (
+                '\n\n  Expecting integer values for "x,y"!'
+                '\n  (window_position input was: "%s")\n'
+                % window_position
+            )
+            raise Exception(message)
+        settings.WINDOW_START_X = win_x
+        settings.WINDOW_START_Y = win_y
     if sb.window_size:
         window_size = sb.window_size
         if window_size.count(",") != 1:
@@ -920,9 +996,11 @@ def get_configured_sb(context):
     sb_config.is_pytest = False
     sb_config.is_nosetest = False
     sb_config.is_context_manager = False
+    sb_config.window_position = sb.window_position
     sb_config.window_size = sb.window_size
     sb_config.maximize_option = sb.maximize_option
     sb_config.xvfb = sb.xvfb
+    sb_config.xvfb_metrics = sb.xvfb_metrics
     sb_config.reuse_class_session = sb._reuse_class_session
     sb_config.save_screenshot = sb.save_screenshot_after_test
     sb_config.no_screenshot = sb.no_screenshot_after_test
@@ -933,6 +1011,7 @@ def get_configured_sb(context):
     sb_config.pdb_option = sb.pdb_option
     sb_config.rec_behave = sb.rec_behave
     sb_config.rec_print = sb.rec_print
+    sb_config.disable_cookies = sb.disable_cookies
     sb_config.disable_js = sb.disable_js
     sb_config.disable_csp = sb.disable_csp
     sb_config.record_sleep = sb.record_sleep
@@ -1144,20 +1223,14 @@ def behave_dashboard_prepare():
         sb_config.item_count_untested = sb_config.item_count
         dash_path = os.path.join(os.getcwd(), "dashboard.html")
         star_len = len("Dashboard: ") + len(dash_path)
-        try:
+        with suppress(Exception):
             terminal_size = os.get_terminal_size().columns
             if terminal_size > 30 and star_len > terminal_size:
                 star_len = terminal_size
-        except Exception:
-            pass
         stars = "*" * star_len
         c1 = ""
         cr = ""
         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))
@@ -1245,7 +1318,7 @@ def _perform_behave_unconfigure_():
 
 
 def do_final_driver_cleanup_as_needed():
-    try:
+    with suppress(Exception):
         if hasattr(sb_config, "last_driver") and sb_config.last_driver:
             if (
                 not is_windows
@@ -1253,8 +1326,6 @@ def do_final_driver_cleanup_as_needed():
                 or sb_config.last_driver.service.process
             ):
                 sb_config.last_driver.quit()
-    except Exception:
-        pass
 
 
 def _perform_behave_terminal_summary_():
@@ -1263,20 +1334,14 @@ def _perform_behave_terminal_summary_():
     )
     dash_path = os.path.join(os.getcwd(), "dashboard.html")
     equals_len = len("Dashboard: ") + len(dash_path)
-    try:
+    with suppress(Exception):
         terminal_size = os.get_terminal_size().columns
         if terminal_size > 30 and equals_len > terminal_size:
             equals_len = terminal_size
-    except Exception:
-        pass
     equals = "=" * (equals_len + 2)
     c2 = ""
     cr = ""
     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/common/decorators.py

@@ -1,11 +1,18 @@
+import colorama
 import logging
 import math
+import sys
 import time
 import warnings
 from contextlib import contextmanager
 from functools import wraps
 from seleniumbase.common.exceptions import TimeoutException
 
+c1 = colorama.Fore.BLUE + colorama.Back.LIGHTYELLOW_EX
+cr = colorama.Style.RESET_ALL
+if "linux" in sys.platform:
+    c1 = cr = ""
+
 
 @contextmanager
 def print_runtime(description=None, limit=None):
@@ -43,23 +50,25 @@ def print_runtime(description=None, limit=None):
     finally:
         end_time = time.time()
         run_time = end_time - start_time
+        name = description
+        info = c1 + "<info>" + cr
         # Print times with a statistically significant number of decimal places
         if run_time < 0.0001:
-            print("  {%s} ran for %.7f seconds." % (description, run_time))
+            print("%s - {%s} ran for %.7f seconds." % (info, name, run_time))
         elif run_time < 0.001:
-            print("  {%s} ran for %.6f seconds." % (description, run_time))
+            print("%s - {%s} ran for %.6f seconds." % (info, name, run_time))
         elif run_time < 0.01:
-            print("  {%s} ran for %.5f seconds." % (description, run_time))
+            print("%s - {%s} ran for %.5f seconds." % (info, name, run_time))
         elif run_time < 0.1:
-            print("  {%s} ran for %.4f seconds." % (description, run_time))
+            print("%s - {%s} ran for %.4f seconds." % (info, name, run_time))
         elif run_time < 1:
-            print("  {%s} ran for %.3f seconds." % (description, run_time))
+            print("%s - {%s} ran for %.3f seconds." % (info, name, run_time))
         else:
-            print("  {%s} ran for %.2f seconds." % (description, run_time))
+            print("%s - {%s} ran for %.2f seconds." % (info, name, run_time))
         if limit and limit > 0 and run_time > limit:
             message = (
                 "\n {%s} duration of %.2fs exceeded the time limit of %.2fs!"
-                % (description, run_time, limit)
+                % (name, run_time, limit)
             )
             if exception:
                 message = exception.msg + "\nAND " + message

seleniumbase/config/proxy_list.py

@@ -23,9 +23,9 @@ you can try finding one from one of following sites:
 """
 
 PROXY_LIST = {
-    "example1": "37.19.220.129:8443",  # (Example) - set your own proxy here
-    "example2": "socks4://104.236.32.53:8915",  # (Example)
-    "example3": "socks5://142.44.212.57:49006",  # (Example)
+    "example1": "35.185.196.38:3128",  # (Example) - set your own proxy here
+    "example2": "129.80.134.71:3128",  # (Example)
+    "example3": "socks5://184.178.172.5:15303",  # (Example)
     "proxy1": None,
     "proxy2": None,
     "proxy3": None,

seleniumbase/config/settings.py

@@ -110,6 +110,10 @@ DISABLE_CSP_ON_CHROME = False
 # (This applies when using --proxy=[PROXY_STRING] for using a proxy server.)
 RAISE_INVALID_PROXY_STRING_EXCEPTION = True
 
+# Default browser coordinates when opening new windows for tests.
+WINDOW_START_X = 20
+WINDOW_START_Y = 54
+
 # Default browser resolutions when opening new windows for tests.
 # (Headless resolutions take priority, and include all browsers.)
 # (Firefox starts maximized by default when running in GUI Mode.)

seleniumbase/console_scripts/logo_helper.py

@@ -3,7 +3,8 @@
     http://www.patorjk.com/software/taag/#p=display&f=Slant&t=SeleniumBase """
 
 import colorama
-import sys
+import os
+from contextlib import suppress
 
 r"""
    ______     __           _                  ____
@@ -15,13 +16,6 @@ r"""
 
 
 def get_seleniumbase_logo():
-    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
@@ -66,4 +60,49 @@ def get_seleniumbase_logo():
     sb += "                        "
     sb += cr
     sb += cr
+    with suppress(Exception):
+        terminal_width = os.get_terminal_size().columns
+        if isinstance(terminal_width, int) and terminal_width >= 66:
+            return sb
+
+    # If the logo is wider than the screen width, use a smaller one:
+    r"""
+     ___      _          _             ___
+    / __| ___| |___ _ _ (_)_  _ _ __  | _ ) __ _ ______
+    \__ \/ -_) / -_) ' \| | \| | '  \ | _ \/ _` (_-< -_)
+    |___/\___|_\___|_||_|_|\_,_|_|_|_\|___/\__,_/__|___|
+    """
+    sb = " "
+    sb += cr
+    sb += "\n"
+    sb += c1
+    sb += " ___      _          _            "
+    sb += c2
+    sb += " ___              "
+    sb += cr
+    sb += "\n"
+    sb += c1
+    sb += "/ __| ___| |___ _ _ (_)_  _ _ __  "
+    sb += c2
+    sb += "| _ ) __ _ ______ "
+    sb += cr
+    sb += "\n"
+    sb += c1
+    sb += "\\__ \\/ -_) / -_) ' \\| | \\| | '  \\ "
+    sb += c2
+    sb += "| _ \\/ _` (_-< -_)"
+    sb += cr
+    sb += "\n"
+    sb += c1
+    sb += "|___/\\___|_\\___|_||_|_|\\_,_|_|_|_\\"
+    sb += c2
+    sb += "|___/\\__,_/__|___|"
+    sb += cr
+    sb += "\n"
+    sb += c3
+    sb += "                                  "
+    sb += c4
+    sb += "                  "
+    sb += cr
+    sb += cr
     return sb