"""
测试分类配置的加载和保存功能
"""
import os
import sys
import json
import logging

# 配置日志记录
logging.basicConfig(level=logging.DEBUG, 
                    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
logger = logging.getLogger(__name__)

# 添加项目根目录到Python路径
project_root = os.path.dirname(os.path.abspath(__file__))
if project_root not in sys.path:
    sys.path.append(project_root)

from llmclipboard.document_processor import DocumentProcessor

def test_categories_persistence():
    """测试分类配置的持久化功能"""
    logger.info("开始测试分类配置的持久化功能")
    
    # 确定配置文件路径
    if getattr(sys, 'frozen', False):
        # 打包环境 - 使用可执行文件所在目录
        config_dir = os.path.dirname(sys.executable)
    else:
        # 开发环境 - 使用项目根目录
        config_dir = project_root
        
    categories_file = os.path.join(config_dir, 'categories.json')
    logger.info(f"分类配置文件路径: {categories_file}")
    
    # 1. 检查配置文件是否存在
    if os.path.exists(categories_file):
        logger.info(f"分类配置文件已存在: {categories_file}")
        with open(categories_file, 'r', encoding='utf-8') as f:
            current_categories = json.load(f)
        logger.info(f"当前分类: {current_categories}")
    else:
        logger.info(f"分类配置文件不存在: {categories_file}")
        current_categories = {}
    
    # 2. 创建文档处理器实例
    doc_processor = DocumentProcessor(config_path=categories_file)
    
    # 3. 获取当前分类
    logger.info(f"文档处理器加载的分类: {doc_processor.categories}")
    
    # 4. 修改分类
    new_categories = {
        "测试": ["测试1", "测试2", "测试3"],
        "工作": ["会议", "项目", "计划", "报告", "任务", "进度"],
        "学习": ["教程", "课程", "学习", "笔记", "知识", "总结"],
        "生活": ["购物", "健康", "旅行", "美食", "娱乐"]
    }
    
    # 5. 更新分类
    logger.info(f"更新分类: {new_categories}")
    doc_processor.update_categories(new_categories)
    
    # 6. 检查配置文件是否已更新
    if os.path.exists(categories_file):
        with open(categories_file, 'r', encoding='utf-8') as f:
            updated_categories = json.load(f)
        logger.info(f"更新后的分类: {updated_categories}")
        
        # 验证更新是否成功
        if updated_categories == new_categories:
            logger.info("分类更新成功!")
        else:
            logger.error("分类更新失败!")
    else:
        logger.error(f"分类配置文件不存在: {categories_file}")
    
    # 7. 创建新的文档处理器实例，验证配置是否能被正确加载
    new_doc_processor = DocumentProcessor(config_path=categories_file)
    logger.info(f"新文档处理器加载的分类: {new_doc_processor.categories}")
    
    # 验证加载是否成功
    if new_doc_processor.categories == new_categories:
        logger.info("分类加载成功!")
    else:
        logger.error("分类加载失败!")
        
    return True

if __name__ == "__main__":
    test_categories_persistence()