Update CDP Mode examples

Author: mdmintz

examples/cdp_mode/raw_async.py

@@ -51,13 +51,14 @@ async def main():
     location = "Amsterdam"
     where_to = 'div[data-automation*="experiences"] input'
     button = 'button[data-automation*="experiences-search"]'
+    sb.wait_for_text("Where to?")
     sb.gui_click_element(where_to)
     sb.press_keys(where_to, location)
     sb.sleep(1)
     sb.gui_click_element(button)
     sb.sleep(3)
     print(sb.get_title())
     print("************")
-    cards = sb.select_all('h2[data-automation*="product-list-card"]')
+    cards = sb.select_all('span[data-automation*="product-list-card"]')
     for card in cards:
         print("* %s" % card.text)

examples/cdp_mode/raw_cdp.py

@@ -16,6 +16,7 @@ def main():
     location = "Amsterdam"
     where_to = 'div[data-automation*="experiences"] input'
     button = 'button[data-automation*="experiences-search"]'
+    sb.wait_for_text("Where to?")
     sb.gui_click_element(where_to)
     sb.press_keys(where_to, location)
     sb.sleep(1)
@@ -26,7 +27,7 @@ def main():
     for i in range(8):
         sb.scroll_down(50)
         sb.sleep(0.2)
-    cards = sb.select_all('h2[data-automation*="product-list-card"]')
+    cards = sb.select_all('span[data-automation*="product-list-card"]')
     for card in cards:
         print("* %s" % card.text)
 

examples/cdp_mode/raw_cdp_with_sb.py

@@ -14,6 +14,7 @@
     location = "Amsterdam"
     where_to = 'div[data-automation*="experiences"] input'
     button = 'button[data-automation*="experiences-search"]'
+    sb.wait_for_text("Where to?")
     sb.cdp.gui_click_element(where_to)
     sb.press_keys(where_to, location)
     sb.sleep(1)
@@ -24,6 +25,6 @@
     for i in range(8):
         sb.cdp.scroll_down(50)
         sb.sleep(0.2)
-    cards = sb.select_all('h2[data-automation*="product-list-card"]')
+    cards = sb.select_all('span[data-automation*="product-list-card"]')
     for card in cards:
         print("* %s" % card.text)

examples/cdp_mode/raw_cookies_async.py

@@ -0,0 +1,38 @@
+"""A script that loads cookies to bypass login."""
+import asyncio
+import time
+from seleniumbase import cdp_driver
+
+
+# Log in to Swag Labs and save cookies
+async def get_login_cookies():
+    url = "https://www.saucedemo.com"
+    driver = await cdp_driver.start_async(incognito=True)
+    page = await driver.get(url)
+    element = await page.select("#user-name")
+    await element.send_keys_async("standard_user")
+    element = await page.select("#password")
+    await element.send_keys_async("secret_sauce")
+    element = await page.select('input[type="submit"]')
+    await element.click_async()
+    cookies = await driver.cookies.get_all()
+    await page.close()
+    return cookies
+
+
+# Load previously saved cookies to bypass login
+async def login_with_cookies(cookies):
+    url_1 = "https://www.saucedemo.com"
+    url_2 = "https://www.saucedemo.com/inventory.html"
+    driver = await cdp_driver.start_async()
+    page = await driver.get(url_1)
+    await driver.cookies.set_all(cookies)
+    await driver.get(url_2)
+    await page.select("div.inventory_list")
+    time.sleep(2)
+
+
+if __name__ == "__main__":
+    loop = asyncio.new_event_loop()
+    cookies = loop.run_until_complete(get_login_cookies())
+    loop.run_until_complete(login_with_cookies(cookies))

examples/cdp_mode/raw_priceline.py

@@ -1,7 +1,6 @@
 from seleniumbase import SB
 
 with SB(uc=True, test=True, locale="en", ad_block=True) as sb:
-    window_handle = sb.driver.current_window_handle
     url = "https://www.priceline.com"
     sb.activate_cdp_mode(url)
     sb.sleep(2.5)
@@ -17,17 +16,14 @@
     sb.sleep(1.5)
     sb.cdp.click('button[aria-label="Dismiss calendar"]')
     sb.sleep(4.5)
-    sb.connect()
-    if len(sb.driver.window_handles) > 1:
-        sb.switch_to_window(window_handle)
-        sb.driver.close()
-        sb.sleep(0.2)
-        sb.switch_to_newest_window()
+    if len(sb.cdp.get_tabs()) > 1:
+        sb.cdp.close_active_tab()
+        sb.cdp.switch_to_newest_tab()
         sb.sleep(0.6)
     sb.sleep(0.8)
     for y in range(1, 9):
         sb.scroll_to_y(y * 400)
-        sb.sleep(0.75)
+        sb.sleep(0.5)
     hotel_names = sb.find_elements('a[data-autobot-element-id*="HOTEL_NAME"]')
     hotel_prices = sb.find_elements('span[font-size="4,,,5"]')
     print("Priceline Hotels in %s:" % location)

examples/presenter/uc_presentation_4.py

@@ -770,7 +770,6 @@ def test_presentation_4(self):
         self.begin_presentation(filename="uc_presentation.html")
 
         with SB(uc=True, test=True, locale="en", ad_block=True) as sb:
-            window_handle = sb.driver.current_window_handle
             url = "https://www.priceline.com"
             sb.activate_cdp_mode(url)
             sb.sleep(2.5)
@@ -786,22 +785,19 @@ def test_presentation_4(self):
             sb.sleep(1.5)
             sb.cdp.click('button[aria-label="Dismiss calendar"]')
             sb.sleep(4.5)
-            sb.connect()
-            if len(sb.driver.window_handles) > 1:
-                sb.switch_to_window(window_handle)
-                sb.driver.close()
-                sb.sleep(0.2)
-                sb.switch_to_newest_window()
+            if len(sb.cdp.get_tabs()) > 1:
+                sb.cdp.close_active_tab()
+                sb.cdp.switch_to_newest_tab()
                 sb.sleep(0.6)
             sb.sleep(0.8)
             for y in range(1, 9):
                 sb.scroll_to_y(y * 400)
-                sb.sleep(0.75)
+                sb.sleep(0.5)
             hotel_names = sb.find_elements(
                 'a[data-autobot-element-id*="HOTEL_NAME"]'
             )
             hotel_prices = sb.find_elements('span[font-size="4,,,5"]')
-            print("\n\nPriceline Hotels in %s:" % location)
+            print("Priceline Hotels in %s:" % location)
             print(sb.get_text('[data-testid="POPOVER-DATE-PICKER"]'))
             if len(hotel_names) == 0:
                 print("No availability over the selected dates!")

help_docs/syntax_formats.md

@@ -1072,7 +1072,7 @@ def main():
     for i in range(8):
         sb.scroll_down(50)
         sb.sleep(0.2)
-    cards = sb.select_all('h2[data-automation*="product-list-card"]')
+    cards = sb.select_all('span[data-automation*="product-list-card"]')
     for card in cards:
         print("* %s" % card.text)