robot-wrapper-sdk 0.2.8__tar.gz → 0.2.10__tar.gz

{robot_wrapper_sdk-0.2.8 → robot_wrapper_sdk-0.2.10}/PKG-INFO

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

{robot_wrapper_sdk-0.2.8 → robot_wrapper_sdk-0.2.10}/pyproject.toml

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

{robot_wrapper_sdk-0.2.8 → robot_wrapper_sdk-0.2.10}/robot_sdk/domain/entities.py

@@ -32,25 +32,23 @@ def normalize_robot_auth_status(status: str | RobotAuthStatus) -> str:
 def ensure_valid_robot_status(status: str) -> None:
     allowed = {s.value for s in RobotStatus}
     if status not in allowed:
-        raise ValueError(f"invalid status: {status}. allowed={sorted(allowed)}")
+        raise ValueError(
+            f"invalid status: {status}. allowed={sorted(allowed)}")
 
 
 def ensure_valid_robot_auth_status(status: str) -> None:
     allowed = {s.value for s in RobotAuthStatus}
     if status not in allowed:
-        raise ValueError(f"invalid auth_status: {status}. allowed={sorted(allowed)}")
+        raise ValueError(
+            f"invalid auth_status: {status}. allowed={sorted(allowed)}")
 
 
 def ensure_valid_worker_robot_auth_status(status: str) -> None:
     ensure_valid_robot_auth_status(status)
-    if status == RobotAuthStatus.LOGIN_ERROR.value:
-        raise ValueError("auth_status login_error is not allowed for this endpoint")
 
 
 def ensure_valid_worker_robot_status(status: str) -> None:
     ensure_valid_robot_status(status)
-    if status == RobotStatus.CHECKPOINT.value:
-        raise ValueError("status checkpoint is not allowed for this endpoint")
 
 
 def ensure_valid_metadata(metadata: Dict[str, Any]) -> None:
@@ -74,16 +72,19 @@ class Robot:
     username: str
     platform: str
     status: str
+    auth_status: str
     metadata: Dict[str, Any] = field(default_factory=dict)
     project_id: Optional[str] = None
     primary_environment: Optional[str] = None
     created_at: Optional[str] = None
 
+
 @dataclass
 class RobotSecrets:
     password: str
     two_fa_secret: str
 
+
 @dataclass
 class RobotActivity:
     id: str
@@ -91,25 +92,30 @@ class RobotActivity:
     status: str
     message: str
 
+
 @dataclass
 class AcquireNeedLoginRequest:
     limit: int = 20
     lock_minutes: int = 30
 
+
 @dataclass
 class AcquireUnhardenedRequest:
     limit: int = 20
     lock_minutes: int = 30
     min_age_days: int = 0
 
+
 @dataclass
 class UpdateAuthStatusRequest:
     auth_status: str
 
+
 @dataclass
 class UpdateSecurityHardenedRequest:
     security_hardened: bool
 
+
 @dataclass
 class StandardResponse:
     status: str

{robot_wrapper_sdk-0.2.8 → robot_wrapper_sdk-0.2.10}/robot_sdk/infrastructure/api_client.py

@@ -3,6 +3,54 @@ import os
 from typing import Dict, Any, Optional
 from httpx import RemoteProtocolError
 
+
+def _build_httpx_client(base_url: str, auth: httpx.Auth, proxy_url: Optional[str]) -> httpx.Client:
+    kwargs: Dict[str, Any] = {"base_url": base_url, "auth": auth}
+    if proxy_url:
+        kwargs["proxy"] = proxy_url
+    try:
+        return httpx.Client(**kwargs)
+    except TypeError:
+        if "proxy" in kwargs:
+            kwargs["proxies"] = kwargs.pop("proxy")
+        return httpx.Client(**kwargs)
+
+
+def _build_async_httpx_client(base_url: str, auth: httpx.Auth, proxy_url: Optional[str]) -> httpx.AsyncClient:
+    kwargs: Dict[str, Any] = {"base_url": base_url, "auth": auth}
+    if proxy_url:
+        kwargs["proxy"] = proxy_url
+    try:
+        return httpx.AsyncClient(**kwargs)
+    except TypeError:
+        if "proxy" in kwargs:
+            kwargs["proxies"] = kwargs.pop("proxy")
+        return httpx.AsyncClient(**kwargs)
+
+
+def _build_plain_httpx_client(proxy_url: Optional[str]) -> httpx.Client:
+    kwargs: Dict[str, Any] = {}
+    if proxy_url:
+        kwargs["proxy"] = proxy_url
+    try:
+        return httpx.Client(**kwargs)
+    except TypeError:
+        if "proxy" in kwargs:
+            kwargs["proxies"] = kwargs.pop("proxy")
+        return httpx.Client(**kwargs)
+
+
+def _build_plain_async_httpx_client(proxy_url: Optional[str]) -> httpx.AsyncClient:
+    kwargs: Dict[str, Any] = {}
+    if proxy_url:
+        kwargs["proxy"] = proxy_url
+    try:
+        return httpx.AsyncClient(**kwargs)
+    except TypeError:
+        if "proxy" in kwargs:
+            kwargs["proxies"] = kwargs.pop("proxy")
+        return httpx.AsyncClient(**kwargs)
+
 def _extract_tokens(payload: Dict[str, Any]) -> tuple[Optional[str], Optional[str]]:
     data = payload.get("data") if isinstance(payload, dict) else None
     if isinstance(data, dict):
