simplex 1.2.36__py3-none-any.whl → 1.2.37__py3-none-any.whl

simplex/cli.py

@@ -63,13 +63,14 @@ def login(website, proxies):
             website = 'https://' + website
             
         animated_print(f"[SIMPLEX] Creating login session for {website}")
-        login_session_url = simplex.create_login_session(website, proxies=proxies)
-        
-        if login_session_url:
-            animated_print("[SIMPLEX] Opening login page in your browser")
-            webbrowser.open_new_tab(login_session_url)
+
+        fileName = simplex.create_login_session(website, proxies=proxies)
+
+        if fileName:
+            animated_print(f"[SIMPLEX] Login session created and saved to {fileName}")
         else:
-            print(f"{Fore.YELLOW}[SIMPLEX] Warning: No login URL returned from the server{Style.RESET_ALL}")
+            print(f"{Fore.YELLOW}[SIMPLEX] Warning: Session data could not be saved{Style.RESET_ALL}")
+        
     except Exception as e:
         raise click.ClickException(str(e))
 

simplex/simplex.py

@@ -1,9 +1,60 @@
 import requests 
 import atexit
 from typing import Optional
+from playwright.sync_api import sync_playwright
+import os
+import json
+
 BASE_URL = "https://api.simplex.sh"
 
-import json
+class Playwright:
+    def __init__(self, simplex_instance):
+        self.simplex = simplex_instance
+        
+    def click(self, locator: str, locator_type: str, exact: Optional[bool] = False, 
+              element_index: Optional[str] = None, nth_index: Optional[int] = None, 
+              locator_options: Optional[dict] = None):
+        
+        if element_index and element_index not in ["first", "last", "nth"]:
+            raise ValueError("element_index must be 'first', 'last', or 'nth'")
+        
+        if element_index=="nth" and not nth_index:
+            raise ValueError("nth_index is required when element_index is 'nth'")
+        
+        data = {
+            'session_id': self.simplex.session_id,
+            'locator': locator,
+            'locator_type': locator_type,
+        }     
+        
+        if element_index:
+            data['element_index'] = element_index
+            data['nth_index'] = nth_index
+
+        if exact:
+            data['exact'] = exact
+            
+        if locator_options:
+            data['locator_options'] = json.dumps(locator_options)
+
+        response = requests.post(
+            f"{BASE_URL}/playwright/click",
+            headers={
+                'x-api-key': self.simplex.api_key
+            },
+            data=data
+        )
+
+        if 'succeeded' not in response.json():
+            print(response.json())
+            raise ValueError(f"It looks like the click action with playwright failed to return a response. Did you set your api_key when creating the Simplex class?")
+        
+        if response.json()["succeeded"]:
+            return
+        else:
+            raise ValueError(f"Failed to click element: {response.json()['error']}")
+        
+            
 
 class Simplex:
     def __init__(self, api_key: str):
@@ -11,6 +62,8 @@ class Simplex:
         self.session_id = None
         atexit.register(self.close_session)
 
+        self.playwright = Playwright(self)
+
     def close_session(self):
         if not self.session_id:
             return
@@ -50,7 +103,7 @@ class Simplex:
         if show_in_console:
             print(f"Livestream URL: {livestream_url}")
 
-        return livestream_url
+        return self.session_id, livestream_url
 
     def goto(self, url: str, cdp_url: str = None):
         if not cdp_url and not self.session_id:
@@ -329,20 +382,81 @@ class Simplex:
         else:
             raise ValueError(f"Failed to wait: {response.json()['error']}")
     
