mirror of
https://github.com/hyb-oyqq/FRAISEMOE-Addons-Installer-NEXT.git
synced 2025-12-21 22:38:35 +00:00
09d6883432
- 在解压线程中添加已解压文件路径参数，支持直接使用已解压的补丁文件，提升解压效率。 - 更新下载管理器，简化下载成功后的处理逻辑，直接进入解压阶段，去除冗余的哈希验证步骤。 - 在离线模式管理器中增强哈希验证功能，确保在解压后进行哈希校验，提升补丁文件的完整性检查。 - 增强日志记录，确保在关键操作中提供详细的调试信息，便于后续排查和用户反馈。
225 lines
10 KiB
Python
225 lines
10 KiB
Python
import os
|
||
import shutil
|
||
from PySide6 import QtWidgets
|
||
from PySide6.QtWidgets import QMessageBox
|
||
from PySide6.QtCore import QTimer
|
||
|
||
from utils.logger import setup_logger
|
||
|
||
# 初始化logger
|
||
logger = setup_logger("extraction_handler")
|
||
|
||
class ExtractionHandler:
|
||
"""解压处理器，负责管理解压任务和结果处理"""
|
||
|
||
def __init__(self, main_window):
|
||
"""初始化解压处理器
|
||
|
||
Args:
|
||
main_window: 主窗口实例，用于访问UI和状态
|
||
"""
|
||
self.main_window = main_window
|
||
self.APP_NAME = main_window.APP_NAME if hasattr(main_window, 'APP_NAME') else ""
|
||
|
||
def start_extraction(self, _7z_path, game_folder, plugin_path, game_version, extracted_path=None):
|
||
"""开始解压任务
|
||
|
||
Args:
|
||
_7z_path: 7z文件路径
|
||
game_folder: 游戏文件夹路径
|
||
plugin_path: 插件路径
|
||
game_version: 游戏版本名称
|
||
extracted_path: 已解压的补丁文件路径，如果提供则直接使用它而不进行解压
|
||
"""
|
||
# 检查是否处于离线模式
|
||
is_offline = False
|
||
if hasattr(self.main_window, 'offline_mode_manager'):
|
||
is_offline = self.main_window.offline_mode_manager.is_in_offline_mode()
|
||
|
||
# 显示解压中的消息窗口
|
||
self.main_window.hash_msg_box = self.main_window.hash_manager.hash_pop_window(
|
||
check_type="offline_extraction" if is_offline else "extraction",
|
||
is_offline=is_offline
|
||
)
|
||
|
||
# 创建并启动解压线程
|
||
self.main_window.extraction_thread = self.main_window.create_extraction_thread(
|
||
_7z_path, game_folder, plugin_path, game_version, extracted_path
|
||
)
|
||
self.main_window.extraction_thread.finished.connect(self.on_extraction_finished_with_hash_check)
|
||
self.main_window.extraction_thread.start()
|
||
|
||
def on_extraction_finished_with_hash_check(self, success, error_message, game_version):
|
||
"""解压完成后进行哈希校验
|
||
|
||
Args:
|
||
success: 是否解压成功
|
||
error_message: 错误信息
|
||
game_version: 游戏版本
|
||
"""
|
||
# 关闭哈希检查窗口
|
||
if self.main_window.hash_msg_box and self.main_window.hash_msg_box.isVisible():
|
||
self.main_window.hash_msg_box.close()
|
||
self.main_window.hash_msg_box = None
|
||
|
||
# 如果解压失败，显示错误并询问是否继续
|
||
if not success:
|
||
# 临时启用窗口以显示错误消息
|
||
self.main_window.setEnabled(True)
|
||
|
||
QtWidgets.QMessageBox.critical(self.main_window, f"错误 - {self.APP_NAME}", error_message)
|
||
self.main_window.installed_status[game_version] = False
|
||
|
||
# 询问用户是否继续其他游戏的安装
|
||
reply = QtWidgets.QMessageBox.question(
|
||
self.main_window,
|
||
f"继续安装? - {self.APP_NAME}",
|
||
f"\n{game_version} 的补丁安装失败。\n\n是否继续安装其他游戏的补丁？\n",
|
||
QtWidgets.QMessageBox.StandardButton.Yes | QtWidgets.QMessageBox.StandardButton.No,
|
||
QtWidgets.QMessageBox.StandardButton.Yes
|
||
)
|
||
|
||
if reply == QtWidgets.QMessageBox.StandardButton.Yes:
|
||
# 继续下一个，重新禁用窗口
|
||
self.main_window.setEnabled(False)
|
||
# 通知DownloadManager继续下一个下载任务
|
||
self.main_window.download_manager.on_extraction_finished(True)
|
||
else:
|
||
# 用户选择停止，保持窗口启用状态
|
||
self.main_window.ui.start_install_text.setText("开始安装")
|
||
# 通知DownloadManager停止下载队列
|
||
self.main_window.download_manager.on_extraction_finished(False)
|
||
return
|
||
|
||
# 解压成功，进行哈希校验
|
||
self._perform_hash_check(game_version)
|
||
|
||
def _perform_hash_check(self, game_version):
|
||
"""解压成功后进行哈希校验
|
||
|
||
Args:
|
||
game_version: 游戏版本
|
||
"""
|
||
# 获取安装路径
|
||
install_paths = {}
|
||
if hasattr(self.main_window, 'game_detector') and hasattr(self.main_window, 'download_manager'):
|
||
game_dirs = self.main_window.game_detector.identify_game_directories_improved(
|
||
self.main_window.download_manager.selected_folder
|
||
)
|
||
|
||
for game, info in self.main_window.GAME_INFO.items():
|
||
if game in game_dirs and game == game_version:
|
||
game_dir = game_dirs[game]
|
||
install_path = os.path.join(game_dir, os.path.basename(info["install_path"]))
|
||
install_paths[game] = install_path
|
||
break
|
||
|
||
if not install_paths:
|
||
# 如果找不到安装路径，直接认为安装成功
|
||
logger.warning(f"未找到 {game_version} 的安装路径，跳过哈希校验")
|
||
self.main_window.installed_status[game_version] = True
|
||
self.main_window.download_manager.on_extraction_finished(True)
|
||
return
|
||
|
||
# 显示哈希校验窗口
|
||
self.main_window.hash_msg_box = self.main_window.hash_manager.hash_pop_window(check_type="post")
|
||
|
||
# 创建并启动哈希线程进行校验
|
||
self.main_window.hash_thread = self.main_window.create_hash_thread(
|
||
"after",
|
||
install_paths,
|
||
self.main_window.plugin_hash,
|
||
self.main_window.installed_status
|
||
)
|
||
self.main_window.hash_thread.after_finished.connect(self.on_hash_check_finished)
|
||
self.main_window.hash_thread.start()
|
||
|
||
def on_hash_check_finished(self, result):
|
||
"""哈希校验完成后的处理
|
||
|
||
Args:
|
||
result: 校验结果，包含通过状态、游戏版本和消息
|
||
"""
|
||
# 关闭哈希检查窗口
|
||
if self.main_window.hash_msg_box and self.main_window.hash_msg_box.isVisible():
|
||
self.main_window.hash_msg_box.close()
|
||
self.main_window.hash_msg_box = None
|
||
|
||
if not result["passed"]:
|
||
# 校验失败，删除已解压的文件并提示重新下载
|
||
game_version = result["game"]
|
||
error_message = result["message"]
|
||
|
||
# 临时启用窗口以显示错误消息
|
||
self.main_window.setEnabled(True)
|
||
|
||
# 获取安装路径
|
||
install_path = None
|
||
if hasattr(self.main_window, 'game_detector') and hasattr(self.main_window, 'download_manager'):
|
||
game_dirs = self.main_window.game_detector.identify_game_directories_improved(
|
||
self.main_window.download_manager.selected_folder
|
||
)
|
||
|
||
if game_version in game_dirs and game_version in self.main_window.GAME_INFO:
|
||
game_dir = game_dirs[game_version]
|
||
install_path = os.path.join(game_dir, os.path.basename(self.main_window.GAME_INFO[game_version]["install_path"]))
|
||
|
||
# 如果找到安装路径，尝试删除已解压的文件
|
||
if install_path and os.path.exists(install_path):
|
||
try:
|
||
os.remove(install_path)
|
||
logger.info(f"已删除校验失败的文件: {install_path}")
|
||
except Exception as e:
|
||
logger.error(f"删除文件失败: {e}")
|
||
|
||
# 显示错误消息并询问是否重试
|
||
reply = QtWidgets.QMessageBox.question(
|
||
self.main_window,
|
||
f"校验失败 - {self.APP_NAME}",
|
||
f"{error_message}\n\n是否重新下载并安装？",
|
||
QtWidgets.QMessageBox.StandardButton.Yes | QtWidgets.QMessageBox.StandardButton.No,
|
||
QtWidgets.QMessageBox.StandardButton.Yes
|
||
)
|
||
|
||
if reply == QtWidgets.QMessageBox.StandardButton.Yes:
|
||
# 重新下载，将游戏重新添加到下载队列
|
||
self.main_window.setEnabled(False)
|
||
self.main_window.installed_status[game_version] = False
|
||
|
||
# 获取游戏目录和下载URL
|
||
if hasattr(self.main_window, 'download_manager') and hasattr(self.main_window, 'game_detector'):
|
||
game_dirs = self.main_window.game_detector.identify_game_directories_improved(
|
||
self.main_window.download_manager.selected_folder
|
||
)
|
||
|
||
if game_version in game_dirs:
|
||
# 重新将游戏添加到下载队列
|
||
self.main_window.download_manager.download_queue.appendleft([game_version])
|
||
# 继续下一个下载任务
|
||
self.main_window.download_manager.next_download_task()
|
||
else:
|
||
# 如果找不到游戏目录，继续下一个
|
||
self.main_window.download_manager.on_extraction_finished(True)
|
||
else:
|
||
# 如果无法重新下载，继续下一个
|
||
self.main_window.download_manager.on_extraction_finished(True)
|
||
else:
|
||
# 用户选择不重试，继续下一个
|
||
self.main_window.installed_status[game_version] = False
|
||
self.main_window.download_manager.on_extraction_finished(True)
|
||
else:
|
||
# 校验通过，更新安装状态
|
||
self.main_window.installed_status[game_version] = True
|
||
# 通知DownloadManager继续下一个下载任务
|
||
self.main_window.download_manager.on_extraction_finished(True)
|
||
|
||
def on_extraction_finished(self, success, error_message, game_version):
|
||
"""兼容旧版本的回调函数
|
||
|
||
Args:
|
||
success: 是否解压成功
|
||
error_message: 错误信息
|
||
game_version: 游戏版本
|
||
"""
|
||
# 调用新的带哈希校验的回调函数
|
||
self.on_extraction_finished_with_hash_check(success, error_message, game_version) |