@@ -69,10 +117,10 @@ class HTTPClient:
         self.access_token: Optional[str] = None
         self.refresh_token: Optional[str] = None
         
-        self.client = httpx.Client(
+        self.client = _build_httpx_client(
             base_url=self.base_url,
             auth=RobotAuth(self._get_or_login, self._refresh),
-            proxy=self.proxy_url
+            proxy_url=self.proxy_url,
         )
 
     def _get_or_login(self) -> str:
@@ -84,7 +132,7 @@ class HTTPClient:
         return token
 
     def _login(self) -> None:
-        with httpx.Client(proxy=self.proxy_url) as c:
+        with _build_plain_httpx_client(self.proxy_url) as c:
             resp = c.post(
                 f"{self.base_url}/api/v1/developer/auth/token",
                 json={"app_id": self.app_id, "app_secret": self.app_secret}
@@ -107,7 +155,7 @@ class HTTPClient:
                 raise RuntimeError("Empty access token after login")
             return token
             
-        with httpx.Client(proxy=self.proxy_url) as c:
+        with _build_plain_httpx_client(self.proxy_url) as c:
             resp = c.post(
                 f"{self.base_url}/api/v1/developer/auth/refresh",
                 json={"refresh_token": self.refresh_token}
@@ -146,10 +194,10 @@ class HTTPClient:
                     raise
                 last_err = err
                 self.client.close()
-                self.client = httpx.Client(
+                self.client = _build_httpx_client(
                     base_url=self.base_url,
                     auth=RobotAuth(self._get_or_login, self._refresh),
-                    proxy=self.proxy_url,
+                    proxy_url=self.proxy_url,
                 )
 
         if resp is None:
@@ -183,10 +231,10 @@ class AsyncHTTPClient:
         self.access_token: Optional[str] = None
         self.refresh_token: Optional[str] = None
         
-        self.client = httpx.AsyncClient(
+        self.client = _build_async_httpx_client(
             base_url=self.base_url,
             auth=AsyncRobotAuth(self._get_or_login, self._refresh),
-            proxy=self.proxy_url
+            proxy_url=self.proxy_url,
         )
 
     async def _get_or_login(self) -> str:
@@ -198,7 +246,7 @@ class AsyncHTTPClient:
         return token
 
     async def _login(self) -> None:
-        async with httpx.AsyncClient(proxy=self.proxy_url) as c:
+        async with _build_plain_async_httpx_client(self.proxy_url) as c:
             resp = await c.post(
                 f"{self.base_url}/api/v1/developer/auth/token",
                 json={"app_id": self.app_id, "app_secret": self.app_secret}
@@ -220,7 +268,7 @@ class AsyncHTTPClient:
                 raise RuntimeError("Empty access token after login")
             return token
             
-        async with httpx.AsyncClient(proxy=self.proxy_url) as c:
+        async with _build_plain_async_httpx_client(self.proxy_url) as c:
             resp = await c.post(
                 f"{self.base_url}/api/v1/developer/auth/refresh",
                 json={"refresh_token": self.refresh_token}

{robot_wrapper_sdk-0.2.8 → robot_wrapper_sdk-0.2.10}/robot_sdk/infrastructure/robot_api_repository.py

@@ -39,13 +39,26 @@ def _extract_items(resp: Dict[str, Any]) -> List[Dict[str, Any]]:
             return [item for item in nested if isinstance(item, dict)]
     return []
 
+
+def _to_robot_payload(item: Dict[str, Any]) -> Dict[str, Any]:
+    payload = dict(item)
+    payload.setdefault("auth_status", "")
+    payload.setdefault("metadata", {})
+    return payload
+
+
+def _to_robots(items: List[Dict[str, Any]]) -> List[Robot]:
+    return [Robot(**_to_robot_payload(item)) for item in items]
+
 class RobotAPIRepository(RobotRepository):
     def __init__(self, client: HTTPClient):
         self.client = client
 
     def find_by_id(self, robot_id: str) -> Optional[Robot]:
         resp = self.client.request("GET", f"/robots/{robot_id}")
-        return Robot(**resp) if resp else None
+        payload = resp.get("data") if isinstance(resp, dict) else None
+        data = payload if isinstance(payload, dict) else resp
+        return Robot(**data) if isinstance(data, dict) else None
 
     def update_status(self, robot_id: str, status: str | RobotStatus) -> None:
         normalized_status = normalize_robot_status(status)
@@ -76,7 +89,7 @@ class RobotAPIRepository(RobotRepository):
             json={"limit": limit, "lock_minutes": lock_minutes},
         )
         data = _extract_items(resp)
-        return [Robot(**r) for r in data]
+        return _to_robots(data)
 
     def acquire_unhardened(self, limit: int = 20, lock_minutes: int = 30, min_age_days: int = 0) -> List[Robot]:
         resp = self.client.request(
@@ -85,7 +98,7 @@ class RobotAPIRepository(RobotRepository):
             json={"limit": limit, "lock_minutes": lock_minutes, "min_age_days": min_age_days},
         )
         data = _extract_items(resp)
-        return [Robot(**r) for r in data]
+        return _to_robots(data)
 
     def update_auth_status(self, robot_id: str, auth_status: str | RobotAuthStatus) -> None:
         normalized_auth_status = normalize_robot_auth_status(auth_status)
@@ -106,7 +119,7 @@ class RobotAPIRepository(RobotRepository):
     def update_metadata(self, robot_id: str, metadata: Dict[str, Any]) -> None:
         self.client.request(
             "PATCH",
-            f"/robots/{robot_id}",
+            f"/robots/{robot_id}/metadata",
             json={"metadata": metadata},
         )
 
@@ -116,7 +129,9 @@ class AsyncRobotAPIRepository(AsyncRobotRepository):
 
     async def find_by_id(self, robot_id: str) -> Optional[Robot]:
         resp = await self.client.request("GET", f"/robots/{robot_id}")
-        return Robot(**resp) if resp else None
+        payload = resp.get("data") if isinstance(resp, dict) else None
+        data = payload if isinstance(payload, dict) else resp
+        return Robot(**data) if isinstance(data, dict) else None
 
     async def update_status(self, robot_id: str, status: str | RobotStatus) -> None:
         normalized_status = normalize_robot_status(status)
@@ -147,7 +162,7 @@ class AsyncRobotAPIRepository(AsyncRobotRepository):
             json={"limit": limit, "lock_minutes": lock_minutes},
         )
         data = _extract_items(resp)
-        return [Robot(**r) for r in data]
+        return _to_robots(data)
 
     async def acquire_unhardened(self, limit: int = 20, lock_minutes: int = 30, min_age_days: int = 0) -> List[Robot]:
         resp = await self.client.request(
@@ -156,7 +171,7 @@ class AsyncRobotAPIRepository(AsyncRobotRepository):
             json={"limit": limit, "lock_minutes": lock_minutes, "min_age_days": min_age_days},
         )
         data = _extract_items(resp)
-        return [Robot(**r) for r in data]
+        return _to_robots(data)
 
     async def update_auth_status(self, robot_id: str, auth_status: str | RobotAuthStatus) -> None:
         normalized_auth_status = normalize_robot_auth_status(auth_status)
@@ -177,6 +192,6 @@ class AsyncRobotAPIRepository(AsyncRobotRepository):
     async def update_metadata(self, robot_id: str, metadata: Dict[str, Any]) -> None:
         await self.client.request(
             "PATCH",
-            f"/robots/{robot_id}",
+            f"/robots/{robot_id}/metadata",
             json={"metadata": metadata},
         )

{robot_wrapper_sdk-0.2.8 → robot_wrapper_sdk-0.2.10}/robot_wrapper_sdk.egg-info/PKG-INFO

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