ok

core/api_data.py

@@ -3,33 +3,37 @@ import requests
 import json
 from astrbot.api import logger
 
-def api_data(api_url, json_data=None, outdata="data"):
+def api_data_post(api_url, json_data=None, outdata=None):
     """
     获取数据的POST请求函数
     
     Args:
         api_url: API地址
         json_data: POST请求的JSON数据
-        outdata: 返回数据中要提取的字段，默认为"data"
+        outdata: 返回数据中要提取的字段
     
     Returns:
         成功时返回outdata字段的数据，失败返回None
     """
     try:
-        response = requests.post(
-            api_url, 
-            json=json_data,  # JSON body数据
-            timeout=10, 
-            verify=False
-        )
-        
+        response = requests.post(api_url, json=json_data, timeout=10, verify=False)
         response.raise_for_status()
         data = response.json()
         
-        if data.get('code') not in [200, "0"]:
-            logger.error(f"API返回错误: {data.get('msg', '未知错误')}")
-            return None
+        # 检查是否有code字段
+        if 'code' in data:
+            # 有code字段时，检查是否成功
+            if data.get('code') not in [200, "0", 0]:
+                logger.error(f"API返回错误:{data.get('code', '未知状态')} {data.get('msg', '未知错误')}")
+                return None
+        else:
+            # 无code字段时，检查数据是否为空
+            if not data:
+                logger.error("API返回空数据")
+                return None
         
+        if outdata is None or outdata == "":
+            return data
         return data.get(outdata, {})
         
     except requests.exceptions.RequestException as e:
@@ -39,27 +43,37 @@ def api_data(api_url, json_data=None, outdata="data"):
         logger.error(f"JSON解析错误: {e}")
         return None
     
-def fetch_jx3_data(api_url=None,outdata="data", **params):
+def api_data_get(api_url,params=None,outdata=None):
     """
     获取数据的GET请求函数
     
     Args:
         api_url: API地址
-        outdata: 返回数据中要提取的字段，默认为"data"
-        **params: 其他查询参数
+        
+        params: 其他查询参数
     
     Returns:
         成功时返回outdata字段的数据，失败返回None
     """
     try:
-        response = requests.get(api_url, params=params, timeout=10, verify=False)
+        response = requests.get(api_url, params, timeout=10, verify=False)
         response.raise_for_status()
         data = response.json()
         
-        if data.get('code') != 200:
-            logger.error(f"API返回错误: {data.get('msg', '未知错误')}")
-            return None
+        # 检查是否有code字段
+        if 'code' in data:
+            # 有code字段时，检查是否成功
+            if data.get('code') not in [200, "0", 0]:
+                logger.error(f"API返回错误:{data.get('code', '未知状态')} {data.get('msg', '未知错误')}")
+                return None
+        else:
+            # 无code字段时，检查数据是否为空
+            if not data:
+                logger.error("API返回空数据")
+                return None
         
+        if outdata is None or outdata == "":
+            return data
         return data.get(outdata, {})
         
     except requests.exceptions.RequestException as e: