376 lines
13 KiB
Python
376 lines
13 KiB
Python
|
"""
|
||
|
|
映射编辑对话框
|
||
|
|
用于创建和编辑映射关系
|
||
|
|
"""
|
||
|
|
from PyQt5.QtWidgets import (
|
||
|
|
QDialog, QVBoxLayout, QHBoxLayout, QFormLayout,
|
||
|
|
QLineEdit, QComboBox, QTextEdit, QPushButton,
|
||
|
|
QLabel, QMessageBox, QGroupBox, QListWidget,
|
||
|
|
QRadioButton, QButtonGroup, QStackedWidget, QWidget
|
||
|
|
)
|
||
|
|
from PyQt5.QtCore import Qt, pyqtSignal
|
||
|
|
|
||
|
|
from models.mapping import OperatorMapping, CodeMapping
|
||
|
|
from models.field import Field
|
||
|
|
from config import OperatorType, LanguageType
|
||
|
|
from controllers.field_controller import FieldController
|
||
|
|
from ui.widgets.code_editor_widget import CodeEditorWidget
|
||
|
|
from utils.logger import get_logger
|
||
|
|
|
||
|
|
logger = get_logger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
class MappingDialog(QDialog):
|
||
|
|
"""映射编辑对话框"""
|
||
|
|
|
||
|
|
mapping_saved = pyqtSignal(object) # OperatorMapping 或 CodeMapping
|
||
|
|
|
||
|
|
def __init__(self, parent=None, mapping=None):
|
||
|
|
super().__init__(parent)
|
||
|
|
|
||
|
|
self.mapping = mapping
|
||
|
|
self.field_controller = FieldController()
|
||
|
|
|
||
|
|
self.init_ui()
|
||
|
|
|
||
|
|
if mapping:
|
||
|
|
self.load_mapping_data(mapping)
|
||
|
|
self.setWindowTitle("编辑映射")
|
||
|
|
else:
|
||
|
|
self.setWindowTitle("新建映射")
|
||
|
|
|
||
|
|
def init_ui(self):
|
||
|
|
"""初始化UI"""
|
||
|
|
self.setModal(True)
|
||
|
|
self.setMinimumSize(700, 600)
|
||
|
|
|
||
|
|
# 主布局
|
||
|
|
main_layout = QVBoxLayout(self)
|
||
|
|
|
||
|
|
# 映射类型选择
|
||
|
|
type_group = QGroupBox("映射类型")
|
||
|
|
type_layout = QHBoxLayout(type_group)
|
||
|
|
|
||
|
|
self.type_button_group = QButtonGroup()
|
||
|
|
|
||
|
|
self.operator_radio = QRadioButton("操作符映射")
|
||
|
|
self.operator_radio.setChecked(True)
|
||
|
|
self.type_button_group.addButton(self.operator_radio, 0)
|
||
|
|
type_layout.addWidget(self.operator_radio)
|
||
|
|
|
||
|
|
self.code_radio = QRadioButton("代码映射")
|
||
|
|
self.type_button_group.addButton(self.code_radio, 1)
|
||
|
|
type_layout.addWidget(self.code_radio)
|
||
|
|
|
||
|
|
type_layout.addStretch()
|
||
|
|
|
||
|
|
main_layout.addWidget(type_group)
|
||
|
|
|
||
|
|
# 堆叠窗口（根据类型切换）
|
||
|
|
self.stacked_widget = QStackedWidget()
|
||
|
|
|
||
|
|
# 操作符映射页面
|
||
|
|
operator_page = self.create_operator_page()
|
||
|
|
self.stacked_widget.addWidget(operator_page)
|
||
|
|
|
||
|
|
# 代码映射页面
|
||
|
|
code_page = self.create_code_page()
|
||
|
|
self.stacked_widget.addWidget(code_page)
|
||
|
|
|
||
|
|
main_layout.addWidget(self.stacked_widget)
|
||
|
|
|
||
|
|
# 连接信号
|
||
|
|
self.type_button_group.buttonClicked.connect(self.on_type_changed)
|
||
|
|
|
||
|
|
# 按钮
|
||
|
|
button_layout = QHBoxLayout()
|
||
|
|
|
||
|
|
self.save_btn = QPushButton("保存")
|
||
|
|
self.save_btn.clicked.connect(self.on_save)
|
||
|
|
self.save_btn.setDefault(True)
|
||
|
|
|
||
|
|
self.cancel_btn = QPushButton("取消")
|
||
|
|
self.cancel_btn.clicked.connect(self.reject)
|
||
|
|
|
||
|
|
button_layout.addStretch()
|
||
|
|
button_layout.addWidget(self.save_btn)
|
||
|
|
button_layout.addWidget(self.cancel_btn)
|
||
|
|
|
||
|
|
main_layout.addLayout(button_layout)
|
||
|
|
|
||
|
|
def create_operator_page(self) -> QWidget:
|
||
|
|
"""创建操作符映射页面"""
|
||
|
|
page = QWidget()
|
||
|
|
layout = QVBoxLayout(page)
|
||
|
|
|
||
|
|
# 字段选择
|
||
|
|
field_group = QGroupBox("字段配置")
|
||
|
|
field_layout = QFormLayout(field_group)
|
||
|
|
|
||
|
|
# 源字段
|
||
|
|
self.source_field_combo = QComboBox()
|
||
|
|
self.load_fields_to_combo(self.source_field_combo)
|
||
|
|
field_layout.addRow("源字段*:", self.source_field_combo)
|
||
|
|
|
||
|
|
# 目标字段
|
||
|
|
self.target_field_combo = QComboBox()
|
||
|
|
self.load_fields_to_combo(self.target_field_combo)
|
||
|
|
field_layout.addRow("目标字段*:", self.target_field_combo)
|
||
|
|
|
||
|
|
layout.addWidget(field_group)
|
||
|
|
|
||
|
|
# 操作符配置
|
||
|
|
operator_group = QGroupBox("操作符配置")
|
||
|
|
operator_layout = QFormLayout(operator_group)
|
||
|
|
|
||
|
|
# 操作符类型
|
||
|
|
self.operator_combo = QComboBox()
|
||
|
|
for op_type in OperatorType:
|
||
|
|
self.operator_combo.addItem(op_type.value, op_type)
|
||
|
|
operator_layout.addRow("操作符*:", self.operator_combo)
|
||
|
|
|
||
|
|
# 操作数
|
||
|
|
self.operand_edit = QLineEdit()
|
||
|
|
self.operand_edit.setPlaceholderText("例如: 10 (用于加减乘除等操作)")
|
||
|
|
operator_layout.addRow("操作数:", self.operand_edit)
|
||
|
|
|
||
|
|
# 描述
|
||
|
|
self.op_description_edit = QTextEdit()
|
||
|
|
self.op_description_edit.setPlaceholderText("映射的详细描述...")
|
||
|
|
self.op_description_edit.setMaximumHeight(80)
|
||
|
|
operator_layout.addRow("描述:", self.op_description_edit)
|
||
|
|
|
||
|
|
layout.addWidget(operator_group)
|
||
|
|
layout.addStretch()
|
||
|
|
|
||
|
|
return page
|
||
|
|
|
||
|
|
def create_code_page(self) -> QWidget:
|
||
|
|
"""创建代码映射页面"""
|
||
|
|
page = QWidget()
|
||
|
|
layout = QVBoxLayout(page)
|
||
|
|
|
||
|
|
# 字段选择
|
||
|
|
field_group = QGroupBox("字段配置")
|
||
|
|
field_layout = QVBoxLayout(field_group)
|
||
|
|
|
||
|
|
# 源字段（可多选）
|
||
|
|
source_label = QLabel("源字段*:")
|
||
|
|
field_layout.addWidget(source_label)
|
||
|
|
|
||
|
|
self.source_fields_list = QListWidget()
|
||
|
|
self.source_fields_list.setSelectionMode(QListWidget.ExtendedSelection)
|
||
|
|
self.source_fields_list.setMaximumHeight(100)
|
||
|
|
self.load_fields_to_list(self.source_fields_list)
|
||
|
|
field_layout.addWidget(self.source_fields_list)
|
||
|
|
|
||
|
|
# 目标字段
|
||
|
|
target_layout = QHBoxLayout()
|
||
|
|
target_label = QLabel("目标字段*:")
|
||
|
|
self.code_target_field_combo = QComboBox()
|
||
|
|
self.load_fields_to_combo(self.code_target_field_combo)
|
||
|
|
target_layout.addWidget(target_label)
|
||
|
|
target_layout.addWidget(self.code_target_field_combo)
|
||
|
|
field_layout.addLayout(target_layout)
|
||
|
|
|
||
|
|
layout.addWidget(field_group)
|
||
|
|
|
||
|
|
# 代码配置
|
||
|
|
code_group = QGroupBox("代码配置")
|
||
|
|
code_layout = QVBoxLayout(code_group)
|
||
|
|
|
||
|
|
# 语言选择
|
||
|
|
lang_layout = QHBoxLayout()
|
||
|
|
lang_label = QLabel("编程语言*:")
|
||
|
|
self.language_combo = QComboBox()
|
||
|
|
for lang in LanguageType:
|
||
|
|
self.language_combo.addItem(lang.value, lang)
|
||
|
|
lang_layout.addWidget(lang_label)
|
||
|
|
lang_layout.addWidget(self.language_combo)
|
||
|
|
lang_layout.addStretch()
|
||
|
|
code_layout.addLayout(lang_layout)
|
||
|
|
|
||
|
|
# 代码编辑器
|
||
|
|
code_label = QLabel("转换代码*:")
|
||
|
|
code_layout.addWidget(code_label)
|
||
|
|
|
||
|
|
self.code_editor = CodeEditorWidget()
|
||
|
|
self.code_editor.setPlaceholderText(
|
||
|
|
"// 示例代码:\n"
|
||
|
|
"// target = source1 * 2 + source2;\n"
|
||
|
|
)
|
||
|
|
code_layout.addWidget(self.code_editor)
|
||
|
|
|
||
|
|
# 描述
|
||
|
|
desc_label = QLabel("描述:")
|
||
|
|
code_layout.addWidget(desc_label)
|
||
|
|
|
||
|
|
self.code_description_edit = QTextEdit()
|
||
|
|
self.code_description_edit.setPlaceholderText("映射的详细描述...")
|
||
|
|
self.code_description_edit.setMaximumHeight(60)
|
||
|
|
code_layout.addWidget(self.code_description_edit)
|
||
|
|
|
||
|
|
layout.addWidget(code_group)
|
||
|
|
|
||
|
|
return page
|
||
|
|
|
||
|
|
def load_fields_to_combo(self, combo: QComboBox):
|
||
|
|
"""加载字段到下拉框"""
|
||
|
|
fields, _ = self.field_controller.get_all_fields(page=1, page_size=1000)
|
||
|
|
|
||
|
|
combo.clear()
|
||
|
|
for field in fields:
|
||
|
|
combo.addItem(f"{field.full_name} ({field.type.value})", field)
|
||
|
|
|
||
|
|
def load_fields_to_list(self, list_widget: QListWidget):
|
||
|
|
"""加载字段到列表"""
|
||
|
|
fields, _ = self.field_controller.get_all_fields(page=1, page_size=1000)
|
||
|
|
|
||
|
|
list_widget.clear()
|
||
|
|
for field in fields:
|
||
|
|
from PyQt5.QtWidgets import QListWidgetItem
|
||
|
|
item = QListWidgetItem(f"{field.full_name} ({field.type.value})")
|
||
|
|
item.setData(Qt.UserRole, field)
|
||
|
|
list_widget.addItem(item)
|
||
|
|
|
||
|
|
def on_type_changed(self, button):
|
||
|
|
"""映射类型改变"""
|
||
|
|
if button == self.operator_radio:
|
||
|
|
self.stacked_widget.setCurrentIndex(0)
|
||
|
|
else:
|
||
|
|
self.stacked_widget.setCurrentIndex(1)
|
||
|
|
|
||
|
|
def load_mapping_data(self, mapping):
|
||
|
|
"""加载映射数据"""
|
||
|
|
if isinstance(mapping, OperatorMapping):
|
||
|
|
self.operator_radio.setChecked(True)
|
||
|
|
self.stacked_widget.setCurrentIndex(0)
|
||
|
|
|
||
|
|
# 设置源字段
|
||
|
|
for i in range(self.source_field_combo.count()):
|
||
|
|
field = self.source_field_combo.itemData(i)
|
||
|
|
if field.id == mapping.source_field.id:
|
||
|
|
self.source_field_combo.setCurrentIndex(i)
|
||
|
|
break
|
||
|
|
|
||
|
|
# 设置目标字段
|
||
|
|
for i in range(self.target_field_combo.count()):
|
||
|
|
field = self.target_field_combo.itemData(i)
|
||
|
|
if field.id == mapping.target_field.id:
|
||
|
|
self.target_field_combo.setCurrentIndex(i)
|
||
|
|
break
|
||
|
|
|
||
|
|
# 设置操作符
|
||
|
|
index = self.operator_combo.findData(mapping.operator)
|
||
|
|
if index >= 0:
|
||
|
|
self.operator_combo.setCurrentIndex(index)
|
||
|
|
|
||
|
|
if mapping.operand:
|
||
|
|
self.operand_edit.setText(mapping.operand)
|
||
|
|
|
||
|
|
if mapping.description:
|
||
|
|
self.op_description_edit.setText(mapping.description)
|
||
|
|
|
||
|
|
elif isinstance(mapping, CodeMapping):
|
||
|
|
self.code_radio.setChecked(True)
|
||
|
|
self.stacked_widget.setCurrentIndex(1)
|
||
|
|
|
||
|
|
# 设置源字段（多选）
|
||
|
|
for i in range(self.source_fields_list.count()):
|
||
|
|
item = self.source_fields_list.item(i)
|
||
|
|
field = item.data(Qt.UserRole)
|
||
|
|
if any(f.id == field.id for f in mapping.source_fields):
|
||
|
|
item.setSelected(True)
|
||
|
|
|
||
|
|
# 设置目标字段
|
||
|
|
for i in range(self.code_target_field_combo.count()):
|
||
|
|
field = self.code_target_field_combo.itemData(i)
|
||
|
|
if field.id == mapping.target_field.id:
|
||
|
|
self.code_target_field_combo.setCurrentIndex(i)
|
||
|
|
break
|
||
|
|
|
||
|
|
# 设置语言
|
||
|
|
index = self.language_combo.findData(mapping.language)
|
||
|
|
if index >= 0:
|
||
|
|
self.language_combo.setCurrentIndex(index)
|
||
|
|
|
||
|
|
# 设置代码
|
||
|
|
self.code_editor.setPlainText(mapping.code)
|
||
|
|
|
||
|
|
if mapping.description:
|
||
|
|
self.code_description_edit.setText(mapping.description)
|
||
|
|
|
||
|
|
def validate_operator_mapping(self) -> tuple[bool, str]:
|
||
|
|
"""验证操作符映射"""
|
||
|
|
if self.source_field_combo.currentIndex() < 0:
|
||
|
|
return False, "请选择源字段"
|
||
|
|
|
||
|
|
if self.target_field_combo.currentIndex() < 0:
|
||
|
|
return False, "请选择目标字段"
|
||
|
|
|
||
|
|
return True, ""
|
||
|
|
|
||
|
|
def validate_code_mapping(self) -> tuple[bool, str]:
|
||
|
|
"""验证代码映射"""
|
||
|
|
if len(self.source_fields_list.selectedItems()) == 0:
|
||
|
|
return False, "请至少选择一个源字段"
|
||
|
|
|
||
|
|
if self.code_target_field_combo.currentIndex() < 0:
|
||
|
|
return False, "请选择目标字段"
|
||
|
|
|
||
|
|
if not self.code_editor.toPlainText().strip():
|
||
|
|
return False, "转换代码不能为空"
|
||
|
|
|
||
|
|
return True, ""
|
||
|
|
|
||
|
|
def on_save(self):
|
||
|
|
"""保存映射"""
|
||
|
|
try:
|
||
|
|
if self.operator_radio.isChecked():
|
||
|
|
# 操作符映射
|
||
|
|
is_valid, error_msg = self.validate_operator_mapping()
|
||
|
|
if not is_valid:
|
||
|
|
QMessageBox.warning(self, "验证失败", error_msg)
|
||
|
|
return
|
||
|
|
|
||
|
|
mapping = OperatorMapping(
|
||
|
|
source_field=self.source_field_combo.currentData(),
|
||
|
|
target_field=self.target_field_combo.currentData(),
|
||
|
|
operator=self.operator_combo.currentData(),
|
||
|
|
operand=self.operand_edit.text().strip() or None,
|
||
|
|
description=self.op_description_edit.toPlainText().strip() or None
|
||
|
|
)
|
||
|
|
|
||
|
|
if self.mapping and hasattr(self.mapping, 'id'):
|
||
|
|
mapping.id = self.mapping.id
|
||
|
|
|
||
|
|
else:
|
||
|
|
# 代码映射
|
||
|
|
is_valid, error_msg = self.validate_code_mapping()
|
||
|
|
if not is_valid:
|
||
|
|
QMessageBox.warning(self, "验证失败", error_msg)
|
||
|
|
return
|
||
|
|
|
||
|
|
source_fields = [
|
||
|
|
item.data(Qt.UserRole)
|
||
|
|
for item in self.source_fields_list.selectedItems()
|
||
|
|
]
|
||
|
|
|
||
|
|
mapping = CodeMapping(
|
||
|
|
source_fields=source_fields,
|
||
|
|
target_field=self.code_target_field_combo.currentData(),
|
||
|
|
language=self.language_combo.currentData(),
|
||
|
|
code=self.code_editor.toPlainText().strip(),
|
||
|
|
description=self.code_description_edit.toPlainText().strip() or None
|
||
|
|
)
|
||
|
|
|
||
|
|
if self.mapping and hasattr(self.mapping, 'id'):
|
||
|
|
mapping.id = self.mapping.id
|
||
|
|
|
||
|
|
self.mapping_saved.emit(mapping)
|
||
|
|
self.accept()
|
||
|
|
|
||
|
|
except Exception as e:
|
||
|
|
QMessageBox.critical(self, "错误", f"创建映射对象失败: {str(e)}")
|
||
|
|
logger.error(f"Failed to create mapping object: {e}")
|