PCM_Report/集成代码示例.py

"""
配置分类功能集成代码示例
直接复制到 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. 测试所有功能是否正常

注意事项：

- 保持代码缩进一致
- 检查方法签名是否匹配
- 确保信号连接正确
- 添加适当的日志记录
- 处理可能的异常情况
"""