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

robot_wrapper_sdk-0.2.6/PKG-INFO

@@ -0,0 +1,171 @@
+Metadata-Version: 2.4
+Name: robot-wrapper-sdk
+Version: 0.2.6
+Summary: Robot Platform API SDK
+Author-email: GH Robot Platform Team <team@ghrobot.com>
+License: MIT
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: httpx[socks]>=0.24.1
+
+# Robot Platform SDK (Python)
+
+Python SDK for Robot Platform with:
+- develop-app authentication (`app_id`/`app_secret`),
+- sync + async clients,
+- auto token refresh,
+- robot operations and login/hardening task flow.
+
+## Installation
+
+```bash
+pip install robot-wrapper-sdk
+```
+
+## Configuration
+
+Set environment variables (recommended):
+
+- `ROBOT_PLATFORM_BASE_URL`
+- `ROBOT_PLATFORM_APP_ID`
+- `ROBOT_PLATFORM_APP_SECRET`
+- `ROBOT_PLATFORM_PROXY_URL` (optional)
+
+Or pass values directly when creating the module.
+
+---
+
+## Quick Start (Sync)
+
+```python
+from robot_sdk import RobotPlatformModule
+
+sdk = RobotPlatformModule(
+    base_url="http://localhost:8085",
+    app_id="your_app_id",
+    app_secret="your_app_secret",
+    # proxy_url="socks5h://192.168.3.100:1080",  # optional
+)
+
+# Read robot
+robot = sdk.get_robot("2047631542552334336")
+print(robot.platform if robot else "not found")
+
+# Update lifecycle status
+sdk.update_status("2047631542552334336", "live")
+
+# Acquire login tasks
+login_tasks = sdk.acquire_need_login(limit=20, lock_minutes=30)
+
+# Update auth status after worker result
+sdk.update_auth_status("2047631542552334336", "authorized")
+# allowed values: authorized | unauthorized | logged_out
+
+# Acquire hardening tasks
+hardening_tasks = sdk.acquire_unhardened(limit=20, lock_minutes=30, min_age_days=0)
+
+# Update hardening status after worker result
+sdk.update_security_hardened("2047631542552334336", True)
+
+sdk.close()
+```
+
+## Quick Start (Async)
+
+```python
+import asyncio
+from robot_sdk import AsyncRobotPlatformModule
+
+async def main():
+    sdk = AsyncRobotPlatformModule(
+        base_url="http://localhost:8085",
+        app_id="your_app_id",
+        app_secret="your_app_secret",
+    )
+
+    robot = await sdk.get_robot("2047631542552334336")
+    print(robot.platform if robot else "not found")
+
+    await sdk.update_auth_status("2047631542552334336", "authorized")
+    await sdk.update_security_hardened("2047631542552334336", True)
+
+    await sdk.close()
+
+asyncio.run(main())
+```
+
+---
+
+## Response Shape (Placeholder)
+
+Acquire endpoints return:
+
+```json
+{
+  "status": "success",
+  "code": 200,
+  "message": "OK",
+  "data": [
+    {
+      "id": "2047631542552334336",
+      "username": "example_user_1",
+      "platform": "facebook",
+      "status": "live",
+      "auth_status": "unauthorized",
+      "metadata": {
+        "security_hardened": false
+      }
+    }
+  ]
+}
+```
+
+Update endpoints return:
+
+```json
+{
+  "status": "success",
+  "code": 200,
+  "message": "OK",
+  "data": null
+}
+```
+
+---
+
+## Main API
+
+### `RobotPlatformModule` (sync)
+- `list_robots(platform=None, status=None, project_id=None, page=1, limit=20)`
+- `get_robot(robot_id)`
+- `delete_robot(robot_id)`
+- `get_secrets(robot_id)`
+- `update_status(robot_id, status)`
+- `acquire_need_login(limit=20, lock_minutes=30)`
+- `acquire_unhardened(limit=20, lock_minutes=30, min_age_days=0)`
+- `update_auth_status(robot_id, auth_status)`
+- `update_security_hardened(robot_id, security_hardened)`
+- `close()`
+
+### `AsyncRobotPlatformModule` (async)
+Async equivalents of all methods above, plus `await close()`.
+
+---
+
+## Documentation
+
+- Usage guide: [`docs/guide.md`](docs/guide.md)
+- Package code: `robot_sdk/`
+
+---
+
+## Build & Publish (Maintainers)
+
+```bash
+make venv
+make install
+make build
+make publish
+```
+
+Requires valid PyPI credentials (Twine).

