mirror of
https://github.com/hyb-oyqq/FRAISEMOE-Addons-Installer-NEXT.git
synced 2025-12-16 03:40:27 +00:00
229 lines
9.5 KiB
Python
229 lines
9.5 KiB
Python
import os
|
||
from PySide6.QtWidgets import (
|
||
QDialog, QVBoxLayout, QLabel, QListWidget, QPushButton, QHBoxLayout,
|
||
QAbstractItemView, QFileDialog, QMessageBox
|
||
)
|
||
from PySide6.QtCore import QObject
|
||
from PySide6.QtGui import QFont
|
||
from utils import msgbox_frame
|
||
|
||
class UninstallHandler(QObject):
|
||
"""
|
||
处理补丁卸载功能的类
|
||
"""
|
||
def __init__(self, main_window):
|
||
"""
|
||
初始化卸载处理程序
|
||
|
||
Args:
|
||
main_window: 主窗口实例，用于访问其他组件
|
||
"""
|
||
super().__init__()
|
||
self.main_window = main_window
|
||
self.debug_manager = main_window.debug_manager
|
||
self.game_detector = main_window.game_detector
|
||
self.patch_manager = main_window.patch_manager
|
||
self.app_name = main_window.patch_manager.app_name
|
||
|
||
def handle_uninstall_button_click(self):
|
||
"""
|
||
处理卸载补丁按钮点击事件
|
||
打开文件选择对话框选择游戏目录，然后卸载对应游戏的补丁
|
||
"""
|
||
# 获取游戏目录
|
||
debug_mode = self.debug_manager._is_debug_mode()
|
||
|
||
# 提示用户选择目录
|
||
file_dialog_info = "选择游戏上级目录" if debug_mode else "选择游戏目录"
|
||
selected_folder = QFileDialog.getExistingDirectory(self.main_window, file_dialog_info, "")
|
||
|
||
if not selected_folder or selected_folder == "":
|
||
return # 用户取消了选择
|
||
|
||
if debug_mode:
|
||
print(f"DEBUG: 卸载功能 - 用户选择了目录: {selected_folder}")
|
||
|
||
# 首先尝试将选择的目录视为上级目录，使用增强的目录识别功能
|
||
game_dirs = self.game_detector.identify_game_directories_improved(selected_folder)
|
||
|
||
if game_dirs and len(game_dirs) > 0:
|
||
self._handle_multiple_games(game_dirs, debug_mode)
|
||
else:
|
||
self._handle_single_game(selected_folder, debug_mode)
|
||
|
||
def _handle_multiple_games(self, game_dirs, debug_mode):
|
||
"""
|
||
处理多个游戏的补丁卸载
|
||
|
||
Args:
|
||
game_dirs: 游戏目录字典
|
||
debug_mode: 是否为调试模式
|
||
"""
|
||
if debug_mode:
|
||
print(f"DEBUG: 卸载功能 - 在上级目录中找到以下游戏: {list(game_dirs.keys())}")
|
||
|
||
# 查找已安装补丁的游戏，只处理那些已安装补丁的游戏
|
||
games_with_patch = {}
|
||
for game_version, game_dir in game_dirs.items():
|
||
if self.patch_manager.check_patch_installed(game_dir, game_version):
|
||
games_with_patch[game_version] = game_dir
|
||
if debug_mode:
|
||
print(f"DEBUG: 卸载功能 - {game_version} 已安装补丁")
|
||
|
||
# 检查是否有已安装补丁的游戏
|
||
if not games_with_patch:
|
||
QMessageBox.information(
|
||
self.main_window,
|
||
f"提示 - {self.app_name}",
|
||
"\n未在选择的目录中找到已安装补丁的游戏。\n请确认您选择了正确的游戏目录，并且该目录中的游戏已安装过补丁。\n",
|
||
QMessageBox.StandardButton.Ok
|
||
)
|
||
return
|
||
|
||
# 显示选择对话框
|
||
selected_games = self._show_game_selection_dialog(games_with_patch)
|
||
|
||
if not selected_games:
|
||
return # 用户取消了选择
|
||
|
||
# 过滤game_dirs，只保留选中的游戏
|
||
selected_game_dirs = {game: games_with_patch[game] for game in selected_games if game in games_with_patch}
|
||
|
||
# 确认卸载
|
||
game_list = '\n'.join(selected_games)
|
||
reply = QMessageBox.question(
|
||
self.main_window,
|
||
f"确认卸载 - {self.app_name}",
|
||
f"\n确定要卸载以下游戏的补丁吗？\n\n{game_list}\n",
|
||
QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No,
|
||
QMessageBox.StandardButton.No
|
||
)
|
||
|
||
if reply == QMessageBox.StandardButton.No:
|
||
return
|
||
|
||
# 使用批量卸载方法
|
||
success_count, fail_count, results = self.patch_manager.batch_uninstall_patches(selected_game_dirs)
|
||
self.patch_manager.show_uninstall_result(success_count, fail_count, results)
|
||
|
||
def _handle_single_game(self, selected_folder, debug_mode):
|
||
"""
|
||
处理单个游戏的补丁卸载
|
||
|
||
Args:
|
||
selected_folder: 选择的游戏目录
|
||
debug_mode: 是否为调试模式
|
||
"""
|
||
# 未找到游戏目录，尝试将选择的目录作为游戏目录
|
||
if debug_mode:
|
||
print(f"DEBUG: 卸载功能 - 未在上级目录找到游戏，尝试将选择的目录视为游戏目录")
|
||
|
||
game_version = self.game_detector.identify_game_version(selected_folder)
|
||
|
||
if game_version:
|
||
if debug_mode:
|
||
print(f"DEBUG: 卸载功能 - 识别为游戏: {game_version}")
|
||
|
||
# 检查是否已安装补丁
|
||
if self.patch_manager.check_patch_installed(selected_folder, game_version):
|
||
# 确认卸载
|
||
reply = QMessageBox.question(
|
||
self.main_window,
|
||
f"确认卸载 - {self.app_name}",
|
||
f"\n确定要卸载 {game_version} 的补丁吗？\n游戏目录: {selected_folder}\n",
|
||
QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No,
|
||
QMessageBox.StandardButton.No
|
||
)
|
||
|
||
if reply == QMessageBox.StandardButton.Yes:
|
||
# 创建单个游戏的目录字典，使用批量卸载流程
|
||
single_game_dir = {game_version: selected_folder}
|
||
success_count, fail_count, results = self.patch_manager.batch_uninstall_patches(single_game_dir)
|
||
self.patch_manager.show_uninstall_result(success_count, fail_count, results)
|
||
else:
|
||
# 没有安装补丁
|
||
QMessageBox.information(
|
||
self.main_window,
|
||
f"提示 - {self.app_name}",
|
||
f"\n未在 {game_version} 中找到已安装的补丁。\n请确认该游戏已经安装过补丁。\n",
|
||
QMessageBox.StandardButton.Ok
|
||
)
|
||
else:
|
||
# 两种方式都未识别到游戏
|
||
if debug_mode:
|
||
print(f"DEBUG: 卸载功能 - 无法识别游戏")
|
||
|
||
msg_box = msgbox_frame(
|
||
f"错误 - {self.app_name}",
|
||
"\n所选目录不是有效的NEKOPARA游戏目录。\n请选择包含游戏可执行文件的目录或其上级目录。\n",
|
||
QMessageBox.StandardButton.Ok,
|
||
)
|
||
msg_box.exec()
|
||
|
||
def _show_game_selection_dialog(self, games_with_patch):
|
||
"""
|
||
显示游戏选择对话框
|
||
|
||
Args:
|
||
games_with_patch: 已安装补丁的游戏目录字典
|
||
|
||
Returns:
|
||
list: 选择的游戏列表
|
||
"""
|
||
dialog = QDialog(self.main_window)
|
||
dialog.setWindowTitle("选择要卸载的游戏补丁")
|
||
dialog.resize(400, 300)
|
||
|
||
layout = QVBoxLayout(dialog)
|
||
|
||
# 添加"已安装补丁的游戏"标签
|
||
already_installed_label = QLabel("已安装补丁的游戏:", dialog)
|
||
already_installed_label.setFont(QFont(already_installed_label.font().family(), already_installed_label.font().pointSize(), QFont.Bold))
|
||
layout.addWidget(already_installed_label)
|
||
|
||
# 添加已安装游戏列表（可选，这里使用静态标签替代，保持一致性）
|
||
installed_games_text = ", ".join(games_with_patch.keys())
|
||
installed_games_label = QLabel(installed_games_text, dialog)
|
||
layout.addWidget(installed_games_label)
|
||
|
||
# 添加一些间距
|
||
layout.addSpacing(10)
|
||
|
||
# 添加"请选择要卸载补丁的游戏"标签
|
||
info_label = QLabel("请选择要卸载补丁的游戏:", dialog)
|
||
info_label.setFont(QFont(info_label.font().family(), info_label.font().pointSize(), QFont.Bold))
|
||
layout.addWidget(info_label)
|
||
|
||
# 添加列表控件，只显示已安装补丁的游戏
|
||
list_widget = QListWidget(dialog)
|
||
list_widget.setSelectionMode(QAbstractItemView.SelectionMode.ExtendedSelection) # 允许多选
|
||
for game in games_with_patch.keys():
|
||
list_widget.addItem(game)
|
||
layout.addWidget(list_widget)
|
||
|
||
# 添加全选按钮
|
||
select_all_btn = QPushButton("全选", dialog)
|
||
select_all_btn.clicked.connect(lambda: list_widget.selectAll())
|
||
layout.addWidget(select_all_btn)
|
||
|
||
# 添加确定和取消按钮
|
||
buttons_layout = QHBoxLayout()
|
||
ok_button = QPushButton("确定", dialog)
|
||
cancel_button = QPushButton("取消", dialog)
|
||
buttons_layout.addWidget(ok_button)
|
||
buttons_layout.addWidget(cancel_button)
|
||
layout.addLayout(buttons_layout)
|
||
|
||
# 连接按钮事件
|
||
ok_button.clicked.connect(dialog.accept)
|
||
cancel_button.clicked.connect(dialog.reject)
|
||
|
||
# 显示对话框并等待用户选择
|
||
result = dialog.exec()
|
||
|
||
if result != QDialog.DialogCode.Accepted or list_widget.selectedItems() == []:
|
||
# 用户取消或未选择任何游戏
|
||
return []
|
||
|
||
# 获取用户选择的游戏
|
||
return [item.text() for item in list_widget.selectedItems()] |