pixelarraythirdparty 1.0.9__tar.gz → 1.1.1__tar.gz

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

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

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

@@ -11,7 +11,7 @@ PixelArray 第三方微服务客户端
 - unified_login: 统一登录模块
 """
 
-__version__ = "1.0.9"
+__version__ = "1.1.1"
 __author__ = "Lu qi"
 __email__ = "qi.lu@pixelarrayai.com"
 
@@ -21,6 +21,4 @@ __all__ = [
     "cron",
     "user",
     "order",
-    "filestorage",
-    "unified_login",
 ]

{pixelarraythirdparty-1.0.9 → pixelarraythirdparty-1.1.1}/pixelarraythirdparty/client.py

@@ -34,3 +34,22 @@ class AsyncClient:
                         # 如果不是JSON响应，返回空
                         pass
                 return {}, False
+
+    async def _request_raw(
+        self, method: str, url: str, **kwargs
+    ) -> Tuple[int, Dict[str, Any]]:
+        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:
+                try:
+                    data = await resp.json()
+                except:
+                    data = {}
+                return resp.status, data

pixelarraythirdparty-1.1.1/pixelarraythirdparty/cron/cron.py

@@ -0,0 +1,113 @@
+from pixelarraythirdparty.client import AsyncClient
+
+
+class CronManagerAsync(AsyncClient):
+    async def get_cron_status(self):
+        """
+        description:
+            获取Cron服务状态，包括已注册的任务、工作节点状态等。
+        return:
+            data(dict): Cron服务状态信息
+                - registered_tasks(dict): 已注册的任务列表，按工作节点分组
+                - worker_stats(dict): 工作节点统计信息
+                - scheduled_task_count(int): 定时任务数量
+                - timestamp(str): 状态获取时间
+            success(bool): 操作是否成功
+        """
+        data, success = await self._request("GET", "/api/cron/status")
+        if not success:
+            return {}, False
+        return data, True
+
+    async def get_cron_tasks(self):
+        """
+        description:
+            获取所有已注册的Cron任务列表。
+        return:
+            data(dict): 任务列表信息
+                - tasks(list): 已注册的任务名称列表
+                - count(int): 任务数量
+                - timestamp(str): 获取时间
+            success(bool): 操作是否成功
+        """
+        data, success = await self._request("GET", "/api/cron/tasks")
+        if not success:
+            return {}, False
+        return data, True
+
+    async def get_cron_tasks_scheduled(self):
+        """
+        description:
+            获取所有配置的定时任务列表，包括任务详情、执行时间、状态等。
+        return:
+            data(dict): 定时任务列表信息
+                - tasks(list): 定时任务列表
+                    - id(str): 任务ID
+                    - name(str): 任务名称
+                    - description(str): 任务描述
+                    - schedule(str): 执行时间
+                    - enabled(bool): 是否启用
+                    - task_name(str): 任务函数名
+                    - module_name(str): 模块名
+                    - function_name(str): 函数名
+                    - file_path(str): 文件路径
+                    - parameters(list): 参数列表
+                    - task_config(dict): 任务配置
+                    - registration_info(dict): 注册信息
+                - count(int): 任务数量
+                - timestamp(str): 获取时间
+            success(bool): 操作是否成功
+        """
+        data, success = await self._request("GET", "/api/cron/tasks/scheduled")
+        if not success:
+            return {}, False
+        return data, True
+
+    async def get_cron_tasks_detail(self, task_name: str):
+        """
+        description:
+            根据任务名称获取指定任务的详细信息。
+        parameters:
+            task_name(str): 任务名称
+        return:
+            data(dict): 任务详细信息
+                - task_name(str): 任务名称
+                - module_name(str): 模块名
+                - function_name(str): 函数名
+                - file_path(str): 文件路径
+                - description(str): 任务描述
+                - parameters(list): 参数列表
+                - task_config(dict): 任务配置
+                - registration_info(dict): 注册信息
+                - timestamp(str): 获取时间
+            success(bool): 操作是否成功
+        """
+        data, success = await self._request("GET", f"/api/cron/tasks/{task_name}")
+        if not success:
+            return {}, False
+        return data, True
+
+    async def trigger_cron_task(self, task_name: str, args: list, kwargs: dict):
+        """
+        description:
+            手动触发指定任务的执行，支持传递参数。
+        parameters:
+            task_name(str): 任务名称
+            args(list): 任务参数列表
+            kwargs(dict): 任务关键字参数
+        return:
+            data(dict): 任务触发信息
+                - task_id(str): 任务ID
+                - task_name(str): 任务名称
+                - status(str): 任务状态，初始为"PENDING"
+                - message(str): 触发消息
+            success(bool): 操作是否成功
+        """
+        data, success = await self._request(
+            "POST",
+            f"/api/cron/tasks/{task_name}/trigger",
+            json={"args": args, "kwargs": kwargs},
+        )
+        if not success:
+            return {}, False
+        return data, True

pixelarraythirdparty-1.1.1/pixelarraythirdparty/order/order.py

@@ -0,0 +1,261 @@
+from pixelarraythirdparty.client import AsyncClient
+
+
+class OrderManagerAsync(AsyncClient):
+    async def create_order(
+        self,
+        product_id: str,
+        body: str = None,
+        remark: str = None,
+        payment_channel: str = "WECHAT",
+    ):
+        """
+        description:
+            根据产品ID创建新的订单，系统会自动生成订单号，获取产品价格信息，并创建对应的支付订单。
+        parameters:
+            product_id(str): 产品ID
+            body(str): 商品描述
+            remark(str): 订单备注
+            payment_channel(str): 支付渠道，可选值："WECHAT"（微信支付）、"ALIPAY"（支付宝支付）
+        return:
+            data(dict): 订单信息
+                - id(int): 订单ID
+                - out_trade_no(str): 商户订单号，格式为"ORD_时间戳_随机字符串"
+                - payment_status(str): 支付状态，初始为"PENDING"（待支付）
+                - payment_channel(str): 支付渠道
+                - product_id(str): 产品ID
+                - amount(str): 订单金额（元），格式为"99.00"
+                - total_fee(int): 订单金额（分），用于支付接口
+                - body(str): 商品描述
+                - remark(str): 订单备注
+                - created_at(str): 订单创建时间
+                - updated_at(str): 订单更新时间
+            success(bool): 操作是否成功
+        """
+        data = {
+            "product_id": product_id,
+            "body": body,
+            "remark": remark,
+            "payment_channel": payment_channel,
+        }
+        data, success = await self._request("POST", "/api/orders/create", json=data)
+        if not success:
+            return {}, False
+        return data, True
+
+    async def list_order(
+        self,
+        page: int = 1,
+        page_size: int = 10,
+        payment_status: str = None,
+        out_trade_no: str = None,
+    ):
+        """
+        description:
+            分页查询订单列表，支持按支付状态和订单号进行筛选。
+        parameters:
+            page(int): 页码
+            page_size(int): 每页数量
+            payment_status(str): 支付状态，可选值："PENDING"（待支付）、"PAID"（已支付）、"REFUNDED"（已退款）、"CANCELLED"（已取消）
+            out_trade_no(str): 订单号，支持模糊匹配
+        return:
+            data(dict): 订单列表信息
+                - orders(list): 订单列表
+                    - id(int): 订单ID
+                    - out_trade_no(str): 商户订单号
+                    - payment_status(str): 支付状态
+                    - payment_channel(str): 支付渠道
+                    - amount(str): 订单金额（元）
+                    - total_fee(int): 订单金额（分）
+                    - created_at(str): 订单创建时间
+                    - updated_at(str): 订单更新时间
+                - total(int): 总订单数量
+                - page(int): 当前页码
+                - page_size(int): 每页数量
+            success(bool): 操作是否成功
+        """
+        params = {
+            "page": page,
+            "page_size": page_size,
+        }
+        # 只添加非None的参数
+        if payment_status is not None:
+            params["payment_status"] = payment_status
+        if out_trade_no is not None:
+            params["out_trade_no"] = out_trade_no
+        data, success = await self._request("GET", "/api/orders/list", params=params)
+        if not success:
+            return {}, False
+        return data, True
+
+    async def get_order_detail(self, out_trade_no: str):
+        """
+        description:
+            根据订单号获取订单的详细信息，包括支付状态、交易信息等。
+        parameters:
+            out_trade_no(str): 商户订单号
+        return:
+            data(dict): 订单详细信息
+                - id(int): 订单ID
+                - out_trade_no(str): 商户订单号
+                - payment_status(str): 支付状态
+                - product_id(str): 产品ID
+                - amount(str): 订单金额（元）
+                - total_fee(int): 订单金额（分）
+                - body(str): 商品描述
+                - transaction_id(str): 微信交易号（支付成功后才有）
+                - openid(str): 用户openid（支付成功后才有）
+                - trade_type(str): 交易类型（支付成功后才有）
+                - bank_type(str): 银行类型（支付成功后才有）
+                - fee_type(str): 货币类型，默认为"CNY"
+                - is_subscribe(str): 是否关注公众号（支付成功后才有）
+                - time_end(str): 支付完成时间（支付成功后才有）
+                - created_at(str): 订单创建时间
+                - updated_at(str): 订单更新时间
+                - paid_at(str): 支付时间（支付成功后才有）
+                - remark(str): 订单备注
+            success(bool): 操作是否成功
+        """
+        data, success = await self._request("GET", f"/api/orders/{out_trade_no}")
+        if not success:
+            return {}, False
+        return data, True
+
+    async def update_order_status(self, out_trade_no: str, payment_status: str):
+        """
+        description:
+            更新指定订单的支付状态，仅支持状态修改，其他字段不可修改。
+        parameters:
+            out_trade_no(str): 商户订单号
+            payment_status(str): 支付状态，可选值："PENDING"（待支付）、"PAID"（已支付）、"REFUNDED"（已退款）、"CANCELLED"（已取消）
+        return:
+            data(dict): 更新后的订单信息
+                - id(int): 订单ID
+                - out_trade_no(str): 商户订单号
+                - payment_status(str): 更新后的支付状态
+                - transaction_id(str): 微信交易号（如果已支付）
+                - openid(str): 用户openid（如果已支付）
+                - trade_type(str): 交易类型（如果已支付）
+                - bank_type(str): 银行类型（如果已支付）
+                - fee_type(str): 货币类型
+                - is_subscribe(str): 是否关注公众号（如果已支付）
+                - time_end(str): 支付完成时间（如果已支付）
+                - paid_at(str): 支付时间（如果已支付）
+                - updated_at(str): 订单更新时间
+                - remark(str): 订单备注
+            success(bool): 操作是否成功
+        """
+        data = {"payment_status": payment_status}
+        data, success = await self._request(
+            "PUT", f"/api/orders/{out_trade_no}/status", json=data
+        )
+        if not success:
+            return {}, False
+        return data, True
+
+    async def delete_order(self, out_trade_no: str):
+        """
+        description:
+            根据订单号删除指定的订单记录。
+        parameters:
+            out_trade_no(str): 商户订单号
+        return:
+            data(None): 删除成功时返回None
+            success(bool): 操作是否成功
+        """
+        data, success = await self._request("DELETE", f"/api/orders/{out_trade_no}")
+        if not success:
+            return {}, False
+        return data, True
+
+    async def get_order_stats(self):
+        """
+        description:
+            获取订单的统计汇总信息，包括总订单数、各状态订单数量、总金额等。
+        return:
+            data(dict): 订单统计信息
+                - total_orders(int): 总订单数量
+                - pending_orders(int): 待支付订单数量
+                - paid_orders(int): 已支付订单数量
+                - refunded_orders(int): 已退款订单数量
+                - total_amount(float): 总订单金额（元）
+                - total_fee(int): 总订单金额（分）
+            success(bool): 操作是否成功
+        """
+        data, success = await self._request("GET", "/api/orders/stats/summary")
+        if not success:
+            return {}, False
+        return data, True
+
+    async def generate_qr_code(self, out_trade_no: str):
+        """
+        description:
+            为指定订单生成支付二维码，支持微信支付和支付宝。二维码会自动上传到OSS并返回访问URL。
+            如果不指定payment_channel，会自动从订单详情中获取支付渠道。
+        parameters:
+            out_trade_no(str): 商户订单号
+        return:
+            data(dict): 二维码信息
+                - qr_code_url(str): 二维码图片URL，可直接用于显示
+                - out_trade_no(str): 商户订单号
+            success(bool): 操作是否成功
+        """
+        order_detail, success = await self.get_order_detail(out_trade_no)
+        print(order_detail)
+        if not success:
+            return {}, False
+
+        if order_detail.get("payment_channel") == "WECHAT":
+            url = "/api/orders/wx_pay/generate_qr_code"
+            request_data = {
+                "out_trade_no": out_trade_no,
+            }
+        elif order_detail.get("payment_channel") == "ALIPAY":
+            url = "/api/orders/ali_pay/generate_qr_code"
+            # 支付宝需要total_fee和subject，从已获取的订单详情中提取
+            request_data = {
+                "out_trade_no": out_trade_no,
+                "total_fee": order_detail.get("total_fee"),
+                "subject": order_detail.get("body", ""),
+            }
+        else:
+            raise ValueError("Invalid payment channel")
+        data, success = await self._request("POST", url, json=request_data)
+        if not success:
+            return {}, False
+        return data, True
+
+    async def refund_order(self, out_trade_no: str):
+        """
+        description:
+            为指定订单申请退款，支持微信支付和支付宝。退款申请提交后，系统会处理退款并更新订单状态。
+        parameters:
+            out_trade_no(str): 商户订单号
+        return:
+            data(dict): 退款信息
+                - out_refund_no(str): 退款单号，格式为"REFUND_订单号_时间戳"
+                - out_trade_no(str): 商户订单号
+                - total_fee(int): 退款金额（分，微信支付）
+                - refund_amount(float): 退款金额（元，支付宝）
+            success(bool): 操作是否成功
+        """
+        order_detail, success = await self.get_order_detail(out_trade_no)
+        if not success:
+            return {}, False
+
+        if order_detail.get("payment_channel") == "WECHAT":
+            url = "/api/orders/wx_pay/refund"
+            request_data = {"out_trade_no": out_trade_no}
+        elif order_detail.get("payment_channel") == "ALIPAY":
+            url = "/api/orders/ali_pay/refund"
+            request_data = {
+                "out_trade_no": out_trade_no,
+                "refund_amount": order_detail.get("total_fee") / 100.0,
+                "refund_reason": "用户退款",
+            }
+        else:
+            raise ValueError("Invalid payment channel")
+        data, success = await self._request("POST", url, json=request_data)
+        if not success:
+            return {}, False
+        return data, True

pixelarraythirdparty-1.1.1/pixelarraythirdparty/product/product.py

@@ -0,0 +1,205 @@
+from pixelarraythirdparty.client import AsyncClient
+
+
+class ProductManagerAsync(AsyncClient):
+    async def create_product(
+        self,
+        name: str,
+        description: str,
+        price: float,
+        category: str,
+        status: str,
+        is_subscription: bool,
+        subscription_period: str,
+        features: str,
+        sort_order: int,
+    ):
+        """
+        description:
+            创建新的产品，支持订阅产品和一次性产品，产品信息包括名称、描述、价格、分类等。
+        parameters:
+            name(str): 产品名称
+            description(str): 产品描述
+            price(float): 产品价格（元）
+            category(str): 产品分类
+            status(str): 产品状态，可选值："ACTIVE"（激活）、"INACTIVE"（停用）
+            is_subscription(bool): 是否为订阅产品
+            subscription_period(str): 订阅周期，可选值："MONTHLY"（月付）、"YEARLY"（年付）
+            features(str): 产品特性，JSON格式字符串
+            sort_order(int): 排序权重
+        return:
+            data(dict): 产品信息
+                - id(int): 产品ID
+                - name(str): 产品名称
+                - description(str): 产品描述
+                - price(float): 产品价格（元）
+                - category(str): 产品分类
+                - status(str): 产品状态
+                - is_subscription(bool): 是否为订阅产品
+                - subscription_period(str): 订阅周期
+                - features(str): 产品特性（JSON格式）
+                - sort_order(int): 排序权重
+                - created_at(str): 产品创建时间
+                - updated_at(str): 产品更新时间
+            success(bool): 操作是否成功
+        """
+        data = {
+            "name": name,
+            "description": description,
+            "price": price,
+            "category": category,
+            "status": status,
+            "is_subscription": is_subscription,
+            "subscription_period": subscription_period,
+            "features": features,
+            "sort_order": sort_order,
+        }
+        data, success = await self._request("POST", "/api/products/create", json=data)
+        if not success:
+            return {}, False
+        return data, True
+
+    async def list_product(
+        self,
+        page: int = 1,
+        page_size: int = 10,
+        status: str = None,
+        category: str = None,
+        name: str = None,
+    ):
+        """
+        description:
+            分页查询产品列表，支持按状态、分类和名称进行筛选。
+        parameters:
+            page(int): 页码
+            page_size(int): 每页数量
+            status(str): 产品状态筛选，可选值："ACTIVE"（激活）、"INACTIVE"（停用）
+            category(str): 产品分类筛选
+            name(str): 产品名称搜索，支持模糊匹配
+        return:
+            data(dict): 产品列表信息
+                - products(list): 产品列表
+                - total(int): 总产品数量
+                - page(int): 当前页码
+                - page_size(int): 每页数量
+            success(bool): 操作是否成功
+        """
+        params = {
+            "page": page,
+            "page_size": page_size,
+        }
+        if status is not None:
+            params["status"] = status
+        if category is not None:
+            params["category"] = category
+        if name is not None:
+            params["name"] = name
+        data, success = await self._request("GET", "/api/products/list", params=params)
+        if not success:
+            return {}, False
+        return data, True
+
+    async def get_product_detail(self, product_id: str):
+        """
+        description:
+            根据产品ID获取产品的详细信息。
+        parameters:
+            product_id(str): 产品ID
+        return:
+            data(dict): 产品详细信息
+            success(bool): 操作是否成功
+        """
+        data, success = await self._request("GET", f"/api/products/{product_id}")
+        if not success:
+            return {}, False
+        return data, True
+
+    async def update_product(
+        self,
+        product_id: str,
+        name: str,
+        description: str,
+        price: float,
+        category: str,
+        status: str,
+        is_subscription: bool,
+        subscription_period: str,
+        features: str,
+        sort_order: int,
+    ):
+        """
+        description:
+            更新指定产品的信息，包括名称、描述、价格、状态等。
+        parameters:
+            product_id(str): 产品ID
+            name(str): 产品名称
+            description(str): 产品描述
+            price(float): 产品价格（元）
+            category(str): 产品分类
+            status(str): 产品状态，可选值："ACTIVE"（激活）、"INACTIVE"（停用）
+            is_subscription(bool): 是否为订阅产品
+            subscription_period(str): 订阅周期，可选值："MONTHLY"（月付）、"YEARLY"（年付）
+            features(str): 产品特性，JSON格式字符串
+            sort_order(int): 排序权重
+        return:
+            data(dict): 更新后的产品信息
+                - id(int): 产品ID
+                - name(str): 产品名称
+                - description(str): 产品描述
+                - price(float): 产品价格（元）
+                - category(str): 产品分类
+                - status(str): 产品状态
+                - is_subscription(bool): 是否为订阅产品
+                - subscription_period(str): 订阅周期
+                - features(str): 产品特性（JSON格式）
+                - sort_order(int): 排序权重
+                - created_at(str): 产品创建时间
+                - updated_at(str): 产品更新时间
+            success(bool): 操作是否成功
+        """
+        data = {
+            "name": name,
+            "description": description,
+            "price": price,
+            "category": category,
+            "status": status,
+            "is_subscription": is_subscription,
+            "subscription_period": subscription_period,
+            "features": features,
+            "sort_order": sort_order,
+        }
+        data, success = await self._request(
+            "PUT", f"/api/products/{product_id}", json=data
+        )
+        if not success:
+            return {}, False
+        return data, True
+
+    async def delete_product(self, product_id: str):
+        """
+        description:
+            根据产品ID删除指定的产品记录。
+        parameters:
+            product_id(str): 产品ID
+        return:
+            data(None): 删除成功时返回None
+            success(bool): 操作是否成功
+        """
+        data, success = await self._request("DELETE", f"/api/products/{product_id}")
+        if not success:
+            return {}, False
+        return data, True
+
+    async def get_product_categories(self):
+        """
+        description:
+            获取所有可用的产品分类列表。
+        return:
+            data(dict): 产品分类信息
+                - categories(list): 产品分类列表，如["subscription", "service", "addon"]
+            success(bool): 操作是否成功
+        """
+        data, success = await self._request("GET", "/api/products/categories/list")
+        if not success:
+            return {}, False
+        return data, True

pixelarraythirdparty-1.1.1/pixelarraythirdparty/unified_login/__init__.py

@@ -0,0 +1,4 @@
+from .unified_login import GoogleLogin, WechatLogin, GitHubLogin
+
+__all__ = ["GoogleLogin", "WechatLogin", "GitHubLogin"]
+