from PyQt5 import QtCore, QtWidgets from PyQt5.QtCore import QSize, Qt from PyQt5.QtGui import QIcon from PyQt5.QtWidgets import QSplitter, QPushButton import qtawesome as qta from .ProjectTable import ProjectTableView from model.ProjectModel.ProjectManage import ProjectManage class ProjectWidgets(QtWidgets.QWidget): def __init__(self): super(ProjectWidgets, self).__init__() self.setAttribute(Qt.WA_StyledBackground, True) self.setupUi() def setupButtonIcon(self, button, icon_name, color): """设置按钮图标,支持不同状态的颜色变化""" # 正常状态图标 normal_icon = qta.icon(icon_name, color=color) # 悬停状态图标(稍微亮一些) hover_color = self.lightenColor(color, 0.2) hover_icon = qta.icon(icon_name, color=hover_color) # 按下状态图标(稍微暗一些) pressed_color = self.darkenColor(color, 0.2) pressed_icon = qta.icon(icon_name, color=pressed_color) # 设置图标 button.setIcon(normal_icon) # 存储不同状态的图标,用于状态切换 button._normal_icon = normal_icon button._hover_icon = hover_icon button._pressed_icon = pressed_icon # 连接事件 button.enterEvent = lambda event: self.onButtonEnter(button, event) button.leaveEvent = lambda event: self.onButtonLeave(button, event) button.mousePressEvent = lambda event: self.onButtonPress(button, event) button.mouseReleaseEvent = lambda event: self.onButtonRelease(button, event) def lightenColor(self, color, factor): """使颜色变亮""" if color.startswith('#'): # 十六进制颜色 r = int(color[1:3], 16) g = int(color[3:5], 16) b = int(color[5:7], 16) r = min(255, int(r + (255 - r) * factor)) g = min(255, int(g + (255 - g) * factor)) b = min(255, int(b + (255 - b) * factor)) return f"#{r:02x}{g:02x}{b:02x}" return color def darkenColor(self, color, factor): """使颜色变暗""" if color.startswith('#'): # 十六进制颜色 r = int(color[1:3], 16) g = int(color[3:5], 16) b = int(color[5:7], 16) r = max(0, int(r * (1 - factor))) g = max(0, int(g * (1 - factor))) b = max(0, int(b * (1 - factor))) return f"#{r:02x}{g:02x}{b:02x}" return color def onButtonEnter(self, button, event): """按钮鼠标进入事件""" if hasattr(button, '_hover_icon'): button.setIcon(button._hover_icon) # 调用原始的enterEvent QPushButton.enterEvent(button, event) def onButtonLeave(self, button, event): """按钮鼠标离开事件""" if hasattr(button, '_normal_icon'): button.setIcon(button._normal_icon) # 调用原始的leaveEvent QPushButton.leaveEvent(button, event) def onButtonPress(self, button, event): """按钮鼠标按下事件""" if hasattr(button, '_pressed_icon'): button.setIcon(button._pressed_icon) # 调用原始的mousePressEvent QPushButton.mousePressEvent(button, event) def onButtonRelease(self, button, event): """按钮鼠标释放事件""" # 检查鼠标是否还在按钮上 if button.rect().contains(event.pos()): if hasattr(button, '_hover_icon'): button.setIcon(button._hover_icon) else: if hasattr(button, '_normal_icon'): button.setIcon(button._normal_icon) # 调用原始的mouseReleaseEvent QPushButton.mouseReleaseEvent(button, event) def setupUi(self): # 新建工程按钮 self.createBtn = QPushButton('新建工程') self.createBtn.setObjectName('createBtn') self.createBtn.setIconSize(QSize(18, 18)) self.createBtn.clicked.connect(self.createProject) self.setupButtonIcon(self.createBtn, 'fa5s.plus', '#2277EF') # 导入工程按钮 self.importBtn = QPushButton('导入工程') self.importBtn.setObjectName('importBtn') self.importBtn.setIconSize(QSize(18, 18)) self.importBtn.clicked.connect(self.importProject) self.setupButtonIcon(self.importBtn, 'fa5s.file-import', '#059669') # 导出工程按钮 self.exportBtn = QPushButton('导出工程') self.exportBtn.setObjectName('exportBtn') self.exportBtn.setIconSize(QSize(18, 18)) self.exportBtn.clicked.connect(self.exportPorject) self.setupButtonIcon(self.exportBtn, 'fa5s.file-export', '#7C3AED') self.projectView = ProjectTableView() self.projectView.setObjectName('projectView') self.gridLayout = QtWidgets.QGridLayout(self) self.gridLayout.addWidget(self.createBtn, 0, 0, 1, 1) self.gridLayout.addWidget(QSplitter(), 0, 1, 1, 20) self.gridLayout.addWidget(self.importBtn, 0, 21, 1, 1) self.gridLayout.addWidget(self.exportBtn, 0, 22, 1, 1) self.gridLayout.addWidget(self.projectView, 1, 0, 10, 23) self.gridLayout.setSpacing(10) self.gridLayout.setContentsMargins(20, 30, 20, 20) self.proxy = QtCore.QSortFilterProxyModel(self) self.proxy.setSourceModel(self.projectView.model) self.projectView.setModel(self.proxy) self.projectView.proxy = self.proxy self.projectView.model.initTable() self.horizontalHeader = self.projectView.horizontalHeader() self.horizontalHeader.sectionClicked.connect(self.on_view_horizontalHeader_sectionClicked) @QtCore.pyqtSlot(int) def on_view_horizontalHeader_sectionClicked(self, logicalIndex): self.logicalIndex = logicalIndex self.on_comboBox_currentIndexChanged(self.logicalIndex) menuName = '_' + str(logicalIndex) + 'Menu' if not hasattr(self, menuName): setattr(self, menuName, QtWidgets.QMenu(self)) self.menuValues = getattr(self, menuName) menuEdit = QtWidgets.QLineEdit() inputAction = QtWidgets.QWidgetAction(self.menuValues) inputAction.setDefaultWidget(menuEdit) self.menuValues.addAction(inputAction) menuEdit.textChanged.connect(self.on_lineEdit_textChanged) self.menuValues = getattr(self, menuName) self.romoveAction(self.menuValues) self.signalMapper = QtCore.QSignalMapper(self) self.menuValues.mouseReleaseEvent = self._menu_mouseReleaseEvent actionAll = QtWidgets.QAction("All", self) actionAll.triggered.connect(self.on_actionAll_triggered) actionAll.setProperty('canHide', True) actionAll.setCheckable(True) self.menuValues.addAction(actionAll) self.menuValues.addSeparator() valuesUnique = [self.proxy.data(self.proxy.index(row, self.logicalIndex)) for row in range(self.proxy.rowCount()) ] for actionNumber, actionName in enumerate(sorted(list(set(valuesUnique)))): action = QtWidgets.QAction(str(actionName), self) self.signalMapper.setMapping(action, actionNumber) action.triggered.connect(self.signalMapper.map) action.setCheckable(True) self.menuValues.addAction(action) self.signalMapper.mapped.connect(self.on_signalMapper_mapped) headerPos =self.projectView.mapToGlobal(self.horizontalHeader.pos()) posY = headerPos.y() + self.horizontalHeader.height() posX = headerPos.x() + self.horizontalHeader.sectionPosition(self.logicalIndex) getattr(self, menuName).exec_(QtCore.QPoint(posX, posY)) @QtCore.pyqtSlot() def on_actionAll_triggered(self): # 显示全部 filterColumn = self.logicalIndex filterString = QtCore.QRegExp( "", QtCore.Qt.CaseInsensitive, QtCore.QRegExp.RegExp ) self.proxy.setFilterRegExp(filterString) self.proxy.setFilterKeyColumn(filterColumn) @QtCore.pyqtSlot(int) def on_signalMapper_mapped(self, i): stringActions = '|'.join([x.text() for x in getattr(self, '_' + str(self.logicalIndex) + 'Menu').actions() if x.isChecked()]) filterColumn = self.logicalIndex filterString = QtCore.QRegExp( stringActions, QtCore.Qt.CaseSensitive, ) self.proxy.setFilterRegExp(filterString) self.proxy.setFilterKeyColumn(filterColumn) @QtCore.pyqtSlot(str) def on_lineEdit_textChanged(self, text): # 搜索框文字变化函数 search = QtCore.QRegExp( text, QtCore.Qt.CaseInsensitive, QtCore.QRegExp.RegExp ) self.proxy.setFilterRegExp(search) @QtCore.pyqtSlot(int) def on_comboBox_currentIndexChanged(self, index): self.proxy.setFilterKeyColumn(index) def _menu_mouseReleaseEvent(self, event): action = self.menuValues.actionAt(event.pos()) if not action: # 没有找到action就交给QMenu自己处理 return QtWidgets.QMenu.mouseReleaseEvent(self.menuValues, event) if action.property('canHide'): # 如果有该属性则给菜单自己处理 return QtWidgets.QMenu.mouseReleaseEvent(self.menuValues, event) # 找到了QAction则只触发Action action.activate(action.Trigger) def romoveAction(self, menu): # 删除输入框之外的按钮1 for action in menu.actions(): if type(action) != QtWidgets.QWidgetAction: menu.removeAction(action) def _checkAction(self): # 三个action都响应该函数 self.labelInfo.setText('\n'.join(['{}\t选中:{}'.format( action.text(), action.isChecked()) for action in self.menuValues.actions()])) def createProject(self): self.projectView.model.append_data(['', '', '', '', '', '']) def exportPorject(self): checks = [i for i,x in enumerate(self.projectView.model.checkList) if x == 'Checked'] if not checks: QtWidgets.QMessageBox.warning(self, "Warning", "请勾选想要导出的工程") return # 弹出一个导出zip文件的文件选择框 zipName, _ = QtWidgets.QFileDialog.getSaveFileName(self, "Export Project", "", "Project Files (*.project)") if not zipName: return projects = [self.projectView.model.datas[i][1] for i in checks] ProjectManage.exportProject(zipName, projects) def importProject(self): # 弹出一个导入zip文件的文件选择框 zipName, _ = QtWidgets.QFileDialog.getOpenFileName(self, "Import Project", "", "Project Files (*.project)") if not zipName: return repeatProject = ProjectManage.importProject(zipName) if isinstance(repeatProject, set): # repeatProject是集合 QtWidgets.QMessageBox.warning(self, "Warning", "{}工程名称重复".format(str(repeatProject)[1:-1])) return else: # repeatProject不是集合 for name in repeatProject: proMes = ProjectManage.getByName(name) proMes.append('') self.projectView.model.append_data(proMes)