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

seleniumbase/fixtures/page_utils.py

@@ -16,7 +16,7 @@ def get_domain_url(url):
     Into this:
     https://blog.xkcd.com
     """
-    if not url.startswith("http://") and not url.startswith("https://"):
+    if not url.startswith(("http://", "https://")):
         return url
     url_header = url.split("://")[0]
     simple_url = url.split("://")[1]
@@ -40,45 +40,25 @@ def swap_selector_and_by_if_reversed(selector, by):
 
 def is_xpath_selector(selector):
     """Determine if a selector is an xpath selector."""
-    if (
-        selector.startswith("/")
-        or selector.startswith("./")
-        or selector.startswith("(")
-    ):
-        return True
-    return False
+    return selector.startswith(("/", "./", "("))
 
 
 def is_link_text_selector(selector):
     """Determine if a selector is a link text selector."""
-    if (
-        selector.startswith("link=")
-        or selector.startswith("link_text=")
-        or selector.startswith("text=")
-    ):
-        return True
-    return False
+    return selector.startswith(("link=", "link_text=", "text="))
 
 
 def is_partial_link_text_selector(selector):
     """Determine if a selector is a partial link text selector."""
-    if (
-        selector.startswith("partial_link=")
-        or selector.startswith("partial_link_text=")
-        or selector.startswith("partial_text=")
-        or selector.startswith("p_link=")
-        or selector.startswith("p_link_text=")
-        or selector.startswith("p_text=")
-    ):
-        return True
-    return False
+    return selector.startswith((
+        "partial_link=", "partial_link_text=", "partial_text=",
+        "p_link=", "p_link_text=", "p_text="
+    ))
 
 
 def is_name_selector(selector):
     """Determine if a selector is a name selector."""
-    if selector.startswith("name=") or selector.startswith("&"):
-        return True
-    return False
+    return selector.startswith(("name=", "&"))
 
 
 def recalculate_selector(selector, by, xp_ok=True):
@@ -108,12 +88,18 @@ def recalculate_selector(selector, by, xp_ok=True):
         name = get_name_from_selector(selector)
         selector = '[name="%s"]' % name
         by = By.CSS_SELECTOR
-    if xp_ok:
-        if ":contains(" in selector and by == By.CSS_SELECTOR:
-            selector = css_to_xpath.convert_css_to_xpath(selector)
-            by = By.XPATH
+    if xp_ok and ":contains(" in selector and by == By.CSS_SELECTOR:
+        selector = css_to_xpath.convert_css_to_xpath(selector)
+        by = By.XPATH
     if by == "":
         by = By.CSS_SELECTOR
+    if not is_valid_by(by):
+        valid_by_options = [
+            "css selector", "link text", "partial link text",
+            "name", "xpath", "id", "tag name", "class name",
+        ]
+        msg = "Choose a `by` from: %s." % valid_by_options
+        raise Exception('Invalid `by`: "%s"\n%s' % (by, msg))
     return (selector, by)
 
 
@@ -123,21 +109,10 @@ def looks_like_a_page_url(url):
     possible typos when calling self.get(url), which will try to
     navigate to the page if a URL is detected, but will instead call
     self.get_element(URL_AS_A_SELECTOR) if the input is not a URL."""
-    if (
-        url.startswith("http:")
-        or url.startswith("https:")
-        or url.startswith("://")
-        or url.startswith("about:")
-        or url.startswith("blob:")
-        or url.startswith("chrome:")
-        or url.startswith("data:")
-        or url.startswith("edge:")
-        or url.startswith("file:")
-        or url.startswith("view-source:")
-    ):
-        return True
-    else:
-        return False
+    return url.startswith((
+        "http:", "https:", "://", "about:", "blob:", "chrome:",
+        "data:", "edge:", "file:", "view-source:"
+    ))
 
 
 def get_link_text_from_selector(selector):
