|
|
from PySide6.QtWidgets import (
|
|
|
QDialog, QVBoxLayout, QHBoxLayout, QTabWidget,
|
|
|
QPushButton, QLabel, QComboBox, QCheckBox,
|
|
|
QGroupBox, QSpinBox, QRadioButton, QMessageBox,
|
|
|
QWidget
|
|
|
)
|
|
|
from PySide6.QtCore import Qt
|
|
|
|
|
|
from src.core.config import Config
|
|
|
from src.core.ocr_engine import OCREngine
|
|
|
|
|
|
class SettingsDialog(QDialog):
|
|
|
"""
|
|
|
设置对话框
|
|
|
|
|
|
提供应用程序各项设置的配置界面，包括常规设置、OCR设置和界面设置。
|
|
|
使用选项卡组织不同类别的设置，提供直观的设置界面。
|
|
|
设置更改后保存到配置文件中，供应用程序其他部分使用。
|
|
|
"""
|
|
|
|
|
|
def __init__(self, parent=None):
|
|
|
"""
|
|
|
初始化设置对话框
|
|
|
|
|
|
创建配置实例，设置窗口基本属性，初始化UI组件。
|
|
|
|
|
|
Args:
|
|
|
parent: 父窗口，默认为None
|
|
|
"""
|
|
|
super().__init__(parent)
|
|
|
self.setWindowTitle("设置")
|
|
|
self.resize(500, 400)
|
|
|
|
|
|
self.config = Config()
|
|
|
self.init_ui()
|
|
|
|
|
|
def init_ui(self):
|
|
|
"""
|
|
|
初始化用户界面
|
|
|
|
|
|
创建选项卡式布局，包含常规、OCR和界面三个选项卡，
|
|
|
以及确定和取消按钮。
|
|
|
"""
|
|
|
# 主布局
|
|
|
main_layout = QVBoxLayout(self)
|
|
|
|
|
|
# 创建选项卡
|
|
|
tab_widget = QTabWidget()
|
|
|
main_layout.addWidget(tab_widget)
|
|
|
|
|
|
# 常规选项卡
|
|
|
general_tab = QWidget()
|
|
|
tab_widget.addTab(general_tab, "常规")
|
|
|
self.setup_general_tab(general_tab)
|
|
|
|
|
|
# OCR选项卡
|
|
|
ocr_tab = QWidget()
|
|
|
tab_widget.addTab(ocr_tab, "OCR")
|
|
|
self.setup_ocr_tab(ocr_tab)
|
|
|
|
|
|
# 界面选项卡
|
|
|
ui_tab = QWidget()
|
|
|
tab_widget.addTab(ui_tab, "界面")
|
|
|
self.setup_ui_tab(ui_tab)
|
|
|
|
|
|
# 按钮区域
|
|
|
button_layout = QHBoxLayout()
|
|
|
button_layout.addStretch()
|
|
|
|
|
|
self.ok_btn = QPushButton("确定")
|
|
|
self.ok_btn.clicked.connect(self.accept)
|
|
|
self.cancel_btn = QPushButton("取消")
|
|
|
self.cancel_btn.clicked.connect(self.reject)
|
|
|
|
|
|
button_layout.addWidget(self.ok_btn)
|
|
|
button_layout.addWidget(self.cancel_btn)
|
|
|
|
|
|
main_layout.addLayout(button_layout)
|
|
|
|
|
|
def setup_general_tab(self, tab):
|
|
|
"""
|
|
|
设置常规选项卡
|
|
|
|
|
|
创建并布局常规设置选项，包括启动选项和文件历史设置。
|
|
|
|
|
|
Args:
|
|
|
tab: 要设置的选项卡控件
|
|
|
"""
|
|
|
layout = QVBoxLayout(tab)
|
|
|
|
|
|
# 启动选项
|
|
|
startup_group = QGroupBox("启动选项")
|
|
|
startup_layout = QVBoxLayout(startup_group)
|
|
|
|
|
|
self.check_update_cb = QCheckBox("启动时检查更新")
|
|
|
self.check_update_cb.setChecked(self.config.get('general.check_update_on_startup', False))
|
|
|
|
|
|
self.show_welcome_cb = QCheckBox("显示欢迎页面")
|
|
|
self.show_welcome_cb.setChecked(self.config.get('general.show_welcome', True))
|
|
|
|
|
|
self.remember_window_cb = QCheckBox("记住窗口大小和位置")
|
|
|
self.remember_window_cb.setChecked(self.config.get('general.remember_window_geometry', True))
|
|
|
|
|
|
startup_layout.addWidget(self.check_update_cb)
|
|
|
startup_layout.addWidget(self.show_welcome_cb)
|
|
|
startup_layout.addWidget(self.remember_window_cb)
|
|
|
|
|
|
# 文件历史
|
|
|
history_group = QGroupBox("文件历史")
|
|
|
history_layout = QVBoxLayout(history_group)
|
|
|
|
|
|
recent_files_layout = QHBoxLayout()
|
|
|
recent_files_layout.addWidget(QLabel("最近文件数量:"))
|
|
|
self.recent_files_spin = QSpinBox()
|
|
|
self.recent_files_spin.setRange(0, 30)
|
|
|
self.recent_files_spin.setValue(self.config.get('general.max_recent_files', 10))
|
|
|
recent_files_layout.addWidget(self.recent_files_spin)
|
|
|
recent_files_layout.addStretch()
|
|
|
|
|
|
self.clear_history_btn = QPushButton("清除历史记录")
|
|
|
self.clear_history_btn.clicked.connect(self.clear_history)
|
|
|
|
|
|
history_layout.addLayout(recent_files_layout)
|
|
|
history_layout.addWidget(self.clear_history_btn)
|
|
|
|
|
|
layout.addWidget(startup_group)
|
|
|
layout.addWidget(history_group)
|
|
|
layout.addStretch()
|
|
|
|
|
|
def setup_ocr_tab(self, tab):
|
|
|
"""
|
|
|
设置OCR选项卡
|
|
|
|
|
|
创建并布局OCR设置选项，包括默认语言设置、处理选项和输出类型设置。
|
|
|
|
|
|
Args:
|
|
|
tab: 要设置的选项卡控件
|
|
|
"""
|
|
|
layout = QVBoxLayout(tab)
|
|
|
|
|
|
# 默认语言
|
|
|
language_group = QGroupBox("默认OCR语言")
|
|
|
language_layout = QVBoxLayout(language_group)
|
|
|
|
|
|
self.language_combo = QComboBox()
|
|
|
self.language_combo.setToolTip("选择默认的OCR识别语言")
|
|
|
|
|
|
# 添加可用的语言
|
|
|
ocr_engine = OCREngine()
|
|
|
# 常用语言列表
|
|
|
common_langs = ['eng', 'chi_sim', 'chi_tra', 'jpn', 'kor']
|
|
|
|
|
|
# 首先添加常用语言
|
|
|
if ocr_engine.available_languages:
|
|
|
# 添加常用语言组
|
|
|
common_available = [lang for lang in common_langs if lang in ocr_engine.available_languages]
|
|
|
if common_available:
|
|
|
self.language_combo.addItem("--- 常用语言 ---", None)
|
|
|
for lang_code in common_available:
|
|
|
lang_name = ocr_engine.get_language_name(lang_code)
|
|
|
self.language_combo.addItem(lang_name, lang_code)
|
|
|
|
|
|
# 添加其他语言组
|
|
|
other_available = [lang for lang in ocr_engine.available_languages
|
|
|
if lang not in common_langs]
|
|
|
if other_available:
|
|
|
self.language_combo.addItem("--- 其他语言 ---", None)
|
|
|
# 按名称排序
|
|
|
other_langs_sorted = sorted(
|
|
|
other_available,
|
|
|
key=lambda x: ocr_engine.get_language_name(x)
|
|
|
)
|
|
|
for lang_code in other_langs_sorted:
|
|
|
lang_name = ocr_engine.get_language_name(lang_code)
|
|
|
self.language_combo.addItem(lang_name, lang_code)
|
|
|
else:
|
|
|
# 如果没有常用语言，直接添加所有语言
|
|
|
for lang_code in ocr_engine.available_languages:
|
|
|
lang_name = ocr_engine.get_language_name(lang_code)
|
|
|
self.language_combo.addItem(lang_name, lang_code)
|
|
|
|
|
|
# 设置当前默认语言
|
|
|
default_lang = self.config.get('default_options.language', 'eng')
|
|
|
index = self.language_combo.findData(default_lang)
|
|
|
if index >= 0:
|
|
|
self.language_combo.setCurrentIndex(index)
|
|
|
|
|
|
# 添加刷新语言列表按钮
|
|
|
lang_buttons_layout = QHBoxLayout()
|
|
|
self.refresh_langs_btn = QPushButton("刷新语言列表")
|
|
|
self.refresh_langs_btn.clicked.connect(self.refresh_languages)
|
|
|
lang_buttons_layout.addWidget(self.refresh_langs_btn)
|
|
|
|
|
|
language_layout.addWidget(self.language_combo)
|
|
|
language_layout.addLayout(lang_buttons_layout)
|
|
|
|
|
|
# 默认选项
|
|
|
options_group = QGroupBox("默认处理选项")
|
|
|
options_layout = QVBoxLayout(options_group)
|
|
|
|
|
|
self.deskew_cb = QCheckBox("自动校正倾斜页面")
|
|
|
self.deskew_cb.setChecked(self.config.get('default_options.deskew', True))
|
|
|
|
|
|
self.rotate_cb = QCheckBox("自动旋转页面")
|
|
|
self.rotate_cb.setChecked(self.config.get('default_options.rotate_pages', True))
|
|
|
|
|
|
self.clean_cb = QCheckBox("清理图像")
|
|
|
self.clean_cb.setChecked(self.config.get('default_options.clean', False))
|
|
|
|
|
|
self.optimize_cb = QCheckBox("优化输出文件大小")
|
|
|
self.optimize_cb.setChecked(self.config.get('default_options.optimize', True))
|
|
|
|
|
|
options_layout.addWidget(self.deskew_cb)
|
|
|
options_layout.addWidget(self.rotate_cb)
|
|
|
options_layout.addWidget(self.clean_cb)
|
|
|
options_layout.addWidget(self.optimize_cb)
|
|
|
|
|
|
# 输出类型
|
|
|
output_group = QGroupBox("默认输出类型")
|
|
|
output_layout = QVBoxLayout(output_group)
|
|
|
|
|
|
self.output_type_combo = QComboBox()
|
|
|
self.output_type_combo.addItems(["pdf", "pdfa", "pdfa-1", "pdfa-2", "pdfa-3"])
|
|
|
self.output_type_combo.setCurrentText(self.config.get('default_options.output_type', 'pdfa'))
|
|
|
|
|
|
output_layout.addWidget(self.output_type_combo)
|
|
|
|
|
|
layout.addWidget(language_group)
|
|
|
layout.addWidget(options_group)
|
|
|
layout.addWidget(output_group)
|
|
|
layout.addStretch()
|
|
|
|
|
|
def setup_ui_tab(self, tab):
|
|
|
"""
|
|
|
设置界面选项卡
|
|
|
|
|
|
创建并布局界面设置选项，包括主题和语言设置。
|
|
|
|
|
|
Args:
|
|
|
tab: 要设置的选项卡控件
|
|
|
"""
|
|
|
layout = QVBoxLayout(tab)
|
|
|
|
|
|
# 主题设置
|
|
|
theme_group = QGroupBox("主题")
|
|
|
theme_layout = QVBoxLayout(theme_group)
|
|
|
|
|
|
self.theme_system_rb = QRadioButton("跟随系统")
|
|
|
self.theme_light_rb = QRadioButton("浅色主题")
|
|
|
self.theme_dark_rb = QRadioButton("深色主题")
|
|
|
|
|
|
current_theme = self.config.get('ui.theme', 'system')
|
|
|
if current_theme == 'light':
|
|
|
self.theme_light_rb.setChecked(True)
|
|
|
elif current_theme == 'dark':
|
|
|
self.theme_dark_rb.setChecked(True)
|
|
|
else:
|
|
|
self.theme_system_rb.setChecked(True)
|
|
|
|
|
|
theme_layout.addWidget(self.theme_system_rb)
|
|
|
theme_layout.addWidget(self.theme_light_rb)
|
|
|
theme_layout.addWidget(self.theme_dark_rb)
|
|
|
|
|
|
# 语言设置
|
|
|
language_group = QGroupBox("界面语言")
|
|
|
language_layout = QVBoxLayout(language_group)
|
|
|
|
|
|
self.ui_language_combo = QComboBox()
|
|
|
self.ui_language_combo.addItem("简体中文", "zh_CN")
|
|
|
self.ui_language_combo.addItem("English", "en_US")
|
|
|
|
|
|
current_lang = self.config.get('ui.language', 'zh_CN')
|
|
|
index = self.ui_language_combo.findData(current_lang)
|
|
|
if index >= 0:
|
|
|
self.ui_language_combo.setCurrentIndex(index)
|
|
|
|
|
|
language_note = QLabel("注：更改语言设置需要重启应用程序才能生效")
|
|
|
language_note.setStyleSheet("color: gray;")
|
|
|
|
|
|
language_layout.addWidget(self.ui_language_combo)
|
|
|
language_layout.addWidget(language_note)
|
|
|
|
|
|
layout.addWidget(theme_group)
|
|
|
layout.addWidget(language_group)
|
|
|
layout.addStretch()
|
|
|
|
|
|
def clear_history(self):
|
|
|
"""
|
|
|
清除历史记录
|
|
|
|
|
|
清空最近使用的文件和输出目录列表，并弹出确认提示。
|
|
|
"""
|
|
|
self.config.set('recent_files', [])
|
|
|
self.config.set('recent_output_dirs', [])
|
|
|
QMessageBox.information(self, "已清除", "已清除所有历史记录")
|
|
|
|
|
|
def accept(self):
|
|
|
"""
|
|
|
确定按钮点击处理
|
|
|
|
|
|
保存所有设置到配置文件中，并关闭对话框。
|
|
|
"""
|
|
|
# 保存常规设置
|
|
|
self.config.set('general.check_update_on_startup', self.check_update_cb.isChecked())
|
|
|
self.config.set('general.show_welcome', self.show_welcome_cb.isChecked())
|
|
|
self.config.set('general.remember_window_geometry', self.remember_window_cb.isChecked())
|
|
|
self.config.set('general.max_recent_files', self.recent_files_spin.value())
|
|
|
|
|
|
# 保存OCR设置
|
|
|
lang_index = self.language_combo.currentIndex()
|
|
|
lang_data = self.language_combo.itemData(lang_index)
|
|
|
|
|
|
# 如果选择了分隔符，尝试找到下一个有效选项
|
|
|
if lang_data is None:
|
|
|
for i in range(lang_index + 1, self.language_combo.count()):
|
|
|
next_data = self.language_combo.itemData(i)
|
|
|
if next_data:
|
|
|
lang_data = next_data
|
|
|
break
|
|
|
# 如果没有找到，使用默认语言
|
|
|
if lang_data is None:
|
|
|
lang_data = 'eng'
|
|
|
|
|
|
self.config.set('default_options.language', lang_data)
|
|
|
self.config.set('default_options.deskew', self.deskew_cb.isChecked())
|
|
|
self.config.set('default_options.rotate_pages', self.rotate_cb.isChecked())
|
|
|
self.config.set('default_options.clean', self.clean_cb.isChecked())
|
|
|
self.config.set('default_options.optimize', self.optimize_cb.isChecked())
|
|
|
self.config.set('default_options.output_type', self.output_type_combo.currentText())
|
|
|
|
|
|
# 保存界面设置
|
|
|
if self.theme_light_rb.isChecked():
|
|
|
theme = 'light'
|
|
|
elif self.theme_dark_rb.isChecked():
|
|
|
theme = 'dark'
|
|
|
else:
|
|
|
theme = 'system'
|
|
|
|
|
|
self.config.set('ui.theme', theme)
|
|
|
self.config.set('ui.language', self.ui_language_combo.currentData())
|
|
|
|
|
|
super().accept()
|
|
|
|
|
|
def refresh_languages(self):
|
|
|
"""
|
|
|
刷新语言列表
|
|
|
|
|
|
重新获取系统中已安装的Tesseract语言包，并更新语言下拉列表。
|
|
|
保存当前选择的语言，并在刷新后尝试恢复选择。
|
|
|
"""
|
|
|
# 保存当前选择的语言
|
|
|
current_lang = self.language_combo.currentData()
|
|
|
|
|
|
# 清空语言下拉列表
|
|
|
self.language_combo.clear()
|
|
|
|
|
|
# 重新获取语言列表
|
|
|
ocr_engine = OCREngine()
|
|
|
# 这会重新检测可用的语言
|
|
|
ocr_engine = OCREngine()
|
|
|
|
|
|
# 重新填充语言下拉列表
|
|
|
common_langs = ['eng', 'chi_sim', 'chi_tra', 'jpn', 'kor']
|
|
|
|
|
|
if ocr_engine.available_languages:
|
|
|
# 添加常用语言组
|
|
|
common_available = [lang for lang in common_langs if lang in ocr_engine.available_languages]
|
|
|
if common_available:
|
|
|
self.language_combo.addItem("--- 常用语言 ---", None)
|
|
|
for lang_code in common_available:
|
|
|
lang_name = ocr_engine.get_language_name(lang_code)
|
|
|
self.language_combo.addItem(lang_name, lang_code)
|
|
|
|
|
|
# 添加其他语言组
|
|
|
other_available = [lang for lang in ocr_engine.available_languages
|
|
|
if lang not in common_langs]
|
|
|
if other_available:
|
|
|
self.language_combo.addItem("--- 其他语言 ---", None)
|
|
|
# 按名称排序
|
|
|
other_langs_sorted = sorted(
|
|
|
other_available,
|
|
|
key=lambda x: ocr_engine.get_language_name(x)
|
|
|
)
|
|
|
for lang_code in other_langs_sorted:
|
|
|
lang_name = ocr_engine.get_language_name(lang_code)
|
|
|
self.language_combo.addItem(lang_name, lang_code)
|
|
|
else:
|
|
|
# 如果没有常用语言，直接添加所有语言
|
|
|
for lang_code in ocr_engine.available_languages:
|
|
|
lang_name = ocr_engine.get_language_name(lang_code)
|
|
|
self.language_combo.addItem(lang_name, lang_code)
|
|
|
|
|
|
# 恢复之前选择的语言
|
|
|
if current_lang:
|
|
|
index = self.language_combo.findData(current_lang)
|
|
|
if index >= 0:
|
|
|
self.language_combo.setCurrentIndex(index)
|
|
|
|
|
|
# 显示刷新结果
|
|
|
QMessageBox.information(
|
|
|
self,
|
|
|
"刷新完成",
|
|
|
f"已刷新语言列表，找到 {len(ocr_engine.available_languages)} 种语言"
|
|
|
)
|
|
|
|
|
|
def download_language_pack(self):
|
|
|
"""下载Tesseract语言包 - 已移除"""
|
|
|
pass
|
