314 lines
13 KiB
Python
314 lines
13 KiB
Python
from PySide6.QtWidgets import QMessageBox, QVBoxLayout
|
||
import logging
|
||
import json
|
||
import os
|
||
from ui.settings_ui import SettingsUI
|
||
from utils.sql_utils import SQLUtils
|
||
from widgets.inspection_settings_widget import InspectionSettingsWidget
|
||
from widgets.pallet_type_settings_widget import PalletTypeSettingsWidget
|
||
from utils.config_loader import ConfigLoader
|
||
|
||
class SettingsWidget(SettingsUI):
|
||
def __init__(self, parent=None):
|
||
super().__init__(parent)
|
||
self.parent = parent
|
||
|
||
logging.info("正在初始化SettingsWidget")
|
||
|
||
# 创建检验设置部件
|
||
logging.info("创建InspectionSettingsWidget实例")
|
||
self.inspection_settings = InspectionSettingsWidget()
|
||
|
||
# 移除临时占位符标签并添加检验设置部件
|
||
if hasattr(self, 'inspection_placeholder'):
|
||
logging.info("移除临时占位符")
|
||
self.inspection_layout.removeWidget(self.inspection_placeholder)
|
||
self.inspection_placeholder.hide()
|
||
self.inspection_placeholder.deleteLater()
|
||
else:
|
||
logging.warning("未找到临时占位符标签")
|
||
|
||
# 检查布局是否可用
|
||
if hasattr(self, 'inspection_layout'):
|
||
logging.info("添加检验设置部件到布局")
|
||
self.inspection_layout.addWidget(self.inspection_settings)
|
||
else:
|
||
logging.error("无法找到inspection_layout布局")
|
||
|
||
# 创建托盘类型设置部件
|
||
logging.info("创建PalletTypeSettingsWidget实例")
|
||
self.pallet_type_settings = PalletTypeSettingsWidget()
|
||
|
||
# 移除临时占位符标签并添加托盘类型设置部件
|
||
if hasattr(self, 'pallet_type_placeholder'):
|
||
logging.info("移除托盘类型临时占位符")
|
||
self.pallet_type_layout.removeWidget(self.pallet_type_placeholder)
|
||
self.pallet_type_placeholder.hide()
|
||
self.pallet_type_placeholder.deleteLater()
|
||
else:
|
||
logging.warning("未找到托盘类型临时占位符标签")
|
||
|
||
# 检查布局是否可用
|
||
if hasattr(self, 'pallet_type_layout'):
|
||
logging.info("添加托盘类型设置部件到布局")
|
||
self.pallet_type_layout.addWidget(self.pallet_type_settings)
|
||
else:
|
||
logging.error("无法找到pallet_type_layout布局")
|
||
|
||
# 加载配置文件
|
||
self.config_loader = ConfigLoader.get_instance()
|
||
|
||
# 连接信号和槽
|
||
self.connect_signals()
|
||
|
||
# 初始化数据库类型UI状态
|
||
self.load_db_config()
|
||
|
||
logging.info("SettingsWidget初始化完成")
|
||
|
||
def connect_signals(self):
|
||
# 数据库类型选择
|
||
self.db_type_combo.currentTextChanged.connect(self.update_db_ui_state)
|
||
|
||
# 更新当前使用的数据源
|
||
self.current_source_combo.currentTextChanged.connect(self.update_default_source)
|
||
|
||
# 按钮动作
|
||
self.test_conn_button.clicked.connect(self.test_connection)
|
||
self.save_db_button.clicked.connect(self.save_db_settings)
|
||
|
||
# 检验配置变更信号
|
||
self.inspection_settings.signal_configs_changed.connect(self.handle_inspection_configs_changed)
|
||
|
||
# 托盘类型配置变更信号
|
||
self.pallet_type_settings.signal_pallet_types_changed.connect(self.handle_pallet_types_changed)
|
||
|
||
# 返回按钮
|
||
if hasattr(self, 'back_button'):
|
||
self.back_button.clicked.connect(self.back_to_main)
|
||
|
||
def load_db_config(self):
|
||
"""加载数据库配置"""
|
||
try:
|
||
# 获取默认数据源
|
||
default_source = self.config_loader.get_value('database.default', 'sqlite').lower()
|
||
|
||
# 设置当前使用的数据源组合框
|
||
self._update_source_combo_items() # 更新组合框项目
|
||
|
||
index = self.current_source_combo.findText(default_source.capitalize(), Qt.MatchFixedString)
|
||
if index >= 0:
|
||
self.current_source_combo.setCurrentIndex(index)
|
||
|
||
# 默认选择当前使用的数据源类型
|
||
index = self.db_type_combo.findText(default_source.capitalize(), Qt.MatchFixedString)
|
||
if index >= 0:
|
||
self.db_type_combo.setCurrentIndex(index)
|
||
|
||
# 更新UI状态
|
||
self.update_db_ui_state()
|
||
|
||
except Exception as e:
|
||
logging.error(f"加载数据库配置失败: {e}")
|
||
|
||
def _update_source_combo_items(self):
|
||
"""更新数据源下拉框项目"""
|
||
try:
|
||
# 清空当前项目
|
||
self.current_source_combo.clear()
|
||
|
||
# 获取所有配置的数据源
|
||
sources = self.config_loader.get_value('database.sources', {})
|
||
|
||
# 添加数据源到下拉框
|
||
for source_name in sources.keys():
|
||
# 处理特殊情况：postgresql显示为PostgreSQL
|
||
display_name = source_name.capitalize()
|
||
self.current_source_combo.addItem(display_name, source_name)
|
||
|
||
except Exception as e:
|
||
logging.error(f"更新数据源下拉框失败: {e}")
|
||
# 添加默认项
|
||
self.current_source_combo.addItem("SQLite", "sqlite")
|
||
|
||
def update_db_ui_state(self):
|
||
"""根据选择的数据库类型更新UI状态"""
|
||
db_type = self.db_type_combo.currentText().lower()
|
||
|
||
# 处理特殊情况：PostgreSQL对应postgresql
|
||
if db_type == "postgresql":
|
||
db_type = "postgresql"
|
||
|
||
# 加载选定类型的数据源配置
|
||
config_path = f"database.sources.{db_type}"
|
||
db_config = {}
|
||
|
||
if db_type == "sqlite":
|
||
# SQLite模式下，只需要数据库文件路径和说明
|
||
path = self.config_loader.get_value(f"{config_path}.path", "db/jtDB.db")
|
||
description = self.config_loader.get_value(f"{config_path}.description", "")
|
||
|
||
self.host_input.setEnabled(False)
|
||
self.host_input.setText("")
|
||
self.user_input.setEnabled(False)
|
||
self.user_input.setText("")
|
||
self.password_input.setEnabled(False)
|
||
self.password_input.setText("")
|
||
self.port_input.setEnabled(False)
|
||
self.port_input.setText("")
|
||
self.database_input.setEnabled(True)
|
||
self.database_input.setText(path)
|
||
self.desc_input.setText(description)
|
||
|
||
elif db_type == "postgresql":
|
||
# PostgreSQL模式下，需要完整的连接信息
|
||
host = self.config_loader.get_value(f"{config_path}.host", "localhost")
|
||
port = self.config_loader.get_value(f"{config_path}.port", "5432")
|
||
user = self.config_loader.get_value(f"{config_path}.user", "postgres")
|
||
password = self.config_loader.get_value(f"{config_path}.password", "")
|
||
name = self.config_loader.get_value(f"{config_path}.name", "jtDB")
|
||
description = self.config_loader.get_value(f"{config_path}.description", "")
|
||
|
||
self.host_input.setEnabled(True)
|
||
self.host_input.setText(host)
|
||
self.user_input.setEnabled(True)
|
||
self.user_input.setText(user)
|
||
self.password_input.setEnabled(True)
|
||
self.password_input.setText(password)
|
||
self.port_input.setEnabled(True)
|
||
self.port_input.setText(port)
|
||
self.database_input.setEnabled(True)
|
||
self.database_input.setText(name)
|
||
self.desc_input.setText(description)
|
||
|
||
elif db_type == "mysql":
|
||
# MySQL模式下，需要完整的连接信息
|
||
host = self.config_loader.get_value(f"{config_path}.host", "localhost")
|
||
port = self.config_loader.get_value(f"{config_path}.port", "3306")
|
||
user = self.config_loader.get_value(f"{config_path}.user", "root")
|
||
password = self.config_loader.get_value(f"{config_path}.password", "")
|
||
name = self.config_loader.get_value(f"{config_path}.name", "jtDB")
|
||
description = self.config_loader.get_value(f"{config_path}.description", "")
|
||
|
||
self.host_input.setEnabled(True)
|
||
self.host_input.setText(host)
|
||
self.user_input.setEnabled(True)
|
||
self.user_input.setText(user)
|
||
self.password_input.setEnabled(True)
|
||
self.password_input.setText(password)
|
||
self.port_input.setEnabled(True)
|
||
self.port_input.setText(port)
|
||
self.database_input.setEnabled(True)
|
||
self.database_input.setText(name)
|
||
self.desc_input.setText(description)
|
||
|
||
def update_default_source(self):
|
||
"""更新默认使用的数据源"""
|
||
default_source = self.current_source_combo.currentText().lower()
|
||
self.config_loader.set_value('database.default', default_source)
|
||
logging.info(f"已更新默认使用的数据源为: {default_source}")
|
||
|
||
def get_db_type(self):
|
||
"""获取当前选择的数据库类型"""
|
||
return self.db_type_combo.currentText().lower()
|
||
|
||
def get_connection_params(self):
|
||
"""获取数据库连接参数"""
|
||
db_type = self.get_db_type()
|
||
params = {
|
||
"database": self.database_input.text().strip()
|
||
}
|
||
|
||
if db_type != "sqlite":
|
||
params.update({
|
||
"host": self.host_input.text().strip(),
|
||
"user": self.user_input.text().strip(),
|
||
"password": self.password_input.text().strip(),
|
||
"port": self.port_input.text().strip()
|
||
})
|
||
|
||
return db_type, params
|
||
|
||
def test_connection(self):
|
||
"""测试数据库连接"""
|
||
db_type, params = self.get_connection_params()
|
||
|
||
try:
|
||
# 创建数据库连接
|
||
db = SQLUtils(db_type, **params)
|
||
|
||
# 测试连接
|
||
db.cursor.execute("SELECT 1")
|
||
|
||
# 关闭连接
|
||
db.close()
|
||
|
||
# 显示成功消息
|
||
QMessageBox.information(self, "连接成功", "数据库连接测试成功!")
|
||
logging.info(f"数据库连接测试成功，类型: {db_type}")
|
||
except Exception as e:
|
||
# 显示错误消息
|
||
QMessageBox.critical(self, "连接失败", f"数据库连接测试失败!\n\n错误: {str(e)}")
|
||
logging.error(f"数据库连接测试失败，类型: {db_type}, 错误: {str(e)}")
|
||
|
||
def save_db_settings(self):
|
||
"""保存数据库设置"""
|
||
db_type = self.get_db_type()
|
||
params = self.get_connection_params()[1]
|
||
desc = self.desc_input.text().strip()
|
||
|
||
try:
|
||
# 构建要保存的配置数据
|
||
config_path = f"database.sources.{db_type}"
|
||
|
||
if db_type == "sqlite":
|
||
self.config_loader.set_value(f"{config_path}.path", params["database"])
|
||
self.config_loader.set_value(f"{config_path}.description", desc)
|
||
else:
|
||
self.config_loader.set_value(f"{config_path}.host", params["host"])
|
||
self.config_loader.set_value(f"{config_path}.port", params["port"])
|
||
self.config_loader.set_value(f"{config_path}.user", params["user"])
|
||
self.config_loader.set_value(f"{config_path}.password", params["password"])
|
||
self.config_loader.set_value(f"{config_path}.name", params["database"])
|
||
self.config_loader.set_value(f"{config_path}.description", desc)
|
||
|
||
# 更新数据源下拉框
|
||
self._update_source_combo_items()
|
||
|
||
# 构建要显示的消息
|
||
settings_info = f"数据库类型: {db_type}\n"
|
||
for key, value in params.items():
|
||
if key != "password":
|
||
settings_info += f"{key}: {value}\n"
|
||
else:
|
||
settings_info += f"{key}: {'*' * len(value)}\n"
|
||
|
||
settings_info += f"说明: {desc}"
|
||
|
||
# 显示成功消息
|
||
QMessageBox.information(self, "设置已保存", f"数据库设置已保存!\n\n{settings_info}")
|
||
logging.info(f"数据库设置已保存，类型: {db_type}")
|
||
|
||
except Exception as e:
|
||
# 显示错误消息
|
||
QMessageBox.critical(self, "保存失败", f"保存数据库设置失败!\n\n错误: {str(e)}")
|
||
logging.error(f"保存数据库设置失败: {str(e)}")
|
||
|
||
def handle_inspection_configs_changed(self):
|
||
"""处理检验配置变更"""
|
||
logging.info("检验配置已更新")
|
||
# 如果有父窗口，通知父窗口更新检验配置
|
||
if self.parent and hasattr(self.parent, 'update_inspection_columns'):
|
||
self.parent.update_inspection_columns()
|
||
|
||
def handle_pallet_types_changed(self):
|
||
"""处理托盘类型配置变更"""
|
||
logging.info("托盘类型配置已更新")
|
||
# 如果有父窗口，通知父窗口更新托盘类型
|
||
if self.parent and hasattr(self.parent, 'update_pallet_types'):
|
||
self.parent.update_pallet_types()
|
||
|
||
def back_to_main(self):
|
||
"""返回主页"""
|
||
if self.parent and hasattr(self.parent, 'show_main_page'):
|
||
self.parent.show_main_page() |