seleniumbase 4.42.4__py3-none-any.whl → 4.42.5__py3-none-any.whl

seleniumbase/__version__.py

@@ -1,2 +1,2 @@
 # seleniumbase package
-__version__ = "4.42.4"
+__version__ = "4.42.5"

seleniumbase/core/browser_launcher.py

@@ -756,12 +756,16 @@ def uc_open_with_cdp_mode(driver, url=None, **kwargs):
     cdp.set_value = CDPM.set_value
     cdp.submit = CDPM.submit
     cdp.evaluate = CDPM.evaluate
+    cdp.execute_script = CDPM.execute_script
     cdp.js_dumps = CDPM.js_dumps
     cdp.maximize = CDPM.maximize
     cdp.minimize = CDPM.minimize
     cdp.medimize = CDPM.medimize
     cdp.set_window_rect = CDPM.set_window_rect
     cdp.reset_window_size = CDPM.reset_window_size
+    cdp.activate_messenger = CDPM.activate_messenger
+    cdp.set_messenger_theme = CDPM.set_messenger_theme
+    cdp.post_message = CDPM.post_message
     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
@@ -877,6 +881,8 @@ def uc_open_with_cdp_mode(driver, url=None, **kwargs):
     cdp.scroll_to_bottom = CDPM.scroll_to_bottom
     cdp.scroll_up = CDPM.scroll_up
     cdp.scroll_down = CDPM.scroll_down
+    cdp.save_page_source = CDPM.save_page_source
+    cdp.save_as_html = CDPM.save_as_html
     cdp.save_screenshot = CDPM.save_screenshot
     cdp.print_to_pdf = CDPM.print_to_pdf
     cdp.save_as_pdf = CDPM.save_as_pdf

seleniumbase/core/sb_cdp.py

@@ -1024,6 +1024,9 @@ class CDPMethods():
             ).strip()
         return self.loop.run_until_complete(self.page.evaluate(expression))
 
+    def execute_script(self, expression):
+        return self.evaluate(expression)
+
     def js_dumps(self, obj_name):
         """Similar to evaluate(), but for dictionary results."""
         if obj_name.startswith("return "):
@@ -1407,6 +1410,64 @@ class CDPMethods():
         mfa_code = self.get_mfa_code(totp_key)
         self.type(selector, mfa_code + "\n", timeout=timeout)
 
+    def activate_messenger(self):
+        js_utils.activate_messenger(self)
+        self.__add_light_pause()
+
+    def set_messenger_theme(
+        self, theme="default", location="default", max_messages="default"
+    ):
+        """Sets a theme for posting messages.
+        Themes: ["flat", "future", "block", "air", "ice"]
+        Locations: ["top_left", "top_center", "top_right",
+                    "bottom_left", "bottom_center", "bottom_right"]
+        max_messages: The limit of concurrent messages to display."""
+        if not theme:
+            theme = "default"  # "flat"
+        if not location:
+            location = "default"  # "bottom_right"
+        if not max_messages:
+            max_messages = "default"  # "8"
+        else:
+            max_messages = str(max_messages)  # Value must be in string format
+        js_utils.set_messenger_theme(
+            self,
+            theme=theme,
+            location=location,
+            max_messages=max_messages,
+        )
+        self.__add_light_pause()
+
+    def post_message(self, message, duration=None, pause=True, style="info"):
+        """Post a message on the screen with Messenger.
+        Arguments:
+            message: The message to display.
+            duration: The time until the message vanishes. (Default: 2.55s)
+            pause: If True, the program waits until the message completes.
+            style: "info", "success", or "error"."""
+        driver = self.driver
+        if hasattr(driver, "cdp_base"):
+            driver = driver.cdp_base
+        if style not in ["info", "success", "error"]:
+            style = "info"
+        if not duration:
+            duration = settings.DEFAULT_MESSAGE_DURATION
+        if (
+            (
+                driver.config.headless
+                or (hasattr(sb_config, "xvfb") and sb_config.xvfb)
+            )
+            and float(duration) > 0.75
+        ):
+            duration = 0.75
+        try:
+            js_utils.post_message(self, message, duration, style=style)
+        except Exception:
+            print(" * %s message: %s" % (style.upper(), message))
+        if pause:
+            duration = float(duration) + 0.15
+            time.sleep(float(duration))
+
     def set_locale(self, locale):
         """(Settings will take effect on the next page load)"""
         self.loop.run_until_complete(self.page.set_locale(locale))
