import typing import qtawesome from PyQt5 import QtGui,QtCore,QtWidgets from PyQt5.QtCore import QAbstractTableModel, QModelIndex, Qt, QVariant, QSize from PyQt5.QtWidgets import QItemDelegate, QHBoxLayout, QWidget,QMessageBox from model.ProjectModel.ProjectManage import ProjectManage from ..LoginWidgets.LoginWidget import LoginWidget from utils import Globals class QPushButton(QtWidgets.QPushButton): def __init__(self, *__args: any,clicked = None): super(QPushButton,self).__init__(*__args) self.setCursor(Qt.PointingHandCursor) self.setIconSize(QSize(25, 25)) self.editableList = [] # 表格中可编辑项 class ProjectTableModel(QAbstractTableModel): def __init__(self, header, data: list, table = None): QAbstractTableModel.__init__(self, parent=None) self.header = header self.datas = data self.checkList = ['Unchecked'] * len(self.datas) # 表格选择列表 self.editableList = [] # 表格中可编辑项 self.supportedDragActions() self.table = table def initTable(self): self.datas = [] projectDatas = ProjectManage.getAllProject() if not projectDatas: self.table.proxy.invalidate() return for x in projectDatas: x.append('') self.datas.append(x) self.checkList = ['Unchecked'] * len(self.datas) self.table.proxy.invalidate() def append_data(self, x): self.datas.append(x) self.checkList = ['Unchecked'] * len(self.datas) self.table.proxy.invalidate() def insert_data(self, x, index): self.datas.insert(index, x) self.checkList = ['Unchecked'] * len(self.datas) self.table.proxy.invalidate() def remove_row(self, row): self.datas.pop(row) self.checkList = ['Unchecked'] * len(self.datas) self.table.proxy.invalidate() def rowCount(self, parent: QModelIndex = ...) -> int: if len(self.datas) > 0: return len(self.datas) return 0 def columnCount(self, parent: QModelIndex = ...) -> int: return len(self.header) def data(self, QModelIndex, role=Qt.DisplayRole): if role == Qt.TextAlignmentRole: return Qt.AlignCenter if not QModelIndex.isValid(): print("行或者列有问题") return QVariant() if role == Qt.BackgroundRole: if str(self.datas[QModelIndex.row()][1]) == str(Globals.getValue('currentPro')): return QtGui.QColor('#00FF7F') # return QtGui.QColor('#FFFFFF') # if role == Qt.TextColorRole: # if QModelIndex.column() == 6: # return QtGui.QColor('#000000') # return QtGui.QColor('#1A1A1A') if role == Qt.CheckStateRole: if QModelIndex.column() == 0: return Qt.Checked if self.checkList[QModelIndex.row()] == 'Checked' else Qt.Unchecked else: return None if role == Qt.ToolTipRole: if QModelIndex.column() == 0: return self.checkList[QModelIndex.row()] if role != Qt.DisplayRole: return QVariant() if role == Qt.DisplayRole or role == Qt.EditRole: if QModelIndex.row() in self.editableList: return self.datas[QModelIndex.row()][QModelIndex.column()] return QVariant(self.datas[QModelIndex.row()][QModelIndex.column()]) def headerData(self, section: int, orientation: Qt.Orientation, role: int = ...) -> typing.Any: if role != Qt.DisplayRole: return None if orientation == Qt.Horizontal: return self.header[section] def setData(self, index, value, role): row = index.row() col = index.column() if role == Qt.CheckStateRole and col == 0: self.checkList[row] = 'Checked' if value == Qt.Checked else 'Unchecked' return True if role == Qt.EditRole: self.datas[row][col] = value return True return False def flags(self, index): if index.column() == 0: return Qt.ItemIsEnabled | Qt.ItemIsUserCheckable if index.row() in self.editableList and index.column() != 3: return Qt.ItemIsEnabled | Qt.ItemIsUserCheckable | Qt.ItemIsEditable return Qt.ItemIsEnabled def headerClick(self, isOn): self.beginResetModel() if isOn: self.checkList = [] self.checkList = ['Checked'] * len(self.datas) else: self.checkList = [] self.checkList = ['UnChecked'] * len(self.datas) self.endResetModel() def dragMoveEvent(self, event): event.setDropAction(QtCore.Qt.MoveAction) event.accept() def moveRow(self, sourceParent: QModelIndex, sourceRow: int, destinationParent: QModelIndex, destinationChild: int) -> bool: if self.datas[destinationChild] == self.datas[sourceRow]: return self.datas[sourceRow], self.datas[destinationChild] = self.datas[destinationChild], self.datas[sourceRow] self.table.proxy.invalidate() class ProjectButtonDelegate(QItemDelegate): """该类用于向单元格中添加按钮 任务表格""" def __init__(self, parent=None): super(ProjectButtonDelegate, self).__init__(parent) def sizeHint(self, option, index): """设置项目的尺寸提示，确保足够的高度""" size = super().sizeHint(option, index) size.setHeight(max(60, size.height())) # 最小高度60px return size def paint(self, painter, option, index): # 首先绘制背景颜色，与BackgroundDelegate保持一致 if index.data(QtCore.Qt.BackgroundRole): height = option.rect.height() top = option.rect.top() # 减少高度调整，保持更多的绘制区域 paint_rect = option.rect paint_rect.setHeight(max(56, height - 4)) # 最小保持56px高度 paint_rect.moveTop(top + 2) painter.fillRect(paint_rect, index.data(QtCore.Qt.BackgroundRole)) # 然后创建按钮部件 if not self.parent().indexWidget(index): button1 = QPushButton( qtawesome.icon('fa.exchange', color='#1fbb6f'), "", self.parent(), clicked=self.exchange_action ) button2 = QPushButton( qtawesome.icon('fa.pencil', color='#4c8cf2'), "", self.parent(), clicked=self.edit_action ) button2.isEdit = True button2.oldName = False button3 = QPushButton( qtawesome.icon('fa.trash', color='#ff6d6d'), "", self.parent(), clicked=self.delete_action ) button1.clicked.connect(self.exchange_action) button2.clicked.connect(self.edit_action) button3.clicked.connect(self.delete_action) button1.index = [index.row(), index.column()] button2.index = [index.row(), index.column()] button3.index = [index.row(), index.column()] data = self.parent().model.datas[index.row()] for x in data: if x != '': break else: button2.isEdit = False button2.setIcon(qtawesome.icon('fa.save', color='#1fbb6f')) self.parent().model.editableList.append(button2.index[0]) h_box_layout = QHBoxLayout() h_box_layout.addWidget(button1) h_box_layout.addWidget(button2) h_box_layout.addWidget(button3) h_box_layout.setContentsMargins(0, 0, 0, 0) h_box_layout.setAlignment(Qt.AlignCenter) widget = QWidget() widget.setLayout(h_box_layout) widget.setObjectName('projectBtnWidget') # 设置widget的背景为透明，让底层的背景颜色显示出来 widget.setStyleSheet(""" QWidget#projectBtnWidget { background-color: transparent; } """) self.parent().setIndexWidget( index, widget ) def exchange_action(self): # 切换工程按钮槽函数 sender = self.sender() projectName = Globals.getValue('currentPro') if projectName == str(self.parent().model.datas[sender.index[0]][1]): reply = QMessageBox.question(self.parent(), '警告', "工程已打开", QMessageBox.Yes) return if sender.index[0] in self.parent().model.editableList: reply = QMessageBox.question(self.parent(), '警告', "请先保存工程", QMessageBox.Yes) return # 初始化读取数据库数据，添加进度条 progress = QtWidgets.QProgressDialog() progress.setWindowTitle("提示") progress.setLabelText("正在加载工程数据...") progress.setCancelButton(None) progress.setMinimum(0) progress.setMaximum(0) progress.setWindowModality(QtCore.Qt.ApplicationModal) progress.setMinimumDuration(0) progress.show() QtWidgets.QApplication.processEvents() ProjectManage.switchProject(str(self.parent().model.datas[sender.index[0]][1])) if not Globals.getValue('switchState'): pass else: Globals.getValue('MainWindows').delecteWidget() Globals.getValue('MainWindows').createWidgets() modelLists = ['ModbusTcpMasterTable', 'ModbusTcpSlaveTable', 'ModbusRtuMasterTable', \ 'ModbusRtuSlaveTable', 'HartTable', 'TcRtdTable', 'AnalogTable', 'HartSimulateTable', 'userTable'] for l in modelLists: # print(l) Globals.getValue(l).model.initTable() QtWidgets.QApplication.processEvents() Globals.clearValue('forceVars')#清除颜色 profibusTabWidgets = ['DP主站', 'DP从站', 'PA主站', 'PA从站'] for widget in profibusTabWidgets: Globals.getValue(widget).switchProject() QtWidgets.QApplication.processEvents() progress.close() QtWidgets.QApplication.processEvents() self.parent().proxy.invalidate() self.loginWidget = LoginWidget() self.loginWidget.show() Globals.setValue('switchState', True)#用来判断是否是第一次切换 def edit_action(self): sender = self.sender() model = self.parent().model # cbRow = str('cb' + str(sender.index[0]) + str(4)) # index = self.parent().model.index(sender.index[0], 4) # delegate = self.parent().itemDelegate(index) # checkbox = getattr(delegate, cbRow) if sender.isEdit: sender.setIcon(qtawesome.icon('fa.save', color='#1fbb6f')) sender.isEdit = False sender.oldName = model.datas[sender.index[0]][1] model.editableList.append(sender.index[0]) else: proMes = model.datas[sender.index[0]] name, des, = str(proMes[1]), str(proMes[2]) if not name : reply = QMessageBox.question(self.parent(), '警告', "有字段为空", QMessageBox.Yes) return if sender.oldName: if ProjectManage.editProject(name = sender.oldName, Nname = name, des = des): reply = QMessageBox.question(self.parent(), '警告', "工程名相同", QMessageBox.Yes) return else: if ProjectManage.createProject(name = name, des= des): reply = QMessageBox.question(self.parent(), '警告', "工程名相同", QMessageBox.Yes) return sender.setIcon(qtawesome.icon('fa.pencil', color='#4c8cf2')) # checkbox.setEnabled(False) sender.isEdit = True model.editableList.remove(sender.index[0]) rowIndex = sender.index[0] proMes = ProjectManage.getByName(name) proMes.append('') model.insert_data(proMes, rowIndex) model.remove_row(rowIndex + 1) self.parent().proxy.invalidate() def delete_action(self): sender = self.sender() model = self.parent().model name = str(model.datas[sender.index[0]][1]) if name == Globals.getValue('currentPro'): QMessageBox.warning(None, '提示', '不能删除当前打开的工程') return ProjectManage.deleteProject(name = name) model.remove_row(sender.index[0]) if name == Globals.getValue('currentPro'): Globals.setValue('currentPro', None) Globals.clearValue('forceVars')#清除颜色 class BackgroundDelegate(QtWidgets.QStyledItemDelegate): def __init__(self, *args): super(BackgroundDelegate, self).__init__(*args) self.setObjectName('item') def sizeHint(self, option, index): """设置项目的尺寸提示，确保足够的高度""" size = super().sizeHint(option, index) size.setHeight(max(60, size.height())) # 最小高度60px return size def paint(self, painter, option, index): # 确保绘制区域有足够的高度 if index.data(QtCore.Qt.BackgroundRole): height = option.rect.height() top = option.rect.top() # 减少高度调整，保持更多的绘制区域 option.rect.setHeight(max(56, height - 4)) # 最小保持56px高度 option.rect.moveTop(top + 2) painter.fillRect(option.rect, index.data(QtCore.Qt.BackgroundRole)) super().paint(painter, option, index)