hyperbrowser 0.28.0__tar.gz → 0.30.0__tar.gz

{hyperbrowser-0.28.0 → hyperbrowser-0.30.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: hyperbrowser
-Version: 0.28.0
+Version: 0.30.0
 Summary: Python SDK for hyperbrowser
 License: MIT
 Author: Nikhil Shahi

{hyperbrowser-0.28.0 → hyperbrowser-0.30.0}/hyperbrowser/client/managers/async_manager/session.py

@@ -2,6 +2,7 @@ from typing import List
 from ....models.session import (
     BasicResponse,
     CreateSessionParams,
+    GetSessionDownloadsUrlResponse,
     GetSessionRecordingUrlResponse,
     SessionDetail,
     SessionListParams,
@@ -47,7 +48,7 @@ class SessionManager:
 
     async def get_recording(self, id: str) -> List[SessionRecording]:
         response = await self._client.transport.get(
-            self._client._build_url(f"/session/{id}/recording")
+            self._client._build_url(f"/session/{id}/recording"), None, True
         )
         return [SessionRecording(**recording) for recording in response.data]
 
@@ -56,3 +57,9 @@ class SessionManager:
             self._client._build_url(f"/session/{id}/recording-url")
         )
         return GetSessionRecordingUrlResponse(**response.data)
+
+    async def get_downloads_url(self, id: str) -> GetSessionDownloadsUrlResponse:
+        response = await self._client.transport.get(
+            self._client._build_url(f"/session/{id}/downloads-url")
+        )
+        return GetSessionDownloadsUrlResponse(**response.data)

{hyperbrowser-0.28.0 → hyperbrowser-0.30.0}/hyperbrowser/client/managers/sync_manager/session.py

@@ -2,6 +2,7 @@ from typing import List
 from ....models.session import (
     BasicResponse,
     CreateSessionParams,
+    GetSessionDownloadsUrlResponse,
     GetSessionRecordingUrlResponse,
     SessionDetail,
     SessionListParams,
@@ -45,7 +46,7 @@ class SessionManager:
 
     def get_recording(self, id: str) -> List[SessionRecording]:
         response = self._client.transport.get(
-            self._client._build_url(f"/session/{id}/recording")
+            self._client._build_url(f"/session/{id}/recording"), None, True
         )
         return [SessionRecording(**recording) for recording in response.data]
 
@@ -54,3 +55,9 @@ class SessionManager:
             self._client._build_url(f"/session/{id}/recording-url")
         )
         return GetSessionRecordingUrlResponse(**response.data)
+
+    def get_downloads_url(self, id: str) -> GetSessionDownloadsUrlResponse:
+        response = self._client.transport.get(
+            self._client._build_url(f"/session/{id}/downloads-url")
+        )
+        return GetSessionDownloadsUrlResponse(**response.data)

{hyperbrowser-0.28.0 → hyperbrowser-0.30.0}/hyperbrowser/models/consts.py

@@ -4,6 +4,12 @@ ScrapeFormat = Literal["markdown", "html", "links", "screenshot"]
 ScrapeWaitUntil = Literal["load", "domcontentloaded", "networkidle"]
 ScrapePageStatus = Literal["completed", "failed", "pending", "running"]
 ScrapeScreenshotFormat = Literal["jpeg", "png", "webp"]
+RecordingStatus = Literal[
+    "not_enabled", "pending", "in_progress", "completed", "failed"
+]
+DownloadsStatus = Literal[
+    "not_enabled", "pending", "in_progress", "completed", "failed"
+]
 
 POLLING_ATTEMPTS = 5
 
@@ -486,3 +492,106 @@ ISO639_1 = Literal[
     "zh",
     "zu",
 ]
+
+State = Literal[
+    "AL",
+    "AK",
+    "AZ",
+    "AR",
+    "CA",
+    "CO",
+    "CT",
+    "DE",
+    "FL",
+    "GA",
+    "HI",
+    "ID",
+    "IL",
+    "IN",
+    "IA",
+    "KS",
+    "KY",
+    "LA",
+    "ME",
+    "MD",
+    "MA",
+    "MI",
+    "MN",
+    "MS",
+    "MO",
+    "MT",
+    "NE",
+    "NV",
+    "NH",
+    "NJ",
+    "NM",
+    "NY",
+    "NC",
+    "ND",
+    "OH",
+    "OK",
+    "OR",
+    "PA",
+    "RI",
+    "SC",
+    "SD",
+    "TN",
+    "TX",
+    "UT",
+    "VT",
+    "VA",
+    "WA",
+    "WV",
+    "WI",
+    "WY",
+    "al",
+    "ak",
+    "az",
+    "ar",
+    "ca",
+    "co",
+    "ct",
+    "de",
+    "fl",
+    "ga",
+    "hi",
+    "id",
+    "il",
+    "in",
+    "ia",
+    "ks",
+    "ky",
+    "la",
+    "me",
+    "md",
+    "ma",
+    "mi",
+    "mn",
+    "ms",
+    "mo",
+    "mt",
+    "ne",
+    "nv",
+    "nh",
+    "nj",
+    "nm",
+    "ny",
+    "nc",
+    "nd",
+    "oh",
+    "ok",
+    "or",
+    "pa",
+    "ri",
+    "sc",
+    "sd",
+    "tn",
+    "tx",
+    "ut",
+    "vt",
+    "va",
+    "wa",
+    "wv",
+    "wi",
+    "wy",
+]

{hyperbrowser-0.28.0 → hyperbrowser-0.30.0}/hyperbrowser/models/extract.py

@@ -24,6 +24,7 @@ class StartExtractJobParams(BaseModel):
     schema_: Optional[Any] = pydantic.Field(
         None, alias="schema", serialization_alias="schema"
     )
+    wait_for: Optional[int] = Field(default=None, serialization_alias="waitFor")
     session_options: Optional[CreateSessionParams] = Field(
         default=None, serialization_alias="sessionOptions"
     )

{hyperbrowser-0.28.0 → hyperbrowser-0.30.0}/hyperbrowser/models/session.py

@@ -3,7 +3,15 @@ from typing import Any, List, Literal, Optional, Union
 
 from pydantic import BaseModel, ConfigDict, Field, field_validator
 
-from hyperbrowser.models.consts import ISO639_1, Country, OperatingSystem, Platform
+from hyperbrowser.models.consts import (
+    ISO639_1,
+    Country,
+    DownloadsStatus,
+    OperatingSystem,
+    Platform,
+    RecordingStatus,
+    State,
+)
 
 SessionStatus = Literal["active", "closed", "error"]
 
@@ -138,8 +146,10 @@ class CreateSessionParams(BaseModel):
         default=None, serialization_alias="proxyServerUsername"
     )
     proxy_country: Optional[Country] = Field(
-        default="US", serialization_alias="proxyCountry"
+        default=None, serialization_alias="proxyCountry"
     )
