pixelarraythirdparty 1.0.0__py3-none-any.whl → 1.0.2__py3-none-any.whl

pixelarraythirdparty/__init__.py

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

pixelarraythirdparty/celery/celery.py

@@ -1,38 +1,37 @@
-import requests
+from pixelarraythirdparty.client import Client
 
 
-class CeleryManager:
-    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, url, **kwargs):
-        resp = requests.request(
-            method, f"{self.base_url}{url}", headers=self.headers, **kwargs
-        )
-        if resp.status_code == 200:
-            return resp.json().get("data", {})
-        return {}
-
+class CeleryManager(Client):
     def get_celery_status(self):
-        return self._request("GET", "/api/celery/status")
+        data, success = self._request("GET", "/api/celery/status")
+        if not success:
+            return {}, False
+        return data, True
 
     def get_celery_tasks(self):
-        return self._request("GET", "/api/celery/tasks")
+        data, success = self._request("GET", "/api/celery/tasks")
+        if not success:
+            return {}, False
+        return data, True
 
     def get_celery_tasks_scheduled(self):
-        return self._request("GET", "/api/celery/tasks/scheduled")
+        data, success = self._request("GET", "/api/celery/tasks/scheduled")
+        if not success:
+            return {}, False
+        return data, True
 
     def get_celery_tasks_detail(self, task_name: str):
-        return self._request("GET", f"/api/celery/tasks/{task_name}")
+        data, success = self._request("GET", f"/api/celery/tasks/{task_name}")
+        if not success:
+            return {}, False
+        return data, True
 
     def trigger_celery_task(self, task_name: str, args: list, kwargs: dict):
-        return self._request(
+        data, success = self._request(
             "POST",
             f"/api/celery/tasks/{task_name}/trigger",
             json={"args": args, "kwargs": kwargs},
         )
+        if not success:
+            return {}, False
+        return data, True

pixelarraythirdparty/client.py

@@ -0,0 +1,22 @@
+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/order/order.py

@@ -1,23 +1,7 @@
-import requests
+from pixelarraythirdparty.client import Client
 
 
-class OrderManager:
-    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, url, **kwargs):
-        resp = requests.request(
-            method, f"{self.base_url}{url}", headers=self.headers, **kwargs
-        )
-        if resp.status_code == 200:
-            return resp.json().get("data", {})
-        return {}
-
+class OrderManager(Client):
     def create_order(
         self,
         product_name: str,
@@ -35,7 +19,10 @@ class OrderManager:
             "remark": remark,
             "payment_channel": payment_channel,
         }
-        return self._request("POST", "/api/orders/create", json=data)
+        data, success = self._request("POST", "/api/orders/create", json=data)
+        if not success:
+            return {}, False
+        return data, True
 
     def list_order(
         self,
@@ -50,10 +37,16 @@ class OrderManager:
             "payment_status": payment_status,
             "order_no": order_no,
         }
-        return self._request("GET", "/api/orders/list", params=params)
+        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):
-        return self._request("GET", f"/api/orders/{order_no}")
+        data, success = self._request("GET", f"/api/orders/{order_no}")
+        if not success:
+            return {}, False
+        return data, True
 
     def update_order(
         self,
@@ -81,10 +74,19 @@ class OrderManager:
             "time_end": time_end,
             "remark": remark,
         }
-        return self._request("PUT", f"/api/orders/{order_no}", json=data)
+        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):
-        return self._request("DELETE", f"/api/orders/{order_no}")
+        data, success = self._request("DELETE", f"/api/orders/{order_no}")
+        if not success:
+            return {}, False
+        return data, True
 
     def get_order_stats(self):
-        return self._request("GET", "/api/orders/stats/summary")
+        data, success = self._request("GET", "/api/orders/stats/summary")
+        if not success:
+            return {}, False
+        return data, True

pixelarraythirdparty/order/pay.py

@@ -1,23 +1,7 @@
-import requests
+from pixelarraythirdparty.client import Client
 
 
-class WeChatPayManager:
-    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, url, **kwargs):
-        resp = requests.request(
-            method, f"{self.base_url}{url}", headers=self.headers, **kwargs
-        )
-        if resp.status_code == 200:
-            return resp.json().get("data", {})
-        return {}
-
+class WeChatPayManager(Client):
     def generate_qr_code(
         self,
         out_trade_no: str,
@@ -26,7 +10,7 @@ class WeChatPayManager:
         product_name: str,
         product_id: str,
     ):
