0627
parent
c1915ae41d
commit
82148397e0
@ -0,0 +1,338 @@
|
||||
import sys
|
||||
from PyQt5 import QtCore
|
||||
from PyQt5.QtCore import QAbstractTableModel, QModelIndex, pyqtSignal, QSize
|
||||
from PyQt5.QtGui import QFont, QBrush, QColor
|
||||
from PyQt5.QtWidgets import (QWidget, QVBoxLayout, QHBoxLayout, QTableView,
|
||||
QPushButton, QLabel, QLineEdit, QComboBox,
|
||||
QDialog, QFormLayout, QDialogButtonBox, QMessageBox,
|
||||
QHeaderView, QMenu, QAction, QToolBar, QSpinBox)
|
||||
import qtawesome as qta
|
||||
from datetime import datetime
|
||||
|
||||
|
||||
class KeywordTableModel(QAbstractTableModel):
|
||||
"""关键词表格模型"""
|
||||
columns = ['ID', '关键词', '操作类型', '描述', '创建时间']
|
||||
|
||||
def __init__(self, dbManager):
|
||||
super().__init__()
|
||||
self.dbManager = dbManager
|
||||
self.keywordData = []
|
||||
self.loadData()
|
||||
|
||||
def loadData(self):
|
||||
"""加载数据"""
|
||||
self.beginResetModel()
|
||||
self.keywordData = self.dbManager.getStepKeywords()
|
||||
self.endResetModel()
|
||||
|
||||
def rowCount(self, parent=QModelIndex()):
|
||||
return len(self.keywordData)
|
||||
|
||||
def columnCount(self, parent=QModelIndex()):
|
||||
return len(self.columns)
|
||||
|
||||
def data(self, index, role=QtCore.Qt.DisplayRole):
|
||||
if not index.isValid():
|
||||
return None
|
||||
|
||||
row = index.row()
|
||||
col = index.column()
|
||||
|
||||
if row >= len(self.keywordData):
|
||||
return None
|
||||
|
||||
keywordInfo = self.keywordData[row]
|
||||
|
||||
if role == QtCore.Qt.DisplayRole:
|
||||
if col == 0:
|
||||
return str(keywordInfo[0]) # ID
|
||||
elif col == 1:
|
||||
return keywordInfo[1] # 关键词
|
||||
elif col == 2:
|
||||
return keywordInfo[2] # 操作类型
|
||||
elif col == 3:
|
||||
return keywordInfo[3] or "" # 描述
|
||||
elif col == 4:
|
||||
return keywordInfo[4] # 创建时间
|
||||
|
||||
elif role == QtCore.Qt.BackgroundRole:
|
||||
if row % 2 == 0:
|
||||
return QBrush(QColor(245, 245, 245))
|
||||
else:
|
||||
return QBrush(QColor(255, 255, 255))
|
||||
|
||||
elif role == QtCore.Qt.FontRole:
|
||||
font = QFont()
|
||||
if col == 1: # 关键词列加粗
|
||||
font.setBold(True)
|
||||
return font
|
||||
|
||||
return None
|
||||
|
||||
def headerData(self, section, orientation, role):
|
||||
if role == QtCore.Qt.DisplayRole and orientation == QtCore.Qt.Horizontal:
|
||||
return self.columns[section]
|
||||
return None
|
||||
|
||||
def getKeywordInfo(self, row):
|
||||
"""获取指定行的关键词信息"""
|
||||
if 0 <= row < len(self.keywordData):
|
||||
return {
|
||||
'id': self.keywordData[row][0],
|
||||
'keyword': self.keywordData[row][1],
|
||||
'operationType': self.keywordData[row][2],
|
||||
'description': self.keywordData[row][3],
|
||||
'createdAt': self.keywordData[row][4]
|
||||
}
|
||||
return None
|
||||
|
||||
|
||||
class KeywordEditDialog(QDialog):
|
||||
"""关键词编辑对话框"""
|
||||
def __init__(self, parent=None, keywordInfo=None):
|
||||
super().__init__(parent)
|
||||
self.keywordInfo = keywordInfo
|
||||
self.isEdit = keywordInfo is not None
|
||||
|
||||
self.setWindowTitle("编辑关键词" if self.isEdit else "添加关键词")
|
||||
self.setFixedSize(400, 200)
|
||||
self.setupUi()
|
||||
|
||||
if self.isEdit:
|
||||
self.loadKeywordInfo()
|
||||
|
||||
def setupUi(self):
|
||||
"""设置界面"""
|
||||
layout = QVBoxLayout()
|
||||
|
||||
formLayout = QFormLayout()
|
||||
|
||||
self.keywordEdit = QLineEdit()
|
||||
self.keywordEdit.setPlaceholderText("请输入关键词")
|
||||
formLayout.addRow("关键词:", self.keywordEdit)
|
||||
|
||||
self.operationTypeCombo = QComboBox()
|
||||
self.operationTypeCombo.addItems(['set', 'check', 'wait', 'deltaT'])
|
||||
formLayout.addRow("操作类型:", self.operationTypeCombo)
|
||||
|
||||
self.descriptionEdit = QLineEdit()
|
||||
self.descriptionEdit.setPlaceholderText("请输入描述（可选）")
|
||||
formLayout.addRow("描述:", self.descriptionEdit)
|
||||
|
||||
layout.addLayout(formLayout)
|
||||
|
||||
buttonBox = QDialogButtonBox(QDialogButtonBox.Ok | QDialogButtonBox.Cancel)
|
||||
buttonBox.accepted.connect(self.accept)
|
||||
buttonBox.rejected.connect(self.reject)
|
||||
layout.addWidget(buttonBox)
|
||||
|
||||
self.setLayout(layout)
|
||||
|
||||
def loadKeywordInfo(self):
|
||||
"""加载关键词信息"""
|
||||
if self.keywordInfo:
|
||||
self.keywordEdit.setText(self.keywordInfo['keyword'])
|
||||
self.operationTypeCombo.setCurrentText(self.keywordInfo['operationType'])
|
||||
self.descriptionEdit.setText(self.keywordInfo['description'] or "")
|
||||
|
||||
def getKeywordData(self):
|
||||
"""获取关键词数据"""
|
||||
return {
|
||||
'keyword': self.keywordEdit.text().strip(),
|
||||
'operationType': self.operationTypeCombo.currentText(),
|
||||
'description': self.descriptionEdit.text().strip()
|
||||
}
|
||||
|
||||
|
||||
class KeywordManagerWidget(QWidget):
|
||||
"""关键词字段库管理界面"""
|
||||
def __init__(self, dbManager, parent=None):
|
||||
super().__init__(parent)
|
||||
self.dbManager = dbManager
|
||||
self.setupUi()
|
||||
self.setupToolbar()
|
||||
self.setupContextMenu()
|
||||
|
||||
def setupUi(self):
|
||||
"""设置界面"""
|
||||
layout = QVBoxLayout()
|
||||
|
||||
# 创建工具栏
|
||||
self.toolbar = QToolBar("关键词管理工具栏")
|
||||
self.toolbar.setIconSize(QSize(24, 24))
|
||||
layout.addWidget(self.toolbar)
|
||||
|
||||
# 创建表格
|
||||
self.tableModel = KeywordTableModel(self.dbManager)
|
||||
self.tableView = QTableView()
|
||||
self.tableView.setModel(self.tableModel)
|
||||
self.tableView.setSelectionBehavior(QTableView.SelectRows)
|
||||
self.tableView.setSelectionMode(QTableView.SingleSelection)
|
||||
self.tableView.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
|
||||
self.tableView.customContextMenuRequested.connect(self.showContextMenu)
|
||||
|
||||
# 设置表格样式
|
||||
self.setupTableStyle()
|
||||
|
||||
layout.addWidget(self.tableView)
|
||||
|
||||
# 创建状态栏
|
||||
self.statusLabel = QLabel("就绪")
|
||||
layout.addWidget(self.statusLabel)
|
||||
|
||||
self.setLayout(layout)
|
||||
|
||||
def setupToolbar(self):
|
||||
"""设置工具栏"""
|
||||
# 添加关键词
|
||||
addAction = QAction(qta.icon('fa5s.plus', color='green'), "添加关键词", self)
|
||||
addAction.triggered.connect(self.addKeyword)
|
||||
self.toolbar.addAction(addAction)
|
||||
|
||||
# 编辑关键词
|
||||
editAction = QAction(qta.icon('fa5s.edit', color='blue'), "编辑关键词", self)
|
||||
editAction.triggered.connect(self.editKeyword)
|
||||
self.toolbar.addAction(editAction)
|
||||
|
||||
# 删除关键词
|
||||
deleteAction = QAction(qta.icon('fa5s.trash', color='red'), "删除关键词", self)
|
||||
deleteAction.triggered.connect(self.deleteKeyword)
|
||||
self.toolbar.addAction(deleteAction)
|
||||
|
||||
self.toolbar.addSeparator()
|
||||
|
||||
# 刷新
|
||||
refreshAction = QAction(qta.icon('fa5s.sync', color='orange'), "刷新", self)
|
||||
refreshAction.triggered.connect(self.refreshData)
|
||||
self.toolbar.addAction(refreshAction)
|
||||
|
||||
def setupTableStyle(self):
|
||||
"""设置表格样式"""
|
||||
header = self.tableView.horizontalHeader()
|
||||
if header:
|
||||
header.setSectionResizeMode(0, QHeaderView.ResizeToContents) # ID
|
||||
header.setSectionResizeMode(1, QHeaderView.Stretch) # 关键词
|
||||
header.setSectionResizeMode(2, QHeaderView.ResizeToContents) # 操作类型
|
||||
header.setSectionResizeMode(3, QHeaderView.Stretch) # 描述
|
||||
header.setSectionResizeMode(4, QHeaderView.ResizeToContents) # 创建时间
|
||||
|
||||
# 设置交替行颜色
|
||||
self.tableView.setAlternatingRowColors(True)
|
||||
|
||||
# 设置表格不可编辑
|
||||
self.tableView.setEditTriggers(QTableView.NoEditTriggers)
|
||||
|
||||
def setupContextMenu(self):
|
||||
"""设置右键菜单"""
|
||||
self.tableView.customContextMenuRequested.connect(self.showContextMenu)
|
||||
|
||||
def showContextMenu(self, pos):
|
||||
"""显示右键菜单"""
|
||||
index = self.tableView.indexAt(pos)
|
||||
if not index.isValid():
|
||||
return
|
||||
|
||||
menu = QMenu()
|
||||
|
||||
editAction = menu.addAction("编辑关键词")
|
||||
editAction.triggered.connect(self.editKeyword)
|
||||
|
||||
deleteAction = menu.addAction("删除关键词")
|
||||
deleteAction.triggered.connect(self.deleteKeyword)
|
||||
|
||||
if hasattr(self.tableView, 'viewport') and hasattr(self.tableView.viewport(), 'mapToGlobal'):
|
||||
menu.exec_(self.tableView.viewport().mapToGlobal(pos))
|
||||
|
||||
def addKeyword(self):
|
||||
"""添加关键词"""
|
||||
dialog = KeywordEditDialog(self)
|
||||
if dialog.exec_() == QDialog.Accepted:
|
||||
keywordData = dialog.getKeywordData()
|
||||
|
||||
if not keywordData['keyword']:
|
||||
QMessageBox.warning(self, "输入错误", "关键词不能为空")
|
||||
return
|
||||
|
||||
keywordId = self.dbManager.addStepKeyword(
|
||||
keywordData['keyword'],
|
||||
keywordData['operationType'],
|
||||
keywordData['description']
|
||||
)
|
||||
|
||||
if keywordId:
|
||||
self.tableModel.loadData()
|
||||
self.statusLabel.setText(f"成功添加关键词: {keywordData['keyword']}")
|
||||
else:
|
||||
QMessageBox.warning(self, "添加失败", "关键词已存在或添加失败")
|
||||
|
||||
def editKeyword(self):
|
||||
"""编辑关键词"""
|
||||
currentRow = self.tableView.currentIndex().row()
|
||||
if currentRow < 0:
|
||||
QMessageBox.information(self, "提示", "请先选择要编辑的关键词")
|
||||
return
|
||||
|
||||
keywordInfo = self.tableModel.getKeywordInfo(currentRow)
|
||||
if not keywordInfo:
|
||||
return
|
||||
|
||||
dialog = KeywordEditDialog(self, keywordInfo)
|
||||
if dialog.exec_() == QDialog.Accepted:
|
||||
keywordData = dialog.getKeywordData()
|
||||
|
||||
if not keywordData['keyword']:
|
||||
QMessageBox.warning(self, "输入错误", "关键词不能为空")
|
||||
return
|
||||
|
||||
success = self.dbManager.updateStepKeyword(
|
||||
keywordInfo['id'],
|
||||
keywordData['keyword'],
|
||||
keywordData['operationType'],
|
||||
keywordData['description']
|
||||
)
|
||||
|
||||
if success:
|
||||
self.tableModel.loadData()
|
||||
self.statusLabel.setText(f"成功更新关键词: {keywordData['keyword']}")
|
||||
else:
|
||||
QMessageBox.warning(self, "更新失败", "关键词已存在或更新失败")
|
||||
|
||||
def deleteKeyword(self):
|
||||
"""删除关键词"""
|
||||
currentRow = self.tableView.currentIndex().row()
|
||||
if currentRow < 0:
|
||||
QMessageBox.information(self, "提示", "请先选择要删除的关键词")
|
||||
return
|
||||
|
||||
keywordInfo = self.tableModel.getKeywordInfo(currentRow)
|
||||
if not keywordInfo:
|
||||
return
|
||||
|
||||
reply = QMessageBox.question(
|
||||
self,
|
||||
"确认删除",
|
||||
f"确定要删除关键词 '{keywordInfo['keyword']}' 吗？",
|
||||
QMessageBox.Yes | QMessageBox.No
|
||||
)
|
||||
|
||||
if reply == QMessageBox.Yes:
|
||||
success = self.dbManager.deleteStepKeyword(keywordInfo['id'])
|
||||
if success:
|
||||
self.tableModel.loadData()
|
||||
self.statusLabel.setText(f"成功删除关键词: {keywordInfo['keyword']}")
|
||||
else:
|
||||
QMessageBox.warning(self, "删除失败", "删除关键词失败")
|
||||
|
||||
def refreshData(self):
|
||||
"""刷新数据"""
|
||||
self.tableModel.loadData()
|
||||
self.statusLabel.setText("数据已刷新")
|
||||
|
||||
def getKeywordByText(self, text):
|
||||
"""根据文本获取关键词"""
|
||||
return self.dbManager.getKeywordByText(text)
|
||||
|
||||
def getOperationTypes(self):
|
||||
"""获取所有操作类型"""
|
||||
return self.dbManager.getOperationTypes()
|
||||
File diff suppressed because it is too large
Load Diff
Loading…
Reference in New Issue