simplex 1.2.35__tar.gz → 1.2.36__tar.gz

{simplex-1.2.35 → simplex-1.2.36}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: simplex
-Version: 1.2.35
+Version: 1.2.36
 Summary: Official Python SDK for Simplex API
 Home-page: https://simplex.sh
 Author: Simplex Labs, Inc.
@@ -17,6 +17,16 @@ Requires-Dist: colorama
 Requires-Dist: requests
 Requires-Dist: python-dotenv
 Requires-Dist: click
+Requires-Dist: playwright
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
 
 # Simplex AI Python SDK
 

{simplex-1.2.35 → simplex-1.2.36}/setup.py

@@ -1,8 +1,23 @@
 from setuptools import setup, find_packages
+from setuptools.command.develop import develop
+from setuptools.command.install import install
+from subprocess import check_call
+
+class PostDevelopCommand(develop):
+    """Post-installation for development mode."""
+    def run(self):
+        develop.run(self)
+        check_call(["playwright", "install"])
+
+class PostInstallCommand(install):
+    """Post-installation for installation mode."""
+    def run(self):
+        install.run(self)
+        check_call(["playwright", "install"])
 
 setup(
     name="simplex",
-    version="1.2.35",
+    version="1.2.36",
     packages=find_packages(),
     package_data={
         "simplex": ["browser_agent/dom/*.js"],  # Include JS files in the dom directory
@@ -12,7 +27,12 @@ setup(
         "requests",
         "python-dotenv",
         "click",
+        "playwright",
     ],
+    cmdclass={
+        'develop': PostDevelopCommand,
+        'install': PostInstallCommand,
+    },
     entry_points={
         'console_scripts': [
             'simplex=simplex.cli:main',

{simplex-1.2.35 → simplex-1.2.36}/simplex/cli.py

@@ -63,14 +63,13 @@ def login(website, proxies):
             website = 'https://' + website
             
         animated_print(f"[SIMPLEX] Creating login session for {website}")
-
-        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: Session data could not be saved{Style.RESET_ALL}")
+        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)
+        else:
+            print(f"{Fore.YELLOW}[SIMPLEX] Warning: No login URL returned from the server{Style.RESET_ALL}")
     except Exception as e:
         raise click.ClickException(str(e))
 

{simplex-1.2.35 → simplex-1.2.36}/simplex/simplex.py

@@ -1,60 +1,9 @@
 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"
 
-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']}")
-        
-            
+import json
 
 class Simplex:
     def __init__(self, api_key: str):
@@ -62,8 +11,6 @@ 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
@@ -82,18 +29,18 @@ class Simplex:
         else:
             raise ValueError(f"Failed to close session: {response.json()['error']}")
 
-    def create_session(self, show_in_console: Optional[bool] = True, proxies: Optional[bool] = True):
+    def create_session(self, show_in_console: Optional[bool] = True, proxies: Optional[bool] = True, session_data: Optional[str] = None):
         response = requests.post(
             f"{BASE_URL}/create_session",
             headers={
                 'x-api-key': self.api_key
             },
-            data={'proxies': proxies}
+            data={'proxies': proxies, 'session_data': session_data}
         )
         # Check for non-200 status code
         if response.status_code != 200:
             raise ValueError(f"Create session request failed with status code {response.status_code}: {response.text}")
-            
+
         response_json = response.json()
         if 'session_id' not in response_json:
             raise ValueError(f"It looks like the session wasn't created successfully. Did you set your api_key when creating the Simplex class?")
@@ -382,81 +329,20 @@ class Simplex:
         else:
             raise ValueError(f"Failed to wait: {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 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 restore_login_session(self, session_data: str, cdp_url: str = None):
         """

{simplex-1.2.35 → simplex-1.2.36}/simplex.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: simplex
-Version: 1.2.35
+Version: 1.2.36
 Summary: Official Python SDK for Simplex API
 Home-page: https://simplex.sh
 Author: Simplex Labs, Inc.
@@ -17,6 +17,16 @@ Requires-Dist: colorama
 Requires-Dist: requests
 Requires-Dist: python-dotenv
 Requires-Dist: click
+Requires-Dist: playwright
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
 
 # Simplex AI Python SDK
 

{simplex-1.2.35 → simplex-1.2.36}/simplex.egg-info/requires.txt

@@ -2,3 +2,4 @@ colorama
 requests
 python-dotenv
 click
+playwright

simplex-1.2.36/tests/test.py

@@ -0,0 +1,75 @@
+from simplex import Simplex
+import os
+from dotenv import load_dotenv
+import time
+from playwright.async_api import Page
+
+load_dotenv()
+
+def login():
+    simplex = Simplex(api_key=os.getenv("SIMPLEX_API_KEY"))
+    simplex.create_session(proxies=True)
+    simplex.goto("https://bill.com")
+    
+    simplex.wait(100000)
+
+import asyncio
+from playwright.async_api import async_playwright
+from hyperbrowser import AsyncHyperbrowser
+from hyperbrowser.models.session import CreateSessionParams, ScreenConfig
+import os
+import dotenv
+
+dotenv.load_dotenv()
+
+async def main():
+    client = AsyncHyperbrowser(api_key=os.getenv("HYPERBROWSER_API_KEY"))
+    # Create a session and connect to it using Pyppeteer
+    session = await client.sessions.create(
+        params=CreateSessionParams(
+            use_stealth=True,
+            use_proxy=True,
+            operating_systems=["macos"],
+            device=["desktop"],
+            locales=["en"],
+        )
+    )
+    print(session.live_url)
+    async with async_playwright() as p:
+        
+        browser = await p.chromium.connect_over_cdp(session.ws_endpoint)
+        
+        # Create context with permissions
+        context = await browser.new_context(
+        )
+        import json
+        page = await context.new_page()
+        
+        session_data = json.load(open("bill_session.json"))
+        await page.goto("https://bill.com")
+        await restore_session_data(page, session_data)
+        await page.wait_for_timeout(10000000)
+    
+async def restore_session_data(page: Page, session_data: dict):
+    # First set cookies
+    if 'cookies' in session_data:
+        await page.context.add_cookies(session_data['cookies'])
+        
+    # Now set localStorage
+    if 'localStorage' in session_data:
+        for key, value in session_data['localStorage'].items():
+            await page.evaluate("""({ key, value }) => {
+                localStorage.setItem(key, value);
+            }""", {"key": key, "value": value})
+    
+    # Set sessionStorage
+    if 'sessionStorage' in session_data:
+        for key, value in session_data['sessionStorage'].items():
+            await page.evaluate("""({ key, value }) => {
+                sessionStorage.setItem(key, value);
+            }""", {"key": key, "value": value})
+    print("restored")
+    
+if __name__ == "__main__":
+    asyncio.run(main())
+    

simplex-1.2.35/tests/test.py

@@ -1,22 +0,0 @@
-from simplex import Simplex
-import os
-from dotenv import load_dotenv
-import time
-
-load_dotenv()
-
-def login():
-    simplex = Simplex(api_key=os.getenv("SIMPLEX_API_KEY"))
-    simplex.create_session(proxies=False)
-    simplex.goto("https://dropbox.com")
-    
-    with open("dropbox_com_session_data.json", "r") as f:
-        session_data = f.read()
-    simplex.restore_login_session(session_data=session_data)
-
-    simplex.wait(1000)
-    print(simplex.extract_text("footer texts"))
-    print(simplex.wait(5000))
-
-if __name__ == "__main__":
-    login()