-        return self._request(
+        data, success = self._request(
             "POST",
             "/api/wx_pay/generate_qr_code",
             json={
@@ -37,15 +21,24 @@ class WeChatPayManager:
                 "product_id": product_id,
             },
         )
+        if not success:
+            return {}, False
+        return data, True
 
     def query_order(self, out_trade_no: str):
-        return self._request(
+        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):
-        return self._request(
+        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

pixelarraythirdparty/product/product.py

@@ -1,23 +1,7 @@
-import requests
+from pixelarraythirdparty.client import Client
 
 
-class ProductManager:
-    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, url, **kwargs):
-        resp = requests.request(
-            method, f"{self.base_url}{url}", headers=self.headers, **kwargs
-        )
-        if resp.status_code == 200:
-            return resp.json().get("data", {})
-        return {}
-
+class ProductManager(Client):
     def create_product(
         self,
         name: str,
@@ -41,7 +25,10 @@ class ProductManager:
             "features": features,
             "sort_order": sort_order,
         }
-        return self._request("POST", "/api/products/create", json=data)
+        data, success = self._request("POST", "/api/products/create", json=data)
+        if not success:
+            return {}, False
+        return data, True
 
     def list_product(
         self,
@@ -64,10 +51,16 @@ class ProductManager:
             params["category"] = category
         if name is not None:
             params["name"] = name
-        return self._request("GET", "/api/products/list", params=params)
+        data, success = self._request("GET", "/api/products/list", params=params)
+        if not success:
+            return {}, False
+        return data, True
 
     def get_product_detail(self, product_id: str):
-        return self._request("GET", f"/api/products/{product_id}")
+        data, success = self._request("GET", f"/api/products/{product_id}")
+        if not success:
+            return {}, False
+        return data, True
 
     def update_product(
         self,
@@ -93,10 +86,19 @@ class ProductManager:
             "features": features,
             "sort_order": sort_order,
         }
-        return self._request("PUT", f"/api/products/{product_id}", json=data)
+        data, success = self._request("PUT", f"/api/products/{product_id}", json=data)
+        if not success:
+            return {}, False
+        return data, True
 
     def delete_product(self, product_id: str):
-        return self._request("DELETE", f"/api/products/{product_id}")
+        data, success = self._request("DELETE", f"/api/products/{product_id}")
+        if not success:
+            return {}, False
+        return data, True
 
     def get_product_categories(self):
-        return self._request("GET", "/api/products/categories/list")
+        data, success = self._request("GET", "/api/products/categories/list")
+        if not success:
+            return {}, False
+        return data, True

pixelarraythirdparty/user/user.py

@@ -1,25 +1,9 @@
-import requests
 import aiohttp
 from typing import Optional
+from pixelarraythirdparty.client import Client
 
 
-class UserManager:
-    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, url, **kwargs):
-        resp = requests.request(
-            method, f"{self.base_url}{url}", headers=self.headers, **kwargs
-        )
-        if resp.status_code == 200:
-            return resp.json().get("data", {})
-        return {}
-
+class UserManager(Client):
     def list_user(
         self,
         page: int = 1,
@@ -32,7 +16,10 @@ class UserManager:
             params["role"] = role
         if is_active is not None:
             params["is_active"] = is_active
-        return self._request("GET", "/api/users/list", params=params)
+        data, success = self._request("GET", "/api/users/list", params=params)
+        if not success:
+            return {}, False
+        return data, True
 
     def create_user(self, username: str, password: str, email: str, role: str):
         data = {
@@ -41,7 +28,10 @@ class UserManager:
             "email": email,
             "role": role,
         }
-        return self._request("POST", "/api/users/create", json=data)
+        data, success = self._request("POST", "/api/users/create", json=data)
+        if not success:
+            return {}, False
+        return data, True
 
     def update_user(
         self, user_id: int, username: str, email: str, role: str, is_active: bool
@@ -52,14 +42,28 @@ class UserManager:
             "role": role,
             "is_active": is_active,
         }
-        return self._request("PUT", f"/api/users/{user_id}", json=data)
+        data, success = self._request("PUT", f"/api/users/{user_id}", json=data)
+        if not success:
+            return {}, False
+        return data, True
 
     def delete_user(self, user_id: int):
-        return self._request("DELETE", f"/api/users/{user_id}")
+        data, success = self._request("DELETE", f"/api/users/{user_id}")
+        if not success:
+            return {}, False
+        return data, True
 
     def get_user_detail(self, user_id: int):
-        return self._request("GET", f"/api/users/{user_id}")
+        data, success = self._request("GET", f"/api/users/{user_id}")
+        if not success:
+            return {}, False
+        return data, True
 
     def reset_user_password(self, user_id: int, new_password: str):
         data = {"new_password": new_password}
-        return self._request("POST", f"/api/users/{user_id}/reset-password", json=data)
+        data, success = self._request(
+            "POST", f"/api/users/{user_id}/reset-password", json=data
+        )
+        if not success:
+            return {}, False
+        return data, True

{pixelarraythirdparty-1.0.0.dist-info → pixelarraythirdparty-1.0.2.dist-info}/METADATA

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

pixelarraythirdparty-1.0.2.dist-info/RECORD

@@ -0,0 +1,16 @@
+pixelarraythirdparty/__init__.py,sha256=WVQdADzRD3tqhJvfxOm3qUInd9L0KRW-u5u37u6xq2c,458
+pixelarraythirdparty/client.py,sha256=dJlIrMlr1M4AGHvaLw191uaPnQbQ12_BGx8DTDTPkZQ,802
+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=51KcwBGfXS9Ico_0jlwcLUqcAgc9xgRutozqtlWwyHY,2616
+pixelarraythirdparty/order/pay.py,sha256=Fq7HKeYS_jk9CWdPldXE5OzuHl66SHHQW4sYlG6yODM,1251
+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.2.dist-info/licenses/LICENSE,sha256=O-g1dUr0U50rSIvmWE9toiVkSgFpVt72_MHITbWvAqA,1067
+pixelarraythirdparty-1.0.2.dist-info/METADATA,sha256=DQSV5A4aLGQ4FVKmol3IgYsZYKLLgyb2zXC4YnhLJPY,993
+pixelarraythirdparty-1.0.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pixelarraythirdparty-1.0.2.dist-info/top_level.txt,sha256=dzG2Ut8j7noUqj_0ZQjcIDAeHYCh_9WtlxjAxtoyufo,21
+pixelarraythirdparty-1.0.2.dist-info/RECORD,,

pixelarraythirdparty-1.0.0.dist-info/RECORD

@@ -1,16 +0,0 @@
-pixelarraythirdparty/__init__.py,sha256=XcGFz8LtH_OWAIVGfT7aymjYiG7F_utG576HIBx3gsw,458
-pixelarraythirdparty/auth.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pixelarraythirdparty/celery/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pixelarraythirdparty/celery/celery.py,sha256=jcK42Qt6KAmWlxVBS4G0saT8nzL2yTcDmPAukYclG8Q,1221
-pixelarraythirdparty/order/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pixelarraythirdparty/order/order.py,sha256=K_uhC9CFvzdW33mFoz5aRypi21V5rd58Q5WyCfe6Wes,2571
-pixelarraythirdparty/order/pay.py,sha256=hIsrOZfzY0wGHDvggac-JoIxQjZpxxw-wiRx-BwGqtE,1470
-pixelarraythirdparty/product/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pixelarraythirdparty/product/product.py,sha256=HA9xKLmv-kINmvSSf-5G86NtmsEWYcR6I5Asa1nIV8k,2984
-pixelarraythirdparty/user/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pixelarraythirdparty/user/user.py,sha256=xHyyypMLzEMbbn-TzhYANc3JftGFi3wAFMdk8qTdakk,2088
-pixelarraythirdparty-1.0.0.dist-info/licenses/LICENSE,sha256=O-g1dUr0U50rSIvmWE9toiVkSgFpVt72_MHITbWvAqA,1067
-pixelarraythirdparty-1.0.0.dist-info/METADATA,sha256=cz1IOAwa3bOzhDEQrVsAD_-1QyiSybdf7nmz2mTuz4g,993
-pixelarraythirdparty-1.0.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pixelarraythirdparty-1.0.0.dist-info/top_level.txt,sha256=dzG2Ut8j7noUqj_0ZQjcIDAeHYCh_9WtlxjAxtoyufo,21
-pixelarraythirdparty-1.0.0.dist-info/RECORD,,