320 lines
15 KiB
Python
320 lines
15 KiB
Python
"""
|
||
配置管理模块
|
||
|
||
负责应用程序的配置管理，包括配置的加载、保存和默认值设置。
|
||
支持文件处理、文本处理、图片处理和文档格式等各类配置。
|
||
"""
|
||
|
||
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.max_sentences_per_paragraph = 0 # 每段最大句子数，0表示不限制
|
||
|
||
# 错别字处理配置
|
||
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)
|
||
# 新增段落句子数控制配置
|
||
self.max_sentences_per_paragraph = section.getint('max_sentences_per_paragraph', self.max_sentences_per_paragraph)
|
||
|
||
# 加载图片处理配置
|
||
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,
|
||
'max_sentences_per_paragraph': str(self.max_sentences_per_paragraph) # 新增配置项
|
||
}
|
||
|
||
# 保存图片处理配置
|
||
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)
|
||
# 新增段落句子数控制配置
|
||
self.max_sentences_per_paragraph = tp.get('max_sentences_per_paragraph', self.max_sentences_per_paragraph)
|
||
|
||
# 图片处理配置
|
||
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.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.max_sentences_per_paragraph = 0 # 每段最大句子数，0表示不限制
|
||
|
||
# 错别字处理配置
|
||
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 # 是否使用自定义样式
|
||
|
||
# 全局配置实例
|
||
CONFIG_FILE_PATH = os.path.join(os.path.expanduser("~"), ".txt2md2docx.ini")
|
||
config = Config()
|
||
|
||
# 自动加载配置
|
||
config.load_from_file(CONFIG_FILE_PATH) |