from PyQt5 import QtCore, QtWidgets from PyQt5.QtCore import QSize, Qt from PyQt5.Qt import * from PyQt5.QtGui import QIcon from PyQt5.QtWidgets import QSplitter from .UserTable import UserTableView from utils.DBModels.UserModels import User from utils import Globals from model.UserModel.UserManage import UserManage class UserWidgets(QtWidgets.QWidget): def __init__(self, parent=None): super(UserWidgets, self).__init__(parent) self.setAttribute(Qt.WA_StyledBackground, True) self.createBtn = QPushButton(QIcon('./Static/add.png'), '添加用户') self.createBtn.setObjectName('forceBtn') self.createBtn.setIconSize(QSize(22, 22)) self.createBtn.clicked.connect(self.createUser) # self.importBtn = QPushButton(QIcon('./Static/import.png'), '导入变量') # self.importBtn.setObjectName('importBtn') # self.importBtn.setIconSize(QSize(22, 22)) self.delBtn = QPushButton(QIcon('./Static/delete.png'), '批量删除') self.delBtn.setObjectName('delBtn') self.delBtn.setIconSize(QSize(22, 22)) self.delBtn.clicked.connect(self.deleteUser) self.userView = UserTableView() self.userView.setObjectName('userView') self.proxy = QtCore.QSortFilterProxyModel(self) self.userView.proxy = self.proxy self.proxy.setSourceModel(self.userView.model) self.userView.setModel(self.proxy) Globals.setValue('userTable', self.userView) # [self.userView.model.append_data([x,'测试','测试','测试','测试','测试','测试','测试','测试','测试']) for x in range(10)] # self.model = VarTableModel(self.dic['header'], self.dic['data']) # self.varView.setModel(self.model) self.gridLayout = QtWidgets.QGridLayout(self) self.gridLayout.addWidget(self.createBtn, 0, 0, 1, 1) self.gridLayout.addWidget(QSplitter(), 0, 1, 1, 21) # self.gridLayout.addWidget(self.importBtn, 0, 21, 1, 1) self.gridLayout.addWidget(self.delBtn, 0, 22, 1, 1) self.gridLayout.addWidget(self.userView, 1, 0, 10, 23) self.gridLayout.setSpacing(10) self.gridLayout.setContentsMargins(20, 30, 20, 20) self.t_list = User.select() #for i in self.t_list: # self.userView.model.append_data([i.id, i.userName, i.userPwd, i.description, i.createTime]) # self.comboBox.currentIndexChanged.connect(self.on_comboBox_currentIndexChanged) self.horizontalHeader = self.userView.horizontalHeader() self.horizontalHeader.sectionClicked.connect(self.on_view_horizontalHeader_sectionClicked) def createUser(self): self.userView.model.append_data(['', '', '', '', '', '0', '0', '0', '0', '']) def deleteUser(self): # print('sss', self.userView.model.checkList) check = [i for i,x in enumerate(self.userView.model.checkList) if x == 'Checked'] check.sort(reverse=True) for i in check: model = self.userView.model #checkrow = self.userView.model.checkList.index(i) # print(i) name = str(model.datas[i][1]) UserManage.deleteUser(name=name) model.remove_row(i) @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.userView.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): # stringAction = self.signalMapper.mapping(i).text() stringActions = '|'.join([x.text() for x in getattr(self, '_' + str(self.logicalIndex) + 'Menu').actions() if x.isChecked()]) filterColumn = self.logicalIndex print(stringActions) filterString = QtCore.QRegExp( stringActions, QtCore.Qt.CaseSensitive, # QtCore.QRegExp.FixedString ) 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()])) if __name__ == "__main__": import sys app = QtWidgets.QApplication(sys.argv) main = UserWidget() main.show() main.resize(400, 600) sys.exit(app.exec_())