DCS_V1.0/UI/UserManage/UserWidget.py

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_())