纳速健身

标题: [pyqt5][整理创新]pyqt5在QTableWidget表头中加入复选框并实现全选功能 [打印本页]
作者: awagink    时间: 2021-1-12 09:15
标题: [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类，设置表头中加入复选框、信号和信号槽方法。

演示效果
(, 下载次数: 0)

(, 下载次数: 0)

代码说明


  2. from PyQt5.QtWidgets import QApplication, QWidget,  QVBoxLayout,  QTableWidget, QCheckBox, QHeaderView, QStyle, QStyleOptionButton, QTableWidgetItem
  3. from PyQt5.QtCore import Qt, pyqtSignal, QRect

  5. # 表头字段，全局变量
  6. header_field = ['全选', '文件名', '文件路径']
  7. # 用来装行表头所有复选框 全局变量
  8. all_header_combobox = []


  11. class CheckBoxHeader(QHeaderView):
  12.     """自定义表头类"""

  14.     # 自定义 复选框全选信号
  15.     select_all_clicked = pyqtSignal(bool)
  16.     # 这4个变量控制列头复选框的样式，位置以及大小
  17.     _x_offset = 0
  18.     _y_offset = 0
  19.     _width = 20
  20.     _height = 20

  22.     def __init__(self, orientation=Qt.Horizontal, parent=None):
  23.         super(CheckBoxHeader, self).__init__(orientation, parent)
  24.         self.isOn = False

  26.     def paintSection(self, painter, rect, logicalIndex):
  27.         painter.save()
  28.         super(CheckBoxHeader, self).paintSection(painter, rect, logicalIndex)
  29.         painter.restore()

  31.         self._y_offset = int((rect.height() - self._width) / 2.)

  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)

  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)

  55.                 self.updateSection(0)
  56.         super(CheckBoxHeader, self).mousePressEvent(event)

  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):

  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中比较合适。

  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.         '''

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

  13.             for row_num in range(len(table_data_list)):  # 注意上面列表中数字加单引号，否则下面不显示(或者下面str方法转化一下即可)

  15.                 item = table_data_list[row_num]

  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)

  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)

  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)

  6.         # 调用UI_init方法，初始化ui
  7.         self.UI_init()

  9.         # 显示gui，没有则不显示窗口
  10.         self.show()
复制代码






欢迎光临 纳速健身 (https://www.nasue.com/) Powered by Discuz! X3.4