+    proxy_state: Optional[State] = Field(default=None, serialization_alias="proxyState")
+    proxy_city: Optional[str] = Field(default=None, serialization_alias="proxyCity")
     operating_systems: Optional[List[OperatingSystem]] = Field(
         default=None, serialization_alias="operatingSystems"
     )
@@ -168,6 +178,9 @@ class CreateSessionParams(BaseModel):
     browser_args: Optional[List[str]] = Field(
         default=None, serialization_alias="browserArgs"
     )
+    save_downloads: Optional[bool] = Field(
+        default=None, serialization_alias="saveDownloads"
+    )
 
 
 class SessionRecording(BaseModel):
@@ -194,4 +207,20 @@ class GetSessionRecordingUrlResponse(BaseModel):
         populate_by_alias=True,
     )
 
-    recording_url: str = Field(alias="recordingUrl")
+    status: RecordingStatus = Field(alias="status")
+    recording_url: Optional[str] = Field(default=None, alias="recordingUrl")
+    error: Optional[str] = Field(default=None, alias="error")
+
+
+class GetSessionDownloadsUrlResponse(BaseModel):
+    """
+    Response containing the signed URL for the session downloads.
+    """
+
+    model_config = ConfigDict(
+        populate_by_alias=True,
+    )
+
+    status: DownloadsStatus = Field(alias="status")
+    downloads_url: Optional[str] = Field(default=None, alias="downloadsUrl")
+    error: Optional[str] = Field(default=None, alias="error")

{hyperbrowser-0.28.0 → hyperbrowser-0.30.0}/hyperbrowser/transport/async_transport.py

@@ -80,11 +80,15 @@ class AsyncTransport(TransportStrategy):
         except Exception as e:
             raise HyperbrowserError("Post request failed", original_error=e)
 
-    async def get(self, url: str, params: Optional[dict] = None) -> APIResponse:
+    async def get(
+        self, url: str, params: Optional[dict] = None, follow_redirects: bool = False
+    ) -> APIResponse:
         if params:
             params = {k: v for k, v in params.items() if v is not None}
         try:
-            response = await self.client.get(url, params=params)
+            response = await self.client.get(
+                url, params=params, follow_redirects=follow_redirects
+            )
             return await self._handle_response(response)
         except HyperbrowserError:
             raise

{hyperbrowser-0.28.0 → hyperbrowser-0.30.0}/hyperbrowser/transport/sync.py

@@ -59,11 +59,15 @@ class SyncTransport(TransportStrategy):
         except Exception as e:
             raise HyperbrowserError("Post request failed", original_error=e)
 
-    def get(self, url: str, params: Optional[dict] = None) -> APIResponse:
+    def get(
+        self, url: str, params: Optional[dict] = None, follow_redirects: bool = False
+    ) -> APIResponse:
         if params:
             params = {k: v for k, v in params.items() if v is not None}
         try:
-            response = self.client.get(url, params=params)
+            response = self.client.get(
+                url, params=params, follow_redirects=follow_redirects
+            )
             return self._handle_response(response)
         except HyperbrowserError:
             raise

{hyperbrowser-0.28.0 → hyperbrowser-0.30.0}/pyproject.toml

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