纳速健身

标题: pyqt5 QTableView 添加 checkbox (一)checkbox选择框状态和第一列不分离版 [打印本页]

---

作者: awagink    时间: 2021-1-11 15:37
标题: pyqt5 QTableView 添加 checkbox (一)checkbox选择框状态和第一列不分离版
参考:
https://blog.csdn.net/yy123xiang/article/details/78739975

代码如下:

1. 
   
2. import sys
   
3. from PyQt5.QtWidgets import (QApplication, QHeaderView, QStyle, QStyleOptionButton, QTableView)
   
4. from PyQt5.QtCore import (pyqtSignal, Qt, QAbstractTableModel, QModelIndex, QRect, QVariant)
   
5. 
   
6. 
   
7. # https://wiki.qt.io/Technical_FAQ#How_can_I_insert_a_checkbox_into_the_header_of_my_view.3F
   
8. # https://stackoverflow.com/questions/30932528/adding-checkbox-as-vertical-header-in-qtableview
   
9. 
   
10. class CheckBoxHeader(QHeaderView):
    
11.     clicked = pyqtSignal(bool)
    
12. 
    
13.     _x_offset = 3
    
14.     _y_offset = 0
    
15.     _width = 20
    
16.     _height = 20
    
17. 
    
18.     def __init__(self, orientation=Qt.Horizontal, parent=None):
    
19.         super(CheckBoxHeader, self).__init__(orientation, parent)
    
20.         self.isOn = False
    
21. 
    
22.     def paintSection(self, painter, rect, logicalIndex):
    
23.         painter.save()
    
24.         super(CheckBoxHeader, self).paintSection(painter, rect, logicalIndex)
    
25.         painter.restore()
    
26. 
    
27.         self._y_offset = int((rect.height()-self._width)/2.)
    
28. 
    
29.         if logicalIndex == 0:
    
30.             option = QStyleOptionButton()
    
31.             option.rect = QRect(rect.x() + self._x_offset, rect.y() + self._y_offset, self._width, self._height)
    
32.             option.state = QStyle.State_Enabled | QStyle.State_Active
    
33.             if self.isOn:
    
34.                 option.state |= QStyle.State_On
    
35.             else:
    
36.                 option.state |= QStyle.State_Off
    
37.             self.style().drawControl(QStyle.CE_CheckBox, option, painter)
    
38. 
    
39.     def mousePressEvent(self, event):
    
40.         index = self.logicalIndexAt(event.pos())
    
41.         if 0 == index:
    
42.             x = self.sectionPosition(index)
    
43.             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:
    
44.                 if self.isOn:
    
45.                     self.isOn = False
    
46.                 else:
    
47.                     self.isOn = True
    
48.                 self.clicked.emit(self.isOn)
    
49.                 self.update()
    
50.         super(CheckBoxHeader, self).mousePressEvent(event)
    
51. 
    
52. 
    
53. class MyModel(QAbstractTableModel):
    
54.     def __init__(self, parent=None):
    
55.         super(MyModel, self).__init__(parent)
    
56.         # Keep track of which object are checked
    
57.         self.checkList = ['Checked', 'Unchecked']
    
58. 
    
59.     def rowCount(self, QModelIndex):
    
60.         return len(self.checkList)
    
61. 
    
62.     def columnCount(self, QModelIndex):
    
63.         return 2
    
64. 
    
65.     def data(self, index, role):
    
66.         row = index.row()
    
67.         col = index.column()
    
68.         if role == Qt.DisplayRole:
    
69.             return 'Row %d, Column %d' % (row + 1, col + 1)
    
70.         elif role == Qt.CheckStateRole:
    
71.             if col == 0:
    
72.                 return Qt.Checked if self.checkList[row] == 'Checked' else Qt.Unchecked
    
73.         elif role == Qt.ToolTipRole:
    
74.             if col == 0:
    
75.                 return self.checkList[row]
    
76.         return QVariant()
    
77. 
    
78.     def setData(self, index, value, role):
    
79.         row = index.row()
    
80.         col = index.column()
    
81.         if role == Qt.CheckStateRole and col == 0:
    
82.             self.checkList[row] = 'Checked' if value == Qt.Checked else 'Unchecked'
    
83.         return True
    
84. 
    
85.     def flags(self, index):
    
86.         if index.column() == 0:
    
87.             return Qt.ItemIsEnabled | Qt.ItemIsUserCheckable
    
88.         return Qt.ItemIsEnabled
    
89. 
    
90.     def headerData(self, section, orientation, role):
    
91.         if role == Qt.DisplayRole:
    
92.             if orientation == Qt.Horizontal:
    
93.                 if section == 0:
    
94.                     return 'Title 1'
    
95.                 elif section == 1:
    
96.                     return 'Title 2'
    
97. 
    
98.     def headerClick(self, isOn):
    
99.         self.beginResetModel()
    
100.         if isOn:
     
101.             self.checkList = ['Checked', 'Checked']
     
102.         else:
     
103.             self.checkList = ['Unchecked', 'Unchecked']
     
104.         self.endResetModel()
     
105. 
     
106. 
     
107. if __name__ == '__main__':
     
108.     a = QApplication(sys.argv)
     
109.     tableView = QTableView()
     
110.     myModel = MyModel()
     
111.     tableView.setModel(myModel)
     
112.     header = CheckBoxHeader()
     
113.     tableView.setHorizontalHeader(header)
     
114.     #tableView.horizontalHeader().setStretchLastSection(True)
     
115.     tableView.horizontalHeader().setStretchLastSection(True)
     
116.     tableView.horizontalHeader().setSectionResizeMode(0, QHeaderView.ResizeToContents)
     
117.     header.clicked.connect(myModel.headerClick)
     
118.     tableView.resize(400, 200)
     
119.     tableView.show()
     
120.     a.exec_()
     
121. 
     

复制代码

效果如下:
(, 下载次数: 0)

上传

点击文件名下载附件

---

欢迎光临 纳速健身 (https://nasue.com/)

Powered by Discuz! X3.4