121 lines
3.4 KiB
Python
121 lines
3.4 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.3
|
||
作者：DLmaster_361
|
||
"""
|
||
|
||
from loguru import logger
|
||
from PySide6.QtCore import QThread, QEventLoop, QTimer
|
||
import time
|
||
import requests
|
||
from pathlib import Path
|
||
|
||
|
||
class _Network(QThread):
|
||
|
||
max_retries = 3
|
||
timeout = 10
|
||
backoff_factor = 0.1
|
||
|
||
def __init__(self) -> None:
|
||
super().__init__()
|
||
|
||
self.if_running = False
|
||
self.mode = None
|
||
self.url = None
|
||
self.loop = QEventLoop()
|
||
self.wait_loop = QEventLoop()
|
||
|
||
@logger.catch
|
||
def run(self) -> None:
|
||
"""运行网络请求线程"""
|
||
|
||
self.if_running = True
|
||
|
||
if self.mode == "get":
|
||
self.get_json(self.url)
|
||
elif self.mode == "get_file":
|
||
self.get_file(self.url, self.path)
|
||
|
||
self.if_running = False
|
||
|
||
def set_info(self, mode: str, url: str, path: Path = None) -> None:
|
||
"""设置网络请求信息"""
|
||
|
||
while self.if_running:
|
||
QTimer.singleShot(self.backoff_factor * 1000, self.wait_loop.quit)
|
||
self.wait_loop.exec()
|
||
|
||
self.mode = mode
|
||
self.url = url
|
||
self.path = path
|
||
|
||
self.stutus_code = None
|
||
self.response_json = None
|
||
self.error_message = None
|
||
|
||
def get_json(self, url: str) -> None:
|
||
"""通过get方法获取json数据"""
|
||
|
||
response = None
|
||
|
||
for _ in range(self.max_retries):
|
||
try:
|
||
response = requests.get(url, timeout=self.timeout)
|
||
self.stutus_code = response.status_code
|
||
self.response_json = response.json()
|
||
self.error_message = None
|
||
break
|
||
except Exception as e:
|
||
self.stutus_code = response.status_code if response else None
|
||
self.response_json = None
|
||
self.error_message = str(e)
|
||
time.sleep(self.backoff_factor)
|
||
|
||
self.loop.quit()
|
||
|
||
def get_file(self, url: str, path: Path) -> None:
|
||
"""通过get方法下载文件"""
|
||
|
||
response = None
|
||
|
||
try:
|
||
response = requests.get(url, timeout=10)
|
||
if response.status_code == 200:
|
||
with open(path, "wb") as file:
|
||
file.write(response.content)
|
||
self.stutus_code = response.status_code
|
||
else:
|
||
self.stutus_code = response.status_code
|
||
self.error_message = "下载失败"
|
||
|
||
except Exception as e:
|
||
self.stutus_code = response.status_code if response else None
|
||
self.error_message = str(e)
|
||
|
||
self.loop.quit()
|
||
|
||
|
||
Network = _Network()
|