Add more methods to CDP Mode

Author: mdmintz

examples/cdp_mode/ReadMe.md

@@ -460,10 +460,14 @@ sb.cdp.get_element_attributes(selector)
 sb.cdp.get_element_attribute(selector, attribute)
 sb.cdp.get_attribute(selector, attribute)
 sb.cdp.get_element_html(selector)
+sb.cdp.get_mfa_code(totp_key=None)
+sb.cdp.enter_mfa_code(selector, totp_key=None, timeout=None)
 sb.cdp.set_locale(locale)
 sb.cdp.set_local_storage_item(key, value)
 sb.cdp.set_session_storage_item(key, value)
 sb.cdp.set_attributes(selector, attribute, value)
+sb.cdp.is_attribute_present(selector, attribute, value=None)
+sb.cdp.is_online()
 sb.cdp.gui_press_key(key)
 sb.cdp.gui_press_keys(keys)
 sb.cdp.gui_write(text)

seleniumbase/core/browser_launcher.py

@@ -766,6 +766,8 @@ def uc_open_with_cdp_mode(driver, url=None, **kwargs):
     cdp.set_local_storage_item = CDPM.set_local_storage_item
     cdp.set_session_storage_item = CDPM.set_session_storage_item
     cdp.set_attributes = CDPM.set_attributes
+    cdp.is_attribute_present = CDPM.is_attribute_present
+    cdp.is_online = CDPM.is_online
     cdp.gui_press_key = CDPM.gui_press_key
     cdp.gui_press_keys = CDPM.gui_press_keys
     cdp.gui_write = CDPM.gui_write
@@ -815,10 +817,12 @@ def uc_open_with_cdp_mode(driver, url=None, **kwargs):
     cdp.get_screen_rect = CDPM.get_screen_rect
     cdp.get_window_rect = CDPM.get_window_rect
     cdp.get_window_size = CDPM.get_window_size
+    cdp.get_mfa_code = CDPM.get_mfa_code
     cdp.nested_click = CDPM.nested_click
     cdp.select_option_by_text = CDPM.select_option_by_text
     cdp.select_option_by_index = CDPM.select_option_by_index
     cdp.select_option_by_value = CDPM.select_option_by_value
+    cdp.enter_mfa_code = CDPM.enter_mfa_code
     cdp.flash = CDPM.flash
     cdp.highlight = CDPM.highlight
     cdp.focus = CDPM.focus

seleniumbase/core/sb_cdp.py

@@ -1393,6 +1393,17 @@ def get_element_html(self, selector):
             )
         )
 
+    def get_mfa_code(self, totp_key=None):
+        """Returns a time-based one-time password based on the Google
+        Authenticator algorithm for multi-factor authentication."""
+        return shared_utils.get_mfa_code(totp_key)
+
+    def enter_mfa_code(self, selector, totp_key=None, timeout=None):
+        if not timeout:
+            timeout = settings.SMALL_TIMEOUT
+        mfa_code = self.get_mfa_code(totp_key)
+        self.type(selector, mfa_code + "\n", timeout=timeout)
+
     def set_locale(self, locale):
         """(Settings will take effect on the next page load)"""
         self.loop.run_until_complete(self.page.set_locale(locale))
@@ -1430,6 +1441,26 @@ def set_attributes(self, selector, attribute, value):
         with suppress(Exception):
             self.loop.run_until_complete(self.page.evaluate(js_code))
 
+    def is_attribute_present(self, selector, attribute, value=None):
+        try:
+            element = self.find_element(selector, timeout=0.1)
+            found_value = element.get_attribute(attribute)
+            if found_value is None:
+                return False
+            if value is not None:
+                if found_value == value:
+                    return True
+                else:
+                    return False
+            else:
+                return True
+        except Exception:
+            return False
+
+    def is_online(self):
+        js_code = "navigator.onLine;"
+        return self.loop.run_until_complete(self.page.evaluate(js_code))
+
     def __make_sure_pyautogui_lock_is_writable(self):
         with suppress(Exception):
             shared_utils.make_writable(constants.MultiBrowser.PYAUTOGUILOCK)