hyperbrowser 0.11.0__tar.gz → 0.12.0__tar.gz

{hyperbrowser-0.11.0 → hyperbrowser-0.12.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hyperbrowser
-Version: 0.11.0
+Version: 0.12.0
 Summary: Python SDK for hyperbrowser
 Home-page: https://github.com/hyperbrowserai/python-sdk
 License: MIT
@@ -19,6 +19,10 @@ Requires-Dist: pydantic (>=2.10.0,<3.0.0)
 Project-URL: Repository, https://github.com/hyperbrowserai/python-sdk
 Description-Content-Type: text/markdown
 
+# Hyperbrowser Python SDK
+
+Checkout the full documentation [here](https://docs.hyperbrowser.ai/)
+
 ## Installation
 
 Currently Hyperbrowser supports creating a browser session in two ways:

{hyperbrowser-0.11.0 → hyperbrowser-0.12.0}/README.md

@@ -1,3 +1,7 @@
+# Hyperbrowser Python SDK
+
+Checkout the full documentation [here](https://docs.hyperbrowser.ai/)
+
 ## Installation
 
 Currently Hyperbrowser supports creating a browser session in two ways:

{hyperbrowser-0.11.0 → hyperbrowser-0.12.0}/hyperbrowser/client/async_client.py

@@ -4,6 +4,7 @@ from .managers.async_manager.profile import ProfileManager
 from .managers.async_manager.session import SessionManager
 from .managers.async_manager.scrape import ScrapeManager
 from .managers.async_manager.crawl import CrawlManager
+from .managers.async_manager.extension import ExtensionManager
 from .base import HyperbrowserBase
 from ..transport.async_transport import AsyncTransport
 from ..config import ClientConfig
@@ -25,6 +26,7 @@ class AsyncHyperbrowser(HyperbrowserBase):
         self.scrape = ScrapeManager(self)
         self.crawl = CrawlManager(self)
         self.profiles = ProfileManager(self)
+        self.extensions = ExtensionManager(self)
 
     async def close(self) -> None:
         await self.transport.close()

hyperbrowser-0.12.0/hyperbrowser/client/managers/async_manager/extension.py

@@ -0,0 +1,40 @@
+import os
+from typing import List
+
+from hyperbrowser.exceptions import HyperbrowserError
+from hyperbrowser.models.extension import CreateExtensionParams, ExtensionResponse
+
+
+class ExtensionManager:
+    def __init__(self, client):
+        self._client = client
+
+    async def create(self, params: CreateExtensionParams) -> ExtensionResponse:
+        file_path = params.file_path
+        params.file_path = None
+
+        # Check if file exists before trying to open it
+        if not os.path.exists(file_path):
+            raise FileNotFoundError(f"Extension file not found at path: {file_path}")
+
+        response = await self._client.transport.post(
+            self._client._build_url("/extensions/add"),
+            data=(
+                {}
+                if params is None
+                else params.model_dump(exclude_none=True, by_alias=True)
+            ),
+            files={"file": open(file_path, "rb")},
+        )
+        return ExtensionResponse(**response.data)
+
+    async def list(self) -> List[ExtensionResponse]:
+        response = await self._client.transport.get(
+            self._client._build_url("/extensions/list"),
+        )
+        if not isinstance(response.data, list):
+            raise HyperbrowserError(
+                f"Expected list response but got {type(response.data)}",
+                original_error=None,
+            )
+        return [ExtensionResponse(**extension) for extension in response.data]

hyperbrowser-0.12.0/hyperbrowser/client/managers/sync_manager/extension.py

@@ -0,0 +1,40 @@
+import os
+from typing import List
+
+from hyperbrowser.exceptions import HyperbrowserError
+from hyperbrowser.models.extension import CreateExtensionParams, ExtensionResponse
+
+
+class ExtensionManager:
+    def __init__(self, client):
+        self._client = client
+
+    def create(self, params: CreateExtensionParams) -> ExtensionResponse:
+        file_path = params.file_path
+        params.file_path = None
+
+        # Check if file exists before trying to open it
+        if not os.path.exists(file_path):
+            raise FileNotFoundError(f"Extension file not found at path: {file_path}")
+
+        response = self._client.transport.post(
+            self._client._build_url("/extensions/add"),
+            data=(
+                {}
+                if params is None
+                else params.model_dump(exclude_none=True, by_alias=True)
+            ),
+            files={"file": open(file_path, "rb")},
+        )
+        return ExtensionResponse(**response.data)
+
+    def list(self) -> List[ExtensionResponse]:
+        response = self._client.transport.get(
+            self._client._build_url("/extensions/list"),
+        )
+        if not isinstance(response.data, list):
+            raise HyperbrowserError(
+                f"Expected list response but got {type(response.data)}",
+                original_error=None,
+            )
+        return [ExtensionResponse(**extension) for extension in response.data]

{hyperbrowser-0.11.0 → hyperbrowser-0.12.0}/hyperbrowser/client/sync.py

@@ -3,6 +3,7 @@ from .managers.sync_manager.profile import ProfileManager
 from .managers.sync_manager.session import SessionManager
 from .managers.sync_manager.scrape import ScrapeManager
 from .managers.sync_manager.crawl import CrawlManager
+from .managers.sync_manager.extension import ExtensionManager
 from .base import HyperbrowserBase
 from ..transport.sync import SyncTransport
 from ..config import ClientConfig
@@ -24,6 +25,7 @@ class Hyperbrowser(HyperbrowserBase):
         self.scrape = ScrapeManager(self)
         self.crawl = CrawlManager(self)
         self.profiles = ProfileManager(self)
+        self.extensions = ExtensionManager(self)
 
     def close(self) -> None:
         self.transport.close()

{hyperbrowser-0.11.0 → hyperbrowser-0.12.0}/hyperbrowser/models/consts.py

@@ -1,6 +1,6 @@
 from typing import Literal
 
-ScrapeFormat = Literal["markdown", "html", "links"]
+ScrapeFormat = Literal["markdown", "html", "links", "screenshot"]
 
 Country = Literal[
     "AD",

hyperbrowser-0.12.0/hyperbrowser/models/extension.py

@@ -0,0 +1,27 @@
+from datetime import datetime
+from typing import List, Literal, Optional
+from pydantic import BaseModel, ConfigDict, Field
+
+
+class CreateExtensionParams(BaseModel):
+    """
+    Parameters for creating a new extension.
+    """
+
+    model_config = ConfigDict(
+        populate_by_alias=True,
+    )
+
+    name: Optional[str] = Field(default=None, serialization_alias="name")
+    file_path: str = Field(serialization_alias="filePath")
+
+
+class ExtensionResponse(BaseModel):
+    model_config = ConfigDict(
+        populate_by_alias=True,
+    )
+
+    id: str = Field(serialization_alias="id")
+    name: str = Field(serialization_alias="name")
+    created_at: datetime = Field(serialization_alias="createdAt", alias="createdAt")
+    updated_at: datetime = Field(serialization_alias="updatedAt", alias="updatedAt")

{hyperbrowser-0.11.0 → hyperbrowser-0.12.0}/hyperbrowser/models/session.py

@@ -145,9 +145,12 @@ class CreateSessionParams(BaseModel):
     trackers: bool = Field(default=False, serialization_alias="trackers")
     annoyances: bool = Field(default=False, serialization_alias="annoyances")
     enable_web_recording: Optional[bool] = Field(
-        default=False, serialization_alias="enableWebRecording"
+        default=True, serialization_alias="enableWebRecording"
     )
     profile: Optional[CreateSessionProfile] = Field(default=None)
+    extension_ids: Optional[List[str]] = Field(
+        default=None, serialization_alias="extensionIds"
+    )
 
 
 class SessionRecording(BaseModel):

{hyperbrowser-0.11.0 → hyperbrowser-0.12.0}/hyperbrowser/transport/async_transport.py

@@ -66,9 +66,14 @@ class AsyncTransport(TransportStrategy):
         except httpx.RequestError as e:
             raise HyperbrowserError("Request failed", original_error=e)
 
-    async def post(self, url: str, data: Optional[dict] = None) -> APIResponse:
+    async def post(
+        self, url: str, data: Optional[dict] = None, files: Optional[dict] = None
+    ) -> APIResponse:
         try:
-            response = await self.client.post(url, json=data)
+            if files:
+                response = await self.client.post(url, data=data, files=files)
+            else:
+                response = await self.client.post(url, json=data)
             return await self._handle_response(response)
         except HyperbrowserError:
             raise

{hyperbrowser-0.11.0 → hyperbrowser-0.12.0}/hyperbrowser/transport/sync.py

@@ -45,9 +45,14 @@ class SyncTransport(TransportStrategy):
     def close(self) -> None:
         self.client.close()
 
-    def post(self, url: str, data: Optional[dict] = None) -> APIResponse:
+    def post(
+        self, url: str, data: Optional[dict] = None, files: Optional[dict] = None
+    ) -> APIResponse:
         try:
-            response = self.client.post(url, json=data)
+            if files:
+                response = self.client.post(url, data=data, files=files)
+            else:
+                response = self.client.post(url, json=data)
             return self._handle_response(response)
         except HyperbrowserError:
             raise

{hyperbrowser-0.11.0 → hyperbrowser-0.12.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "hyperbrowser"
-version = "0.11.0"
+version = "0.12.0"
 description = "Python SDK for hyperbrowser"
 authors = ["Nikhil Shahi <nshahi1998@gmail.com>"]
 license = "MIT"