robot_wrapper_sdk-0.2.6/README.md

@@ -0,0 +1,161 @@
+# Robot Platform SDK (Python)
+
+Python SDK for Robot Platform with:
+- develop-app authentication (`app_id`/`app_secret`),
+- sync + async clients,
+- auto token refresh,
+- robot operations and login/hardening task flow.
+
+## Installation
+
+```bash
+pip install robot-wrapper-sdk
+```
+
+## Configuration
+
+Set environment variables (recommended):
+
+- `ROBOT_PLATFORM_BASE_URL`
+- `ROBOT_PLATFORM_APP_ID`
+- `ROBOT_PLATFORM_APP_SECRET`
+- `ROBOT_PLATFORM_PROXY_URL` (optional)
+
+Or pass values directly when creating the module.
+
+---
+
+## Quick Start (Sync)
+
+```python
+from robot_sdk import RobotPlatformModule
+
+sdk = RobotPlatformModule(
+    base_url="http://localhost:8085",
+    app_id="your_app_id",
+    app_secret="your_app_secret",
+    # proxy_url="socks5h://192.168.3.100:1080",  # optional
+)
+
+# Read robot
+robot = sdk.get_robot("2047631542552334336")
+print(robot.platform if robot else "not found")
+
+# Update lifecycle status
+sdk.update_status("2047631542552334336", "live")
+
+# Acquire login tasks
+login_tasks = sdk.acquire_need_login(limit=20, lock_minutes=30)
+
+# Update auth status after worker result
+sdk.update_auth_status("2047631542552334336", "authorized")
+# allowed values: authorized | unauthorized | logged_out
+
+# Acquire hardening tasks
+hardening_tasks = sdk.acquire_unhardened(limit=20, lock_minutes=30, min_age_days=0)
+
+# Update hardening status after worker result
+sdk.update_security_hardened("2047631542552334336", True)
+
+sdk.close()
+```
+
+## Quick Start (Async)
+
+```python
+import asyncio
+from robot_sdk import AsyncRobotPlatformModule
+
+async def main():
+    sdk = AsyncRobotPlatformModule(
+        base_url="http://localhost:8085",
+        app_id="your_app_id",
+        app_secret="your_app_secret",
+    )
+
+    robot = await sdk.get_robot("2047631542552334336")
+    print(robot.platform if robot else "not found")
+
+    await sdk.update_auth_status("2047631542552334336", "authorized")
+    await sdk.update_security_hardened("2047631542552334336", True)
+
+    await sdk.close()
+
+asyncio.run(main())
+```
+
+---
+
+## Response Shape (Placeholder)
+
+Acquire endpoints return:
+
+```json
+{
+  "status": "success",
+  "code": 200,
+  "message": "OK",
+  "data": [
+    {
+      "id": "2047631542552334336",
+      "username": "example_user_1",
+      "platform": "facebook",
+      "status": "live",
+      "auth_status": "unauthorized",
+      "metadata": {
+        "security_hardened": false
+      }
+    }
+  ]
+}
+```
+
+Update endpoints return:
+
+```json
+{
+  "status": "success",
+  "code": 200,
+  "message": "OK",
+  "data": null
+}
+```
+
+---
+
+## Main API
+
+### `RobotPlatformModule` (sync)
+- `list_robots(platform=None, status=None, project_id=None, page=1, limit=20)`
+- `get_robot(robot_id)`
+- `delete_robot(robot_id)`
+- `get_secrets(robot_id)`
+- `update_status(robot_id, status)`
+- `acquire_need_login(limit=20, lock_minutes=30)`
+- `acquire_unhardened(limit=20, lock_minutes=30, min_age_days=0)`
+- `update_auth_status(robot_id, auth_status)`
+- `update_security_hardened(robot_id, security_hardened)`
+- `close()`
+
+### `AsyncRobotPlatformModule` (async)
+Async equivalents of all methods above, plus `await close()`.
+
+---
+
+## Documentation
+
+- Usage guide: [`docs/guide.md`](docs/guide.md)
+- Package code: `robot_sdk/`
+
+---
+
+## Build & Publish (Maintainers)
+
+```bash
+make venv
+make install
+make build
+make publish
+```
+
+Requires valid PyPI credentials (Twine).

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.4 → 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.4"
+version = "0.2.6"
 description = "Robot Platform API SDK"
 readme = "README.md"
 requires-python = ">=3.10"

