smooth-py 0.3.0.dev20251014__tar.gz → 0.3.1.dev20251027__tar.gz

{smooth_py-0.3.0.dev20251014 → smooth_py-0.3.1.dev20251027}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: smooth-py
-Version: 0.3.0.dev20251014
+Version: 0.3.1.dev20251027
 Summary: 
 Author: Luca Pinchetti
 Author-email: luca@circlemind.co

{smooth_py-0.3.0.dev20251014 → smooth_py-0.3.1.dev20251027}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "smooth-py"
-version = "0.3.0.dev20251014"
+version = "0.3.1.dev20251027"
 description = ""
 authors = [
     {name = "Luca Pinchetti",email = "luca@circlemind.co"}

{smooth_py-0.3.0.dev20251014 → smooth_py-0.3.1.dev20251027}/src/smooth/__init__.py

@@ -1,6 +1,7 @@
 """Smooth python SDK."""
 
 import asyncio
+import base64
 import io
 import logging
 import os
@@ -8,7 +9,7 @@ import time
 import urllib.parse
 import warnings
 from pathlib import Path
-from typing import Any, Literal, Type
+from typing import Any, Literal, NotRequired, Type, TypedDict
 
 import httpx
 import requests
@@ -33,7 +34,47 @@ def _encode_url(url: str, interactive: bool = True, embed: bool = False) -> str:
 
 
 # --- Models ---
-# These models define the data structures for API requests and responses.
+
+class Certificate(TypedDict):
+  """Client certificate for accessing secure websites.
+
+  Attributes:
+      file: p12 file object to be uploaded (e.g., open("cert.p12", "rb")).
+      password: Password to decrypt the certificate file. Optional.
+  """
+
+  file: str | io.IOBase  # Required - base64 string or binary IO
+  password: NotRequired[str]  # Optional
+  filters: NotRequired[list[str]]  # Optional - TODO: Reserved for future use to specify URL patterns where the certificate should be applied.
+
+
+def _process_certificates(certificates: list[Certificate] | None) -> list[dict[str, Any]] | None:
+  """Process certificates, converting binary IO to base64-encoded strings.
+
+  Args:
+      certificates: List of certificates with file field as string or binary IO.
+
+  Returns:
+      List of certificates with file field as base64-encoded string, or None if input is None.
+  """
+  if certificates is None:
+    return None
+
+  processed_certs = []
+  for cert in certificates:
+    processed_cert = dict(cert)  # Create a copy
+
+    file_content = processed_cert["file"]
+    if isinstance(file_content, io.IOBase):
+      # Read the binary content and encode to base64
+      binary_data = file_content.read()
+      processed_cert["file"] = base64.b64encode(binary_data).decode("utf-8")
+    elif not isinstance(file_content, str):
+      raise TypeError(f"Certificate file must be a string or binary IO, got {type(file_content)}")
+
+    processed_certs.append(processed_cert)
+
+  return processed_certs
 
 
 class TaskResponse(BaseModel):
@@ -48,6 +89,7 @@ class TaskResponse(BaseModel):
   device: Literal["desktop", "mobile"] | None = Field(default=None, description="The device type used for the task.")
   live_url: str | None = Field(default=None, description="The URL to view and interact with the task execution.")
   recording_url: str | None = Field(default=None, description="The URL to view the task recording.")
+  created_at: int | None = Field(default=None, description="The timestamp when the task was created.")
 
 
 class TaskRequest(BaseModel):
@@ -82,7 +124,11 @@ class TaskRequest(BaseModel):
     description=("Browser profile ID to use. Each profile maintains its own state, such as cookies and login credentials."),
   )
   profile_read_only: bool = Field(
-    default=False, description="If true, the profile specified by `profile_id` will be loaded in read-only mode."
+    default=False,
+    description=(
+      "If true, the profile specified by `profile_id` will be loaded in read-only mode. "
+      "Changes made during the task will not be saved back to the profile."
+    ),
   )
   stealth_mode: bool = Field(default=False, description="Run the browser in stealth mode.")
   proxy_server: str | None = Field(
@@ -93,6 +139,15 @@ class TaskRequest(BaseModel):
   )
   proxy_username: str | None = Field(default=None, description="Proxy server username.")
   proxy_password: str | None = Field(default=None, description="Proxy server password.")
+  certificates: list[dict[str, Any]] | None = Field(
+    default=None,
+    description=(
+      "List of client certificates to use when accessing secure websites. "
+      "Each certificate is a dictionary with the following fields:\n"
+      " - `file`: p12 file object to be uploaded (e.g., open('cert.p12', 'rb')).\n"
+      " - `password` (optional): Password to decrypt the certificate file."
+    ),
+  )
   experimental_features: dict[str, Any] | None = Field(
     default=None, description="Experimental features to enable for the task."
   )
@@ -330,6 +385,10 @@ class BrowserSessionHandle(BaseModel):
       return _encode_url(self.browser_session.live_url, interactive=interactive, embed=embed)
     return None
 
+  def live_id(self):
+    """Returns the live ID for the browser session."""
+    return self.browser_session.live_id
+
 
 class TaskHandle:
   """A handle to a running task."""
@@ -393,6 +452,14 @@ class TaskHandle:
       task_response = self._client._get_task(self.id())
       self._task_response = task_response
       if task_response.recording_url is not None:
+        if not task_response.recording_url:
+          raise ApiError(
+            status_code=404,
+            detail=(
+              f"Recording URL not available for task {self.id()}."
+              " Set `enable_recording=True` when creating the task to enable it."
+            )
+          )
         return task_response.recording_url
       time.sleep(1)
     raise TimeoutError(f"Recording URL not available for task {self.id()}.")
