mirror of
https://github.com/hyb-oyqq/FRAISEMOE-Addons-Installer-NEXT.git
synced 2025-12-17 12:30:27 +00:00
a97cdf4c23
- 更新主窗口信号连接，使用私有方法处理关闭和最小化按钮点击事件，增强代码可读性。 - 根据离线模式和配置状态统一管理开始安装按钮的状态，简化逻辑。 - 增强日志记录，确保在用户操作时提供详细的调试信息，便于后续排查和用户反馈。 - 优化卸载处理程序的日志记录，提升用户体验和系统稳定性。
351 lines
21 KiB
Python
351 lines
21 KiB
Python
import os
|
||
import shutil
|
||
import py7zr
|
||
import tempfile
|
||
import traceback
|
||
from PySide6.QtCore import QThread, Signal
|
||
from config.config import PLUGIN, GAME_INFO
|
||
import time # 用于时间计算
|
||
import threading
|
||
import queue
|
||
from concurrent.futures import TimeoutError
|
||
|
||
class ExtractionThread(QThread):
|
||
finished = Signal(bool, str, str) # success, error_message, game_version
|
||
progress = Signal(int, str) # 添加进度信号，传递进度百分比和状态信息
|
||
|
||
def __init__(self, _7z_path, game_folder, plugin_path, game_version, parent=None, extracted_path=None):
|
||
super().__init__(parent)
|
||
self._7z_path = _7z_path
|
||
self.game_folder = game_folder
|
||
self.plugin_path = plugin_path
|
||
self.game_version = game_version
|
||
self.extracted_path = extracted_path # 添加已解压文件路径参数
|
||
|
||
def run(self):
|
||
try:
|
||
# 确保游戏目录存在
|
||
os.makedirs(self.game_folder, exist_ok=True)
|
||
|
||
def update_progress(percent: int, message: str):
|
||
try:
|
||
self.progress.emit(percent, message)
|
||
except Exception:
|
||
pass
|
||
|
||
# 记录调试信息
|
||
from utils.logger import setup_logger
|
||
debug_logger = setup_logger("extraction_thread")
|
||
debug_logger.info(f"====== 开始处理 {self.game_version} 补丁文件 ======")
|
||
debug_logger.info(f"压缩包路径: {self._7z_path}")
|
||
debug_logger.info(f"游戏目录: {self.game_folder}")
|
||
debug_logger.info(f"插件路径: {self.plugin_path}")
|
||
|
||
update_progress(0, f"开始处理 {self.game_version} 的补丁文件...")
|
||
|
||
# 支持外部请求中断
|
||
if self.isInterruptionRequested():
|
||
self.finished.emit(False, "操作已取消", self.game_version)
|
||
return
|
||
|
||
# 如果提供了已解压文件路径，直接使用它
|
||
if self.extracted_path and os.path.exists(self.extracted_path):
|
||
update_progress(20, f"正在复制 {self.game_version} 的补丁文件...\n(在此过程中可能会卡顿或无响应，请不要关闭软件)")
|
||
|
||
# 直接复制已解压的文件到游戏目录
|
||
target_file = os.path.join(self.game_folder, os.path.basename(self.plugin_path))
|
||
shutil.copy(self.extracted_path, target_file)
|
||
|
||
update_progress(60, f"正在完成 {self.game_version} 的补丁安装...")
|
||
|
||
# 对于NEKOPARA After，还需要复制签名文件
|
||
if self.game_version == "NEKOPARA After":
|
||
try:
|
||
update_progress(70, f"正在处理 {self.game_version} 的签名文件...")
|
||
# 从已解压文件的目录中获取签名文件
|
||
extracted_dir = os.path.dirname(self.extracted_path)
|
||
sig_filename = os.path.basename(GAME_INFO[self.game_version]["sig_path"])
|
||
sig_path = os.path.join(extracted_dir, sig_filename)
|
||
|
||
# 尝试多种可能的签名文件路径
|
||
if not os.path.exists(sig_path):
|
||
# 尝试在同级目录查找
|
||
sig_path = os.path.join(os.path.dirname(extracted_dir), sig_filename)
|
||
|
||
# 如果签名文件存在，则复制它
|
||
if os.path.exists(sig_path):
|
||
target_sig = os.path.join(self.game_folder, sig_filename)
|
||
shutil.copy(sig_path, target_sig)
|
||
update_progress(80, f"签名文件复制完成")
|
||
else:
|
||
# 如果签名文件不存在，则使用原始路径
|
||
sig_path = os.path.join(PLUGIN, GAME_INFO[self.game_version]["sig_path"])
|
||
if os.path.exists(sig_path):
|
||
target_sig = os.path.join(self.game_folder, os.path.basename(sig_path))
|
||
shutil.copy(sig_path, target_sig)
|
||
update_progress(80, f"使用内置签名文件完成")
|
||
else:
|
||
update_progress(80, f"未找到签名文件，继续安装主补丁文件")
|
||
except Exception as sig_err:
|
||
# 签名文件处理失败时记录错误但不中断主流程
|
||
update_progress(80, f"签名文件处理失败: {str(sig_err)}")
|
||
|
||
update_progress(100, f"{self.game_version} 补丁文件处理完成")
|
||
self.finished.emit(True, "", self.game_version)
|
||
return
|
||
|
||
# 否则解压源压缩包到临时目录，再复制目标文件
|
||
update_progress(10, f"正在打开 {self.game_version} 的补丁压缩包...")
|
||
|
||
with py7zr.SevenZipFile(self._7z_path, mode="r") as archive:
|
||
# 获取压缩包内的文件列表
|
||
file_list = archive.getnames()
|
||
|
||
# 详细记录压缩包中的所有文件
|
||
debug_logger.debug(f"压缩包内容分析:")
|
||
debug_logger.debug(f"- 文件总数: {len(file_list)}")
|
||
for i, f in enumerate(file_list):
|
||
debug_logger.debug(f" {i+1}. {f} - 类型: {'文件夹' if f.endswith('/') or f.endswith('\\') else '文件'}")
|
||
|
||
update_progress(20, f"正在分析 {self.game_version} 的补丁文件...")
|
||
|
||
update_progress(30, f"正在解压 {self.game_version} 的补丁文件...\n(在此过程中可能会卡顿或无响应，请不要关闭软件)")
|
||
|
||
with tempfile.TemporaryDirectory() as temp_dir:
|
||
# 查找主补丁文件和签名文件
|
||
target_filename = os.path.basename(self.plugin_path)
|
||
# 只有NEKOPARA After版本才需要查找签名文件
|
||
if self.game_version == "NEKOPARA After":
|
||
sig_filename = target_filename + ".sig" # 签名文件名
|
||
debug_logger.debug(f"查找主补丁文件: {target_filename}")
|
||
debug_logger.debug(f"查找签名文件: {sig_filename}")
|
||
else:
|
||
sig_filename = None
|
||
debug_logger.debug(f"查找主补丁文件: {target_filename}")
|
||
debug_logger.debug(f"{self.game_version} 不需要签名文件")
|
||
|
||
target_file_in_archive = None
|
||
sig_file_in_archive = None
|
||
|
||
# 对于NEKOPARA After，增加特殊处理
|
||
if self.game_version == "NEKOPARA After":
|
||
# 增加专门的检查，同时识别主补丁和签名文件
|
||
debug_logger.debug("执行NEKOPARA After特殊补丁文件识别")
|
||
|
||
# 查找主补丁和签名文件
|
||
for file_path in file_list:
|
||
basename = os.path.basename(file_path)
|
||
|
||
# 查找主补丁文件
|
||
if basename == "afteradult.xp3" and not basename.endswith('.sig'):
|
||
target_file_in_archive = file_path
|
||
debug_logger.debug(f"找到精确匹配的After主补丁文件: {target_file_in_archive}")
|
||
|
||
# 查找签名文件
|
||
elif basename == "afteradult.xp3.sig" or basename.endswith('.sig'):
|
||
sig_file_in_archive = file_path
|
||
debug_logger.debug(f"找到After签名文件: {sig_file_in_archive}")
|
||
|
||
# 如果没找到主补丁文件，寻找可能的替代文件
|
||
if not target_file_in_archive:
|
||
for file_path in file_list:
|
||
if "afteradult.xp3" in file_path and not file_path.endswith('.sig'):
|
||
target_file_in_archive = file_path
|
||
debug_logger.debug(f"找到备选After主补丁文件: {target_file_in_archive}")
|
||
break
|
||
else:
|
||
# 标准处理逻辑
|
||
for file_path in file_list:
|
||
basename = os.path.basename(file_path)
|
||
|
||
# 查找主补丁文件
|
||
if basename == target_filename and not basename.endswith('.sig'):
|
||
target_file_in_archive = file_path
|
||
debug_logger.debug(f"在压缩包中找到主补丁文件: {target_file_in_archive}")
|
||
|
||
# 查找签名文件
|
||
elif basename == sig_filename:
|
||
sig_file_in_archive = file_path
|
||
debug_logger.debug(f"在压缩包中找到签名文件: {sig_file_in_archive}")
|
||
|
||
# 如果没有找到精确匹配的主补丁文件，使用更宽松的搜索
|
||
if not target_file_in_archive:
|
||
debug_logger.warning(f"没有找到精确匹配的主补丁文件，尝试更宽松的搜索")
|
||
for file_path in file_list:
|
||
if target_filename in file_path and not file_path.endswith('.sig'):
|
||
target_file_in_archive = file_path
|
||
debug_logger.info(f"在压缩包中找到可能的主补丁文件: {target_file_in_archive}")
|
||
break
|
||
|
||
# 如果找不到主补丁文件，使用回退方案：提取全部内容
|
||
if not target_file_in_archive:
|
||
debug_logger.warning(f"未能识别正确的主补丁文件，将提取所有文件并尝试查找")
|
||
|
||
# 提取所有文件到临时目录
|
||
update_progress(30, f"正在解压所有文件...")
|
||
archive.extractall(path=temp_dir)
|
||
debug_logger.debug(f"已提取所有文件到临时目录")
|
||
|
||
# 在提取的文件中查找主补丁文件和签名文件
|
||
found_main = False
|
||
found_sig = False
|
||
|
||
for root, dirs, files in os.walk(temp_dir):
|
||
for file in files:
|
||
# 查找主补丁文件
|
||
if file == target_filename and not file.endswith('.sig'):
|
||
extracted_file_path = os.path.join(root, file)
|
||
file_size = os.path.getsize(extracted_file_path)
|
||
debug_logger.debug(f"在提取的文件中找到主补丁文件: {extracted_file_path}, 大小: {file_size} 字节")
|
||
|
||
# 复制到目标位置
|
||
target_path = os.path.join(self.game_folder, target_filename)
|
||
shutil.copy2(extracted_file_path, target_path)
|
||
debug_logger.debug(f"已复制主补丁文件到: {target_path}")
|
||
found_main = True
|
||
|
||
# 查找签名文件
|
||
elif file == sig_filename or file.endswith('.sig'):
|
||
extracted_sig_path = os.path.join(root, file)
|
||
sig_size = os.path.getsize(extracted_sig_path)
|
||
debug_logger.debug(f"在提取的文件中找到签名文件: {extracted_sig_path}, 大小: {sig_size} 字节")
|
||
|
||
# 复制到目标位置
|
||
sig_target = os.path.join(self.game_folder, sig_filename)
|
||
shutil.copy2(extracted_sig_path, sig_target)
|
||
debug_logger.debug(f"已复制签名文件到: {sig_target}")
|
||
found_sig = True
|
||
|
||
# 如果两个文件都找到，可以停止遍历
|
||
if found_main and found_sig:
|
||
debug_logger.debug("已找到所有需要的文件，停止遍历")
|
||
break
|
||
|
||
if found_main and found_sig:
|
||
break
|
||
|
||
if not found_main:
|
||
debug_logger.error(f"无法找到主补丁文件，安装失败")
|
||
raise FileNotFoundError(f"在压缩包中未找到主补丁文件 {target_filename}")
|
||
|
||
# 只有NEKOPARA After版本才需要处理签名文件
|
||
if self.game_version == "NEKOPARA After":
|
||
# 签名文件没找到不影响主流程，但记录警告
|
||
if not found_sig:
|
||
debug_logger.warning(f"未找到签名文件 {sig_filename}，但继续安装主补丁文件")
|
||
else:
|
||
debug_logger.info(f"{self.game_version} 不需要签名文件，跳过签名文件处理")
|
||
else:
|
||
# 准备要解压的文件列表
|
||
files_to_extract = [target_file_in_archive]
|
||
# 只有NEKOPARA After版本才需要解压签名文件
|
||
if self.game_version == "NEKOPARA After" and sig_file_in_archive:
|
||
files_to_extract.append(sig_file_in_archive)
|
||
debug_logger.debug(f"将同时解压主补丁文件和签名文件: {files_to_extract}")
|
||
else:
|
||
debug_logger.debug(f"将仅解压主补丁文件: {files_to_extract}")
|
||
|
||
# 解压选定的文件到临时目录
|
||
debug_logger.debug(f"开始解压选定文件到临时目录: {temp_dir}")
|
||
|
||
# 设置解压超时时间（秒）
|
||
extract_timeout = 180 # 3分钟超时
|
||
debug_logger.debug(f"设置解压超时: {extract_timeout}秒")
|
||
|
||
# 创建子线程执行解压
|
||
import threading
|
||
import queue
|
||
|
||
extract_result = queue.Queue()
|
||
|
||
def extract_files():
|
||
try:
|
||
archive.extract(path=temp_dir, targets=files_to_extract)
|
||
extract_result.put(("success", None))
|
||
except Exception as e:
|
||
extract_result.put(("error", e))
|
||
|
||
extract_thread = threading.Thread(target=extract_files)
|
||
extract_thread.daemon = True
|
||
extract_thread.start()
|
||
|
||
# 每5秒更新一次进度，最多等待设定的超时时间
|
||
total_waited = 0
|
||
while extract_thread.is_alive() and total_waited < extract_timeout:
|
||
update_progress(30 + int(30 * total_waited / extract_timeout),
|
||
f"正在解压文件...已等待{total_waited}秒")
|
||
extract_thread.join(5) # 等待5秒
|
||
total_waited += 5
|
||
|
||
# 检查是否超时
|
||
if extract_thread.is_alive():
|
||
debug_logger.error(f"解压超时（超过{extract_timeout}秒）")
|
||
raise TimeoutError(f"解压超时（超过{extract_timeout}秒），请检查补丁文件是否完整")
|
||
|
||
# 检查解压结果
|
||
if not extract_result.empty():
|
||
status, error = extract_result.get()
|
||
if status == "error":
|
||
debug_logger.error(f"解压错误: {error}")
|
||
raise error
|
||
|
||
debug_logger.debug(f"文件解压完成")
|
||
|
||
update_progress(60, f"正在复制 {self.game_version} 的补丁文件...")
|
||
|
||
# 复制主补丁文件到游戏目录
|
||
extracted_file_path = os.path.join(temp_dir, target_file_in_archive)
|
||
|
||
# 检查解压后的文件是否存在及其大小
|
||
if os.path.exists(extracted_file_path):
|
||
file_size = os.path.getsize(extracted_file_path)
|
||
debug_logger.debug(f"解压后的主补丁文件存在: {extracted_file_path}, 大小: {file_size} 字节")
|
||
else:
|
||
debug_logger.error(f"解压后的主补丁文件不存在: {extracted_file_path}")
|
||
raise FileNotFoundError(f"解压后的文件不存在: {extracted_file_path}")
|
||
|
||
# 构建目标路径并复制
|
||
target_path = os.path.join(self.game_folder, target_filename)
|
||
debug_logger.debug(f"复制主补丁文件: {extracted_file_path} 到 {target_path}")
|
||
shutil.copy2(extracted_file_path, target_path)
|
||
|
||
# 验证主补丁文件是否成功复制
|
||
if os.path.exists(target_path):
|
||
target_size = os.path.getsize(target_path)
|
||
debug_logger.debug(f"主补丁文件成功复制: {target_path}, 大小: {target_size} 字节")
|
||
else:
|
||
debug_logger.error(f"主补丁文件复制失败: {target_path}")
|
||
raise FileNotFoundError(f"目标文件复制失败: {target_path}")
|
||
|
||
# 只有NEKOPARA After版本才需要处理签名文件
|
||
if self.game_version == "NEKOPARA After":
|
||
# 如果有找到签名文件，也复制它
|
||
if sig_file_in_archive:
|
||
update_progress(80, f"正在复制签名文件...")
|
||
extracted_sig_path = os.path.join(temp_dir, sig_file_in_archive)
|
||
|
||
if os.path.exists(extracted_sig_path):
|
||
sig_size = os.path.getsize(extracted_sig_path)
|
||
debug_logger.debug(f"解压后的签名文件存在: {extracted_sig_path}, 大小: {sig_size} 字节")
|
||
|
||
# 复制签名文件到游戏目录
|
||
sig_target = os.path.join(self.game_folder, sig_filename)
|
||
shutil.copy2(extracted_sig_path, sig_target)
|
||
debug_logger.debug(f"签名文件成功复制: {sig_target}")
|
||
else:
|
||
debug_logger.warning(f"解压后的签名文件不存在: {extracted_sig_path}")
|
||
else:
|
||
debug_logger.warning(f"压缩包中没有找到签名文件，但继续安装主补丁文件")
|
||
else:
|
||
debug_logger.info(f"{self.game_version} 不需要签名文件，跳过签名文件处理")
|
||
|
||
update_progress(100, f"{self.game_version} 补丁文件解压完成")
|
||
self.finished.emit(True, "", self.game_version)
|
||
except (py7zr.Bad7zFile, FileNotFoundError, Exception) as e:
|
||
try:
|
||
self.progress.emit(100, f"处理 {self.game_version} 的补丁文件失败")
|
||
except Exception:
|
||
pass
|
||
self.finished.emit(False, f"\n文件操作失败，请重试\n\n【错误信息】：{e}\n", self.game_version)
|
||
|
||
|