-    def create_login_session(self, url: str, proxies: Optional[bool] = True):
-        response = requests.post(
-            f"{BASE_URL}/create_login_session",
-            headers={
-                'x-api-key': self.api_key
-            },
-            data={'url': url, 'proxies': proxies}
-        )
-        if 'succeeded' not in response.json():
-            raise ValueError(f"It looks like the create_login_session action failed to return a response. Did you set your api_key when creating the Simplex class?")
-        if response.json()["succeeded"]:
-            return response.json()["login_session_url"]
-        else:
-            raise ValueError(f"Failed to create login session: {response.json()['error']}")
+    def create_login_session(self, url: str, save_directory: Optional[str] = None, proxies: Optional[bool] = True):
+        def get_website_name(url: str) -> str:
+            """Extract website name from URL"""
+            from urllib.parse import urlparse
+            netloc = urlparse(url).netloc
+            # Remove www. if present
+            if netloc.startswith('www.'):
+                netloc = netloc[4:]
+            return netloc.replace(".", "_")
+
+        with sync_playwright() as p:
+            browser = p.chromium.launch_persistent_context(
+                user_data_dir="",  
+                channel="chrome",  
+                headless=False,
+                args=[
+                    '--disable-blink-features=AutomationControlled',
+                    '--disable-automation',
+                ],
+                ignore_default_args=['--enable-automation']
+            )
+            
+            # Close the default about:blank page
+            if len(browser.pages) > 0:
+                browser.pages[0].close()
+            
+            # Create main page for login
+            main_page = browser.new_page()
+            main_page.goto(url)
+            
+            # Create control page
+            control_page = browser.new_page()
+            control_page.set_content("""
+                <html>
+                <body style="display: flex; justify-content: center; align-items: center; height: 100vh; margin: 0; background-color: #f5f5f5;">
+                    <button id="capture-btn" style="
+                        padding: 20px 40px;
+                        font-size: 18px;
+                        cursor: pointer;
+                        background-color: #2196F3;
+                        color: white;
+                        border: none;
+                        border-radius: 5px;
+                        box-shadow: 0 2px 4px rgba(0,0,0,0.2);
+                    ">Click here when logged in to capture session</button>
+                </body>
+                </html>
+            """)
+            
+            # Wait for button click using Playwright's wait_for_event
+            control_page.wait_for_selector('#capture-btn')
+            control_page.evaluate("""
+                () => {
+                    return new Promise((resolve) => {
+                        document.getElementById('capture-btn').onclick = () => {
+                            resolve(true);
+                        }
+                    });
+                }
+            """)
+            
+            storage = browser.storage_state()
+            if save_directory:
+                filename = os.path.join(save_directory, get_website_name(url) + "_session_data.json")
+            else:
+                filename = get_website_name(url) + "_session_data.json"
+            
+            with open(filename, 'w') as f:
+                json.dump(storage, f, indent=2)
+                
+            print(f"Session data saved to '{filename}'")
+            
+            browser.close()
+
+        return filename
     
     def restore_login_session(self, session_data: str, cdp_url: str = None):
         """

{simplex-1.2.36.dist-info → simplex-1.2.37.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: simplex
-Version: 1.2.36
+Version: 1.2.37
 Summary: Official Python SDK for Simplex API
 Home-page: https://simplex.sh
 Author: Simplex Labs, Inc.

simplex-1.2.37.dist-info/RECORD

@@ -0,0 +1,11 @@
+simplex/__init__.py,sha256=L_5i__xt_ZDkr6e-Wx9cr84t9sXpioOT7j01NJYJCTE,75
+simplex/cli.py,sha256=DxZMiX-mqhy1cWJhf75YQs49GfZediueFuxyEhBFpLk,2324
+simplex/simplex.py,sha256=8JcmId0SQ32GamEAQTWQlafQpa8oswmXM6E3Plfn-9k,21853
+simplex/deploy/__init__.py,sha256=_JQ81F_Nu7hSAfMA691gzs6a4-8oZ-buJ9h3Au12BKw,96
+simplex/deploy/push.py,sha256=hRAbtFZaECKnBljaOLQ5nzJ6hk7tZgc1c7QdgxKQFoY,6123
+simplex-1.2.37.dist-info/LICENSE,sha256=Xh0SJjYZfNI71pCNMB40aKlBLLuOB0blx5xkTtufFNQ,1075
+simplex-1.2.37.dist-info/METADATA,sha256=DTpFDlzDtuQsYAjR-2I4CmGUlObOq3MD814DTqCAQSs,964
+simplex-1.2.37.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+simplex-1.2.37.dist-info/entry_points.txt,sha256=3veL2w3c5vxb3dm8I_M8Fs-370n1ZnvD8uu1nSsL7z8,45
+simplex-1.2.37.dist-info/top_level.txt,sha256=cbMH1bYpN0A3gP-ecibPRHasHoqB-01T_2BUFS8p0CE,8
+simplex-1.2.37.dist-info/RECORD,,

simplex-1.2.36.dist-info/RECORD

@@ -1,11 +0,0 @@
-simplex/__init__.py,sha256=L_5i__xt_ZDkr6e-Wx9cr84t9sXpioOT7j01NJYJCTE,75
-simplex/cli.py,sha256=FlZPWx014wpIy3TWYxhv8pANoPG9UWoq-ipkEIaiJAU,2389
-simplex/simplex.py,sha256=dVFAW1Q83gSTpn20G6DOsKyvnILQsOIeJqOUYySgwjU,17707
-simplex/deploy/__init__.py,sha256=_JQ81F_Nu7hSAfMA691gzs6a4-8oZ-buJ9h3Au12BKw,96
-simplex/deploy/push.py,sha256=hRAbtFZaECKnBljaOLQ5nzJ6hk7tZgc1c7QdgxKQFoY,6123
-simplex-1.2.36.dist-info/LICENSE,sha256=Xh0SJjYZfNI71pCNMB40aKlBLLuOB0blx5xkTtufFNQ,1075
-simplex-1.2.36.dist-info/METADATA,sha256=th6kl09ViBxgKGYrIqgQej82XwrVnieyCgl5BJXtMiw,964
-simplex-1.2.36.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-simplex-1.2.36.dist-info/entry_points.txt,sha256=3veL2w3c5vxb3dm8I_M8Fs-370n1ZnvD8uu1nSsL7z8,45
-simplex-1.2.36.dist-info/top_level.txt,sha256=cbMH1bYpN0A3gP-ecibPRHasHoqB-01T_2BUFS8p0CE,8
-simplex-1.2.36.dist-info/RECORD,,