@@ -466,7 +533,7 @@ class SmoothClient(BaseClient):
     max_steps: int = 32,
     device: Literal["desktop", "mobile"] = "mobile",
     allowed_urls: list[str] | None = None,
-    enable_recording: bool = False,
+    enable_recording: bool = True,
     session_id: str | None = None,
     profile_id: str | None = None,
     profile_read_only: bool = False,
@@ -474,6 +541,7 @@ class SmoothClient(BaseClient):
     proxy_server: str | None = None,
     proxy_username: str | None = None,
     proxy_password: str | None = None,
+    certificates: list[Certificate] | None = None,
     experimental_features: dict[str, Any] | None = None,
   ) -> TaskHandle:
     """Runs a task and returns a handle to the task.
@@ -500,6 +568,10 @@ class SmoothClient(BaseClient):
         proxy_server: Proxy server url to route browser traffic through.
         proxy_username: Proxy server username.
         proxy_password: Proxy server password.
+        certificates: List of client certificates to use when accessing secure websites.
+          Each certificate is a dictionary with the following fields:
+          - `file` (required): p12 file object to be uploaded (e.g., open("cert.p12", "rb")).
+          - `password` (optional): Password to decrypt the certificate file, if password-protected.
         experimental_features: Experimental features to enable for the task.
 
     Returns:
@@ -525,6 +597,7 @@ class SmoothClient(BaseClient):
       proxy_server=proxy_server,
       proxy_username=proxy_username,
       proxy_password=proxy_password,
+      certificates=_process_certificates(certificates),
       experimental_features=experimental_features,
     )
     initial_response = self._submit_task(payload)
@@ -577,7 +650,7 @@ class SmoothClient(BaseClient):
         ApiException: If the API request fails.
     """
     try:
-      response = self._session.get(f"{self.base_url}/browser/session")
+      response = self._session.get(f"{self.base_url}/browser/profile")
       data = self._handle_response(response)
       return BrowserProfilesResponse(**data["r"])
     except requests.exceptions.RequestException as e:
@@ -592,7 +665,7 @@ class SmoothClient(BaseClient):
   def delete_profile(self, profile_id: str):
     """Delete a browser profile."""
     try:
-      response = self._session.delete(f"{self.base_url}/browser/session/{profile_id}")
+      response = self._session.delete(f"{self.base_url}/browser/profile/{profile_id}")
       self._handle_response(response)
     except requests.exceptions.RequestException as e:
       logger.error(f"Request failed: {e}")
@@ -711,6 +784,14 @@ class AsyncTaskHandle:
       task_response = await self._client._get_task(self.id())
       self._task_response = task_response
       if task_response.recording_url is not None:
+        if not task_response.recording_url:
+          raise ApiError(
+            status_code=404,
+            detail=(
+              f"Recording URL not available for task {self.id()}."
+              " Set `enable_recording=True` when creating the task to enable it."
+            )
+          )
         return task_response.recording_url
       await asyncio.sleep(1)
 
@@ -779,7 +860,7 @@ class SmoothAsyncClient(BaseClient):
     max_steps: int = 32,
     device: Literal["desktop", "mobile"] = "mobile",
     allowed_urls: list[str] | None = None,
-    enable_recording: bool = False,
+    enable_recording: bool = True,
     session_id: str | None = None,
     profile_id: str | None = None,
     profile_read_only: bool = False,
@@ -787,6 +868,7 @@ class SmoothAsyncClient(BaseClient):
     proxy_server: str | None = None,
     proxy_username: str | None = None,
     proxy_password: str | None = None,
+    certificates: list[Certificate] | None = None,
     experimental_features: dict[str, Any] | None = None,
   ) -> AsyncTaskHandle:
     """Runs a task and returns a handle to the task asynchronously.
@@ -813,6 +895,10 @@ class SmoothAsyncClient(BaseClient):
         proxy_server: Proxy server url to route browser traffic through.
         proxy_username: Proxy server username.
         proxy_password: Proxy server password.
+        certificates: List of client certificates to use when accessing secure websites.
+          Each certificate is a dictionary with the following fields:
+          - `file` (required): p12 file object to be uploaded (e.g., open("cert.p12", "rb")).
+          - `password` (optional): Password to decrypt the certificate file.
         experimental_features: Experimental features to enable for the task.
 
     Returns:
@@ -838,6 +924,7 @@ class SmoothAsyncClient(BaseClient):
       proxy_server=proxy_server,
       proxy_username=proxy_username,
       proxy_password=proxy_password,
+      certificates=_process_certificates(certificates),
       experimental_features=experimental_features,
     )
 
@@ -890,7 +977,7 @@ class SmoothAsyncClient(BaseClient):
         ApiException: If the API request fails.
     """
     try:
-      response = await self._client.get(f"{self.base_url}/browser/session")
+      response = await self._client.get(f"{self.base_url}/browser/profile")
       data = self._handle_response(response)
       return BrowserProfilesResponse(**data["r"])
     except httpx.RequestError as e:
@@ -905,7 +992,7 @@ class SmoothAsyncClient(BaseClient):
   async def delete_profile(self, profile_id: str):
     """Delete a browser profile."""
     try:
-      response = await self._client.delete(f"{self.base_url}/browser/session/{profile_id}")
+      response = await self._client.delete(f"{self.base_url}/browser/profile/{profile_id}")
       self._handle_response(response)
     except httpx.RequestError as e:
       logger.error(f"Request failed: {e}")
@@ -976,6 +1063,7 @@ __all__ = [
   "BrowserSessionResponse",
   "BrowserSessionsResponse",
   "UploadFileResponse",
+  "Certificate",
   "ApiError",
   "TimeoutError",
 ]