robot-wrapper-sdk 0.2.6__tar.gz → 0.2.7__tar.gz

{robot_wrapper_sdk-0.2.6 → robot_wrapper_sdk-0.2.7}/PKG-INFO

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

{robot_wrapper_sdk-0.2.6 → robot_wrapper_sdk-0.2.7}/pyproject.toml

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

{robot_wrapper_sdk-0.2.6 → robot_wrapper_sdk-0.2.7}/robot_sdk/infrastructure/api_client.py

@@ -1,6 +1,7 @@
 import httpx
 import os
 from typing import Dict, Any, Optional
+from httpx import RemoteProtocolError
 
 def _extract_tokens(payload: Dict[str, Any]) -> tuple[Optional[str], Optional[str]]:
     data = payload.get("data") if isinstance(payload, dict) else None
@@ -132,7 +133,30 @@ class HTTPClient:
             full_path = f"/api/v1/developer{path}"
 
         print(f"🚀 Requesting: {self.base_url}{full_path}")
-        resp = self.client.request(method, full_path, **kwargs)
+        last_err: Optional[Exception] = None
+        resp = None
+        for attempt in range(2):
+            try:
+                resp = self.client.request(method, full_path, **kwargs)
+                break
+            except Exception as err:
+                msg = str(err)
+                is_disconnect = isinstance(err, RemoteProtocolError) or "Server disconnected without sending a response" in msg
+                if not is_disconnect or attempt == 1:
+                    raise
+                last_err = err
+                self.client.close()
+                self.client = httpx.Client(
+                    base_url=self.base_url,
+                    auth=RobotAuth(self._get_or_login, self._refresh),
+                    proxy=self.proxy_url,
+                )
+
+        if resp is None:
+            if last_err is not None:
+                raise last_err
+            raise RuntimeError("Request failed without response")
+
         resp.raise_for_status()
         try:
             return dict(resp.json())

{robot_wrapper_sdk-0.2.6 → robot_wrapper_sdk-0.2.7}/robot_sdk/infrastructure/robot_api_repository.py

@@ -3,6 +3,33 @@ from .api_client import HTTPClient, AsyncHTTPClient
 from ..domain.repositories import RobotRepository, AsyncRobotRepository
 from ..domain.entities import Robot, RobotSecrets
 
+
+def _extract_list_payload(resp: Dict[str, Any]) -> Tuple[List[Dict[str, Any]], int]:
+    container = resp.get("data") if isinstance(resp, dict) else None
+    if isinstance(container, dict):
+        items = container.get("data")
+        total = container.get("total", 0)
+    else:
+        items = container
+        total = resp.get("total", 0) if isinstance(resp, dict) else 0
+
+    if not isinstance(items, list):
+        return [], int(total) if isinstance(total, int) else 0
+
+    mapped = [item for item in items if isinstance(item, dict)]
+    return mapped, int(total) if isinstance(total, int) else 0
+
+
+def _extract_items(resp: Dict[str, Any]) -> List[Dict[str, Any]]:
+    container = resp.get("data") if isinstance(resp, dict) else None
+    if isinstance(container, list):
+        return [item for item in container if isinstance(item, dict)]
+    if isinstance(container, dict):
+        nested = container.get("data")
+        if isinstance(nested, list):
+            return [item for item in nested if isinstance(item, dict)]
+    return []
+
 class RobotAPIRepository(RobotRepository):
     def __init__(self, client: HTTPClient):
         self.client = client
@@ -28,8 +55,7 @@ class RobotAPIRepository(RobotRepository):
         if project_id: params["project_id"] = project_id
 
         resp = self.client.request("GET", "/robots", params=params)
-        data = resp.get("data", [])
-        total = resp.get("total", 0)
+        data, total = _extract_list_payload(resp)
         return [Robot(**r) for r in data], total
 
     def acquire_need_login(self, limit: int = 20, lock_minutes: int = 30) -> List[Robot]:
@@ -38,7 +64,7 @@ class RobotAPIRepository(RobotRepository):
             "/robots/login/acquire",
             json={"limit": limit, "lock_minutes": lock_minutes},
         )
-        data = resp.get("data", [])
+        data = _extract_items(resp)
         return [Robot(**r) for r in data]
 
     def acquire_unhardened(self, limit: int = 20, lock_minutes: int = 30, min_age_days: int = 0) -> List[Robot]:
@@ -47,7 +73,7 @@ class RobotAPIRepository(RobotRepository):
             "/robots/security-hardened/acquire",
             json={"limit": limit, "lock_minutes": lock_minutes, "min_age_days": min_age_days},
         )
-        data = resp.get("data", [])
+        data = _extract_items(resp)
         return [Robot(**r) for r in data]
 
     def update_auth_status(self, robot_id: str, auth_status: str) -> None:
@@ -89,8 +115,7 @@ class AsyncRobotAPIRepository(AsyncRobotRepository):
         if project_id: params["project_id"] = project_id
 
         resp = await self.client.request("GET", "/robots", params=params)
-        data = resp.get("data", [])
-        total = resp.get("total", 0)
+        data, total = _extract_list_payload(resp)
         return [Robot(**r) for r in data], total
 
     async def acquire_need_login(self, limit: int = 20, lock_minutes: int = 30) -> List[Robot]:
@@ -99,7 +124,7 @@ class AsyncRobotAPIRepository(AsyncRobotRepository):
             "/robots/login/acquire",
             json={"limit": limit, "lock_minutes": lock_minutes},
         )
-        data = resp.get("data", [])
+        data = _extract_items(resp)
         return [Robot(**r) for r in data]
 
     async def acquire_unhardened(self, limit: int = 20, lock_minutes: int = 30, min_age_days: int = 0) -> List[Robot]:
@@ -108,7 +133,7 @@ class AsyncRobotAPIRepository(AsyncRobotRepository):
             "/robots/security-hardened/acquire",
             json={"limit": limit, "lock_minutes": lock_minutes, "min_age_days": min_age_days},
         )
-        data = resp.get("data", [])
+        data = _extract_items(resp)
         return [Robot(**r) for r in data]
 
     async def update_auth_status(self, robot_id: str, auth_status: str) -> None:

{robot_wrapper_sdk-0.2.6 → robot_wrapper_sdk-0.2.7}/robot_wrapper_sdk.egg-info/PKG-INFO

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