@@ -1510,7 +1571,10 @@ class CDPMethods():
                 except Exception:
                     if (
                         shared_utils.is_linux()
-                        and (not sb_config.headed or sb_config.xvfb)
+                        and (
+                            not sb_config.headed
+                            or (hasattr(sb_config, "xvfb") and sb_config.xvfb)
+                        )
                         and not driver.config.headless
                         and (
                             not hasattr(sb_config, "_virtual_display")
@@ -2611,6 +2675,40 @@ class CDPMethods():
         self.loop.run_until_complete(self.page.scroll_down(amount))
         self.loop.run_until_complete(self.page.wait())
 
+    def save_page_source(self, name, folder=None):
+        import codecs
+        from seleniumbase.core import log_helper
+        if not name.endswith(".html"):
+            name = name + ".html"
+        if folder:
+            abs_path = os.path.abspath(".")
+            file_path = os.path.join(abs_path, folder)
+            if not os.path.exists(file_path):
+                os.makedirs(file_path)
+            html_file_path = os.path.join(file_path, name)
+        else:
+            html_file_path = name
+        page_source = self.get_page_source()
+        last_page = self.get_current_url()
+        meta_charset = '<meta charset="utf-8">'
+        rendered_source = ""
+        if "://" in last_page:
+            base_href_html = log_helper.get_base_href_html(last_page)
+            if ' charset="' not in page_source:
+                rendered_source = "%s\n%s\n%s" % (
+                    base_href_html, meta_charset, page_source
+                )
+            else:
+                rendered_source = "%s\n%s" % (base_href_html, page_source)
+        else:
+            rendered_source = page_source
+        html_file = codecs.open(html_file_path, "w+", "utf-8")
+        html_file.write(rendered_source)
+        html_file.close()
+
+    def save_as_html(self, *args, **kwargs):
+        self.save_page_source(*args, **kwargs)
+
     def save_screenshot(self, name, folder=None, selector=None):
         filename = name
         if folder:

seleniumbase/fixtures/base_case.py

@@ -9205,6 +9205,10 @@ class BaseCase(unittest.TestCase):
         """Same as self.switch_to_newest_window()"""
         self.switch_to_newest_window()
 
+    def save_as_html(self, name, folder=None):
+        """Same as self.save_page_source()"""
+        self.save_page_source(name, folder=folder)
+
     def input(
         self, selector, text, by="css selector", timeout=None, retry=False
     ):
@@ -9654,9 +9658,11 @@ class BaseCase(unittest.TestCase):
 
     def activate_messenger(self):
         self.__check_scope()
-        self._check_browser()
+        if not self.__is_cdp_swap_needed():
+            self._check_browser()
         js_utils.activate_messenger(self.driver)
-        self.wait_for_ready_state_complete()
+        if not self.__is_cdp_swap_needed():
+            self.wait_for_ready_state_complete()
 
     def set_messenger_theme(
         self, theme="default", location="default", max_messages="default"

seleniumbase/fixtures/js_utils.py

@@ -29,6 +29,8 @@ def wait_for_ready_state_complete(driver, timeout=settings.LARGE_TIMEOUT):
     If the timeout is exceeded, the test will still continue
       because readyState == "interactive" may be good enough.
     (Previously, tests would fail immediately if exceeding the timeout.)"""
+    if hasattr(driver, "_swap_driver"):
+        return
     if hasattr(settings, "SKIP_JS_WAITS") and settings.SKIP_JS_WAITS:
         return
     start_ms = time.time() * 1000.0
@@ -54,8 +56,12 @@ def wait_for_ready_state_complete(driver, timeout=settings.LARGE_TIMEOUT):
 
 
 def execute_async_script(driver, script, timeout=settings.LARGE_TIMEOUT):
-    driver.set_script_timeout(timeout)
-    return driver.execute_async_script(script)
+    if hasattr(driver, "set_script_timeout"):
+        driver.set_script_timeout(timeout)
+    if hasattr(driver, "execute_async_script"):
+        return driver.execute_async_script(script)
+    else:
+        return None
 
 
 def wait_for_angularjs(driver, timeout=settings.LARGE_TIMEOUT, **kwargs):
@@ -937,7 +943,10 @@ def post_message(driver, message, msg_dur=None, style="info"):
         execute_script(driver, messenger_script)
     except TypeError as e:
         if (
-            shared_utils.is_cdp_swap_needed(driver)
+            (
+                shared_utils.is_cdp_swap_needed(driver)
+                or hasattr(driver, "_swap_driver")
+            )
             and "cannot unpack non-iterable" in str(e)
         ):
             pass

seleniumbase/fixtures/page_actions.py

@@ -1528,10 +1528,10 @@ def save_page_source(driver, name, folder=None):
         page_source = driver.cdp.get_page_source()
     else:
         page_source = driver.page_source
-    html_file = codecs.open(html_file_path, "w+", "utf-8")
     rendered_source = log_helper.get_html_source_with_base_href(
         driver, page_source
     )
+    html_file = codecs.open(html_file_path, "w+", "utf-8")
     html_file.write(rendered_source)
     html_file.close()
 

seleniumbase/undetected/cdp_driver/cdp_util.py

@@ -335,6 +335,8 @@ async def start(
             xvfb = True
         else:
             xvfb = False
+    if not hasattr(sb_config, "xvfb"):
+        sb_config.xvfb = xvfb
     if incognito is None:
         if "--incognito" in sys_argv:
             incognito = True

seleniumbase/undetected/cdp_driver/config.py

@@ -202,6 +202,7 @@ class Config:
             "InsecureDownloadWarnings,DownloadBubble,DownloadBubbleV2,"
             "OptimizationTargetPrediction,OptimizationGuideModelDownloading,"
             "SidePanelPinning,UserAgentClientHint,PrivacySandboxSettings4,"
+            "OptimizationHintsFetching,InterestFeedContentSuggestions,"
             "DisableLoadExtensionCommandLineSwitch"
         ]
         if self.proxy:

{seleniumbase-4.42.4.dist-info → seleniumbase-4.42.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: seleniumbase
-Version: 4.42.4
+Version: 4.42.5
 Summary: A complete web automation framework for end-to-end testing.
 Home-page: https://github.com/seleniumbase/SeleniumBase
 Author: Michael Mintz
@@ -75,7 +75,7 @@ Requires-Dist: filelock~=3.16.1; python_version < "3.9"
 Requires-Dist: filelock~=3.19.1; python_version >= "3.9" and python_version < "3.10"
 Requires-Dist: filelock>=3.20.0; python_version >= "3.10"
 Requires-Dist: fasteners>=0.20
-Requires-Dist: mycdp>=1.2.0
+Requires-Dist: mycdp>=1.2.1
 Requires-Dist: pynose>=1.5.5
 Requires-Dist: platformdirs~=4.3.6; python_version < "3.9"
 Requires-Dist: platformdirs~=4.4.0; python_version >= "3.9" and python_version < "3.10"
@@ -127,7 +127,8 @@ Requires-Dist: pytest-html==4.0.2
 Requires-Dist: pytest-metadata==3.1.1
 Requires-Dist: pytest-ordering==0.6
 Requires-Dist: pytest-rerunfailures==14.0; python_version < "3.9"
-Requires-Dist: pytest-rerunfailures==16.0.1; python_version >= "3.9"
+Requires-Dist: pytest-rerunfailures==16.0.1; python_version >= "3.9" and python_version < "3.10"
+Requires-Dist: pytest-rerunfailures==16.1; python_version >= "3.10"
 Requires-Dist: pytest-xdist==3.6.1; python_version < "3.9"
 Requires-Dist: pytest-xdist==3.8.0; python_version >= "3.9"
 Requires-Dist: parameterized==0.9.0

{seleniumbase-4.42.4.dist-info → seleniumbase-4.42.5.dist-info}/RECORD

@@ -3,7 +3,7 @@ sbase/__main__.py,sha256=G0bVB1-DM4PGwQ1KyOupaWCs4ePbChZNNWuX2htim5U,647
 sbase/steps.py,sha256=wiPSWZhFpBlWvkqXRJ18btpBu3nQaw9K5AzIJNAX5RM,43521
 seleniumbase/__init__.py,sha256=JFEY9P5QJqsa1M6ghzLMH2eIPQyh85iglCaQwg8Y8z4,2498
 seleniumbase/__main__.py,sha256=dn1p6dgCchmcH1zzTzzQvFwwdQQqnTGH6ULV9m4hv24,654
-seleniumbase/__version__.py,sha256=XfqlisgEGdUvw8FkT-NC7OyGmKmt_1-6RFkqr_Vt3DI,46
+seleniumbase/__version__.py,sha256=QxAoP8OngQTSxxVddd7kUktH7MikCkbIaCqpWS1-a-Y,46
 seleniumbase/behave/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 seleniumbase/behave/behave_helper.py,sha256=lJtagtivSbEpbRj0EKV4l4PuPU6NANONJJAnYwgVCe0,24379
 seleniumbase/behave/behave_sb.py,sha256=guLihFsr1uJ4v2AR3r3Vy_BTeHrHwy2JEJxhp-MVnZA,59872
@@ -36,7 +36,7 @@ seleniumbase/console_scripts/sb_print.py,sha256=tNy-bMDgwHJO3bZxMpmo9weSE8uhbH0C
 seleniumbase/console_scripts/sb_recorder.py,sha256=DH-n2fN7N9qyHMl7wjtn8MiliBgfw-1kwgmfg1GUuhk,10772
 seleniumbase/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 seleniumbase/core/application_manager.py,sha256=e_0sjtI8cjY5BNyZj1QBR0j6_oCScxGmSXYEpcYwuZE,576
-seleniumbase/core/browser_launcher.py,sha256=i2Y8_cYpR103zWf65cRTCwhnJQQEpbVhYH9UQKXYF90,250688
+seleniumbase/core/browser_launcher.py,sha256=B1kj4DfxB3IKZZs1SmDcCxhK--8W8tznpwcXIHzJoKE,250972
 seleniumbase/core/capabilities_parser.py,sha256=meIS2uHapTCq2ldfNAToC7r0cKmZDRXuYNKExM1GHDY,6038
 seleniumbase/core/colored_traceback.py,sha256=DrRWfg7XEnKcgY59Xj7Jdk09H-XqHYBSUpB-DiZt6iY,2020
 seleniumbase/core/create_db_tables.sql,sha256=VWPtrdiW_HQ6yETHjqTu-VIrTwvd8I8o1NfBeaVSHpU,972
@@ -50,7 +50,7 @@ seleniumbase/core/proxy_helper.py,sha256=pZ1NboNfziHU3vWZLOZLX-qkfM3oKSnpc3omQf9
 seleniumbase/core/recorder_helper.py,sha256=gDION28OK4NAYsE-7ohKZ9Z3sxQQ0FEjf859LDpqsg4,25320
 seleniumbase/core/report_helper.py,sha256=AIl6Qava2yW1uSzbLpJBlPlYDz0KE-rVhogh8hsGWBo,12201
 seleniumbase/core/s3_manager.py,sha256=z_4qx2jI_gtK5r3niGXgEOBpfMUicUCOciowai50MP4,3529
-seleniumbase/core/sb_cdp.py,sha256=E_7sbQjyqDAO38IiGYSSB57xrTqxAi8nnqUEWIAXYt8,104551
+seleniumbase/core/sb_cdp.py,sha256=9xK3NoiX2knPmBJkIU-zvUHJ9EGfnSBSxrmnQM7MUVo,108271
 seleniumbase/core/sb_driver.py,sha256=-IQsskc7HpXQbcBL04IPjmGpyYchyo7v9OPF2WcahDw,14159
 seleniumbase/core/session_helper.py,sha256=s9zD3PVZEWVzG2h81cCUskbNWLfdjC_LwwQjKptHCak,558
 seleniumbase/core/settings_parser.py,sha256=gqVohHVlE_5L5Cqe2L24uYrRzvoK-saX8E_Df7_-_3I,7609
@@ -67,12 +67,12 @@ seleniumbase/extensions/disable_csp.zip,sha256=5RvomXnm2PdivUVcxTV6jfvD8WhTEsQYH
 seleniumbase/extensions/recorder.zip,sha256=JEE_FVEvlS63cFQbVLEroIyPSS91nWCDL0MhjVrmIpk,11813
 seleniumbase/extensions/sbase_ext.zip,sha256=3s1N8zrVaMz8RQEOIoBzC3KDjtmHwVZRvVsX25Odr_s,8175
 seleniumbase/fixtures/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-seleniumbase/fixtures/base_case.py,sha256=wHXUqjQmALFdWTk5Lf7GX0zWLji3FqzjhnJ91RYrOII,741117
+seleniumbase/fixtures/base_case.py,sha256=GjFttS041ML2mUrquiFpF47fuoIv0FeZBIRYh9RUJeU,741358
 seleniumbase/fixtures/constants.py,sha256=WMrItuNyKq3XVJ64NluLIRc4gJCxDw8K8qXED0b9S2w,13752
 seleniumbase/fixtures/css_to_xpath.py,sha256=9ouDB1xl4MJ2os6JOgTIAyHKOQfuxtxvXC3O5hSnEKA,1954
 seleniumbase/fixtures/errors.py,sha256=KyxuEVx_e3MPhVrJfNIa_3ltMpbCFxfy_jxK8RFNTns,555
-seleniumbase/fixtures/js_utils.py,sha256=Ykt019DFP8S27o8Kai_ihT01qQWo3RqHUeljqOEUdFU,52247
-seleniumbase/fixtures/page_actions.py,sha256=bkCo3s-t2hglV_SYTPPt849R9l0h1gKKboTmZar0n9Y,73213
+seleniumbase/fixtures/js_utils.py,sha256=waXwlyMx7-rIFI8mJ_R8cL_mvih_gXlcJ_wT5h38HbI,52517
+seleniumbase/fixtures/page_actions.py,sha256=DvT0CzqFnpFGLfvmDk7zZ2XMcX_RxLYhNX6Nxxj4rRo,73213
 seleniumbase/fixtures/page_utils.py,sha256=H1iV8f9vDyEy87DBntyiBXC_tg8HskcebUOAJVn0hxE,12160
 seleniumbase/fixtures/shared_utils.py,sha256=LFXGGdvXhNZCShsRxyuEOcO2dcEbSkMSW5seQR2oE08,9950
 seleniumbase/fixtures/unittest_helper.py,sha256=sfZ92rZeBAn_sF_yQ3I6_I7h3lyU5-cV_UMegBNoEm8,1294
@@ -118,8 +118,8 @@ seleniumbase/undetected/webelement.py,sha256=OOpUYbEiOG52KsYYyuDW9tYLdA2SMnukvwQ
 seleniumbase/undetected/cdp_driver/__init__.py,sha256=Ga9alwuaZZy4_XOShc0HjgFnNqpPdrcbjAicz5gE7a4,215
 seleniumbase/undetected/cdp_driver/_contradict.py,sha256=lP4b0h5quAy573ETn_TBbYV889cL1AuPLVInpJ0ZkiU,3183
 seleniumbase/undetected/cdp_driver/browser.py,sha256=JQlwMuwZgK0vWlvH4SU6DA7LcZo2I4hJRUIcv4gbZCQ,35609
-seleniumbase/undetected/cdp_driver/cdp_util.py,sha256=83NOk962PCiERr6YQ_QjTS5yS8Uwcnsyaw9BdgnGoAQ,33481
-seleniumbase/undetected/cdp_driver/config.py,sha256=pjcGq0azNQpNb3XT2sTjxShLEmcIPdzTzB_mitu6GkM,13557
+seleniumbase/undetected/cdp_driver/cdp_util.py,sha256=fs8c3kz9zvESd2xM29ovw0tsffMUFB5VZTrphWHT3g4,33552
+seleniumbase/undetected/cdp_driver/config.py,sha256=B5Wf0E5xvCmIuLO_Y06oyKYd04yM2auj--JyGKUKsls,13630
 seleniumbase/undetected/cdp_driver/connection.py,sha256=WgZ4QamXSdTzP4Xfgkn8mxv-JFivMG6hqbyHy2_LQlg,25717
 seleniumbase/undetected/cdp_driver/element.py,sha256=FIC6v7OmumLCT-_vIc3H4oju_oBbaLpWJUJIKm2c_q4,40467
 seleniumbase/undetected/cdp_driver/tab.py,sha256=t7Ucn0pmm7imwdCM-5KmIJNU2MCeMuIl6G3T2VMrbxU,53170
@@ -137,9 +137,9 @@ seleniumbase/utilities/selenium_grid/start-grid-hub.bat,sha256=Ftq-GrAKRYH2ssDPr
 seleniumbase/utilities/selenium_grid/start-grid-hub.sh,sha256=KADv0RUHONLL2_I443QFK8PryBpDmKn5Gy0s4o0vDSM,106
 seleniumbase/utilities/selenium_ide/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 seleniumbase/utilities/selenium_ide/convert_ide.py,sha256=pZFnqEJQEKZPyNFjkLD29s2HPQgCrWW9XJWpCPhWOoM,31691
-seleniumbase-4.42.4.dist-info/licenses/LICENSE,sha256=BRblZsX7HyPUjQmYTiyWr_e9tzWvmR3R4SFclM2R3W0,1085
-seleniumbase-4.42.4.dist-info/METADATA,sha256=rZJaeZmWtedSbZT_0SihwgvcSlfa5NEuVfmF3uIB40U,89479
-seleniumbase-4.42.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-seleniumbase-4.42.4.dist-info/entry_points.txt,sha256=CNrh2EKNaHYEhO6pP1RJyVLB99LkDDYX7TnUK8xfjqk,623
-seleniumbase-4.42.4.dist-info/top_level.txt,sha256=4N97aBOQ8ETCnDnokBsWb07lJfTaq3C1ZzYRxvLMxqU,19
-seleniumbase-4.42.4.dist-info/RECORD,,
+seleniumbase-4.42.5.dist-info/licenses/LICENSE,sha256=BRblZsX7HyPUjQmYTiyWr_e9tzWvmR3R4SFclM2R3W0,1085
+seleniumbase-4.42.5.dist-info/METADATA,sha256=tMx-4iFxgYOZpmEj-DZwbYvzZi2_tN8kXK0Plmc9Pus,89575
+seleniumbase-4.42.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+seleniumbase-4.42.5.dist-info/entry_points.txt,sha256=CNrh2EKNaHYEhO6pP1RJyVLB99LkDDYX7TnUK8xfjqk,623
+seleniumbase-4.42.5.dist-info/top_level.txt,sha256=4N97aBOQ8ETCnDnokBsWb07lJfTaq3C1ZzYRxvLMxqU,19
+seleniumbase-4.42.5.dist-info/RECORD,,