""" 配置分类功能集成代码示例 直接复制到 ui_main.py 中使用 """ # ============================================================ # 1. 在文件顶部添加导入 # ============================================================ from config_category_widget import ConfigCategoryWidget from config_category import get_default_category_manager, ConfigCategory # ============================================================ # 2. 在 MainWindow.__init__ 中添加分类选择器 # ============================================================ # 在工具栏布局中，self.cfg_btn 之前添加： # 配置分类选择器 self.category_widget = ConfigCategoryWidget(on_change=self.schedule_autosave) self.category_widget.category_changed.connect(self._on_category_changed) toolbar_layout.addWidget(self.category_widget) # 添加分隔符（可选） separator = QFrame() separator.setFrameShape(QFrame.VLine) separator.setFrameShadow(QFrame.Sunken) toolbar_layout.addWidget(separator) # ============================================================ # 3. 添加分类切换处理方法 # ============================================================ def _on_category_changed(self, category_name: str, config_path: Path, template_path) -> None: """ 分类切换时的处理 Args: category_name: 分类名称 config_path: 配置文件路径 template_path: 模板文件路径（可能为None） """ try: self.logger.info(f"切换到配置分类: {category_name}") # 加载配置 self.config = AppConfig.load(config_path) self._autosave_path = config_path # 加载模板 if template_path and template_path.exists(): self.template_path = template_path self.logger.info(f"已加载模板: {template_path}") else: self.logger.warning(f"分类 {category_name} 没有模板文件") # 更新UI显示 self.apply_config_to_ui() # 保存状态 self._save_last_state(template=self.template_path, config=self._autosave_path) # 提示用户 msg = f"已切换到配置分类: {category_name}\n" msg += f"配置文件: {config_path.name}\n" if template_path: msg += f"模板文件: {template_path.name}" else: msg += "模板文件: (未配置)" QMessageBox.information(self, "分类切换", msg) except Exception as e: self.logger.error(f"切换配置分类失败: {e}") QMessageBox.warning(self, "错误", f"切换配置分类失败:\n{e}") # ============================================================ # 4. 修改 _do_autosave 方法 # ============================================================ def _do_autosave(self) -> None: """自动保存配置""" # 如果没有设置保存路径，尝试使用当前分类的配置路径 if self._autosave_path is None: category = self.category_widget.get_current_category() if category: self._autosave_path = category.config_path self.logger.info(f"使用分类配置路径: {self._autosave_path}") # 原有的保存逻辑 if self._autosave_path: try: self.apply_ui_to_config() self.config.save(self._autosave_path) self.logger.info(f"配置已自动保存到: {self._autosave_path}") except Exception as e: self.logger.error(f"自动保存配置失败: {e}") # ============================================================ # 5. 修改 _load_default_config_if_exists 方法 # ============================================================ def _load_default_config_if_exists(self) -> None: """加载默认配置，优先使用分类配置""" # 检查是否有可用的配置分类 if self.category_widget.has_categories(): # 有分类，使用第一个分类的配置 category = self.category_widget.get_current_category() if category: try: self.config = AppConfig.load(category.config_path) self._autosave_path = category.config_path if category.template_path: self.template_path = category.template_path self.logger.info(f"已加载分类配置: {category.name}") return except Exception as e: self.logger.error(f"加载分类配置失败: {e}") # 没有分类或加载失败，使用根目录的 default.json（向后兼容） default_path = APP_DIR / DEFAULT_CONFIG_FILENAME if default_path.exists(): try: self.config = AppConfig.load(default_path) self._autosave_path = default_path self.logger.info("已加载根目录配置: default.json") except Exception as e: self.logger.error(f"加载default.json失败: {e}") else: self.logger.info("未找到配置文件，等待用户配置") # ============================================================ # 6. 在配置对话框中显示当前分类信息（可选） # ============================================================ def open_config_dialog(self, preview: bool = False) -> None: # ... 现有代码 ... # 在对话框顶部显示当前分类信息 if hasattr(self, 'category_widget'): category = self.category_widget.get_current_category() if category: category_info = QLabel(f"当前分类: {category.name}") category_info.setStyleSheet( "background-color: #e3f2fd; padding: 8px; " "border-radius: 4px; font-weight: bold; color: #1976d2;" ) # 将此标签添加到对话框布局的顶部 # dlg_layout.insertWidget(0, category_info) # ============================================================ # 7. 完整的集成检查清单 # ============================================================ """ 集成检查清单： □ 1. 已添加导入语句 □ 2. 已在工具栏添加分类选择器 □ 3. 已添加 _on_category_changed 方法 □ 4. 已修改 _do_autosave 方法 □ 5. 已修改 _load_default_config_if_exists 方法 □ 6. 已运行 setup_config_categories.py 初始化 □ 7. 已运行 test_config_category.py 验证 □ 8. 已测试分类切换功能 □ 9. 已测试配置保存功能 □ 10. 已测试程序启动加载 测试步骤： 1. 启动程序，检查分类选择器是否显示 2. 切换分类，检查配置和模板是否正确加载 3. 修改配置，检查是否保存到正确的分类文件 4. 重启程序，检查是否加载上次使用的分类 5. 开始实验，检查是否使用当前分类的配置 6. 生成报告，检查是否使用当前分类的模板 """ # ============================================================ # 8. 调试辅助代码（可选） # ============================================================ def _debug_category_info(self): """打印当前分类信息（用于调试）""" if hasattr(self, 'category_widget'): category = self.category_widget.get_current_category() if category: print(f"当前分类: {category.name}") print(f"配置路径: {category.config_path}") print(f"模板路径: {category.template_path}") print(f"自动保存路径: {self._autosave_path}") else: print("未选择分类") else: print("分类选择器未初始化") # ============================================================ # 9. 错误处理示例 # ============================================================ def _safe_load_category_config(self, category: ConfigCategory) -> bool: """ 安全加载分类配置（带错误处理） Returns: 是否成功加载 """ try: # 检查配置文件是否存在 if not category.config_path.exists(): self.logger.error(f"配置文件不存在: {category.config_path}") QMessageBox.warning( self, "配置错误", f"分类 {category.name} 的配置文件不存在" ) return False # 加载配置 self.config = AppConfig.load(category.config_path) self._autosave_path = category.config_path # 加载模板（可选） if category.template_path and category.template_path.exists(): self.template_path = category.template_path # 更新UI self.apply_config_to_ui() self.logger.info(f"成功加载分类配置: {category.name}") return True except Exception as e: self.logger.error(f"加载分类配置失败: {e}") QMessageBox.critical( self, "加载失败", f"加载分类 {category.name} 的配置失败:\n{e}" ) return False # ============================================================ # 10. 使用提示 # ============================================================ """ 使用提示： 1. 复制需要的代码片段到 ui_main.py 对应位置 2. 根据实际情况调整代码（如变量名、方法名等） 3. 确保导入语句在文件顶部 4. 确保方法添加到 MainWindow 类中 5. 运行初始化脚本创建配置分类 6. 测试所有功能是否正常 注意事项： - 保持代码缩进一致 - 检查方法签名是否匹配 - 确保信号连接正确 - 添加适当的日志记录 - 处理可能的异常情况 """