pixelarraythirdparty 1.0.2__tar.gz → 1.0.4__tar.gz

{pixelarraythirdparty-1.0.2 → pixelarraythirdparty-1.0.4}/PKG-INFO

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

{pixelarraythirdparty-1.0.2 → pixelarraythirdparty-1.0.4}/pixelarraythirdparty/__init__.py

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

pixelarraythirdparty-1.0.4/pixelarraythirdparty/client.py

@@ -0,0 +1,47 @@
+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
+        self.headers = {
+            "Content-Type": "application/json",
+            "X-API-Key": self.api_key,
+        }
+
+    def _request(self, method: str, url: str, **kwargs) -> Tuple[Dict[str, Any], bool]:
+        """统一的请求方法"""
+        resp = requests.request(
+            method, f"{self.base_url}{url}", headers=self.headers, **kwargs
+        )
+        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-1.0.4/pixelarraythirdparty/order/order.py

@@ -0,0 +1,205 @@
+from pixelarraythirdparty.client import Client, AsyncClient
+
+
+class OrderManager(Client):
+    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 = self._request("POST", "/api/orders/create", json=data)
+        if not success:
+            return {}, False
+        return data, True
+
+    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 = self._request("GET", "/api/orders/list", params=params)
+        if not success:
+            return {}, False
+        return data, True
+
+    def get_order_detail(self, out_trade_no: str):
+        data, success = self._request("GET", f"/api/orders/{out_trade_no}")
+        if not success:
+            return {}, False
+        return data, True
+
+    def update_order_status(
+        self,
+        out_trade_no: str,
+        payment_status: str,
+    ):
+        data = {
+            "payment_status": payment_status,
+        }
+        data, success = self._request(
+            "PUT", f"/api/orders/{out_trade_no}/status", json=data
+        )
+        if not success:
+            return {}, False
+        return data, True
+
+    def delete_order(self, out_trade_no: str):
+        data, success = self._request("DELETE", f"/api/orders/{out_trade_no}")
+        if not success:
+            return {}, False
+        return data, True
+
+    def get_order_stats(self):
+        data, success = self._request("GET", "/api/orders/stats/summary")
+        if not success:
+            return {}, False
+        return data, True
+
+    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 = self._request(
+            "POST",
+            url,
+            json={
+                "out_trade_no": out_trade_no,
+            },
+        )
+        if not success:
+            return {}, False
+        return data, True
+
+    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 = self._request(
+            "POST",
+            url,
+            json={"out_trade_no": out_trade_no},
+        )
+        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.2 → pixelarraythirdparty-1.0.4}/pixelarraythirdparty.egg-info/PKG-INFO

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

{pixelarraythirdparty-1.0.2 → pixelarraythirdparty-1.0.4}/pixelarraythirdparty.egg-info/SOURCES.txt

@@ -12,7 +12,6 @@ pixelarraythirdparty/celery/__init__.py
 pixelarraythirdparty/celery/celery.py
 pixelarraythirdparty/order/__init__.py
 pixelarraythirdparty/order/order.py
-pixelarraythirdparty/order/pay.py
 pixelarraythirdparty/product/__init__.py
 pixelarraythirdparty/product/product.py
 pixelarraythirdparty/user/__init__.py

{pixelarraythirdparty-1.0.2 → pixelarraythirdparty-1.0.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "pixelarraythirdparty"
-version = "1.0.2"
+version = "1.0.4"
 authors = [
     {name = "Lu qi", email = "qi.lu@pixelarrayai.com"},
 ]

pixelarraythirdparty-1.0.2/pixelarraythirdparty/client.py

@@ -1,22 +0,0 @@
-import requests
-from typing import Dict, Any, Optional, Tuple
-
-
-class Client:
-    """基础认证类，提供公共的 API key 认证和请求方法"""
-    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,
-        }
-
-    def _request(self, method: str, url: str, **kwargs) -> Tuple[Dict[str, Any], bool]:
-        """统一的请求方法"""
-        resp = requests.request(
-            method, f"{self.base_url}{url}", headers=self.headers, **kwargs
-        )
-        if resp.status_code == 200 and resp.json().get("success") is True:
-            return resp.json().get("data", {}), True
-        return {}, False

