pixelarraythirdparty 1.0.3__py3-none-any.whl → 1.0.4__py3-none-any.whl

pixelarraythirdparty/__init__.py

@@ -10,7 +10,7 @@ PixelArray 第三方微服务客户端
 - user: 用户管理模块
 """
 
-__version__ = "1.0.3"
+__version__ = "1.0.4"
 __author__ = "Lu qi"
 __email__ = "qi.lu@pixelarrayai.com"
 

pixelarraythirdparty/client.py

@@ -1,9 +1,12 @@
 import requests
+import aiohttp
 from typing import Dict, Any, Optional, Tuple
+import asyncio
 
 
 class Client:
     """基础认证类，提供公共的 API key 认证和请求方法"""
+
     def __init__(self, api_key: str):
         self.base_url = "https://thirdparty.pixelarrayai.com"
         self.api_key = api_key
@@ -20,3 +23,25 @@ class Client:
         if resp.status_code == 200 and resp.json().get("success") is True:
             return resp.json().get("data", {}), True
         return {}, False
+
+
+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]:
+        async with aiohttp.ClientSession() as session:
+            req_method = getattr(session, method.lower())
+            async with req_method(
+                f"{self.base_url}{url}", headers=self.headers, **kwargs
+            ) as resp:
+                if resp.status == 200:
+                    result = await resp.json()
+                    if result.get("success") is True:
+                        return result.get("data", {}), True
+                return {}, False

pixelarraythirdparty/order/order.py

@@ -1,4 +1,4 @@
-from pixelarraythirdparty.client import Client
+from pixelarraythirdparty.client import Client, AsyncClient
 
 
 class OrderManager(Client):
@@ -108,3 +108,98 @@ class OrderManager(Client):
         if not success:
             return {}, False
         return data, True
+
+
+class OrderManagerAsync(AsyncClient):
+    async def create_order(
+        self,
+        product_id: str,
+        body: str = None,
+        remark: str = None,
+        payment_channel: str = "WECHAT",
+    ):
+        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,
+    ):
+        params = {
+            "page": page,
+            "page_size": page_size,
+            "payment_status": payment_status,
+            "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):
+        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):
+        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):
+        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):
+        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, payment_channel: str = "WECHAT"
+    ):
+        if payment_channel == "WECHAT":
+            url = "/api/orders/wx_pay/generate_qr_code"
+        elif payment_channel == "ALIPAY":
+            url = "/api/orders/ali_pay/generate_qr_code"
+        else:
+            raise ValueError("Invalid payment channel")
+        data, success = await self._request(
+            "POST", url, json={"out_trade_no": out_trade_no}
+        )
+        if not success:
+            return {}, False
+        return data, True
+
+    async def refund_order(self, out_trade_no: str, payment_channel: str = "WECHAT"):
+        if payment_channel == "WECHAT":
+            url = "/api/orders/wx_pay/refund"
+        elif payment_channel == "ALIPAY":
+            url = "/api/orders/ali_pay/refund"
+        else:
+            raise ValueError("Invalid payment channel")
+        data, success = await self._request(
+            "POST", url, json={"out_trade_no": out_trade_no}
+        )
+        if not success:
+            return {}, False
+        return data, True

{pixelarraythirdparty-1.0.3.dist-info → pixelarraythirdparty-1.0.4.dist-info}/METADATA

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

{pixelarraythirdparty-1.0.3.dist-info → pixelarraythirdparty-1.0.4.dist-info}/RECORD

@@ -1,15 +1,15 @@
-pixelarraythirdparty/__init__.py,sha256=TYuj96ZwZ4VXhB9Xn4KoEWpoJVsChUjzoWSRZONmC-U,458
-pixelarraythirdparty/client.py,sha256=dJlIrMlr1M4AGHvaLw191uaPnQbQ12_BGx8DTDTPkZQ,802
+pixelarraythirdparty/__init__.py,sha256=EShklfQHjt1FKP_lKD4GTfwFXZCueZt31qiGnQN6oh8,458
+pixelarraythirdparty/client.py,sha256=r63EDeyPhf3Y1sORW24tDSIqo2c8E3pwbik7r0Xy5zo,1678
 pixelarraythirdparty/celery/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pixelarraythirdparty/celery/celery.py,sha256=Oaa7pBztL-FM0X5NSCjjfZP80AUsnnme5Z3RxXIJsZg,1178
 pixelarraythirdparty/order/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pixelarraythirdparty/order/order.py,sha256=kmJNAvfiXxhjOhRbD3JTl1saE8aetMZ0TFR9NjPsaIM,3208
+pixelarraythirdparty/order/order.py,sha256=77UYvDteQdkVvEVG588ehmN5C9bpGAYkJsLpZqCH1Mg,6336
 pixelarraythirdparty/product/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pixelarraythirdparty/product/product.py,sha256=eHU0VKal1QpJs6F4_VxLoBYeeqQ19elEogH2Qpw7IDo,3029
 pixelarraythirdparty/user/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pixelarraythirdparty/user/user.py,sha256=tbcFsrNEdWz4op7V_zJ5o8neBuI4ogl4dgHS-uSIAX0,2155
-pixelarraythirdparty-1.0.3.dist-info/licenses/LICENSE,sha256=O-g1dUr0U50rSIvmWE9toiVkSgFpVt72_MHITbWvAqA,1067
-pixelarraythirdparty-1.0.3.dist-info/METADATA,sha256=_BcKEA0W8joCfUZcvDqyKBkBXKGyWUwIDVI16boT7s4,993
-pixelarraythirdparty-1.0.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pixelarraythirdparty-1.0.3.dist-info/top_level.txt,sha256=dzG2Ut8j7noUqj_0ZQjcIDAeHYCh_9WtlxjAxtoyufo,21
-pixelarraythirdparty-1.0.3.dist-info/RECORD,,
+pixelarraythirdparty-1.0.4.dist-info/licenses/LICENSE,sha256=O-g1dUr0U50rSIvmWE9toiVkSgFpVt72_MHITbWvAqA,1067
+pixelarraythirdparty-1.0.4.dist-info/METADATA,sha256=nefI-ENI5vYKvBvGGNFc5aakZh7ptR12S6a7Yk7DpSY,993
+pixelarraythirdparty-1.0.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pixelarraythirdparty-1.0.4.dist-info/top_level.txt,sha256=dzG2Ut8j7noUqj_0ZQjcIDAeHYCh_9WtlxjAxtoyufo,21
+pixelarraythirdparty-1.0.4.dist-info/RECORD,,