[pyqt5][整理创新]pyqt5在QTableWidget表头中加入复选框并实现全选功能

我试图在tablewidget的水平(列)标题中放置一个复选框。百度很久后我试过这个：
失败方法一：

1. item = QtGui.QTableWidgetItem()
   
2. item.setCheckState(QtCore.Qt.Checked)
   
3. self.tableWidget.setHorizontalHeaderItem(1, item)

复制代码

我也试过这个：
失败方法二：

1. self.tableWidget.horizontalHeaderItem(1).setCheckState(QtCore.Qt.Checked)

复制代码

上面两个方法，这些都不会在表头水平标题中生成一个复选框。

后来查到需要用tableView的方法来设置表头，首先定义一个CheckBoxHeader类，设置表头中加入复选框、信号和信号槽方法。

演示效果




代码说明

1. 
   
2. from PyQt5.QtWidgets import QApplication, QWidget,  QVBoxLayout,  QTableWidget, QCheckBox, QHeaderView, QStyle, QStyleOptionButton, QTableWidgetItem
   
3. from PyQt5.QtCore import Qt, pyqtSignal, QRect
   
4. 
   
5. # 表头字段，全局变量
   
6. header_field = ['全选', '文件名', '文件路径']
   
7. # 用来装行表头所有复选框 全局变量
   
8. all_header_combobox = []
   
9. 
   
10. 
    
11. class CheckBoxHeader(QHeaderView):
    
12.     """自定义表头类"""
    
13. 
    
14.     # 自定义 复选框全选信号
    
15.     select_all_clicked = pyqtSignal(bool)
    
16.     # 这4个变量控制列头复选框的样式，位置以及大小
    
17.     _x_offset = 0
    
18.     _y_offset = 0
    
19.     _width = 20
    
20.     _height = 20
    
21. 
    
22.     def __init__(self, orientation=Qt.Horizontal, parent=None):
    
23.         super(CheckBoxHeader, self).__init__(orientation, parent)
    
24.         self.isOn = False
    
25. 
    
26.     def paintSection(self, painter, rect, logicalIndex):
    
27.         painter.save()
    
28.         super(CheckBoxHeader, self).paintSection(painter, rect, logicalIndex)
    
29.         painter.restore()
    
30. 
    
31.         self._y_offset = int((rect.height() - self._width) / 2.)
    
32. 
    
33.         if logicalIndex == 0:
    
34.             option = QStyleOptionButton()
    
35.             option.rect = QRect(rect.x() + self._x_offset, rect.y() + self._y_offset, self._width, self._height)
    
36.             option.state = QStyle.State_Enabled | QStyle.State_Active
    
37.             if self.isOn:
    
38.                 option.state |= QStyle.State_On
    
39.             else:
    
40.                 option.state |= QStyle.State_Off
    
41.             self.style().drawControl(QStyle.CE_CheckBox, option, painter)
    
42. 
    
43.     def mousePressEvent(self, event):
    
44.         index = self.logicalIndexAt(event.pos())
    
45.         if 0 == index:
    
46.             x = self.sectionPosition(index)
    
47.             if x + self._x_offset < event.pos().x() < x + self._x_offset + self._width and self._y_offset < event.pos().y() < self._y_offset + self._height:
    
48.                 if self.isOn:
    
49.                     self.isOn = False
    
50.                 else:
    
51.                     self.isOn = True
    
52.                     # 当用户点击了行表头复选框，发射 自定义信号 select_all_clicked()
    
53.                 self.select_all_clicked.emit(self.isOn)
    
54. 
    
55.                 self.updateSection(0)
    
56.         super(CheckBoxHeader, self).mousePressEvent(event)
    
57. 
    
58.     # 自定义信号 select_all_clicked 的槽方法
    
59.     def change_state(self, isOn):
    
60.         # 如果行表头复选框为勾选状态
    
61.         if isOn:
    
62.             # 将所有的复选框都设为勾选状态
    
63.             for i in all_header_combobox:
    
64.                 i.setCheckState(Qt.Checked)
    
65.         else:
    
66.             for i in all_header_combobox:
    
67.                 i.setCheckState(Qt.Unchecked)

复制代码

然后在UI_init函数中引入

1.     def UI_init(self):
   
2. 
   
3.         self.ui = Ui_MainWindow()
   
4.         self.ui.setupUi(self)
   
5.         # 方法一：
   
6.         header = CheckBoxHeader()               # 实例化自定义表头
   
7.         self.ui.tableWidget.setHorizontalHeader(header)            # 设置表头
   
8.         header.select_all_clicked.connect(header.change_state)        # 行表头复选框单击信号与槽
   

复制代码

在设置table数据的时候第一列加入checkbox复选框，这里也可以使用方法一，方法一的checkbox列在设置只能选中行选中的时候，选中效果兼容后面的选中列，不是一个颜色。这里经过研究在tableWedgit里采用方法二比较合适，方法一在tableView中比较合适。

1. 
   
2.     def show_items_in_table_widget(self,table_data_list):
   
3.         '''
   
4.         table_data_list的数据再table中显示,list格式，[[1,2],[1,2],[1,2]]
   
5.         '''
   
6. 
   
7.         # self.ui.tableWidget.clearContents()
   
8.         if table_data_list:
   
9.             # 重置tablewidget的行数和列数达到清零表格的目的
   
10.             self.ui.tableWidget.setRowCount(len(table_data_list))
    
11.             self.ui.tableWidget.setColumnCount(len(header_field))
    
12. 
    
13.             for row_num in range(len(table_data_list)):  # 注意上面列表中数字加单引号，否则下面不显示(或者下面str方法转化一下即可)
    
14. 
    
15.                 item = table_data_list[row_num]
    
16. 
    
17.                 # # 方法一：setCellWidget方法，每一行的第一列加入checkbox，需要self.ui.tableWidget.cellWidget(0,0).isChecked()来检测选中状态
    
18.                 # checkbox = QCheckBox()
    
19.                 # # 将所有的复选框都添加到 全局变量 all_header_combobox 中
    
20.                 # all_header_combobox.append(checkbox)
    
21.                 # self.ui.tableWidget.setCellWidget(row_num, 0, checkbox)
    
22. 
    
23.                 # 方法二：setItem,setCheckState方法，需要self.ui.tableWidget.item(row_num,0).checkState()来检测选中状态
    
24.                 check_box = QTableWidgetItem()
    
25.                 check_box.setCheckState(QtCore.Qt.Unchecked)
    
26.                 all_header_combobox.append(check_box)
    
27.                 self.ui.tableWidget.setItem(row_num, 0, check_box)
    
28. 
    
29.                 # 向table中第二列和第三列写入数据
    
30.                 for index in range(len(item)):
    
31.                     temp_item = QTableWidgetItem(str(table_data_list[row_num][index]))
    
32.                     # index+1第一列为checkbox在第二列写入数据
    
33.                     self.ui.tableWidget.setItem(row_num , index+1, temp_item)
    

复制代码

1. class MainWindow(QMainWindow):
   
2.     '''程序主函数，名称也可以为其他内容，需要在if name=__main__函数中更改对应名称'''
   
3.     def __init__(self, parent=None):
   
4.         super(MainWindow, self).__init__(parent)
   
5. 
   
6.         # 调用UI_init方法，初始化ui
   
7.         self.UI_init()
   
8. 
   
9.         # 显示gui，没有则不显示窗口
   
10.         self.show()

复制代码