@@ -190,12 +165,9 @@ def is_valid_url(url):
     )
     if (
         regex.match(url)
-        or url.startswith("about:")
-        or url.startswith("blob:")
-        or url.startswith("chrome:")
-        or url.startswith("data:")
-        or url.startswith("edge:")
-        or url.startswith("file:")
+        or url.startswith((
+            "about:", "blob:", "chrome:", "data:", "edge:", "file:"
+        ))
     ):
         return True
     else:

seleniumbase/fixtures/shared_utils.py

@@ -1,8 +1,11 @@
 """Shared utility methods"""
+import colorama
 import os
+import pathlib
 import platform
 import sys
 import time
+from contextlib import suppress
 from seleniumbase import config as sb_config
 from seleniumbase.fixtures import constants
 
@@ -67,6 +70,89 @@ def get_terminal_width():
     return width
 
 
+def fix_colorama_if_windows():
+    if is_windows():
+        colorama.just_fix_windows_console()
+
+
+def fix_url_as_needed(url):
+    if not url:
+        url = "data:,"
+    elif url.startswith("//"):
+        url = "https:" + url
+    elif ":" not in url:
+        url = "https://" + url
+    return url
+
+
+def reconnect_if_disconnected(driver):
+    if (
+        hasattr(driver, "_is_using_uc")
+        and driver._is_using_uc
+        and hasattr(driver, "is_connected")
+        and not driver.is_connected()
+    ):
+        with suppress(Exception):
+            driver.connect()
+
+
+def is_cdp_swap_needed(driver):
+    """
+    When someone is using CDP Mode with a disconnected webdriver,
+    but they forget to reconnect before calling a webdriver method,
+    this method is used to substitute the webdriver method for a
+    CDP Mode method instead, which keeps CDP Stealth Mode enabled.
+    For other webdriver methods, SeleniumBase will reconnect first.
+    """
+    return (
+        hasattr(driver, "is_cdp_mode_active")
+        and driver.is_cdp_mode_active()
+        and hasattr(driver, "is_connected")
+        and not driver.is_connected()
+    )
+
+
+def is_chrome_130_or_newer(self, binary_location=None):
+    from seleniumbase.core import detect_b_ver
+
+    """Due to changes in Chrome-130, UC Mode freezes at start-up
+    unless the user-data-dir already exists and is populated."""
+    with suppress(Exception):
+        if not binary_location:
+            ver = detect_b_ver.get_browser_version_from_os("google-chrome")
+        else:
+            ver = detect_b_ver.get_browser_version_from_binary(
+                binary_location
+            )
+        if ver and len(ver) > 3 and int(ver.split(".")[0]) >= 130:
+            return True
+    return False
+
+
+def make_dir_files_writable(dir_path):
+    # Make all files in the given directory writable.
+    for file_path in pathlib.Path(dir_path).glob("*"):
+        if file_path.is_file():
+            mode = os.stat(file_path).st_mode
+            mode |= (mode & 0o444) >> 1  # copy R bits to W
+            with suppress(Exception):
+                os.chmod(file_path, mode)
+
+
+def make_writable(file_path):
+    # Set permissions to: "If you can read it, you can write it."
+    mode = os.stat(file_path).st_mode
+    mode |= (mode & 0o444) >> 1  # copy R bits to W
+    os.chmod(file_path, mode)
+
+
+def make_executable(file_path):
+    # Set permissions to: "If you can read it, you can execute it."
+    mode = os.stat(file_path).st_mode
+    mode |= (mode & 0o444) >> 2  # copy R bits to X
+    os.chmod(file_path, mode)
+
+
 def format_exc(exception, message):
     """Formats an exception message to make the output cleaner."""
     from selenium.common.exceptions import ElementNotVisibleException
@@ -81,46 +167,46 @@ def format_exc(exception, message):
     from seleniumbase.common.exceptions import TextNotVisibleException
     from seleniumbase.common import exceptions
 
-    if exception == Exception:
+    if exception is Exception:
         exc = Exception
         return exc, message
