simplex 1.2.31__py3-none-any.whl → 1.2.33__py3-none-any.whl

simplex/simplex.py

@@ -22,7 +22,8 @@ class Simplex:
             data={'session_id': self.session_id}
         )
         self.session_id = None
-
+        if 'succeeded' not in response.json():
+            raise ValueError(f"It looks like the close_session action failed to return a response. Did you set your api_key when creating the Simplex class?")
         if response.json()["succeeded"]:
             return
         else:
@@ -36,6 +37,10 @@ class Simplex:
             },
             data={'proxies': proxies}
         )
+        # 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?")
@@ -68,6 +73,9 @@ class Simplex:
             },
             data=data
         )
+        if 'succeeded' not in response.json():
+            raise ValueError(f"It looks like the goto action failed to return a response. Did you set your api_key when creating the Simplex class?")
+    
         if response.json()["succeeded"]:
             return
         else:
@@ -91,6 +99,8 @@ class Simplex:
             },
             data=data
         )
+        if 'succeeded' not in response.json():
+            raise ValueError(f"It looks like the click action failed to return a response. Did you set your api_key when creating the Simplex class?")
         if response.json()["succeeded"]:
             return response.json()["element_clicked"]
         else:
@@ -137,6 +147,8 @@ class Simplex:
             },
             data=data
         )
+        if 'succeeded' not in response.json():
+            raise ValueError(f"It looks like the press_enter action failed to return a response. Did you set your api_key when creating the Simplex class?")
         if response.json()["succeeded"]:
             return
         else:
@@ -160,6 +172,8 @@ class Simplex:
             },
             data=data
         )
+        if 'succeeded' not in response.json():
+            raise ValueError(f"It looks like the press_tab action failed to return a response. Did you set your api_key when creating the Simplex class?")
         if response.json()["succeeded"]:
             return
         else:
@@ -183,6 +197,8 @@ class Simplex:
             },
             data=data
         )
+        if 'succeeded' not in response.json():
+            raise ValueError(f"It looks like the delete_text action failed to return a response. Did you set your api_key when creating the Simplex class?")
         if response.json()["succeeded"]:
             return
         else:
@@ -206,6 +222,8 @@ class Simplex:
             },
             params=data
         )
+        if 'succeeded' not in response.json():
+            raise ValueError(f"It looks like the extract_bbox action failed to return a response. Did you set your api_key when creating the Simplex class?")
         if response.json()["succeeded"]:
             return response.json()["bbox"]
         else:
@@ -229,6 +247,8 @@ class Simplex:
             },
             params=data
         )
+        if 'succeeded' not in response.json():
+            raise ValueError(f"It looks like the extract_text action failed to return a response. Did you set your api_key when creating the Simplex class?")
         if response.json()["succeeded"]:
             return response.json()["text"]
         else:
@@ -252,6 +272,8 @@ class Simplex:
             },
             params=data
         )
+        if 'succeeded' not in response.json():
+            raise ValueError(f"It looks like the extract_image action failed to return a response. Did you set your api_key when creating the Simplex class?")
         if response.json()["succeeded"]:
             return response.json()["image"]
         else:
@@ -275,6 +297,8 @@ class Simplex:
             },
             data=data
         )
+        if 'succeeded' not in response.json():
+            raise ValueError(f"It looks like the scroll action failed to return a response. Did you set your api_key when creating the Simplex class?")
         if response.json()["succeeded"]:
             return
         else:
@@ -298,6 +322,8 @@ class Simplex:
             },
             data=data
         )
+        if 'succeeded' not in response.json():
+            raise ValueError(f"It looks like the wait action failed to return a response. Did you set your api_key when creating the Simplex class?")
         if response.json()["succeeded"]:
             return
         else:
@@ -311,20 +337,34 @@ class Simplex:
             },
             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_filepath: str, cdp_url: str = None):
-        filename = session_data_filepath
+    def restore_login_session(self, session_data: str, cdp_url: str = None):
+        """
+        Restore a login session from either a file path or a JSON string.
         
-        with open(filename, 'r') as f:
-            session_data = json.load(f)
+        Args:
+            session_data: Either a file path to JSON file or a JSON string
+            cdp_url: Optional CDP URL for remote debugging
+        """
+        try:
+            # Try to parse as JSON string first
+            session_data_dict = json.loads(session_data)
+        except json.JSONDecodeError:
+            # If parsing fails, treat as file path
+            try:
+                with open(session_data, 'r') as f:
+                    session_data_dict = json.load(f)
+            except Exception as e:
+                raise ValueError(f"Failed to load session data. Input must be valid JSON string or path to JSON file. Error: {str(e)}")
         
-        # Serialize the data to JSON string
         data = {
-            'session_data': json.dumps(session_data)  # Serialize the session_data
+            'session_data': json.dumps(session_data_dict)
         }
         if cdp_url:
             data['cdp_url'] = cdp_url
@@ -338,6 +378,8 @@ class Simplex:
             },
             data=data
         )
+        if 'succeeded' not in response.json():
+            raise ValueError(f"It looks like the restore_login_session action failed to return a response. Did you set your api_key when creating the Simplex class?")
         if response.json()["succeeded"]:
             return
         else:
@@ -365,6 +407,8 @@ class Simplex:
             files=files,
             data=data
         )
+        if 'succeeded' not in response.json():
+            raise ValueError(f"It looks like the click_and_upload action failed to return a response. Did you set your api_key when creating the Simplex class?")
         if response.json()["succeeded"]:
             return
         else:
@@ -387,6 +431,8 @@ class Simplex:
             },
             data=data
         )
+        if 'succeeded' not in response.json():
+            raise ValueError(f"It looks like the click_and_download action failed to return a response. Did you set your api_key when creating the Simplex class?")
         if response.json()["succeeded"]:
             return response.json()["b64"], response.json()["filename"]
         else:
@@ -410,6 +456,8 @@ class Simplex:
             },
             data=data
         )
+        if 'succeeded' not in response.json():
+            raise ValueError(f"It looks like the exists action failed to return a response. Did you set your api_key when creating the Simplex class?")
         response_json = response.json()
         if response_json['succeeded']:
             return response_json['exists'], response_json['reasoning']

{simplex-1.2.31.dist-info → simplex-1.2.33.dist-info}/METADATA

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

simplex-1.2.33.dist-info/RECORD

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

simplex-1.2.31.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=rFISsAihefu1WbktzAPAy2GDTIKpGkLZPV7wriej3w4,13621
-simplex/deploy/__init__.py,sha256=_JQ81F_Nu7hSAfMA691gzs6a4-8oZ-buJ9h3Au12BKw,96
-simplex/deploy/push.py,sha256=hRAbtFZaECKnBljaOLQ5nzJ6hk7tZgc1c7QdgxKQFoY,6123
-simplex-1.2.31.dist-info/LICENSE,sha256=Xh0SJjYZfNI71pCNMB40aKlBLLuOB0blx5xkTtufFNQ,1075
-simplex-1.2.31.dist-info/METADATA,sha256=J-7fusHtfk2fHX25p7n8w5ol2edKIxoJkQOz7IqCDho,938
-simplex-1.2.31.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-simplex-1.2.31.dist-info/entry_points.txt,sha256=3veL2w3c5vxb3dm8I_M8Fs-370n1ZnvD8uu1nSsL7z8,45
-simplex-1.2.31.dist-info/top_level.txt,sha256=cbMH1bYpN0A3gP-ecibPRHasHoqB-01T_2BUFS8p0CE,8
-simplex-1.2.31.dist-info/RECORD,,