{robot_wrapper_sdk-0.2.4 → robot_wrapper_sdk-0.2.6}/robot_sdk/__init__.py

@@ -31,6 +31,18 @@ class RobotPlatformModule:
     def update_status(self, robot_id: str, status: str) -> None:
         self.robot_usecase.update_status(robot_id, status)
 
+    def acquire_need_login(self, limit: int = 20, lock_minutes: int = 30):
+        return self.robot_usecase.acquire_need_login(limit, lock_minutes)
+
+    def acquire_unhardened(self, limit: int = 20, lock_minutes: int = 30, min_age_days: int = 0):
+        return self.robot_usecase.acquire_unhardened(limit, lock_minutes, min_age_days)
+
+    def update_auth_status(self, robot_id: str, auth_status: str) -> None:
+        self.robot_usecase.update_auth_status(robot_id, auth_status)
+
+    def update_security_hardened(self, robot_id: str, security_hardened: bool) -> None:
+        self.robot_usecase.update_security_hardened(robot_id, security_hardened)
+
 
 class AsyncRobotPlatformModule:
     """Asynchronous Facade for the Robot Platform SDK."""
@@ -57,4 +69,16 @@ class AsyncRobotPlatformModule:
     async def update_status(self, robot_id: str, status: str) -> None:
         await self.robot_usecase.update_status(robot_id, status)
 
-__all__ = ["RobotPlatformModule", "AsyncRobotPlatformModule"]
+    async def acquire_need_login(self, limit: int = 20, lock_minutes: int = 30):
+        return await self.robot_usecase.acquire_need_login(limit, lock_minutes)
+
+    async def acquire_unhardened(self, limit: int = 20, lock_minutes: int = 30, min_age_days: int = 0):
+        return await self.robot_usecase.acquire_unhardened(limit, lock_minutes, min_age_days)
+
+    async def update_auth_status(self, robot_id: str, auth_status: str) -> None:
+        await self.robot_usecase.update_auth_status(robot_id, auth_status)
+
+    async def update_security_hardened(self, robot_id: str, security_hardened: bool) -> None:
+        await self.robot_usecase.update_security_hardened(robot_id, security_hardened)
+
+__all__ = ["RobotPlatformModule", "AsyncRobotPlatformModule", "Robot", "RobotSecrets"]

{robot_wrapper_sdk-0.2.4 → robot_wrapper_sdk-0.2.6}/robot_sdk/application/usecases.py

@@ -21,12 +21,24 @@ class RobotUsecase:
     def list_robots(self, platform: Optional[str] = None, status: Optional[str] = None, project_id: Optional[int] = None, page: int = 1, limit: int = 20) -> Dict[str, Any]:
         items, total = self.repo.list_robots(platform, status, project_id, page, limit)
         return {
-            "data": [r.__dict__ for r in items], 
+            "data": [r.__dict__ for r in items],
             "total": total,
             "page": page,
             "limit": limit
         }
 
