seleniumbase 4.44.2__py3-none-any.whl → 4.45.10__py3-none-any.whl

seleniumbase/__version__.py

@@ -1,2 +1,2 @@
 # seleniumbase package
-__version__ = "4.44.2"
+__version__ = "4.45.10"

seleniumbase/behave/behave_sb.py

@@ -113,6 +113,7 @@ import sys
 from contextlib import suppress
 from seleniumbase import config as sb_config
 from seleniumbase.config import settings
+from seleniumbase.core import detect_b_ver
 from seleniumbase.core import download_helper
 from seleniumbase.core import log_helper
 from seleniumbase.core import proxy_helper
@@ -493,6 +494,12 @@ def get_configured_sb(context):
             sb.binary_location = binary_location
             sb_config.binary_location = binary_location
             continue
+        # Handle: -D use-chromium
+        if low_key in ["use-chromium"] and not sb_config.binary_location:
+            binary_location = "_chromium_"
+            sb.binary_location = binary_location
+            sb_config.binary_location = binary_location
+            continue
         # Handle: -D cft
         if low_key in ["cft"] and not sb_config.binary_location:
             binary_location = "cft"
@@ -890,6 +897,13 @@ def get_configured_sb(context):
             "\nOnly ONE default browser is allowed!\n"
             "%s browsers were selected: %s" % (len(browsers), browsers)
         )
+    if sb.browser in ["opera", "brave", "comet", "atlas"]:
+        bin_loc = detect_b_ver.get_binary_location(sb.browser)
+        if bin_loc and os.path.exists(bin_loc):
+            sb_config._cdp_browser = sb.browser
+            sb_config._cdp_bin_loc = bin_loc
+            sb_config.binary_location = bin_loc
+            sb.binary_location = bin_loc
     # Recorder Mode can still optimize scripts in "-D headless2" mode.
     if sb.recorder_ext and sb.headless:
         sb.headless = False

seleniumbase/console_scripts/run.py

@@ -248,6 +248,7 @@ def show_mkdir_usage():
     print("     sbase mkdir ui_tests")
     print("  Options:")
     print("     -b / --basic  (Only config files. No tests added.)")
+    print("     --gha  (Include GitHub Actions YML with defaults.)")
     print("  Output:")
     print("     Creates a new folder for running SBase scripts.")
     print("     The new folder contains default config files,")

seleniumbase/console_scripts/sb_install.py

@@ -19,6 +19,7 @@ Examples:
     sbase get chromedriver stable
     sbase get chromedriver beta
     sbase get chromedriver -p
+    sbase get chromium
     sbase get cft 131
     sbase get chs
 Output:
@@ -46,16 +47,21 @@ from seleniumbase import config as sb_config
 from seleniumbase import drivers  # webdriver storage folder for SeleniumBase
 from seleniumbase.drivers import cft_drivers  # chrome-for-testing
 from seleniumbase.drivers import chs_drivers  # chrome-headless-shell
+from seleniumbase.drivers import chromium_drivers  # base chromium
 
 urllib3.disable_warnings()
 ARCH = platform.architecture()[0]
 IS_ARM_MAC = shared_utils.is_arm_mac()
 IS_MAC = shared_utils.is_mac()
+IS_ARM_LINUX = shared_utils.is_arm_linux()
 IS_LINUX = shared_utils.is_linux()
 IS_WINDOWS = shared_utils.is_windows()
 DRIVER_DIR = os.path.dirname(os.path.realpath(drivers.__file__))
 DRIVER_DIR_CFT = os.path.dirname(os.path.realpath(cft_drivers.__file__))
 DRIVER_DIR_CHS = os.path.dirname(os.path.realpath(chs_drivers.__file__))
+DRIVER_DIR_CHROMIUM = os.path.dirname(
+    os.path.realpath(chromium_drivers.__file__)
+)
 LOCAL_PATH = "/usr/local/bin/"  # On Mac and Linux systems
 DEFAULT_CHROMEDRIVER_VERSION = "114.0.5735.90"  # (If can't find LATEST_STABLE)
 DEFAULT_GECKODRIVER_VERSION = "v0.36.0"
