seleniumbase 4.49.8__tar.gz → 4.49.10__tar.gz

{seleniumbase-4.49.8/seleniumbase.egg-info → seleniumbase-4.49.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: seleniumbase
-Version: 4.49.8
+Version: 4.49.10
 Summary: SeleniumBase is a framework for web crawling, scraping, and testing. Supports pytest. CDP Mode adds stealth. Includes many tools.
 Home-page: https://github.com/seleniumbase/SeleniumBase
 Author: Michael Mintz
@@ -71,7 +71,7 @@ Requires-Dist: exceptiongroup>=1.3.1
 Requires-Dist: websockets~=15.0.1; python_version < "3.10"
 Requires-Dist: websockets>=16.0; python_version >= "3.10"
 Requires-Dist: filelock~=3.19.1; python_version < "3.10"
-Requires-Dist: filelock>=3.29.1; python_version >= "3.10"
+Requires-Dist: filelock>=3.29.3; python_version >= "3.10"
 Requires-Dist: fasteners>=0.20
 Requires-Dist: mycdp>=1.3.7
 Requires-Dist: pynose>=1.5.5
@@ -187,8 +187,8 @@ Dynamic: summary
 <!-- SeleniumBase Docs -->
 
 <meta property="og:site_name" content="SeleniumBase">
-<meta property="og:title" content="SeleniumBase: Stealthy Chromium Automation and E2E Testing with Python." />
-<meta property="og:description" content="Stealthy Chromium Automation and E2E Testing with Python." />
+<meta property="og:title" content="SeleniumBase: Stealthy Chromium Automation with Python; and E2E Testing." />
+<meta property="og:description" content="Stealthy Chromium Automation, including fast, easy, and reliable Web/UI testing with Python." />
 <meta property="og:keywords" content="Python, pytest, selenium, webdriver, testing, automation, seleniumbase, framework, dashboard, recorder, reports, screenshots, playwright, stealth, CAPTCHA">
 <meta property="og:image" content="https://seleniumbase.github.io/cdn/img/mac_sb_logo_5b.png" />
 <link rel="icon" href="https://seleniumbase.github.io/img/logo6.png" />
@@ -306,8 +306,8 @@ sb.quit()
 ```python
 from seleniumbase import sb_cdp
 
-url = "https://news.ycombinator.com/submitted?id=seleniumbase"
-sb = sb_cdp.Chrome(url)
+sb = sb_cdp.Chrome()
+sb.goto("https://news.ycombinator.com/submitted?id=seleniumbase")
 elements = sb.find_elements("span.titleline > a")
 for element in elements:
     print("* " + element.text)
@@ -346,7 +346,7 @@ Google Chrome is the default browser. Only unbranded Chromium and Chrome-for-Tes
 The Chromium browser can also be set via method args, eg: `cft=True`, `use_chromium=True`, `browser="edge"`, `browser="brave"`, etc. Eg:
 
 ```python
-sb = sb_cdp.Chrome(url, use_chromium=True)
+sb = sb_cdp.Chrome(use_chromium=True)
 ```
 
 --------
@@ -404,12 +404,13 @@ with SB(uc=True, test=True, locale="en") as sb:
 ```python
 from seleniumbase import sb_cdp
 
-url = "https://gitlab.com/users/sign_in"
-sb = sb_cdp.Chrome(url, incognito=True)
+sb = sb_cdp.Chrome(incognito=True)
+sb.goto("https://gitlab.com/users/sign_in")
 sb.sleep(2)
 sb.solve_captcha()
 sb.highlight('h1:contains("GitLab")')
 sb.highlight('button:contains("Sign in")')
+sb.quit()
 ```
 
 --------

{seleniumbase-4.49.8 → seleniumbase-4.49.10}/README.md

@@ -1,8 +1,8 @@
 <!-- SeleniumBase Docs -->
 
 <meta property="og:site_name" content="SeleniumBase">
-<meta property="og:title" content="SeleniumBase: Stealthy Chromium Automation and E2E Testing with Python." />
-<meta property="og:description" content="Stealthy Chromium Automation and E2E Testing with Python." />
+<meta property="og:title" content="SeleniumBase: Stealthy Chromium Automation with Python; and E2E Testing." />
+<meta property="og:description" content="Stealthy Chromium Automation, including fast, easy, and reliable Web/UI testing with Python." />
 <meta property="og:keywords" content="Python, pytest, selenium, webdriver, testing, automation, seleniumbase, framework, dashboard, recorder, reports, screenshots, playwright, stealth, CAPTCHA">
 <meta property="og:image" content="https://seleniumbase.github.io/cdn/img/mac_sb_logo_5b.png" />
 <link rel="icon" href="https://seleniumbase.github.io/img/logo6.png" />
@@ -120,8 +120,8 @@ sb.quit()
 ```python
 from seleniumbase import sb_cdp
 
-url = "https://news.ycombinator.com/submitted?id=seleniumbase"
-sb = sb_cdp.Chrome(url)
+sb = sb_cdp.Chrome()
+sb.goto("https://news.ycombinator.com/submitted?id=seleniumbase")
 elements = sb.find_elements("span.titleline > a")
 for element in elements:
     print("* " + element.text)
@@ -160,7 +160,7 @@ Google Chrome is the default browser. Only unbranded Chromium and Chrome-for-Tes
 The Chromium browser can also be set via method args, eg: `cft=True`, `use_chromium=True`, `browser="edge"`, `browser="brave"`, etc. Eg:
 
 ```python
-sb = sb_cdp.Chrome(url, use_chromium=True)
+sb = sb_cdp.Chrome(use_chromium=True)
 ```
 
 --------
@@ -218,12 +218,13 @@ with SB(uc=True, test=True, locale="en") as sb:
 ```python
 from seleniumbase import sb_cdp
 
-url = "https://gitlab.com/users/sign_in"
-sb = sb_cdp.Chrome(url, incognito=True)
+sb = sb_cdp.Chrome(incognito=True)
+sb.goto("https://gitlab.com/users/sign_in")
 sb.sleep(2)
 sb.solve_captcha()
 sb.highlight('h1:contains("GitLab")')
 sb.highlight('button:contains("Sign in")')
+sb.quit()
 ```
 
 --------

{seleniumbase-4.49.8 → seleniumbase-4.49.10}/requirements.txt

@@ -10,7 +10,7 @@ exceptiongroup>=1.3.1
 websockets~=15.0.1;python_version<"3.10"
 websockets>=16.0;python_version>="3.10"
 filelock~=3.19.1;python_version<"3.10"
-filelock>=3.29.1;python_version>="3.10"
+filelock>=3.29.3;python_version>="3.10"
 fasteners>=0.20
 mycdp>=1.3.7
 pynose>=1.5.5

seleniumbase-4.49.10/seleniumbase/__version__.py

@@ -0,0 +1,2 @@
+# seleniumbase package
+__version__ = "4.49.10"

{seleniumbase-4.49.8 → seleniumbase-4.49.10}/seleniumbase/core/browser_launcher.py

@@ -18,11 +18,7 @@ from selenium.common.exceptions import ElementClickInterceptedException
 from selenium.common.exceptions import InvalidSessionIdException
 from selenium.common.exceptions import SessionNotCreatedException
 from selenium.webdriver.chrome.service import Service as ChromeService
-from selenium.webdriver.common.options import ArgOptions
-from selenium.webdriver.common.service import utils as service_utils
-from selenium.webdriver.edge.service import Service as EdgeService
-from selenium.webdriver.firefox.service import Service as FirefoxService
-from selenium.webdriver.safari.service import Service as SafariService
+from selenium.webdriver.common import utils as common_utils
 from seleniumbase import config as sb_config
 from seleniumbase import decorators
 from seleniumbase import drivers  # webdriver storage folder for SeleniumBase
@@ -793,6 +789,7 @@ def uc_open_with_cdp_mode(driver, url=None, **kwargs):
     cdp.grant_permissions = CDPM.grant_permissions
     cdp.grant_all_permissions = CDPM.grant_all_permissions
     cdp.reset_permissions = CDPM.reset_permissions
+    cdp.get_all_urls = CDPM.get_all_urls
     cdp.get_all_cookies = CDPM.get_all_cookies
     cdp.set_all_cookies = CDPM.set_all_cookies
     cdp.save_cookies = CDPM.save_cookies
@@ -831,6 +828,11 @@ def uc_open_with_cdp_mode(driver, url=None, **kwargs):
     cdp.activate_messenger = CDPM.activate_messenger
     cdp.set_messenger_theme = CDPM.set_messenger_theme
     cdp.post_message = CDPM.post_message
+    cdp.download_file = CDPM.download_file
+    cdp.save_file_as = CDPM.save_file_as
+    cdp.assert_downloaded_file = CDPM.assert_downloaded_file
+    cdp.get_path_of_downloaded_file = CDPM.get_path_of_downloaded_file
+    cdp.set_download_path = CDPM.set_download_path
     cdp.set_locale = CDPM.set_locale
     cdp.set_local_storage_item = CDPM.set_local_storage_item
     cdp.set_session_storage_item = CDPM.set_session_storage_item
@@ -2668,7 +2670,7 @@ def _set_chrome_options(
         args = " ".join(sys.argv)
         debug_port = 9222
         if ("-n" in sys.argv or " -n=" in args or args == "-c"):
-            debug_port = service_utils.free_port()
+            debug_port = common_utils.free_port()
         chrome_options.add_argument("--remote-debugging-port=%s" % debug_port)
     if swiftshader:
         chrome_options.add_argument("--use-gl=angle")
@@ -3672,6 +3674,7 @@ def get_remote_driver(
     device_height,
     device_pixel_ratio,
 ):
+    from selenium.webdriver.common.options import ArgOptions
     # Construct the address for connecting to a Selenium Grid
     if servername.startswith("https://"):
         protocol = "https"
@@ -4130,6 +4133,7 @@ def get_local_driver(
     b_path = binary_location
     use_uc = is_using_uc(undetectable, browser_name)
     if browser_name == constants.Browser.FIREFOX:
+        from selenium.webdriver.firefox.service import Service as FFService
         firefox_options = _set_firefox_options(
             downloads_path,
             headless,
@@ -4184,7 +4188,7 @@ def get_local_driver(
                         sys.argv = sys_args  # Put back original sys args
         # Launch Firefox
         if os.path.exists(local_geckodriver):
-            service = FirefoxService(
+            service = FFService(
                 executable_path=local_geckodriver,
                 log_output=os.devnull,
             )
@@ -4229,7 +4233,7 @@ def get_local_driver(
                 else:
                     raise  # Not an obvious fix.
         else:
-            service = FirefoxService(log_output=os.devnull)
+            service = FFService(log_output=os.devnull)
             try:
                 driver = webdriver.Firefox(
                     service=service,
@@ -4337,6 +4341,7 @@ def get_local_driver(
             driver = webdriver.Ie(service=service, options=ie_options)
             return extend_driver(driver)
     elif browser_name == constants.Browser.EDGE:
+        from selenium.webdriver.edge.service import Service as EdgeService
         prefs = {
             "download.default_directory": downloads_path,
             "download.directory_upgrade": True,
@@ -4756,7 +4761,7 @@ def get_local_driver(
             args = " ".join(sys.argv)
             free_port = 9222
             if ("-n" in sys.argv or " -n=" in args or args == "-c"):
-                free_port = service_utils.free_port()
+                free_port = common_utils.free_port()
             edge_options.add_argument("--remote-debugging-port=%s" % free_port)
         if swiftshader:
             edge_options.add_argument("--use-gl=angle")
@@ -4866,7 +4871,7 @@ def get_local_driver(
                 args = " ".join(sys.argv)
                 free_port = 9222
                 if ("-n" in sys.argv or " -n=" in args or args == "-c"):
-                    free_port = service_utils.free_port()
+                    free_port = common_utils.free_port()
                 edge_options.add_argument(
                     "--remote-debugging-port=%s" % free_port
                 )
@@ -4904,6 +4909,7 @@ def get_local_driver(
             raise Exception("Can't run Safari tests in multithreaded mode!")
         warnings.simplefilter("ignore", category=DeprecationWarning)
         from selenium.webdriver.safari.options import Options as SafariOptions
+        from selenium.webdriver.safari.service import Service as SafariService
         service = SafariService(quiet=False)
         options = SafariOptions()
         if (

{seleniumbase-4.49.8 → seleniumbase-4.49.10}/seleniumbase/core/sb_cdp.py

@@ -152,6 +152,17 @@ class CDPMethods():
         except Exception:
             pass
 
+    def activate_cdp_mode(self, url=None, **kwargs):
+        """If calling this method from here, then it was likely done
+        by accident because CDP Mode is already active at this point.
+        If URL, then navigate there with any kwargs that were passed.
+        If no URL (but kwargs are included) then refresh with kwargs.
+        If no URL and no kwargs either, then do nothing."""
+        if url:
+            self.get(url, **kwargs)
+        elif kwargs:
+            self.get(self.get_current_url(), **kwargs)
+
     def open(self, url, **kwargs):
         self.get(url, **kwargs)
 
@@ -752,6 +763,18 @@ class CDPMethods():
             driver = driver.cdp_base
         return self.loop.run_until_complete(driver.reset_permissions())
 
+    def get_all_urls(self, absolute=True):
+        """
+        Convenience function that returns all links (a,link,img,script,meta).
+        :param absolute:
+         Try to build all the links in absolute form
+         instead of "as is", often relative.
+        :return: List of URLs.
+        """
+        return self.loop.run_until_complete(
+            self.page.get_all_urls(absolute=absolute)
+        )
+
     def get_all_cookies(self, *args, **kwargs):
         driver = self.driver
         if hasattr(driver, "cdp_base"):
@@ -1374,10 +1397,15 @@ class CDPMethods():
         return self.find_element(selector).text_all
 
     def get_title(self):
+        """Returns the title of the current web page."""
         return self.loop.run_until_complete(
             self.page.evaluate("document.title")
         )
 
+    def get_page_title(self):
+        """Same as get_title(), which returns the current page title."""
+        return self.get_title()
+
     def get_current_url(self):
         return self.loop.run_until_complete(
             self.page.evaluate("window.location.href")
@@ -1725,6 +1753,80 @@ class CDPMethods():
             duration = float(duration) + 0.15
             time.sleep(float(duration))
 
+    def download_file(self, file_url):
+        """Download a file from a URL.
+        The default download location is: "./downloaded_files/"."""
+        self.loop.run_until_complete(
+            self.page.download_file(file_url)
+        )
+
+    def save_file_as(self, file_url, new_file_name):
+        """Download a file from a URL and rename it.
+        The default download location is: "./downloaded_files/"."""
+        self.loop.run_until_complete(
+            self.page.download_file(file_url, new_file_name)
+        )
+
+    def assert_downloaded_file(self, file, timeout=None):
+        """Asserts that the file exists in SeleniumBase's [Downloads Folder].
+        For browser click-initiated downloads, SeleniumBase will override
+            the system [Downloads Folder] to be "./downloaded_files/".
+        @Params
+        file - The filename of the downloaded file.
+        timeout - The time (seconds) to wait for the download to complete.
+        browser - If True, uses the path set by click-initiated downloads."""
+        downloads_folder = constants.Files.DOWNLOADS_FOLDER
+        abs_path = os.path.abspath(".")
+        downloads_path = os.path.join(abs_path, downloads_folder)
+        if not timeout:
+            timeout = settings.LARGE_TIMEOUT
+        start_ms = time.time() * 1000.0
+        stop_ms = start_ms + (timeout * 1000.0)
+        downloaded_file_path = os.path.join(downloads_path, file)
+        found = False
+        for x in range(int(timeout)):
+            shared_utils.check_if_time_limit_exceeded()
+            try:
+                self.assert_true(
+                    os.path.exists(downloaded_file_path),
+                    "File [%s] was not found in the downloads folder [%s]!"
+                    % (file, downloads_path),
+                )
+                found = True
+                break
+            except Exception:
+                now_ms = time.time() * 1000.0
+                if now_ms >= stop_ms:
+                    break
+                time.sleep(1)
+        if not found and not os.path.exists(downloaded_file_path):
+            plural = "s"
+            if timeout == 1:
+                plural = ""
+            message = (
+                "File {%s} was not found in the downloads folder {%s} "
+                "after %s second%s! (Or the download didn't complete!)"
+                % (file, downloads_path, timeout, plural)
+            )
+            from seleniumbase.common.exceptions import NoSuchFileException
+            raise NoSuchFileException(message)
+
+    def get_path_of_downloaded_file(self, file):
+        """This assumes the default location of SeleniumBase downloads,
+        which is the "./downloaded_files/" folder where scripts run."""
+        downloads_folder = constants.Files.DOWNLOADS_FOLDER
+        abs_path = os.path.abspath(".")
+        downloads_path = os.path.join(abs_path, downloads_folder)
+        return os.path.join(downloads_path, file)
+
+    def set_download_path(self, path):
+        """Set a new download path for click-initiated downloads.
+        (For Pure CDP Mode sync format only! -> sb_cdp.)
+        The default download location is: "./downloaded_files/".
+        Convenience methods such as assert_downloaded_file(file)
+        will still expect the default location."""
+        self.loop.run_until_complete(self.page.set_download_path(path))
+
     def set_locale(self, locale):
         """(Settings will take effect on the next page load)"""
         self.loop.run_until_complete(self.page.set_locale(locale))
@@ -3233,13 +3335,23 @@ class CDPMethods():
         )
         return True
 
-    def assert_true(self, expression):
+    def assert_true(self, expression, msg=None):
         if not expression:
-            raise AssertionError("%s is not true" % expression)
+            if not msg:
+                raise AssertionError("%s is not true" % expression)
+            else:
+                raise AssertionError(
+                    "%s is not true. (%s)" % (expression, msg)
+                )
 
-    def assert_false(self, expression):
+    def assert_false(self, expression, msg=None):
         if expression:
-            raise AssertionError("%s is not false" % expression)
+            if not msg:
+                raise AssertionError("%s is not false" % expression)
+            else:
+                raise AssertionError(
+                    "%s is not false. (%s)" % (expression, msg)
+                )
 
     def assert_equal(self, first, second):
         if first != second:

{seleniumbase-4.49.8 → seleniumbase-4.49.10}/seleniumbase/fixtures/base_case.py

@@ -5101,14 +5101,48 @@ class BaseCase(unittest.TestCase):
             self.close_active_tab = self.cdp.close_active_tab
         if hasattr(self.cdp, "find_element_by_text"):
             self.find_element_by_text = self.cdp.find_element_by_text
+        if hasattr(self.cdp, "find_elements_by_text"):
+            self.find_elements_by_text = self.cdp.find_elements_by_text
         if hasattr(self.cdp, "get_active_tab"):
             self.get_active_tab = self.cdp.get_active_tab
+        if hasattr(self.cdp, "get_all_cookies"):
+            self.get_all_cookies = self.cdp.get_all_cookies
+        if hasattr(self.cdp, "get_all_urls"):
+            self.get_all_urls = self.cdp.get_all_urls
+        if hasattr(self.cdp, "get_document"):
+            self.get_document = self.cdp.get_document
+        if hasattr(self.cdp, "get_element_attribute"):
+            self.get_element_attribute = self.cdp.get_element_attribute
+        if hasattr(self.cdp, "get_element_attributes"):
+            self.get_element_attributes = self.cdp.get_element_attributes
+        if hasattr(self.cdp, "get_element_html"):
+            self.get_element_html = self.cdp.get_element_html
         if hasattr(self.cdp, "get_endpoint_url"):
             self.get_endpoint_url = self.cdp.get_endpoint_url
         if hasattr(self.cdp, "get_event_loop"):
             self.get_event_loop = self.cdp.get_event_loop
+        if hasattr(self.cdp, "get_flattened_document"):
+            self.get_flattened_document = self.cdp.get_flattened_document
+        if hasattr(self.cdp, "get_navigation_history"):
+            self.get_navigation_history = self.cdp.get_navigation_history
+        if hasattr(self.cdp, "get_nested_element"):
+            self.get_nested_element = self.cdp.get_nested_element
+        if hasattr(self.cdp, "get_port"):
+            self.get_port = self.cdp.get_port
+        if hasattr(self.cdp, "get_rd_host"):
+            self.get_rd_host = self.cdp.get_rd_host
+        if hasattr(self.cdp, "get_rd_port"):
+            self.get_rd_port = self.cdp.get_rd_port
+        if hasattr(self.cdp, "get_rd_url"):
+            self.get_rd_url = self.cdp.get_rd_url
         if hasattr(self.cdp, "get_tabs"):
             self.get_tabs = self.cdp.get_tabs
+        if hasattr(self.cdp, "get_websocket_url"):
+            self.get_websocket_url = self.cdp.get_websocket_url
+        if hasattr(self.cdp, "grant_all_permissions"):
+            self.grant_all_permissions = self.cdp.grant_all_permissions
+        if hasattr(self.cdp, "grant_permissions"):
+            self.grant_permissions = self.cdp.grant_permissions
         if hasattr(self.cdp, "gui_click_and_hold"):
             self.gui_click_and_hold = self.cdp.gui_click_and_hold
         if hasattr(self.cdp, "gui_click_element"):
@@ -5119,6 +5153,26 @@ class BaseCase(unittest.TestCase):
             self.gui_drag_drop_points = self.cdp.gui_drag_drop_points
         if hasattr(self.cdp, "highlight_overlay"):
             self.highlight_overlay = self.cdp.highlight_overlay
+        if hasattr(self.cdp, "js_dumps"):
+            self.js_dumps = self.cdp.js_dumps
+        if hasattr(self.cdp, "js_scroll_into_view"):
+            self.js_scroll_into_view = self.cdp.js_scroll_into_view
+        if hasattr(self.cdp, "medimize"):
+            self.medimize = self.cdp.medimize
+        if hasattr(self.cdp, "mouse_click"):
+            self.mouse_click = self.cdp.mouse_click
+        if hasattr(self.cdp, "nested_click"):
+            self.nested_click = self.cdp.nested_click
+        if hasattr(self.cdp, "remove_from_dom"):
+            self.remove_from_dom = self.cdp.remove_from_dom
+        if hasattr(self.cdp, "reset_permissions"):
+            self.reset_permissions = self.cdp.reset_permissions
+        if hasattr(self.cdp, "set_all_cookies"):
+            self.set_all_cookies = self.cdp.set_all_cookies
+        if hasattr(self.cdp, "tile_windows"):
+            self.tile_windows = self.cdp.tile_windows
+        if hasattr(self.cdp, "get_window"):
+            self.get_window = self.cdp.get_window
         if getattr(self.driver, "_is_using_auth", None):
             with suppress(Exception):
                 self.cdp.loop.run_until_complete(self.cdp.page.wait(0.25))

{seleniumbase-4.49.8 → seleniumbase-4.49.10}/seleniumbase/undetected/cdp_driver/browser.py

@@ -380,6 +380,7 @@ class Browser:
                     filter(lambda item: item.type_ == "page", self.targets)
                 )
                 await connection.sleep(0.005)
+        _cdp_downloads_path = None
         _cdp_timezone = None
         _cdp_user_agent = ""
         _cdp_locale = None
@@ -389,6 +390,8 @@ class Browser:
         _cdp_mobile_mode = None
         _cdp_recorder = None
         _cdp_ad_block = None
+        if getattr(sb_config, "_cdp_downloads_path", None):
+            _cdp_downloads_path = sb_config._cdp_downloads_path
         if getattr(sb_config, "_cdp_timezone", None):
             _cdp_timezone = sb_config._cdp_timezone
         if getattr(sb_config, "_cdp_user_agent", None):
@@ -405,6 +408,8 @@ class Browser:
             _cdp_ad_block = sb_config.ad_block_on
         if getattr(sb_config, "disable_csp", None):
             _cdp_disable_csp = sb_config.disable_csp
+        if "downloads_path" in kwargs:
+            _cdp_downloads_path = kwargs["downloads_path"]
         if "timezone" in kwargs:
             _cdp_timezone = kwargs["timezone"]
         elif "tzone" in kwargs:
@@ -439,6 +444,7 @@ class Browser:
         await connection.sleep(0.01)
         await connection.send(cdp.network.enable())
         await connection.sleep(0.01)
+        await connection.set_downloads_folder(_cdp_downloads_path)
         if _cdp_timezone:
             await connection.set_timezone(_cdp_timezone)
         if _cdp_locale:

{seleniumbase-4.49.8 → seleniumbase-4.49.10}/seleniumbase/undetected/cdp_driver/cdp_util.py

@@ -744,6 +744,8 @@ async def start(
         sb_config._cdp_user_agent = kwargs["user_agent"]
     else:
         sb_config._cdp_user_agent = None
+    if "downloads_path" in kwargs:
+        sb_config._cdp_downloads_path = kwargs["downloads_path"]
     if "platform" in kwargs:
         sb_config._cdp_platform = kwargs["platform"]
     elif "plat" in kwargs:

{seleniumbase-4.49.8 → seleniumbase-4.49.10}/seleniumbase/undetected/cdp_driver/connection.py

@@ -5,6 +5,7 @@ import inspect
 import itertools
 import json
 import logging
+import os
 import sys
 import types
 import warnings
@@ -19,6 +20,7 @@ from typing import (
 )
 import websockets
 from websockets.protocol import State
+from seleniumbase.fixtures import constants
 from . import cdp_util as util
 import mycdp as cdp
 import mycdp.network
@@ -335,6 +337,17 @@ class Connection(metaclass=CantTouchThis):
                 # No listener created yet.
                 pass
 
+    async def set_downloads_folder(self, downloads_path):
+        if not downloads_path:
+            downloads_dir = constants.Files.DOWNLOADS_FOLDER
+            abs_path = os.path.abspath(".")
+            downloads_path = os.path.join(abs_path, downloads_dir)
+        await self.send(
+            cdp.browser.set_download_behavior(
+                behavior="allow", download_path=str(downloads_path)
+            )
+        )
+
     async def set_locale(self, locale: Optional[str] = None):
         """Sets the Language Locale code via set_user_agent_override."""
         await self.set_user_agent(user_agent="", accept_language=locale)

{seleniumbase-4.49.8 → seleniumbase-4.49.10}/seleniumbase/undetected/cdp_driver/element.py

@@ -373,7 +373,9 @@ class Element:
             object_id=self._remote_object.object_id
         )]
         script = 'sessionStorage.getItem("pxsid") !== null;'
-        using_px = await self.tab.evaluate(script)
+        using_px = True
+        with suppress(Exception):
+            using_px = await self.tab.evaluate(script)
         if not using_px:
             await self.flash_async(0.25)
         await self._tab.send(
@@ -505,7 +507,9 @@ class Element:
             return
         logger.debug("Clicking on location: %.2f, %.2f" % center)
         script = 'sessionStorage.getItem("pxsid") !== null;'
-        using_px = await self.tab.evaluate(script)
+        using_px = True
+        with suppress(Exception):
+            using_px = await self.tab.evaluate(script)
         if not using_px:
             asyncio.create_task(self.flash_async(0.25))
         asyncio.create_task(
@@ -567,7 +571,9 @@ class Element:
         else:
             logger.debug("Clicking on location: %.2f, %.2f" % (x_pos, y_pos))
         script = 'sessionStorage.getItem("pxsid") !== null;'
-        using_px = await self.tab.evaluate(script)
+        using_px = True
+        with suppress(Exception):
+            using_px = await self.tab.evaluate(script)
         if not using_px:
             asyncio.create_task(
                 self.flash_async(