TxT2Docx/config.py

"""
配置管理模块

负责应用程序的配置管理，包括配置的加载、保存和默认值设置。
支持文件处理、文本处理、图片处理和文档格式等各类配置。
"""

import os
import configparser
from typing import Dict, Any


class Config:
    """配置管理类，统一管理应用程序的所有配置项"""
    
    def __init__(self):
        """初始化配置，设置所有默认值"""
        # 文件处理配置
        self.txt_encoding = "utf-8"
        self.match_pattern = "exact"  # exact: 完全匹配, prefix: 前缀匹配, contains: 包含
        self.output_location = "txt_folder"  # txt_folder or custom
        
        # 最近使用的文件夹路径
        self.last_txt_folder = ""
        self.last_images_root = ""
        self.last_output_root = ""
        
        # 文字处理配置
        self.reverse_text_order = False  # 转换文字顺序开关
        self.replace_punctuation = False  # 是否替换标点符号
        self.add_disclaimer = False  # 是否添加免责声明
        
        # 错别字处理配置
        self.enable_char_errors = False  # 是否启用错别字处理
        self.char_error_intensity = 0.3  # 错别字强度 0.0-1.0
        self.char_error_db_path = "data/error_chars.json"  # 错别字库路径
        
        # 图片处理配置
        self.image_sort_by = "name"  # name or time
        self.image_resize = "none"  # none or width
        self.image_width = 6  # 英寸
        self.image_alignment = "center"  # left, center, right
        self.image_strategy = "cycle"  # cycle, truncate, repeat_last
        self.image_insert_position = "after_title"  # before_title, after_title (有标题时)
        self.image_insert_interval = 5  # 无标题时每隔几段插入一张图片
        
        # 文档格式配置
        self.line_spacing = 1.5
        self.title_levels = 6  # 支持的最大标题层级
        
        # 排版样式配置
        self.current_style = "爆款文章风格"  # 当前选中的样式
        self.use_custom_style = False  # 是否使用自定义样式

    def load_from_file(self, file_path: str) -> bool:
        """
        从配置文件加载配置
        
        Args:
            file_path: 配置文件路径
            
        Returns:
            bool: 是否成功加载
        """
        if not os.path.exists(file_path):
            return False

        try:
            config_parser = configparser.ConfigParser()
            config_parser.read(file_path, encoding='utf-8')

            # 加载文件处理配置
            if 'FileHandling' in config_parser:
                section = config_parser['FileHandling']
                self.txt_encoding = section.get('txt_encoding', self.txt_encoding)
                self.match_pattern = section.get('match_pattern', self.match_pattern)
                self.output_location = section.get('output_location', self.output_location)
                self.last_txt_folder = section.get('last_txt_folder', self.last_txt_folder)
                self.last_images_root = section.get('last_images_root', self.last_images_root)
                self.last_output_root = section.get('last_output_root', self.last_output_root)

            # 加载文字处理配置
            if 'TextProcessing' in config_parser:
                section = config_parser['TextProcessing']
                self.reverse_text_order = section.getboolean('reverse_text_order', self.reverse_text_order)
                self.replace_punctuation = section.getboolean('replace_punctuation', self.replace_punctuation)
                self.add_disclaimer = section.getboolean('add_disclaimer', self.add_disclaimer)
                self.enable_char_errors = section.getboolean('enable_char_errors', self.enable_char_errors)
                self.char_error_intensity = section.getfloat('char_error_intensity', self.char_error_intensity)
                self.char_error_db_path = section.get('char_error_db_path', self.char_error_db_path)

            # 加载图片处理配置
            if 'ImageProcessing' in config_parser:
                section = config_parser['ImageProcessing']
                self.image_sort_by = section.get('image_sort_by', self.image_sort_by)
                self.image_resize = section.get('image_resize', self.image_resize)
                self.image_width = section.getfloat('image_width', self.image_width)
                self.image_alignment = section.get('image_alignment', self.image_alignment)
                self.image_strategy = section.get('image_strategy', self.image_strategy)
                # 新增的图片插入配置
                self.image_insert_position = section.get('image_insert_position', self.image_insert_position)
                self.image_insert_interval = section.getint('image_insert_interval', self.image_insert_interval)

            # 加载文档格式配置
            if 'DocumentFormat' in config_parser:
                section = config_parser['DocumentFormat']
                self.line_spacing = section.getfloat('line_spacing', self.line_spacing)
                self.title_levels = section.getint('title_levels', self.title_levels)
                self.current_style = section.get('current_style', self.current_style)
                self.use_custom_style = section.getboolean('use_custom_style', self.use_custom_style)

            return True
            
        except Exception as e:
            print(f"加载配置文件失败: {e}")
            return False

    def save_to_file(self, file_path: str) -> bool:
        """
        保存配置到文件
        
        Args:
            file_path: 配置文件路径
            
        Returns:
            bool: 是否成功保存
        """
        try:
            config_parser = configparser.ConfigParser()

            # 保存文件处理配置
            config_parser['FileHandling'] = {
                'txt_encoding': self.txt_encoding,
                'match_pattern': self.match_pattern,
                'output_location': self.output_location,
                'last_txt_folder': self.last_txt_folder,
                'last_images_root': self.last_images_root,
                'last_output_root': self.last_output_root
            }

            # 保存文字处理配置
            config_parser['TextProcessing'] = {
                'reverse_text_order': str(self.reverse_text_order),
                'replace_punctuation': str(self.replace_punctuation),
                'add_disclaimer': str(self.add_disclaimer),
                'enable_char_errors': str(self.enable_char_errors),
                'char_error_intensity': str(self.char_error_intensity),
                'char_error_db_path': self.char_error_db_path
            }

            # 保存图片处理配置
            config_parser['ImageProcessing'] = {
                'image_sort_by': self.image_sort_by,
                'image_resize': self.image_resize,
                'image_width': str(self.image_width),
                'image_alignment': self.image_alignment,
                'image_strategy': self.image_strategy,
                'image_insert_position': self.image_insert_position,
                'image_insert_interval': str(self.image_insert_interval)
            }

            # 保存文档格式配置
            config_parser['DocumentFormat'] = {
                'line_spacing': str(self.line_spacing),
                'title_levels': str(self.title_levels),
                'current_style': self.current_style,
                'use_custom_style': str(self.use_custom_style)
            }

            # 确保目录存在
            os.makedirs(os.path.dirname(file_path), exist_ok=True)
            
            with open(file_path, 'w', encoding='utf-8') as f:
                config_parser.write(f)

            return True
            
        except Exception as e:
            print(f"保存配置文件失败: {e}")
            return False

    def to_dict(self) -> Dict[str, Any]:
        """
        将配置转换为字典格式
        
        Returns:
            Dict[str, Any]: 配置字典
        """
        return {
            'file_handling': {
                'txt_encoding': self.txt_encoding,
                'match_pattern': self.match_pattern,
                'output_location': self.output_location,
                'last_txt_folder': self.last_txt_folder,
                'last_images_root': self.last_images_root,
                'last_output_root': self.last_output_root
            },
            'text_processing': {
                'reverse_text_order': self.reverse_text_order,
                'replace_punctuation': self.replace_punctuation,
                'add_disclaimer': self.add_disclaimer,
                'enable_char_errors': self.enable_char_errors,
                'char_error_intensity': self.char_error_intensity,
                'char_error_db_path': self.char_error_db_path
            },
            'image_processing': {
                'image_sort_by': self.image_sort_by,
                'image_resize': self.image_resize,
                'image_width': self.image_width,
                'image_alignment': self.image_alignment,
                'image_strategy': self.image_strategy,
                'image_insert_position': self.image_insert_position,
                'image_insert_interval': self.image_insert_interval
            },
            'document_format': {
                'line_spacing': self.line_spacing,
                'title_levels': self.title_levels,
                'current_style': self.current_style,
                'use_custom_style': self.use_custom_style
            }
        }

    def from_dict(self, config_dict: Dict[str, Any]) -> None:
        """
        从字典加载配置
        
        Args:
            config_dict: 配置字典
        """
        # 文件处理配置
        if 'file_handling' in config_dict:
            fh = config_dict['file_handling']
            self.txt_encoding = fh.get('txt_encoding', self.txt_encoding)
            self.match_pattern = fh.get('match_pattern', self.match_pattern)
            self.output_location = fh.get('output_location', self.output_location)
            self.last_txt_folder = fh.get('last_txt_folder', self.last_txt_folder)
            self.last_images_root = fh.get('last_images_root', self.last_images_root)
            self.last_output_root = fh.get('last_output_root', self.last_output_root)

        # 文字处理配置
        if 'text_processing' in config_dict:
            tp = config_dict['text_processing']
            self.reverse_text_order = tp.get('reverse_text_order', self.reverse_text_order)
            self.replace_punctuation = tp.get('replace_punctuation', self.replace_punctuation)
            self.add_disclaimer = tp.get('add_disclaimer', self.add_disclaimer)
            self.enable_char_errors = tp.get('enable_char_errors', self.enable_char_errors)
            self.char_error_intensity = tp.get('char_error_intensity', self.char_error_intensity)
            self.char_error_db_path = tp.get('char_error_db_path', self.char_error_db_path)

        # 图片处理配置
        if 'image_processing' in config_dict:
            ip = config_dict['image_processing']
            self.image_sort_by = ip.get('image_sort_by', self.image_sort_by)
            self.image_resize = ip.get('image_resize', self.image_resize)
            self.image_width = ip.get('image_width', self.image_width)
            self.image_alignment = ip.get('image_alignment', self.image_alignment)
            self.image_strategy = ip.get('image_strategy', self.image_strategy)
            self.image_insert_position = ip.get('image_insert_position', self.image_insert_position)
            self.image_insert_interval = ip.get('image_insert_interval', self.image_insert_interval)

        # 文档格式配置
        if 'document_format' in config_dict:
            df = config_dict['document_format']
            self.line_spacing = df.get('line_spacing', self.line_spacing)
            self.title_levels = df.get('title_levels', self.title_levels)
            self.current_style = df.get('current_style', self.current_style)
            self.use_custom_style = df.get('use_custom_style', self.use_custom_style)

    def reset_to_defaults(self) -> None:
        """重置所有配置为默认值"""
        self.__init__()


# 全局配置实例
CONFIG_FILE_PATH = os.path.join(os.path.expanduser("~"), ".txt2md2docx.ini")
config = Config()

# 自动加载配置
config.load_from_file(CONFIG_FILE_PATH)