@@ -177,7 +183,7 @@ def requests_get_with_retry(url):
 
 
 def get_cft_known_good_versions():
-    if hasattr(sb_config, "cft_kgv_json") and sb_config.cft_kgv_json:
+    if getattr(sb_config, "cft_kgv_json", None):
         return sb_config.cft_kgv_json
     cft_ngv_url = (
         "https://googlechromelabs.github.io/"
@@ -188,7 +194,7 @@ def get_cft_known_good_versions():
 
 
 def get_cft_latest_versions_per_milestone():
-    if hasattr(sb_config, "cft_lvpm_json") and sb_config.cft_lvpm_json:
+    if getattr(sb_config, "cft_lvpm_json", None):
         return sb_config.cft_lvpm_json
     cft_lvpm_url = (
         "https://googlechromelabs.github.io/"
@@ -203,9 +209,62 @@ def get_cft_latest_version_from_milestone(milestone):
     return url_request.json()["milestones"][milestone]["version"]
 
 
+def get_chromium_channel_revision(platform_code, channel):
+    """Snapshots only exist for revisions where a build occurred.
+    Therefore, not all found revisions will lead to snapshots."""
+    platform_key = None
+    if platform_code in ["Mac_Arm", "Mac"]:
+        platform_key = "Mac"
+    elif platform_code in ["Linux_x64"]:
+        platform_key = "Linux"
+    elif platform_code in ["Win_x64"]:
+        platform_key = "Windows"
+    elif platform_code in ["Win"]:
+        platform_key = "Win32"
+    channel_key = None
+    if channel.lower() == "stable":
+        channel_key = "Stable"
+    elif channel.lower() == "beta":
+        channel_key = "Beta"
+    elif channel.lower() == "dev":
+        channel_key = "Dev"
+    elif channel.lower() == "canary":
+        channel_key = "Canary"
+    base_url = "https://chromiumdash.appspot.com/fetch_releases"
+    url = f"{base_url}?channel={channel_key}&platform={platform_key}&num=1"
+    url_request = requests_get_with_retry(url)
+    data = None
+    if url_request.ok:
+        data = url_request.text
+    else:
+        raise Exception("Could not determine Chromium revision!")
+    if data:
+        try:
+            import ast
+
+            result = ast.literal_eval(data)
+            revision = result[0]["chromium_main_branch_position"]
+            return str(revision)
+        except Exception:
+            return get_latest_chromedriver_version(platform_code)
+    else:
+        return get_latest_chromedriver_version(platform_code)
+
+
+def get_chromium_latest_revision(platform_code):
+    base_url = "https://storage.googleapis.com/chromium-browser-snapshots"
+    url = f"{base_url}/{platform_code}/LAST_CHANGE"
+    url_request = requests_get_with_retry(url)
+    if url_request.ok:
+        latest_revision = url_request.text
+    else:
+        raise Exception("Could not determine latest Chromium revision!")
+    return latest_revision
+
+
 def get_latest_chromedriver_version(channel="Stable"):
     try:
-        if hasattr(sb_config, "cft_lkgv_json") and sb_config.cft_lkgv_json:
+        if getattr(sb_config, "cft_lkgv_json", None):
             return sb_config.cft_lkgv_json["channels"][channel]["version"]
         req = requests_get(
             "https://googlechromelabs.github.io/"
@@ -239,10 +298,7 @@ def get_latest_canary_chromedriver_version():
 def log_d(message):
     """If setting sb_config.settings.HIDE_DRIVER_DOWNLOADS to True,
     output from driver downloads are logged instead of printed."""
-    if (
-        hasattr(sb_config.settings, "HIDE_DRIVER_DOWNLOADS")
-        and sb_config.settings.HIDE_DRIVER_DOWNLOADS
-    ):
+    if getattr(sb_config.settings, "HIDE_DRIVER_DOWNLOADS", None):
         logging.debug(message)
     else:
         print(message)
@@ -251,7 +307,7 @@ def log_d(message):
 def main(override=None, intel_for_uc=None, force_uc=None):
     if override:
         found_proxy = None
-        if hasattr(sb_config, "proxy_driver") and sb_config.proxy_driver:
+        if getattr(sb_config, "proxy_driver", None):
             if " --proxy=" in " ".join(sys.argv):
                 for arg in sys.argv:
                     if arg.startswith("--proxy="):
@@ -277,6 +333,11 @@ def main(override=None, intel_for_uc=None, force_uc=None):
         elif override.startswith("iedriver "):
             extra = override.split("iedriver ")[1]
             sys.argv = ["seleniumbase", "get", "iedriver", extra]
+        elif override == "chromium":
+            sys.argv = ["seleniumbase", "get", "chromium"]
+        elif override.startswith("chromium "):
+            extra = override.split("chromium ")[1]
+            sys.argv = ["seleniumbase", "get", "chromium", extra]
         elif override == "cft":
             sys.argv = ["seleniumbase", "get", "cft"]
         elif override.startswith("cft "):
@@ -311,8 +372,7 @@ def main(override=None, intel_for_uc=None, force_uc=None):
     downloads_folder = DRIVER_DIR
     if (
         hasattr(sb_config, "settings")
-        and hasattr(sb_config.settings, "NEW_DRIVER_DIR")
-        and sb_config.settings.NEW_DRIVER_DIR
+        and getattr(sb_config.settings, "NEW_DRIVER_DIR", None)
         and os.path.exists(sb_config.settings.NEW_DRIVER_DIR)
     ):
         downloads_folder = sb_config.settings.NEW_DRIVER_DIR
@@ -320,6 +380,8 @@ def main(override=None, intel_for_uc=None, force_uc=None):
         downloads_folder = DRIVER_DIR_CFT
     elif override == "chs" or name == "chs":
         downloads_folder = DRIVER_DIR_CHS
+    elif override == "chromium":
+        downloads_folder = DRIVER_DIR_CHROMIUM
     expected_contents = None
     platform_code = None
     copy_to_path = False
@@ -647,6 +709,31 @@ def main(override=None, intel_for_uc=None, force_uc=None):
             "https://storage.googleapis.com/chrome-for-testing-public/"
             "%s/%s/%s" % (use_version, platform_code, file_name)
         )
+    elif name == "chromium":
+        if IS_MAC:
+            if IS_ARM_MAC:
+                platform_code = "Mac_Arm"
+            else:
+                platform_code = "Mac"
+            file_name = "chrome-mac.zip"
+        elif IS_LINUX:
+            platform_code = "Linux_x64"
+            file_name = "chrome-linux.zip"
+        elif IS_WINDOWS:
+            if "64" in ARCH:
+                platform_code = "Win_x64"
+            else:
+                platform_code = "Win"
+            file_name = "chrome-win.zip"
+        revision = get_chromium_latest_revision(platform_code)
+        msg = c2 + "Chromium revision to download" + cr
+        p_version = c3 + revision + cr
+        log_d("\n*** %s = %s" % (msg, p_version))
+        download_url = (
+            "https://storage.googleapis.com/chromium-browser-snapshots/"
+            "%s/%s/%s" % (platform_code, revision, file_name)
+        )
+        downloads_folder = DRIVER_DIR_CHROMIUM
     elif name == "chrome-headless-shell" or name == "chs":
         set_version = None
         found_version = None
@@ -1007,12 +1094,12 @@ def main(override=None, intel_for_uc=None, force_uc=None):
     remote_file = requests_get_with_retry(download_url)
     with open(file_path, "wb") as file:
         file.write(remote_file.content)
-    log_d("%sDownload Complete!%s\n" % (c1, cr))
 
     if file_name.endswith(".zip"):
         zip_file_path = file_path
         zip_ref = zipfile.ZipFile(zip_file_path, "r")
         contents = zip_ref.namelist()
+        log_d("%sDownload Complete!%s\n" % (c1, cr))
         if (
             len(contents) >= 1
             and name in ["chromedriver", "uc_driver", "geckodriver"]
@@ -1253,6 +1340,48 @@ def main(override=None, intel_for_uc=None, force_uc=None):
                 "Chrome for Testing was saved inside:\n%s%s\n%s\n"
                 % (pr_base_path, pr_sep, pr_folder_name)
             )
+        elif name == "chromium":
+            # Zip file is valid. Proceed.
+            driver_path = None
+            driver_file = None
+            base_path = os.sep.join(zip_file_path.split(os.sep)[:-1])
+            folder_name = contents[0].split("/")[0]
+            folder_path = os.path.join(base_path, folder_name)
+            if IS_MAC or IS_LINUX:
+                if (
+                    "chromium" in folder_path
+                    and "drivers" in folder_path
+                    and os.path.exists(folder_path)
+                ):
+                    shutil.rmtree(folder_path)
+                subprocess.run(
+                    ["unzip", zip_file_path, "-d", downloads_folder]
+                )
+            elif IS_WINDOWS:
+                subprocess.run(
+                    [
+                        "powershell",
+                        "Expand-Archive",
+                        "-Path",
+                        zip_file_path,
+                        "-DestinationPath",
+                        downloads_folder,
+                        "-Force",
+                    ]
+                )
+            else:
+                zip_ref.extractall(downloads_folder)
+                zip_ref.close()
+            with suppress(Exception):
+                os.remove(zip_file_path)
+            log_d("%sUnzip Complete!%s\n" % (c2, cr))
+            pr_base_path = c3 + base_path + cr
+            pr_sep = c3 + os.sep + cr
+            pr_folder_name = c3 + folder_name + cr
+            log_d(
+                "Chromium was saved inside:\n%s%s\n%s\n"
+                % (pr_base_path, pr_sep, pr_folder_name)
+            )
         elif name == "chrome-headless-shell" or name == "chs":
             # Zip file is valid. Proceed.
             driver_path = None
@@ -1304,6 +1433,7 @@ def main(override=None, intel_for_uc=None, force_uc=None):
         tar = tarfile.open(file_path)
         contents = tar.getnames()
         if len(contents) == 1:
+            log_d("%sDownload Complete!%s\n" % (c1, cr))
             for f_name in contents:
                 # Remove existing version if exists
                 new_file = os.path.join(downloads_folder, str(f_name))
@@ -1341,6 +1471,7 @@ def main(override=None, intel_for_uc=None, force_uc=None):
     else:
         # Not a .zip file or a .tar.gz file. Just a direct download.
         if "Driver" in file_name or "driver" in file_name:
+            log_d("%sDownload Complete!%s\n" % (c1, cr))
             log_d("Making [%s] executable ..." % file_name)
             make_executable(file_path)
             log_d("%s[%s] is now ready for use!%s" % (c1, file_name, cr))

seleniumbase/console_scripts/sb_mkdir.py

@@ -10,6 +10,7 @@ Example:
 
 Options:
     -b / --basic  (Only config files. No tests added.)
+    --gha  (Include GitHub Actions YML with defaults.)
 
 Output:
     Creates a new folder for running SBase scripts.
@@ -32,6 +33,7 @@ def invalid_run_command(msg=None):
     exp += "     sbase mkdir ui_tests\n"
     exp += "  Options:\n"
     exp += "     -b / --basic  (Only config files. No tests added.)\n"
+    exp += "     --gha  (Include GitHub Actions YML with defaults.)\n"
     exp += "  Output:\n"
     exp += "     Creates a new folder for running SBase scripts.\n"
     exp += "     The new folder contains default config files,\n"
@@ -58,6 +60,7 @@ def main():
         c7 = colorama.Fore.BLACK + colorama.Back.MAGENTA
         cr = colorama.Style.RESET_ALL
 
+    gha = False
     basic = False
     help_me = False
     error_msg = None
@@ -89,6 +92,8 @@ def main():
                 help_me = True
             elif option == "-b" or option == "--basic":
                 basic = True
+            elif option == "--gha":
+                gha = True
             else:
                 invalid_cmd = "\n===> INVALID OPTION: >> %s <<\n" % option
                 invalid_cmd = invalid_cmd.replace(">> ", ">>" + c5 + " ")
@@ -246,6 +251,12 @@ def main():
     data.append("msedgedriver.exe")
     data.append("operadriver.exe")
     data.append("uc_driver.exe")
+    data.append("chrome-mac.zip")
+    data.append("chrome-linux.zip")
+    data.append("chrome-win.zip")
+    data.append("chrome-mac")
+    data.append("chrome-linux")
+    data.append("chrome-win")
     data.append("chrome-mac-arm64.zip")
     data.append("chrome-mac-x64.zip")
     data.append("chrome-linux64.zip")
@@ -315,9 +326,69 @@ def main():
     file.writelines("\r\n".join(data))
     file.close()
 
+    if gha:
+        dir_name_b = dir_name + "/" + ".github"
+        os.mkdir(dir_name_b)
+        dir_name_c = dir_name_b + "/" + "workflows"
+        os.mkdir(dir_name_c)
+
+        data = []
+        data.append("name: CI build")
+        data.append("on:")
+        data.append("  push:")
+        data.append("    branches:")
+        data.append("  pull_request:")
+        data.append("    branches:")
+        data.append("  workflow_dispatch:")
+        data.append("    branches:")
+        data.append("jobs:")
+        data.append("  build:")
+        data.append("    env:")
+        data.append('      PY_COLORS: "1"')
+        data.append("    strategy:")
+        data.append("      fail-fast: false")
+        data.append("      max-parallel: 15")
+        data.append("      matrix:")
+        data.append("        os: [ubuntu-latest]")
+        data.append('        python-version: ["3.x"]')
+        data.append("    runs-on: ${{ matrix.os }}")
+        data.append("    steps:")
+        data.append("    - uses: actions/checkout@v6")
+        data.append("    - name: Set up Python ${{ matrix.python-version }}")
+        data.append("      uses: actions/setup-python@v6")
+        data.append("      with:")
+        data.append("        python-version: ${{ matrix.python-version }}")
+        data.append("    - name: Install dependencies")
+        data.append("      run: |")
+        data.append("        python -m pip install --upgrade pip")
+        data.append("        pip install -r requirements.txt")
+        data.append("    - name: Install Chrome")
+        data.append("      if: matrix.os == 'ubuntu-latest'")
+        data.append("      run: sudo apt install google-chrome-stable")
+        data.append("    - name: Download chromedriver")
+        data.append("      run: sbase get chromedriver")
+        data.append("    - name: Run pytest")
+        data.append("      run: pytest -v -s --rs --crumbs --reruns=1")
+        data.append("    - name: Upload artifacts")
+        data.append("      if: ${{ always() }}")
+        data.append("      uses: actions/upload-artifact@v6")
+        data.append("      with:")
+        data.append("        name: seleniumbase-artifacts")
+        data.append("        path: ./latest_logs/")
+        data.append("        if-no-files-found: ignore")
+        data.append("")
+        file_path = "%s/%s" % (dir_name_c, "python-package.yml")
+        file = open(file_path, mode="w+", encoding="utf-8")
+        file.writelines("\r\n".join(data))
+        file.close()
+
     if basic:
         data = []
         data.append("  %s/" % dir_name)
+        if gha:
+            data.append("  ├── .github")
+            data.append("  │   └── workflows/")
+            data.append("  │      └── python-package.yml")
         data.append("  ├── __init__.py")
         data.append("  ├── pytest.ini")
         data.append("  ├── requirements.txt")
@@ -497,6 +568,7 @@ def main():
     )
     data.append('        self.click(\'mark:contains("%s")\' % keyword)')
     data.append('        self.assert_title_contains(title_text)')
+    data.append('        self.save_screenshot_to_logs()')
     data.append("")
     file_path = "%s/%s" % (dir_name, "parameterized_test.py")
     file = open(file_path, mode="w+", encoding="utf-8")
@@ -734,6 +806,10 @@ def main():
 
     data = []
     data.append("  %s/" % dir_name)
+    if gha:
+        data.append("  ├── .github")
+        data.append("  │   └── workflows/")
+        data.append("  │      └── python-package.yml")
     data.append("  ├── __init__.py")
     data.append("  ├── my_first_test.py")
     data.append("  ├── parameterized_test.py")

seleniumbase/console_scripts/sb_mkrec.py

@@ -84,6 +84,11 @@ def main():
     error_msg = None
     invalid_cmd = None
     use_edge = False
+    use_opera = False
+    use_brave = False
+    use_comet = False
+    use_atlas = False
+    use_chromium = False
     use_uc = False
     esc_end = False
     start_page = None
@@ -138,6 +143,16 @@ def main():
                 help_me = True
             elif option.lower() == "--edge":
                 use_edge = True
+            elif option.lower() == "--opera":
+                use_opera = True
+            elif option.lower() == "--brave":
+                use_brave = True
+            elif option.lower() == "--comet":
+                use_comet = True
+            elif option.lower() == "--atlas":
+                use_atlas = True
+            elif option.lower() == "--use-chromium":
+                use_chromium = True
             elif option.lower() == "--ee":
                 esc_end = True
             elif option.lower() in ("--gui", "--headed"):
@@ -275,6 +290,16 @@ def main():
             )
     if use_edge:
         run_cmd += " --edge"
