mirror of
https://github.com/hyb-oyqq/FRAISEMOE-Addons-Installer-NEXT.git
synced 2025-12-17 12:30:27 +00:00
185 lines
7.2 KiB
Python
185 lines
7.2 KiB
Python
import os
|
||
import sys
|
||
import subprocess
|
||
import re
|
||
from urllib.parse import urlparse
|
||
from PySide6 import QtCore, QtWidgets
|
||
from PySide6.QtCore import (Qt, Signal, QThread, QTimer)
|
||
from PySide6.QtWidgets import (QLabel, QProgressBar, QVBoxLayout, QDialog)
|
||
from utils import resource_path
|
||
from data.config import APP_NAME, UA
|
||
|
||
# 下载线程类
|
||
class DownloadThread(QThread):
|
||
progress = Signal(dict)
|
||
finished = Signal(bool, str)
|
||
|
||
def __init__(self, url, _7z_path, game_version, parent=None):
|
||
super().__init__(parent)
|
||
self.url = url
|
||
self._7z_path = _7z_path
|
||
self.game_version = game_version
|
||
self.process = None
|
||
self._is_running = True
|
||
|
||
def stop(self):
|
||
if self.process and self.process.poll() is None:
|
||
self._is_running = False
|
||
try:
|
||
# 使用 taskkill 强制终止进程及其子进程，并隐藏窗口
|
||
subprocess.run(['taskkill', '/F', '/T', '/PID', str(self.process.pid)], check=True, creationflags=subprocess.CREATE_NO_WINDOW)
|
||
except (subprocess.CalledProcessError, FileNotFoundError) as e:
|
||
print(f"停止下载进程时出错: {e}")
|
||
|
||
|
||
def run(self):
|
||
try:
|
||
if not self._is_running:
|
||
self.finished.emit(False, "下载已手动停止。")
|
||
return
|
||
|
||
aria2c_path = resource_path("aria2c.exe")
|
||
download_dir = os.path.dirname(self._7z_path)
|
||
file_name = os.path.basename(self._7z_path)
|
||
|
||
parsed_url = urlparse(self.url)
|
||
referer = f"{parsed_url.scheme}://{parsed_url.netloc}/"
|
||
|
||
command = [
|
||
aria2c_path,
|
||
]
|
||
|
||
command.extend([
|
||
'--dir', download_dir,
|
||
'--out', file_name,
|
||
'--user-agent', UA,
|
||
'--referer', referer,
|
||
'--header', f'Origin: {referer.rstrip("/")}',
|
||
'--header', 'Accept: */*',
|
||
'--header', 'Accept-Language: zh-CN,zh;q=0.9,en;q=0.8',
|
||
'--header', 'Accept-Encoding: gzip, deflate, br',
|
||
'--header', 'Cache-Control: no-cache',
|
||
'--header', 'Pragma: no-cache',
|
||
'--header', 'DNT: 1',
|
||
'--header', 'Sec-Fetch-Dest: empty',
|
||
'--header', 'Sec-Fetch-Mode: cors',
|
||
'--header', 'Sec-Fetch-Site: same-origin',
|
||
'--http-accept-gzip=true',
|
||
'--console-log-level=info',
|
||
'--summary-interval=1',
|
||
'--log-level=info',
|
||
'--max-tries=3',
|
||
'--retry-wait=2',
|
||
'--connect-timeout=60',
|
||
'--timeout=60',
|
||
'--auto-file-renaming=false',
|
||
'--allow-overwrite=true',
|
||
'--split=16',
|
||
'--max-connection-per-server=16'
|
||
])
|
||
|
||
# 证书验证现在总是需要，因为我们依赖hosts文件
|
||
command.append('--check-certificate=false')
|
||
|
||
command.append(self.url)
|
||
|
||
# 打印将要执行的命令，用于调试
|
||
print(f"即将执行的 Aria2c 命令: {' '.join(command)}")
|
||
|
||
creation_flags = subprocess.CREATE_NO_WINDOW if sys.platform == 'win32' else 0
|
||
self.process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True, encoding='utf-8', errors='replace', creationflags=creation_flags)
|
||
|
||
# 正则表达式用于解析aria2c的输出
|
||
# 例如: #1 GID[...]( 5%) CN:1 DL:10.5MiB/s ETA:1m30s
|
||
progress_pattern = re.compile(r'\((\d{1,3})%\).*?CN:(\d+).*?DL:\s*([^\s]+).*?ETA:\s*([^\s]+)')
|
||
|
||
full_output = []
|
||
while self._is_running and self.process.poll() is None:
|
||
if self.process.stdout:
|
||
line = self.process.stdout.readline()
|
||
if not line:
|
||
break
|
||
else:
|
||
break
|
||
|
||
full_output.append(line)
|
||
print(line.strip()) # 在控制台输出实时日志
|
||
|
||
match = progress_pattern.search(line)
|
||
if match:
|
||
percent = int(match.group(1))
|
||
threads = match.group(2)
|
||
speed = match.group(3)
|
||
eta = match.group(4)
|
||
self.progress.emit({
|
||
"game": self.game_version,
|
||
"percent": percent,
|
||
"threads": threads,
|
||
"speed": speed,
|
||
"eta": eta
|
||
})
|
||
|
||
return_code = self.process.wait()
|
||
|
||
if not self._is_running: # 如果是手动停止的
|
||
self.finished.emit(False, "下载已手动停止。")
|
||
return
|
||
|
||
if return_code == 0:
|
||
self.progress.emit({
|
||
"game": self.game_version,
|
||
"percent": 100,
|
||
"threads": "N/A",
|
||
"speed": "N/A",
|
||
"eta": "完成"
|
||
})
|
||
self.finished.emit(True, "")
|
||
else:
|
||
error_message = f"\nAria2c下载失败，退出码: {return_code}\n\n--- Aria2c 输出 ---\n{''.join(full_output)}\n---------------------\n"
|
||
self.finished.emit(False, error_message)
|
||
|
||
except Exception as e:
|
||
if self._is_running:
|
||
self.finished.emit(False, f"\n下载时发生未知错误\n\n【错误信息】: {e}\n")
|
||
|
||
# 下载进度窗口类
|
||
class ProgressWindow(QDialog):
|
||
def __init__(self, parent=None):
|
||
super(ProgressWindow, self).__init__(parent)
|
||
self.setWindowTitle(f"下载进度 - {APP_NAME}")
|
||
self.resize(450, 180)
|
||
self.setWindowFlags(self.windowFlags() & ~Qt.WindowType.WindowCloseButtonHint)
|
||
self.setWindowFlags(self.windowFlags() & ~Qt.WindowType.WindowSystemMenuHint)
|
||
|
||
layout = QVBoxLayout()
|
||
self.game_label = QLabel("正在启动下载，请稍后...")
|
||
self.progress_bar = QProgressBar()
|
||
self.progress_bar.setValue(0)
|
||
self.stats_label = QLabel("速度: - | 线程: - | 剩余时间: -")
|
||
self.stop_button = QtWidgets.QPushButton("停止下载")
|
||
|
||
layout.addWidget(self.game_label)
|
||
layout.addWidget(self.progress_bar)
|
||
layout.addWidget(self.stats_label)
|
||
layout.addWidget(self.stop_button)
|
||
self.setLayout(layout)
|
||
|
||
def update_progress(self, data):
|
||
game_version = data.get("game", "未知游戏")
|
||
percent = data.get("percent", 0)
|
||
speed = data.get("speed", "-")
|
||
threads = data.get("threads", "-")
|
||
eta = data.get("eta", "-")
|
||
|
||
self.game_label.setText(f"正在下载: {game_version}")
|
||
self.progress_bar.setValue(int(percent))
|
||
self.stats_label.setText(f"速度: {speed} | 线程: {threads} | 剩余时间: {eta}")
|
||
|
||
if percent == 100:
|
||
self.stop_button.setEnabled(False)
|
||
self.stop_button.setText("下载完成")
|
||
QTimer.singleShot(1500, self.accept)
|
||
|
||
def closeEvent(self, event):
|
||
# 覆盖默认的关闭事件，防止用户通过其他方式关闭窗口
|
||
event.ignore() |