pixelarraythirdparty-1.0.2/pixelarraythirdparty/order/order.py

@@ -1,92 +0,0 @@
-from pixelarraythirdparty.client import Client
-
-
-class OrderManager(Client):
-    def create_order(
-        self,
-        product_name: str,
-        product_id: str,
-        amount: float,
-        body: str,
-        remark: str,
-        payment_channel: str,
-    ):
-        data = {
-            "product_name": product_name,
-            "product_id": product_id,
-            "amount": amount,
-            "body": body,
-            "remark": remark,
-            "payment_channel": payment_channel,
-        }
-        data, success = self._request("POST", "/api/orders/create", json=data)
-        if not success:
-            return {}, False
-        return data, True
-
-    def list_order(
-        self,
-        page: int = 1,
-        page_size: int = 10,
-        payment_status: str = None,
-        order_no: str = None,
-    ):
-        params = {
-            "page": page,
-            "page_size": page_size,
-            "payment_status": payment_status,
-            "order_no": order_no,
-        }
-        data, success = self._request("GET", "/api/orders/list", params=params)
-        if not success:
-            return {}, False
-        return data, True
-
-    def get_order_detail(self, order_no: str):
-        data, success = self._request("GET", f"/api/orders/{order_no}")
-        if not success:
-            return {}, False
-        return data, True
-
-    def update_order(
-        self,
-        order_no: str,
-        payment_status: str,
-        wx_order_no: str,
-        transaction_id: str,
-        openid: str,
-        trade_type: str,
-        bank_type: str,
-        fee_type: str,
-        is_subscribe: str,
-        time_end: str,
-        remark: str,
-    ):
-        data = {
-            "payment_status": payment_status,
-            "wx_order_no": wx_order_no,
-            "transaction_id": transaction_id,
-            "openid": openid,
-            "trade_type": trade_type,
-            "bank_type": bank_type,
-            "fee_type": fee_type,
-            "is_subscribe": is_subscribe,
-            "time_end": time_end,
-            "remark": remark,
-        }
-        data, success = self._request("PUT", f"/api/orders/{order_no}", json=data)
-        if not success:
-            return {}, False
-        return data, True
-
-    def delete_order(self, order_no: str):
-        data, success = self._request("DELETE", f"/api/orders/{order_no}")
-        if not success:
-            return {}, False
-        return data, True
-
-    def get_order_stats(self):
-        data, success = self._request("GET", "/api/orders/stats/summary")
-        if not success:
-            return {}, False
-        return data, True

pixelarraythirdparty-1.0.2/pixelarraythirdparty/order/pay.py

@@ -1,44 +0,0 @@
-from pixelarraythirdparty.client import Client
-
-
-class WeChatPayManager(Client):
-    def generate_qr_code(
-        self,
-        out_trade_no: str,
-        total_fee: int,
-        body: str,
-        product_name: str,
-        product_id: str,
-    ):
-        data, success = self._request(
-            "POST",
-            "/api/wx_pay/generate_qr_code",
-            json={
-                "out_trade_no": out_trade_no,
-                "total_fee": total_fee,
-                "body": body,
-                "product_name": product_name,
-                "product_id": product_id,
-            },
-        )
-        if not success:
-            return {}, False
-        return data, True
-
-    def query_order(self, out_trade_no: str):
-        data, success = self._request(
-            "POST", "/api/wx_pay/query_order", json={"out_trade_no": out_trade_no}
-        )
-        if not success:
-            return {}, False
-        return data, True
-
-    def refund(self, out_trade_no: str, total_fee: int):
-        data, success = self._request(
-            "POST",
-            "/api/wx_pay/refund",
-            json={"out_trade_no": out_trade_no, "total_fee": total_fee},
-        )
-        if not success:
-            return {}, False
-        return data, True