robot-wrapper-sdk 0.2.24__tar.gz → 0.2.26__tar.gz

{robot_wrapper_sdk-0.2.24 → robot_wrapper_sdk-0.2.26}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: robot-wrapper-sdk
-Version: 0.2.24
+Version: 0.2.26
 Summary: Robot Platform API SDK
 Author-email: GH Robot Platform Team <team@ghrobot.com>
 License: MIT

{robot_wrapper_sdk-0.2.24 → robot_wrapper_sdk-0.2.26}/pyproject.toml

@@ -8,7 +8,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "robot-wrapper-sdk"
-version = "0.2.24"
+version = "0.2.26"
 description = "Robot Platform API SDK"
 readme = "README.md"
 requires-python = ">=3.10"

{robot_wrapper_sdk-0.2.24 → robot_wrapper_sdk-0.2.26}/robot_sdk/__init__.py

@@ -9,7 +9,7 @@ from .infrastructure.api_client import AsyncHTTPClient, HTTPClient
 from .infrastructure.robot_api_repository import (AsyncRobotAPIRepository,
                                                   RobotAPIRepository)
 
-__version__ = "0.2.24"
+__version__ = "0.2.26"
 class RobotPlatformModule:
     """Synchronous Facade for the Robot Platform SDK."""
 
@@ -51,6 +51,9 @@ class RobotPlatformModule:
     def unlock_hardening_tasks(self, robot_ids: List[str]) -> None:
         self.robot_usecase.unlock_hardening_tasks(robot_ids)
 
+    def unlock_acquire_tasks(self, robot_ids: List[str]) -> None:
+        self.robot_usecase.unlock_acquire_tasks(robot_ids)
+
     def update_auth_status(self, robot_id: str, auth_status: str | RobotAuthStatus) -> None:
         self.robot_usecase.update_auth_status(robot_id, auth_status)
 
@@ -127,6 +130,9 @@ class AsyncRobotPlatformModule:
     async def unlock_hardening_tasks(self, robot_ids: List[str]) -> None:
         await self.robot_usecase.unlock_hardening_tasks(robot_ids)
 
+    async def unlock_acquire_tasks(self, robot_ids: List[str]) -> None:
+        await self.robot_usecase.unlock_acquire_tasks(robot_ids)
+
     async def update_auth_status(self, robot_id: str, auth_status: str | RobotAuthStatus) -> None:
         await self.robot_usecase.update_auth_status(robot_id, auth_status)
 

{robot_wrapper_sdk-0.2.24 → robot_wrapper_sdk-0.2.26}/robot_sdk/application/usecases.py

@@ -45,6 +45,9 @@ class RobotUsecase:
     def unlock_login_tasks(self, robot_ids: List[str]) -> None:
         self.repo.unlock_login_tasks(robot_ids)
 
+    def unlock_acquire_tasks(self, robot_ids: List[str]) -> None:
+        self.repo.unlock_acquire_tasks(robot_ids)
+
     def unlock_hardening_tasks(self, robot_ids: List[str]) -> None:
         self.repo.unlock_hardening_tasks(robot_ids)
 
@@ -122,6 +125,9 @@ class AsyncRobotUsecase:
     async def unlock_hardening_tasks(self, robot_ids: List[str]) -> None:
         await self.repo.unlock_hardening_tasks(robot_ids)
 
+    async def unlock_acquire_tasks(self, robot_ids: List[str]) -> None:
+        await self.repo.unlock_acquire_tasks(robot_ids)
+
     async def update_auth_status(self, robot_id: str, auth_status: str | RobotAuthStatus) -> None:
         await self.repo.update_auth_status(robot_id, auth_status)
 

{robot_wrapper_sdk-0.2.24 → robot_wrapper_sdk-0.2.26}/robot_sdk/domain/entities.py

@@ -73,6 +73,9 @@ class Robot:
     platform: str
     status: str
     auth_status: str
+    first_name: Optional[str] = None
+    last_name: Optional[str] = None
+    email: Optional[str] = None
     metadata: Dict[str, Any] = field(default_factory=dict)
     project_id: Optional[str] = None
     primary_environment: Optional[str] = None

{robot_wrapper_sdk-0.2.24 → robot_wrapper_sdk-0.2.26}/robot_sdk/domain/repositories.py

@@ -48,6 +48,10 @@ class RobotRepository(ABC):
     def unlock_hardening_tasks(self, robot_ids: List[str]) -> None:
         pass
 
+    @abstractmethod
+    def unlock_acquire_tasks(self, robot_ids: List[str]) -> None:
+        pass
+
     @abstractmethod
     def update_auth_status(self, robot_id: str, auth_status: str | RobotAuthStatus) -> None:
         pass
@@ -134,6 +138,10 @@ class AsyncRobotRepository(ABC):
     async def unlock_hardening_tasks(self, robot_ids: List[str]) -> None:
         pass
 
+    @abstractmethod
+    async def unlock_acquire_tasks(self, robot_ids: List[str]) -> None:
+        pass
+
     @abstractmethod
     async def update_auth_status(self, robot_id: str, auth_status: str | RobotAuthStatus) -> None:
         pass

{robot_wrapper_sdk-0.2.24 → robot_wrapper_sdk-0.2.26}/robot_sdk/infrastructure/robot_api_repository.py

@@ -1,4 +1,4 @@
-from dataclasses import asdict
+from dataclasses import asdict, is_dataclass
 from typing import Any, Dict, List, Optional, Tuple
 
 from ..domain.entities import (AcquireRobotsMetadataFilter, BrowserCapture,
@@ -185,7 +185,15 @@ def _acquire_robots_payload(limit: int, lock_minutes: int, project_name: Optiona
     if environment:
         payload["environment"] = environment
     if metadata:
-        payload["metadata"] = [asdict(m) for m in metadata]
+        if isinstance(metadata, list):
+            payload["metadata"] = [
+                asdict(m) if is_dataclass(m) else m
+                for m in metadata
+            ]
+        elif isinstance(metadata, dict):
+            payload["metadata"] = metadata
+        else:
+            payload["metadata"] = asdict(metadata) if is_dataclass(metadata) else metadata
     return payload
 
 
@@ -448,6 +456,13 @@ class AsyncRobotAPIRepository(AsyncRobotRepository):
             json={"robot_ids": robot_ids},
         )
 
+    async def unlock_acquire_tasks(self, robot_ids: List[str]) -> None:
+        await self.client.request(
+            "POST",
+            "/robots/acquire/unlock",
+            json={"robot_ids": robot_ids},
+        )
+
     async def update_auth_status(self, robot_id: str, auth_status: str | RobotAuthStatus) -> None:
         normalized_auth_status = normalize_robot_auth_status(auth_status)
         ensure_valid_worker_robot_auth_status(normalized_auth_status)

{robot_wrapper_sdk-0.2.24 → robot_wrapper_sdk-0.2.26}/robot_wrapper_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: robot-wrapper-sdk
-Version: 0.2.24
+Version: 0.2.26
 Summary: Robot Platform API SDK
 Author-email: GH Robot Platform Team <team@ghrobot.com>
 License: MIT