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

sbase/steps.py

@@ -1191,6 +1191,15 @@ def set_attributes(context, selector, attribute, value):
     sb.set_attributes(selector, attribute, value)
 
 
+@step("Save as PDF to logs")
+@step("Save as PDF to the logs")
+@step("User saves page as PDF to logs")
+@step("User saves page as PDF to the logs")
+def save_as_pdf_to_logs(context):
+    sb = context.sb
+    sb.save_as_pdf_to_logs()
+
+
 @step("Save page source to logs")
 @step("Save the page source to the logs")
 @step("User saves page source to logs")

seleniumbase/__version__.py

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

seleniumbase/behave/behave_helper.py

@@ -522,6 +522,8 @@ def generate_gherkin(srt_actions):
                 )
         elif action[0] == "ss_tl":
             sb_actions.append("Save screenshot to logs")
+        elif action[0] == "pdftl":
+            sb_actions.append("Save as PDF to logs")
         elif action[0] == "spstl":
             sb_actions.append("Save page source to logs")
         elif action[0] == "sh_fc":

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
@@ -1205,7 +1219,6 @@ def dashboard_pre_processing():
 
 
 def _create_dashboard_assets_():
-    import codecs
     from seleniumbase.js_code.live_js import live_js
     from seleniumbase.core.style_sheet import get_pytest_style
 
@@ -1217,24 +1230,24 @@ def _create_dashboard_assets_():
     add_pytest_style_css = True
     if os.path.exists(pytest_style_css):
         existing_pytest_style = None
-        with open(pytest_style_css, "r") as f:
+        with open(pytest_style_css, mode="r") as f:
             existing_pytest_style = f.read()
         if existing_pytest_style == get_pytest_style():
             add_pytest_style_css = False
     if add_pytest_style_css:
-        out_file = codecs.open(pytest_style_css, "w+", encoding="utf-8")
+        out_file = open(pytest_style_css, mode="w+", encoding="utf-8")
         out_file.writelines(get_pytest_style())
         out_file.close()
     live_js_file = os.path.join(assets_folder, "live.js")
     add_live_js_file = True
     if os.path.exists(live_js_file):
         existing_live_js = None
-        with open(live_js_file, "r") as f:
+        with open(live_js_file, mode="r") as f:
             existing_live_js = f.read()
         if existing_live_js == live_js:
             add_live_js_file = False
     if add_live_js_file:
-        out_file = codecs.open(live_js_file, "w+", encoding="utf-8")
+        out_file = open(live_js_file, mode="w+", encoding="utf-8")
         out_file.writelines(live_js)
         out_file.close()
 
@@ -1307,7 +1320,7 @@ def _perform_behave_unconfigure_():
         # Part 1: Finalizing the dashboard / integrating html report
         if os.path.exists(dashboard_path):
             the_html_d = None
-            with open(dashboard_path, "r", encoding="utf-8") as f:
+            with open(dashboard_path, mode="r", encoding="utf-8") as f:
                 the_html_d = f.read()
             if sb_config._multithreaded and "-c" in sys.argv:
                 # Threads have "-c" in sys.argv, except for the last
@@ -1318,7 +1331,7 @@ def _perform_behave_unconfigure_():
                 if os.path.exists(pie_path):
                     import json
 
-                    with open(pie_path, "r") as f:
+                    with open(pie_path, mode="r") as f:
                         dash_pie = f.read().strip()
                     sb_config._saved_dashboard_pie = json.loads(dash_pie)
             # If the test run doesn't complete by itself, stop refresh
@@ -1327,7 +1340,7 @@ def _perform_behave_unconfigure_():
             the_html_d = the_html_d.replace(find_it_3, swap_with_3)
             the_html_d = the_html_d.replace(find_it_4, swap_with_4)
             the_html_d += stamp
-            with open(dashboard_path, "w", encoding="utf-8") as f:
+            with open(dashboard_path, mode="w", encoding="utf-8") as f:
                 f.write(the_html_d)  # Finalize the dashboard
     except KeyboardInterrupt:
         pass

seleniumbase/common/decorators.py

@@ -53,7 +53,9 @@ def print_runtime(description=None, limit=None):
         name = description
         info = c1 + "<info>" + cr
         # Print times with a statistically significant number of decimal places
-        if run_time < 0.0001:
+        if run_time < 0.00001:
+            print("%s - {%s} ran for %.8f seconds." % (info, name, run_time))
+        elif run_time < 0.0001:
             print("%s - {%s} ran for %.7f seconds." % (info, name, run_time))
         elif run_time < 0.001:
             print("%s - {%s} ran for %.6f seconds." % (info, name, run_time))

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_caseplans.py

@@ -15,7 +15,6 @@ Examples:
 Output:
       Launches the SeleniumBase Case Plans Generator.
 """
-import codecs
 import colorama
 import os
 import subprocess
@@ -135,7 +134,7 @@ def generate_case_plan_boilerplates(
         file_name = case_id
         file_path = os.path.join(full_folder_path, file_name)
         if not os.path.exists(file_path):
-            out_file = codecs.open(file_path, "w+", "utf-8")
+            out_file = open(file_path, mode="w+", encoding="utf-8")
             out_file.writelines("\r\n".join(data))
             out_file.close()
             new_plans += 1
@@ -183,7 +182,7 @@ def view_summary_of_existing_case_plans(root, tests):
         else:
             case_path = os.path.join(folder_path, "case_plans", case_id)
         if os.path.exists(case_path):
-            f = open(case_path, "r")
+            f = open(case_path, mode="r")
             case_data = f.read()
             f.close()
             case_data_storage.append(case_data)
@@ -316,7 +315,7 @@ def view_summary_of_existing_case_plans(root, tests):
     full_plan = plan_head
 
     file_path = "case_summary.md"
-    file = codecs.open(file_path, "w+", "utf-8")
+    file = open(file_path, mode="w+", encoding="utf-8")
     file.writelines("\r\n".join(full_plan))
     file.close()
 

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_mkchart.py

@@ -22,7 +22,6 @@ Output:
     and use a "sky" theme with "slide" transition.
     The chart can be used as a basic boilerplate.
 """
-import codecs
 import colorama
 import os
 import sys
@@ -254,7 +253,7 @@ def main():
                 continue
             new_data.append(line)
     data = new_data
-    file = codecs.open(file_path, "w+", "utf-8")
+    file = open(file_path, mode="w+", encoding="utf-8")
     file.writelines("\r\n".join(data))
     file.close()
     if " " not in file_name: