✨ 更新数据库逻辑
This commit is contained in:
+72
-16
@@ -1,31 +1,87 @@
|
||||
from email.mime import base
|
||||
from unittest import result
|
||||
|
||||
import requests
|
||||
import httpx
|
||||
|
||||
from zhenxun.configs.config import Config
|
||||
from zhenxun.services.log import logger
|
||||
|
||||
|
||||
class CRequestManager:
|
||||
|
||||
@classmethod
|
||||
async def post(cls, url, json_data=None, form_data=None):
|
||||
"""发送 POST 请求(支持 JSON/Form-Data 格式)"""
|
||||
async def download(cls, url: str, savePath: str, fileName: str) -> bool:
|
||||
"""下载文件到指定路径
|
||||
|
||||
Args:
|
||||
url (str): 文件的下载链接
|
||||
savePath (str): 保存文件的文件夹路径
|
||||
fileName (str): 保存后的文件名
|
||||
|
||||
Returns:
|
||||
bool: 是否下载成功
|
||||
"""
|
||||
try:
|
||||
headers = {"Content-Type": "application/json"} if json_data else None
|
||||
response = requests.post(
|
||||
url,
|
||||
json=json_data,
|
||||
data=form_data,
|
||||
headers=headers,
|
||||
timeout=5
|
||||
)
|
||||
response.raise_for_status()
|
||||
return response.json()
|
||||
except requests.exceptions.RequestException as e:
|
||||
print(f"请求失败: {e}")
|
||||
async with httpx.AsyncClient(timeout=10.0) as client:
|
||||
response = await client.get(url)
|
||||
if response.status_code == 200:
|
||||
fullPath = savePath.rstrip("/") + "/" + fileName
|
||||
with open(fullPath, "wb") as f:
|
||||
f.write(response.content)
|
||||
return True
|
||||
else:
|
||||
logger.warning(f"文件下载失败: HTTP {response.status_code} {response.text}")
|
||||
return False
|
||||
except Exception as e:
|
||||
logger.warning(f"下载文件异常: {e}")
|
||||
return False
|
||||
|
||||
@classmethod
|
||||
async def post(cls, endpoint: str, name: str = "", jsonData: dict = None, formData: dict = None) -> dict:
|
||||
"""发送POST请求到指定接口,供其他方法统一调用
|
||||
|
||||
Args:
|
||||
endpoint (str): 请求的接口路径
|
||||
name (str, optional): 操作名称用于日志记录
|
||||
jsonData (dict, optional): 以JSON格式发送的数据
|
||||
formData (dict, optional): 以表单格式发送的数据
|
||||
|
||||
Raises:
|
||||
ValueError: 当jsonData和formData都未提供时抛出
|
||||
|
||||
Returns:
|
||||
dict: 返回请求结果的JSON数据
|
||||
"""
|
||||
if jsonData is None and formData is None:
|
||||
raise ValueError("post请求必须提供jsonData或formData其中之一")
|
||||
|
||||
baseUrl = Config.get_config("zhenxun_plugin_farm", "服务地址")
|
||||
|
||||
url = f"{baseUrl.rstrip('/')}/{endpoint.lstrip('/')}"
|
||||
|
||||
try:
|
||||
async with httpx.AsyncClient(timeout=5.0) as client:
|
||||
if jsonData is not None:
|
||||
response = await client.post(url, json=jsonData)
|
||||
else:
|
||||
response = await client.post(url, data=formData)
|
||||
|
||||
if response.status_code == 200:
|
||||
return response.json()
|
||||
else:
|
||||
logger.warning(f"真寻农场{name}请求失败: HTTP {response.status_code} {response.text}")
|
||||
return {}
|
||||
except httpx.RequestError as e:
|
||||
logger.warning(f"真寻农场{name}请求异常: {e}")
|
||||
return {}
|
||||
except Exception as e:
|
||||
logger.warning(f"真寻农场{name}处理异常: {e}")
|
||||
return {}
|
||||
|
||||
|
||||
@classmethod
|
||||
async def sign(cls, uid: str) -> str:
|
||||
a = await cls.post("http://diuse.work:9099/testPost", json_data={"level":3})
|
||||
a = await cls.post("http://diuse.work:9099/testPost", jsonData={"level":3})
|
||||
|
||||
result = ""
|
||||
|
||||
|
||||
Reference in New Issue
Block a user