+    def acquire_need_login(self, limit: int = 20, lock_minutes: int = 30) -> List[Robot]:
+        return self.repo.acquire_need_login(limit, lock_minutes)
+
+    def acquire_unhardened(self, limit: int = 20, lock_minutes: int = 30, min_age_days: int = 0) -> List[Robot]:
+        return self.repo.acquire_unhardened(limit, lock_minutes, min_age_days)
+
+    def update_auth_status(self, robot_id: str, auth_status: str) -> None:
+        self.repo.update_auth_status(robot_id, auth_status)
+
+    def update_security_hardened(self, robot_id: str, security_hardened: bool) -> None:
+        self.repo.update_security_hardened(robot_id, security_hardened)
+
 
 class AsyncRobotUsecase:
     def __init__(self, repo: AsyncRobotRepository):
@@ -47,8 +59,20 @@ class AsyncRobotUsecase:
     async def list_robots(self, platform: Optional[str] = None, status: Optional[str] = None, project_id: Optional[int] = None, page: int = 1, limit: int = 20) -> Dict[str, Any]:
         items, total = await self.repo.list_robots(platform, status, project_id, page, limit)
         return {
-            "data": [r.__dict__ for r in items], 
+            "data": [r.__dict__ for r in items],
             "total": total,
             "page": page,
             "limit": limit
         }
+
+    async def acquire_need_login(self, limit: int = 20, lock_minutes: int = 30) -> List[Robot]:
+        return await self.repo.acquire_need_login(limit, lock_minutes)
+
+    async def acquire_unhardened(self, limit: int = 20, lock_minutes: int = 30, min_age_days: int = 0) -> List[Robot]:
+        return await self.repo.acquire_unhardened(limit, lock_minutes, min_age_days)
+
+    async def update_auth_status(self, robot_id: str, auth_status: str) -> None:
+        await self.repo.update_auth_status(robot_id, auth_status)
+
+    async def update_security_hardened(self, robot_id: str, security_hardened: bool) -> None:
+        await self.repo.update_security_hardened(robot_id, security_hardened)

{robot_wrapper_sdk-0.2.4 → robot_wrapper_sdk-0.2.6}/robot_sdk/domain/entities.py

@@ -23,3 +23,29 @@ class RobotActivity:
     activity_type: str
     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
+    code: int
+    message: str
+    data: Any = None

{robot_wrapper_sdk-0.2.4 → robot_wrapper_sdk-0.2.6}/robot_sdk/domain/repositories.py

@@ -23,6 +23,22 @@ class RobotRepository(ABC):
     def list_robots(self, platform: Optional[str] = None, status: Optional[str] = None, project_id: Optional[int] = None, page: int = 1, limit: int = 20) -> Tuple[List[Robot], int]:
         pass
 
+    @abstractmethod
+    def acquire_need_login(self, limit: int = 20, lock_minutes: int = 30) -> List[Robot]:
+        pass
+
+    @abstractmethod
+    def acquire_unhardened(self, limit: int = 20, lock_minutes: int = 30, min_age_days: int = 0) -> List[Robot]:
+        pass
+
+    @abstractmethod
+    def update_auth_status(self, robot_id: str, auth_status: str) -> None:
+        pass
+
+    @abstractmethod
+    def update_security_hardened(self, robot_id: str, security_hardened: bool) -> None:
+        pass
+
 class AsyncRobotRepository(ABC):
     @abstractmethod
     async def find_by_id(self, robot_id: str) -> Optional[Robot]:
@@ -43,3 +59,19 @@ class AsyncRobotRepository(ABC):
     @abstractmethod
     async def list_robots(self, platform: Optional[str] = None, status: Optional[str] = None, project_id: Optional[int] = None, page: int = 1, limit: int = 20) -> Tuple[List[Robot], int]:
         pass
+
+    @abstractmethod
+    async def acquire_need_login(self, limit: int = 20, lock_minutes: int = 30) -> List[Robot]:
+        pass
+
+    @abstractmethod
+    async def acquire_unhardened(self, limit: int = 20, lock_minutes: int = 30, min_age_days: int = 0) -> List[Robot]:
+        pass
+
+    @abstractmethod
+    async def update_auth_status(self, robot_id: str, auth_status: str) -> None:
+        pass
+
+    @abstractmethod
+    async def update_security_hardened(self, robot_id: str, security_hardened: bool) -> None:
+        pass