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
import qtawesome as qta

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.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.createUser)
        self.setupButtonIcon(self.createBtn, 'fa5s.user-plus', '#2277EF')

        # 批量删除按钮
        self.delBtn = QPushButton('批量删除')
        self.delBtn.setObjectName('delBtn')
        self.delBtn.setIconSize(QSize(18, 18))
        self.delBtn.clicked.connect(self.deleteUser)
        self.setupButtonIcon(self.delBtn, 'fa5s.user-minus', '#DC2626')

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