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

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

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

robot_wrapper_sdk-0.2.6/examples/main.py

@@ -0,0 +1,80 @@
+import asyncio
+import os
+import sys
+
+sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
+from robot_sdk import RobotPlatformModule, AsyncRobotPlatformModule
+
+
+def setup_environ():
+    # Update theo develop-app auth flow
+    os.environ["ROBOT_PLATFORM_BASE_URL"] = "http://localhost:8085"
+    os.environ["ROBOT_PLATFORM_APP_ID"] = "your_app_id"
+    os.environ["ROBOT_PLATFORM_APP_SECRET"] = "your_app_secret"
+    # os.environ["ROBOT_PLATFORM_PROXY_URL"] = "socks5h://192.168.3.100:1080"
+
+
+def test_sync():
+    print("--- SYNC EXECUTION ---")
+    module = RobotPlatformModule()
+    try:
+        robot_id = "2047631542552334336"
+
+        robot = module.get_robot(robot_id)
+        if robot:
+            print("Robot:", robot.username, robot.platform)
+        else:
+            print("Robot not found")
+
+        login_tasks = module.acquire_need_login(limit=20, lock_minutes=30)
+        print("Acquire login tasks:", len(login_tasks))
+
+        module.update_auth_status(robot_id, "authorized")
+        print("Updated auth_status -> authorized")
+
+        hardening_tasks = module.acquire_unhardened(limit=20, lock_minutes=30, min_age_days=0)
+        print("Acquire hardening tasks:", len(hardening_tasks))
+
+        module.update_security_hardened(robot_id, True)
+        print("Updated security_hardened -> true")
+
+    except Exception as e:
+        print("Sync demo error:", e)
+    finally:
+        module.close()
+
+
+async def test_async():
+    print("--- ASYNC EXECUTION ---")
+    module = AsyncRobotPlatformModule()
+    try:
+        robot_id = "2047631542552334336"
+
+        robot = await module.get_robot(robot_id)
+        if robot:
+            print("Robot:", robot.username, robot.platform)
+        else:
+            print("Robot not found")
+
+        login_tasks = await module.acquire_need_login(limit=20, lock_minutes=30)
+        print("Acquire login tasks:", len(login_tasks))
+
+        await module.update_auth_status(robot_id, "authorized")
+        print("Updated auth_status -> authorized")
+
+        hardening_tasks = await module.acquire_unhardened(limit=20, lock_minutes=30, min_age_days=0)
+        print("Acquire hardening tasks:", len(hardening_tasks))
+
+        await module.update_security_hardened(robot_id, True)
+        print("Updated security_hardened -> true")
+
+    except Exception as e:
+        print("Async demo error:", e)
+    finally:
+        await module.close()
+
+
+if __name__ == "__main__":
+    setup_environ()
+    test_sync()
+    asyncio.run(test_async())

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

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

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

@@ -2,6 +2,19 @@ import httpx
 import os
 from typing import Dict, Any, Optional
 
+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):
+        access_token = data.get("access_token")
+        refresh_token = data.get("refresh_token")
+    else:
+        access_token = payload.get("access_token") if isinstance(payload, dict) else None
+        refresh_token = payload.get("refresh_token") if isinstance(payload, dict) else None
+
+    access_token = str(access_token).strip() if access_token else None
+    refresh_token = str(refresh_token).strip() if refresh_token else None
+    return access_token, refresh_token
+
 class RobotAuth(httpx.Auth):
     def __init__(self, login_func, refresh_func):
         self.login_func = login_func
@@ -9,11 +22,15 @@ class RobotAuth(httpx.Auth):
 
     def auth_flow(self, request):
         token = self.login_func()
+        if not token:
+            raise RuntimeError("Missing access token. Authenticate first.")
         request.headers["Authorization"] = f"Bearer {token}"
         response = yield request
 
         if response.status_code == 401:
             token = self.refresh_func()
+            if not token:
+                raise RuntimeError("Token refresh failed: empty access token")
             request.headers["Authorization"] = f"Bearer {token}"
             yield request
 
@@ -24,11 +41,15 @@ class AsyncRobotAuth(httpx.Auth):
 
     async def async_auth_flow(self, request):
         token = await self.login_func()
+        if not token:
+            raise RuntimeError("Missing access token. Authenticate first.")
         request.headers["Authorization"] = f"Bearer {token}"
         response = yield request
 
         if response.status_code == 401:
             token = await self.refresh_func()
+            if not token:
+                raise RuntimeError("Token refresh failed: empty access token")
             request.headers["Authorization"] = f"Bearer {token}"
             yield request
 
@@ -56,7 +77,10 @@ class HTTPClient:
     def _get_or_login(self) -> str:
         if not self.access_token:
             self._login()
-        return str(self.access_token) if self.access_token else ""
+        token = str(self.access_token).strip() if self.access_token else ""
+        if not token:
+            raise RuntimeError("Empty access token after login")
+        return token
 
     def _login(self) -> None:
         with httpx.Client(proxy=self.proxy_url) as c:
@@ -67,8 +91,7 @@ class HTTPClient:
         resp.raise_for_status()
         try:
             data = resp.json()
-            self.access_token = data.get("access_token")
-            self.refresh_token = data.get("refresh_token")
+            self.access_token, self.refresh_token = _extract_tokens(data)
         except Exception as e:
             print(f"❌ Failed to parse JSON from _login: {e}")
             print(f"📄 Login Response Content: {resp.text[:500]}")