+    elif use_opera:
+        run_cmd += " --opera"
+    elif use_brave:
+        run_cmd += " --brave"
+    elif use_comet:
+        run_cmd += " --comet"
+    elif use_atlas:
+        run_cmd += " --atlas"
+    elif use_chromium:
+        run_cmd += " --use-chromium"
     if force_gui:
         run_cmd += " --gui"
     if use_uc:

seleniumbase/console_scripts/sb_recorder.py

@@ -144,6 +144,16 @@ def do_recording(file_name, url, overwrite_enabled, use_chrome, window):
             )
         if not use_chrome:
             command += " --edge"
+        elif "--opera" in command_args:
+            command += " --opera"
+        elif "--brave" in command_args:
+            command += " --brave"
+        elif "--comet" in command_args:
+            command += " --comet"
+        elif "--atlas" in command_args:
+            command += " --atlas"
+        elif "--use-chromium" in command_args:
+            command += " --use-chromium"
         if (
             "--uc" in command_args
             or "--cdp" in command_args
@@ -226,10 +236,37 @@ def create_tkinter_gui():
     chk = tk.Checkbutton(window, text="Overwrite existing files", variable=cbx)
     chk.pack()
     chk.select()
+    use_stealth = False
+    command_args = sys.argv[2:]
+    if (
+        "--uc" in command_args
+        or "--cdp" in command_args
+        or "--undetected" in command_args
+        or "--undetectable" in command_args
+    ):
+        use_stealth = True
+    browser_display = "Use Chrome over Edge"
+    if "--opera" in command_args:
+        browser_display = "Use Opera over Edge"
+    elif "--brave" in command_args:
+        browser_display = "Use Brave over Edge"
+    elif "--comet" in command_args:
+        browser_display = "Use Comet over Edge"
+    elif "--atlas" in command_args:
+        browser_display = "Use Atlas over Edge"
     cbb = tk.IntVar()
-    chkb = tk.Checkbutton(window, text="Use Chrome over Edge", variable=cbb)
-    chkb.pack()
-    chkb.select()
+    if not use_stealth:
+        chkb = tk.Checkbutton(window, text=browser_display, variable=cbb)
+        chkb.pack()
+        if "--edge" not in command_args:
+            chkb.select()
+    else:
+        chkb = tk.Checkbutton(
+            window, text="Stealthy Chrome Mode", variable=cbb
+        )
+        chkb.pack()
+        chkb.select()
+        chkb.config(state=tk.DISABLED)
     tk.Label(window, text="").pack()
     url = tk.StringVar()
     tk.Label(window, text="Enter the URL to start recording on:").pack()