pixelarraythirdparty 1.0.7__tar.gz → 1.0.9__tar.gz

{pixelarraythirdparty-1.0.7 → pixelarraythirdparty-1.0.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pixelarraythirdparty
-Version: 1.0.7
+Version: 1.0.9
 Summary: PixelArray 第三方微服务客户端
 Author-email: Lu qi <qi.lu@pixelarrayai.com>
 License-Expression: MIT

{pixelarraythirdparty-1.0.7 → pixelarraythirdparty-1.0.9}/pixelarraythirdparty/__init__.py

@@ -8,9 +8,10 @@ PixelArray 第三方微服务客户端
 - product: 产品管理模块
 - cron: 定时任务管理模块
 - user: 用户管理模块
+- unified_login: 统一登录模块
 """
 
-__version__ = "1.0.7"
+__version__ = "1.0.9"
 __author__ = "Lu qi"
 __email__ = "qi.lu@pixelarrayai.com"
 
@@ -20,4 +21,6 @@ __all__ = [
     "cron",
     "user",
     "order",
+    "filestorage",
+    "unified_login",
 ]

pixelarraythirdparty-1.0.9/pixelarraythirdparty/client.py

@@ -0,0 +1,36 @@
+import aiohttp
+from typing import Dict, Any, Tuple
+
+
+class AsyncClient:
+    def __init__(self, api_key: str):
+        self.base_url = "https://thirdparty.pixelarrayai.com"
+        self.api_key = api_key
+        self.headers = {
+            "Content-Type": "application/json",
+            "X-API-Key": self.api_key,
+        }
+
+    async def _request(
+        self, method: str, url: str, **kwargs
+    ) -> Tuple[Dict[str, Any], bool]:
+        # 如果kwargs中有headers，则合并headers
+        headers = self.headers.copy()
+        if "headers" in kwargs:
+            headers.update(kwargs["headers"])
+            kwargs = {k: v for k, v in kwargs.items() if k != "headers"}
+
+        async with aiohttp.ClientSession() as session:
+            req_method = getattr(session, method.lower())
+            async with req_method(
+                f"{self.base_url}{url}", headers=headers, **kwargs
+            ) as resp:
+                if resp.status == 200:
+                    try:
+                        result = await resp.json()
+                        if result.get("success") is True:
+                            return result.get("data", {}), True
+                    except:
+                        # 如果不是JSON响应，返回空
+                        pass
+                return {}, False

{pixelarraythirdparty-1.0.7 → pixelarraythirdparty-1.0.9}/pixelarraythirdparty/cron/cron.py

@@ -1,17 +1,17 @@
-from pixelarraythirdparty.client import Client
+from pixelarraythirdparty.client import AsyncClient
 
 
-class CronManager(Client):
-    def get_cron_status(self):
+class CronManagerAsync(AsyncClient):
+    async def get_cron_status(self):
         """
-        获取Cron服务状态
-        
+        获取Cron服务状态（异步版本）
+
         功能说明：
             获取Cron服务的运行状态，包括已注册的任务、工作节点状态等。
-        
+
         输入参数：
             无
-        
+
         返回字段：
             data (dict): Cron服务状态信息
                 - registered_tasks (dict): 已注册的任务列表，按工作节点分组
@@ -19,50 +19,50 @@ class CronManager(Client):
                 - scheduled_task_count (int): 定时任务数量
                 - timestamp (str): 状态获取时间
             success (bool): 操作是否成功
-        
+
         异常情况：
             - 获取Cron状态失败：返回错误信息"获取Cron状态失败"
         """
-        data, success = self._request("GET", "/api/cron/status")
+        data, success = await self._request("GET", "/api/cron/status")
         if not success:
             return {}, False
         return data, True
 
-    def get_cron_tasks(self):
+    async def get_cron_tasks(self):
         """
-        获取已注册任务列表
-        
+        获取已注册任务列表（异步版本）
+
         功能说明：
             获取所有已注册的Cron任务列表。
-        
+
         输入参数：
             无
-        
+
         返回字段：
             data (dict): 任务列表信息
                 - tasks (list): 已注册的任务名称列表
                 - count (int): 任务数量
                 - timestamp (str): 获取时间
             success (bool): 操作是否成功
-        
+
         异常情况：
             - 获取任务列表失败：返回错误信息"获取任务列表失败"
         """
-        data, success = self._request("GET", "/api/cron/tasks")
+        data, success = await self._request("GET", "/api/cron/tasks")
         if not success:
             return {}, False
         return data, True
 
-    def get_cron_tasks_scheduled(self):
+    async def get_cron_tasks_scheduled(self):
         """
-        获取定时任务列表
-        
+        获取定时任务列表（异步版本）
+
         功能说明：
             获取所有配置的定时任务列表，包括任务详情、执行时间、状态等。
-        
+
         输入参数：
             无
-        
+
         返回字段：
             data (dict): 定时任务列表信息
                 - tasks (list): 定时任务列表
@@ -81,25 +81,25 @@ class CronManager(Client):
                 - count (int): 任务数量
                 - timestamp (str): 获取时间
             success (bool): 操作是否成功
-        
+
         异常情况：
             - 获取定时任务列表失败：返回错误信息"获取定时任务列表失败"
         """
-        data, success = self._request("GET", "/api/cron/tasks/scheduled")
+        data, success = await self._request("GET", "/api/cron/tasks/scheduled")
         if not success:
             return {}, False
         return data, True
 
-    def get_cron_tasks_detail(self, task_name: str):
+    async def get_cron_tasks_detail(self, task_name: str):
         """
-        获取任务详情
-        
+        获取任务详情（异步版本）
+
         功能说明：
             根据任务名称获取指定任务的详细信息。
-        
+
         输入参数：
             task_name (str): 任务名称，必填，需要URL编码
-        
+
         返回字段：
             data (dict): 任务详细信息
                 - task_name (str): 任务名称
@@ -112,28 +112,28 @@ class CronManager(Client):
                 - registration_info (dict): 注册信息
                 - timestamp (str): 获取时间
             success (bool): 操作是否成功
-        
+
         异常情况：
             - 任务不存在：返回错误信息"任务不存在"
             - 获取任务详情失败：返回错误信息"获取任务详情失败"
         """
-        data, success = self._request("GET", f"/api/cron/tasks/{task_name}")
+        data, success = await self._request("GET", f"/api/cron/tasks/{task_name}")
         if not success:
             return {}, False
         return data, True
 
-    def trigger_cron_task(self, task_name: str, args: list, kwargs: dict):
+    async def trigger_cron_task(self, task_name: str, args: list, kwargs: dict):
         """
-        触发任务执行
-        
+        触发任务执行（异步版本）
+
         功能说明：
             手动触发指定任务的执行，支持传递参数。
-        
+
         输入参数：
             task_name (str): 任务名称，必填，需要URL编码
             args (list): 任务参数列表，可选
             kwargs (dict): 任务关键字参数，可选
-        
+
         返回字段：
             data (dict): 任务触发信息
                 - task_id (str): 任务ID
@@ -141,12 +141,12 @@ class CronManager(Client):
                 - status (str): 任务状态，初始为"PENDING"
                 - message (str): 触发消息
             success (bool): 操作是否成功
-        
+
         异常情况：
             - 任务不存在：返回错误信息"任务不存在"
             - 任务触发失败：返回错误信息"任务触发失败"
         """
-        data, success = self._request(
+        data, success = await self._request(
             "POST",
             f"/api/cron/tasks/{task_name}/trigger",
             json={"args": args, "kwargs": kwargs},

pixelarraythirdparty-1.0.9/pixelarraythirdparty/filestorage/__init__.py

@@ -0,0 +1,6 @@
+from pixelarraythirdparty.filestorage.filestorage import (
+    FileStorageManagerAsync,
+)
+
+__all__ = ["FileStorageManagerAsync"]
+