@@ -78,7 +101,10 @@ class HTTPClient:
         # Developer tokens currently don't use refresh flow in the same way, but keeping structure
         if not self.refresh_token:
             self._login()
-            return str(self.access_token) if self.access_token else ""
+            token = str(self.access_token).strip() if self.access_token else ""
+            if not token:
+                raise RuntimeError("Empty access token after login")
+            return token
             
         with httpx.Client(proxy=self.proxy_url) as c:
             resp = c.post(
@@ -89,20 +115,22 @@ class HTTPClient:
             self._login()
         else:
             data = resp.json()
-            self.access_token = data.get("access_token")
-            self.refresh_token = data.get("refresh_token")
-            
-        return str(self.access_token) if self.access_token else ""
+            self.access_token, self.refresh_token = _extract_tokens(data)
+
+        token = str(self.access_token).strip() if self.access_token else ""
+        if not token:
+            raise RuntimeError("Token refresh failed: empty access token")
+        return token
 
     def request(self, method: str, path: str, **kwargs: Any) -> Dict[str, Any]:
-        # Intelligent path prefixing
         full_path = path
-        if not path.startswith("/api/v1/developer"):
-            # If path doesn't start with /api, add it
+        if path.startswith("/api/"):
+            full_path = path
+        else:
             if not path.startswith("/"):
                 path = f"/{path}"
             full_path = f"/api/v1/developer{path}"
-             
+
         print(f"🚀 Requesting: {self.base_url}{full_path}")
         resp = self.client.request(method, full_path, **kwargs)
         resp.raise_for_status()
@@ -140,7 +168,10 @@ class AsyncHTTPClient:
     async def _get_or_login(self) -> str:
         if not self.access_token:
             await self._login()
-        return str(self.access_token) if self.access_token else ""
+        token = str(self.access_token).strip() if self.access_token else ""
+        if not token:
+            raise RuntimeError("Empty access token after login")
+        return token
 
     async def _login(self) -> None:
         async with httpx.AsyncClient(proxy=self.proxy_url) as c:
@@ -151,8 +182,7 @@ class AsyncHTTPClient:
         resp.raise_for_status()
         try:
             data = resp.json()
-            self.access_token = data.get("access_token")
-            self.refresh_token = data.get("refresh_token")
+            self.access_token, self.refresh_token = _extract_tokens(data)
         except Exception as e:
             print(f"❌ Failed to parse JSON from async _login: {e}")
             print(f"📄 Login Response Content: {resp.text[:500]}")
@@ -161,7 +191,10 @@ class AsyncHTTPClient:
     async def _refresh(self) -> str:
         if not self.refresh_token:
             await self._login()
-            return str(self.access_token) if self.access_token else ""
+            token = str(self.access_token).strip() if self.access_token else ""
+            if not token:
+                raise RuntimeError("Empty access token after login")
+            return token
             
         async with httpx.AsyncClient(proxy=self.proxy_url) as c:
             resp = await c.post(
@@ -172,10 +205,12 @@ class AsyncHTTPClient:
             await self._login()
         else:
             data = resp.json()
-            self.access_token = data.get("access_token")
-            self.refresh_token = data.get("refresh_token")
-            
-        return str(self.access_token) if self.access_token else ""
+            self.access_token, self.refresh_token = _extract_tokens(data)
+
+        token = str(self.access_token).strip() if self.access_token else ""
+        if not token:
+            raise RuntimeError("Token refresh failed: empty access token")
+        return token
 
     async def request(self, method: str, path: str, **kwargs: Any) -> Dict[str, Any]:
         # Intelligent path prefixing

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

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

robot_wrapper_sdk-0.2.5/examples/main.py

@@ -1,42 +0,0 @@
-import asyncio
-import os
-import sys
-
-# Mở rộng đường dẫn tránh lỗi module khi chạy local
-sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
-from robot_sdk import RobotPlatformModule, AsyncRobotPlatformModule
-
-def setup_environ():
-    os.environ["ROBOT_PLATFORM_BASE_URL"] = "http://localhost:8080"
-    os.environ["ROBOT_PLATFORM_USERNAME"] = "testuser"
-    os.environ["ROBOT_PLATFORM_PASSWORD"] = "testpass"
-
-def test_sync():
-    """Ví dụ sử dụng thư viện đồng bộ."""
-    print("--- SYNC EXECUTION ---")
-    module = RobotPlatformModule()
-    try:
-        robot = module.robot_usecase.get_robot_profile("1234")
-        print("Tải thông tin thành công:", robot.username)
-    except Exception as e:
-        print("Lỗi Demo:", e)
-    finally:
-        module.close()
-
-async def test_async():
-    """Ví dụ sử dụng thư viện bất đồng bộ."""
-    print("--- ASYNC EXECUTION ---")
-    module = AsyncRobotPlatformModule()
-    try:
-        robot = await module.robot_usecase.get_robot_profile("1234")
-        print("Tải thông tin thành công:", robot.username)
-    except Exception as e:
-        print("Lỗi Demo:", e)
-    finally:
-        await module.close()
-
-if __name__ == "__main__":
-    setup_environ()
-    
-    test_sync()
-    asyncio.run(test_async())