309 lines
9.3 KiB
Python
309 lines
9.3 KiB
Python
# AUTO_MAA:A MAA Multi Account Management and Automation Tool
|
||
# Copyright © 2024-2025 DLmaster361
|
||
|
||
# This file is part of AUTO_MAA.
|
||
|
||
# AUTO_MAA is free software: you can redistribute it and/or modify
|
||
# it under the terms of the GNU General Public License as published
|
||
# by the Free Software Foundation, either version 3 of the License,
|
||
# or (at your option) any later version.
|
||
|
||
# AUTO_MAA is distributed in the hope that it will be useful,
|
||
# but WITHOUT ANY WARRANTY; without even the implied warranty
|
||
# of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See
|
||
# the GNU General Public License for more details.
|
||
|
||
# You should have received a copy of the GNU General Public License
|
||
# along with AUTO_MAA. If not, see <https://www.gnu.org/licenses/>.
|
||
|
||
# Contact: DLmaster_361@163.com
|
||
|
||
"""
|
||
AUTO_MAA
|
||
AUTO_MAA网络请求线程
|
||
v4.4
|
||
作者：DLmaster_361
|
||
"""
|
||
|
||
from PySide6.QtCore import QObject, QThread, QEventLoop
|
||
import re
|
||
import time
|
||
import requests
|
||
import truststore
|
||
from pathlib import Path
|
||
from typing import Dict
|
||
|
||
from .logger import logger
|
||
|
||
|
||
class NetworkThread(QThread):
|
||
"""网络请求线程类"""
|
||
|
||
max_retries = 3
|
||
timeout = 10
|
||
backoff_factor = 0.1
|
||
|
||
def __init__(
|
||
self,
|
||
mode: str,
|
||
url: str,
|
||
path: Path = None,
|
||
files: Dict = None,
|
||
data: Dict = None,
|
||
) -> None:
|
||
super().__init__()
|
||
|
||
self.setObjectName(
|
||
f"NetworkThread-{mode}-{re.sub(r'(&cdk=)[^&]+(&)', r'\1******\2', url)}"
|
||
)
|
||
|
||
logger.info(f"创建网络请求线程: {self.objectName()}", module="网络请求子线程")
|
||
|
||
self.mode = mode
|
||
self.url = url
|
||
self.path = path
|
||
self.files = files
|
||
self.data = data
|
||
|
||
from .config import Config
|
||
|
||
self.proxies = {
|
||
"http": Config.get(Config.update_ProxyAddress),
|
||
"https": Config.get(Config.update_ProxyAddress),
|
||
}
|
||
|
||
self.status_code = None
|
||
self.response_json = None
|
||
self.error_message = None
|
||
|
||
self.loop = QEventLoop()
|
||
|
||
truststore.inject_into_ssl() # 信任系统证书
|
||
|
||
@logger.catch
|
||
def run(self) -> None:
|
||
"""运行网络请求线程"""
|
||
|
||
if self.mode == "get":
|
||
self.get_json(self.url)
|
||
elif self.mode == "get_file":
|
||
self.get_file(self.url, self.path)
|
||
elif self.mode == "upload_file":
|
||
self.upload_file(self.url, self.files, self.data)
|
||
|
||
def get_json(self, url: str) -> None:
|
||
"""
|
||
通过get方法获取json数据
|
||
|
||
:param url: 请求的URL
|
||
"""
|
||
|
||
logger.info(f"子线程 {self.objectName()} 开始网络请求", module="网络请求子线程")
|
||
|
||
response = None
|
||
|
||
for _ in range(self.max_retries):
|
||
try:
|
||
response = requests.get(url, timeout=self.timeout, proxies=self.proxies)
|
||
self.status_code = response.status_code
|
||
self.response_json = response.json()
|
||
self.error_message = None
|
||
break
|
||
except Exception as e:
|
||
self.status_code = response.status_code if response else None
|
||
self.response_json = None
|
||
self.error_message = str(e)
|
||
logger.exception(
|
||
f"子线程 {self.objectName()} 网络请求失败：{e}，第{_+1}次尝试",
|
||
module="网络请求子线程",
|
||
)
|
||
time.sleep(self.backoff_factor)
|
||
|
||
self.loop.quit()
|
||
|
||
def get_file(self, url: str, path: Path) -> None:
|
||
"""
|
||
通过get方法下载文件到指定路径
|
||
|
||
:param url: 请求的URL
|
||
:param path: 下载文件的保存路径
|
||
"""
|
||
|
||
logger.info(f"子线程 {self.objectName()} 开始下载文件", module="网络请求子线程")
|
||
|
||
response = None
|
||
|
||
try:
|
||
response = requests.get(url, timeout=self.timeout, proxies=self.proxies)
|
||
if response.status_code == 200:
|
||
with open(path, "wb") as file:
|
||
file.write(response.content)
|
||
self.status_code = response.status_code
|
||
self.error_message = None
|
||
else:
|
||
self.status_code = response.status_code
|
||
self.error_message = f"下载失败，状态码: {response.status_code}"
|
||
|
||
except Exception as e:
|
||
self.status_code = response.status_code if response else None
|
||
self.error_message = str(e)
|
||
logger.exception(
|
||
f"子线程 {self.objectName()} 网络请求失败：{e}", module="网络请求子线程"
|
||
)
|
||
|
||
self.loop.quit()
|
||
|
||
def upload_file(self, url: str, files: Dict, data: Dict = None) -> None:
|
||
"""
|
||
通过POST方法上传文件
|
||
|
||
:param url: 请求的URL
|
||
:param files: 文件字典，格式为 {'file': ('filename', file_obj, 'content_type')}
|
||
:param data: 表单数据字典
|
||
"""
|
||
|
||
logger.info(f"子线程 {self.objectName()} 开始上传文件", module="网络请求子线程")
|
||
|
||
response = None
|
||
|
||
for _ in range(self.max_retries):
|
||
try:
|
||
response = requests.post(
|
||
url,
|
||
files=files,
|
||
data=data,
|
||
timeout=self.timeout,
|
||
proxies=self.proxies,
|
||
)
|
||
self.status_code = response.status_code
|
||
|
||
# 尝试解析JSON响应
|
||
try:
|
||
self.response_json = response.json()
|
||
except ValueError:
|
||
# 如果不是JSON格式，保存文本内容
|
||
self.response_json = {"text": response.text}
|
||
|
||
self.error_message = None
|
||
break
|
||
|
||
except Exception as e:
|
||
self.status_code = response.status_code if response else None
|
||
self.response_json = None
|
||
self.error_message = str(e)
|
||
logger.exception(
|
||
f"子线程 {self.objectName()} 文件上传失败：{e}，第{_+1}次尝试",
|
||
module="网络请求子线程",
|
||
)
|
||
time.sleep(self.backoff_factor)
|
||
|
||
self.loop.quit()
|
||
|
||
|
||
class _Network(QObject):
|
||
"""网络请求线程管理类"""
|
||
|
||
def __init__(self) -> None:
|
||
super().__init__()
|
||
|
||
self.task_queue = []
|
||
|
||
def add_task(
|
||
self,
|
||
mode: str,
|
||
url: str,
|
||
path: Path = None,
|
||
files: Dict = None,
|
||
data: Dict = None,
|
||
) -> NetworkThread:
|
||
"""
|
||
添加网络请求任务
|
||
|
||
:param mode: 请求模式，支持 "get", "get_file", "upload_file"
|
||
:param url: 请求的URL
|
||
:param path: 下载文件的保存路径，仅在 mode 为 "get_file" 时有效
|
||
:param files: 上传文件字典，仅在 mode 为 "upload_file" 时有效
|
||
:param data: 表单数据字典，仅在 mode 为 "upload_file" 时有效
|
||
:return: 返回创建的 NetworkThread 实例
|
||
"""
|
||
|
||
logger.info(f"添加网络请求任务: {mode} {url} {path}", module="网络请求")
|
||
|
||
network_thread = NetworkThread(mode, url, path, files, data)
|
||
|
||
self.task_queue.append(network_thread)
|
||
|
||
network_thread.start()
|
||
|
||
return network_thread
|
||
|
||
def upload_config_file(
|
||
self, file_path: Path, username: str = "", description: str = ""
|
||
) -> NetworkThread:
|
||
"""
|
||
上传配置文件到分享服务器
|
||
|
||
:param file_path: 要上传的文件路径
|
||
:param username: 用户名（可选）
|
||
:param description: 文件描述（必填）
|
||
:return: 返回创建的 NetworkThread 实例
|
||
"""
|
||
|
||
if not file_path.exists():
|
||
raise FileNotFoundError(f"文件不存在: {file_path}")
|
||
|
||
if not description:
|
||
raise ValueError("文件描述不能为空")
|
||
|
||
# 准备上传的文件
|
||
with open(file_path, "rb") as f:
|
||
files = {"file": (file_path.name, f.read(), "application/json")}
|
||
|
||
# 准备表单数据
|
||
data = {"description": description}
|
||
|
||
if username:
|
||
data["username"] = username
|
||
|
||
url = "http://221.236.27.82:10023/api/upload/share"
|
||
|
||
logger.info(
|
||
f"准备上传配置文件: {file_path.name}，用户: {username or '匿名'}，描述: {description}",
|
||
extra={"module": "网络请求"},
|
||
)
|
||
|
||
return self.add_task("upload_file", url, files=files, data=data)
|
||
|
||
def get_result(self, network_thread: NetworkThread) -> dict:
|
||
"""
|
||
获取网络请求结果
|
||
|
||
:param network_thread: 网络请求线程实例
|
||
:return: 包含状态码、响应JSON和错误信息的字典
|
||
"""
|
||
|
||
result = {
|
||
"status_code": network_thread.status_code,
|
||
"response_json": network_thread.response_json,
|
||
"error_message": (
|
||
re.sub(r"(&cdk=)[^&]+(&)", r"\1******\2", network_thread.error_message)
|
||
if network_thread.error_message
|
||
else None
|
||
),
|
||
}
|
||
|
||
network_thread.quit()
|
||
network_thread.wait()
|
||
self.task_queue.remove(network_thread)
|
||
network_thread.deleteLater()
|
||
|
||
logger.info(
|
||
f"网络请求结果: {result['status_code']}，请求子线程已结束",
|
||
module="网络请求",
|
||
)
|
||
|
||
return result
|
||
|
||
|
||
Network = _Network()
|