import os from PySide6.QtWidgets import ( QDialog, QVBoxLayout, QLabel, QListWidget, QPushButton, QHBoxLayout, QAbstractItemView, QRadioButton, QButtonGroup, QFileDialog, QMessageBox ) from PySide6.QtCore import QObject from PySide6.QtGui import QFont from utils import msgbox_frame class PatchToggleHandler(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_toggle_patch_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): # 检查补丁当前状态（是否禁用） is_disabled, disabled_path = self.patch_manager.check_patch_disabled(game_dir, game_version) status = "已禁用" if is_disabled else "已启用" games_with_patch[game_version] = { "dir": game_dir, "disabled": is_disabled, "status": status } if debug_mode: print(f"DEBUG: 禁/启用功能 - {game_version} 已安装补丁，当前状态: {status}") # 检查是否有已安装补丁的游戏 if not games_with_patch: QMessageBox.information( self.main_window, f"提示 - {self.app_name}", "\n未在选择的目录中找到已安装补丁的游戏。\n请确认您选择了正确的游戏目录，并且该目录中的游戏已安装过补丁。\n", QMessageBox.StandardButton.Ok ) return # 显示选择对话框 selected_games, operation = self._show_multi_game_dialog(games_with_patch) if not selected_games: return # 用户取消了操作 # 过滤games_with_patch，只保留选中的游戏 selected_game_dirs = {} for game in selected_games: if game in games_with_patch: selected_game_dirs[game] = games_with_patch[game]["dir"] # 确认操作 operation_text = "禁用" if operation == "disable" else "启用" if operation == "enable" else "切换" game_list = '\n'.join([f"{game} ({games_with_patch[game]['status']})" for game in selected_games]) reply = QMessageBox.question( self.main_window, f"确认{operation_text}操作 - {self.app_name}", f"\n确定要{operation_text}以下游戏补丁吗？\n\n{game_list}\n", QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No, QMessageBox.StandardButton.No ) if reply == QMessageBox.StandardButton.No: return # 执行批量操作 self._execute_batch_toggle(selected_game_dirs, operation, debug_mode) 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): # 检查补丁当前状态 is_disabled, disabled_path = self.patch_manager.check_patch_disabled(selected_folder, game_version) current_status = "已禁用" if is_disabled else "已启用" # 显示单游戏操作对话框 operation = self._show_single_game_dialog(game_version, current_status, is_disabled) if not operation: return # 用户取消了操作 # 执行操作 result = self.patch_manager.toggle_patch(selected_folder, game_version, operation=operation) if not result["success"]: # 操作失败的消息已在toggle_patch中显示 pass 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_multi_game_dialog(self, games_with_patch): """ 显示多游戏选择对话框 Args: games_with_patch: 已安装补丁的游戏信息 Returns: tuple: (选择的游戏列表, 操作类型) """ dialog = QDialog(self.main_window) dialog.setWindowTitle("选择要操作的游戏补丁") dialog.resize(400, 400) # 增加高度以适应新增的单选按钮 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) # 添加游戏列表和状态 games_status_text = "" for game, info in games_with_patch.items(): games_status_text += f"{game} ({info['status']})\n" games_status_label = QLabel(games_status_text.strip(), dialog) layout.addWidget(games_status_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, info in games_with_patch.items(): list_widget.addItem(f"{game} ({info['status']})") layout.addWidget(list_widget) # 添加全选按钮 select_all_btn = QPushButton("全选", dialog) select_all_btn.clicked.connect(lambda: list_widget.selectAll()) layout.addWidget(select_all_btn) # 添加操作选择单选按钮 operation_label = QLabel("请选择要执行的操作:", dialog) operation_label.setFont(QFont(operation_label.font().family(), operation_label.font().pointSize(), QFont.Bold)) layout.addWidget(operation_label) # 创建单选按钮组 radio_button_group = QButtonGroup(dialog) # 添加"自动切换状态"单选按钮（默认选中） auto_toggle_radio = QRadioButton("自动切换状态（禁用<->启用）", dialog) auto_toggle_radio.setChecked(True) radio_button_group.addButton(auto_toggle_radio, 0) layout.addWidget(auto_toggle_radio) # 添加"全部禁用"单选按钮 disable_all_radio = QRadioButton("禁用选中的补丁", dialog) radio_button_group.addButton(disable_all_radio, 1) layout.addWidget(disable_all_radio) # 添加"全部启用"单选按钮 enable_all_radio = QRadioButton("启用选中的补丁", dialog) radio_button_group.addButton(enable_all_radio, 2) layout.addWidget(enable_all_radio) # 添加确定和取消按钮 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 [], None # 获取用户选择的游戏 selected_items = [item.text() for item in list_widget.selectedItems()] selected_games = [] # 从选中项文本中提取游戏名称 for item in selected_items: # 去除状态后缀 " (已启用)" 或 " (已禁用)" game_name = item.split(" (")[0] selected_games.append(game_name) # 获取选中的操作类型 operation = None if radio_button_group.checkedId() == 1: # 禁用选中的补丁 operation = "disable" elif radio_button_group.checkedId() == 2: # 启用选中的补丁 operation = "enable" # 否则为None，表示自动切换状态 return selected_games, operation def _show_single_game_dialog(self, game_version, current_status, is_disabled): """ 显示单游戏操作对话框 Args: game_version: 游戏版本 current_status: 当前补丁状态 is_disabled: 是否已禁用 Returns: str: 操作类型，"enable"或"disable"，或None表示取消 """ dialog = QDialog(self.main_window) dialog.setWindowTitle(f"{game_version} 补丁操作") dialog.resize(300, 200) layout = QVBoxLayout(dialog) # 添加当前状态标签 status_label = QLabel(f"当前补丁状态: {current_status}", dialog) status_label.setFont(QFont(status_label.font().family(), status_label.font().pointSize(), QFont.Bold)) layout.addWidget(status_label) # 添加操作选择单选按钮 operation_label = QLabel("请选择要执行的操作:", dialog) layout.addWidget(operation_label) # 创建单选按钮组 radio_button_group = QButtonGroup(dialog) # 添加可选操作 if is_disabled: # 当前是禁用状态，显示启用选项 enable_radio = QRadioButton("启用补丁", dialog) enable_radio.setChecked(True) radio_button_group.addButton(enable_radio, 0) layout.addWidget(enable_radio) else: # 当前是启用状态，显示禁用选项 disable_radio = QRadioButton("禁用补丁", dialog) disable_radio.setChecked(True) radio_button_group.addButton(disable_radio, 0) layout.addWidget(disable_radio) # 添加确定和取消按钮 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: # 用户取消 return None # 根据当前状态确定操作 return "enable" if is_disabled else "disable" def _execute_batch_toggle(self, selected_game_dirs, operation, debug_mode): """ 执行批量补丁切换操作 Args: selected_game_dirs: 选择的游戏目录 operation: 操作类型 debug_mode: 是否为调试模式 """ success_count = 0 fail_count = 0 results = [] for game_version, game_dir in selected_game_dirs.items(): try: # 使用静默模式进行操作 result = self.patch_manager.toggle_patch(game_dir, game_version, operation=operation, silent=True) if result["success"]: success_count += 1 else: fail_count += 1 results.append({ "version": game_version, "success": result["success"], "message": result["message"], "action": result["action"] }) except Exception as e: if debug_mode: print(f"DEBUG: 切换 {game_version} 补丁状态时出错: {str(e)}") fail_count += 1 results.append({ "version": game_version, "success": False, "message": f"操作出错: {str(e)}", "action": "none" }) # 显示操作结果 self.patch_manager.show_toggle_result(success_count, fail_count, results)