-    elif exception == ElementNotVisibleException:
+    elif exception is ElementNotVisibleException:
         exc = exceptions.ElementNotVisibleException
     elif exception == "ElementNotVisibleException":
         exc = exceptions.ElementNotVisibleException
-    elif exception == LinkTextNotFoundException:
+    elif exception is LinkTextNotFoundException:
         exc = exceptions.LinkTextNotFoundException
     elif exception == "LinkTextNotFoundException":
         exc = exceptions.LinkTextNotFoundException
-    elif exception == NoSuchElementException:
+    elif exception is NoSuchElementException:
         exc = exceptions.NoSuchElementException
     elif exception == "NoSuchElementException":
         exc = exceptions.NoSuchElementException
-    elif exception == TextNotVisibleException:
+    elif exception is TextNotVisibleException:
         exc = exceptions.TextNotVisibleException
     elif exception == "TextNotVisibleException":
         exc = exceptions.TextNotVisibleException
-    elif exception == NoAlertPresentException:
+    elif exception is NoAlertPresentException:
         exc = exceptions.NoAlertPresentException
     elif exception == "NoAlertPresentException":
         exc = exceptions.NoAlertPresentException
-    elif exception == NoSuchAttributeException:
+    elif exception is NoSuchAttributeException:
         exc = exceptions.NoSuchAttributeException
     elif exception == "NoSuchAttributeException":
         exc = exceptions.NoSuchAttributeException
-    elif exception == NoSuchFrameException:
+    elif exception is NoSuchFrameException:
         exc = exceptions.NoSuchFrameException
     elif exception == "NoSuchFrameException":
         exc = exceptions.NoSuchFrameException
-    elif exception == NoSuchWindowException:
+    elif exception is NoSuchWindowException:
         exc = exceptions.NoSuchWindowException
     elif exception == "NoSuchWindowException":
         exc = exceptions.NoSuchWindowException
-    elif exception == NoSuchFileException:
+    elif exception is NoSuchFileException:
         exc = exceptions.NoSuchFileException
     elif exception == "NoSuchFileException":
         exc = exceptions.NoSuchFileException
-    elif exception == NoSuchOptionException:
+    elif exception is NoSuchOptionException:
         exc = exceptions.NoSuchOptionException
     elif exception == "NoSuchOptionException":
         exc = exceptions.NoSuchOptionException

seleniumbase/js_code/active_css_js.py

@@ -103,7 +103,7 @@ function hasDigit(str) {
     return /\d/.test(str);
 };
 function isGen(str) {
-    return /[_-]\d/.test(str);
+    return /[_-]\d/.test(str) || /\d[a-z]/.test(str);
 };
 function tagName(el) {
     return el.tagName.toLowerCase();

seleniumbase/js_code/recorder_js.py

@@ -103,7 +103,7 @@ function hasDigit(str) {
     return /\d/.test(str);
 };
 function isGen(str) {
-    return /[_-]\d/.test(str);
+    return /[_-]\d/.test(str) || /\d[a-z]/.test(str);
 };
 function tagName(el) {
     return el.tagName.toLowerCase();

seleniumbase/plugins/base_plugin.py

@@ -2,6 +2,7 @@
 import ast
 import sys
 import time
+from contextlib import suppress
 from nose.plugins import Plugin
 from seleniumbase import config as sb_config
 from seleniumbase.config import settings
@@ -305,14 +306,12 @@ class Base(Plugin):
         if python3_11_or_newer and py311_patch2:
             # Handle a bug on Python 3.11 where exceptions aren't seen
             sb_config._browser_version = None
-            try:
+            with suppress(Exception):
                 test._BaseCase__set_last_page_screenshot()
                 test._BaseCase__set_last_page_url()
                 test._BaseCase__set_last_page_source()
                 sb_config._browser_version = test._get_browser_version()
                 test._log_fail_data()
-            except Exception:
-                pass
             sb_config._excinfo_tb = err
             log_path = None
             if hasattr(sb_config, "_test_logpath"):