You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
47 lines
1.7 KiB
Python
47 lines
1.7 KiB
Python
2 years ago
|
from PyQt5.QtWidgets import QStyle, QStyleOptionButton
|
||
|
|
from PyQt5.QtCore import pyqtSignal, Qt, QRect
|
||
|
|
from PyQt5.QtWidgets import QHeaderView
|
||
|
|
|
||
|
|
|
||
|
|
class CheckBoxHeader(QHeaderView):
|
||
|
|
clicked = pyqtSignal(bool)
|
||
|
|
|
||
|
|
_x_offset = 3
|
||
|
|
_y_offset = 0
|
||
|
|
_width = 20
|
||
|
|
_height = 20
|
||
|
|
|
||
|
|
def __init__(self, orientation=Qt.Horizontal, parent=None):
|
||
|
|
super(CheckBoxHeader, self).__init__(orientation, parent)
|
||
|
|
self.isOn = False
|
||
|
|
|
||
|
|
def paintSection(self, painter, rect, logicalIndex):
|
||
|
|
painter.save()
|
||
|
|
super(CheckBoxHeader, self).paintSection(painter, rect, logicalIndex)
|
||
|
|
painter.restore()
|
||
|
|
|
||
|
|
self._y_offset = int((rect.height()-self._width)/2.)
|
||
|
|
|
||
|
|
if logicalIndex == 0:
|
||
|
|
option = QStyleOptionButton()
|
||
|
|
option.rect = QRect(rect.x() + self._x_offset, rect.y() + self._y_offset, self._width, self._height)
|
||
|
|
option.state = QStyle.State_Enabled | QStyle.State_Active
|
||
|
|
if self.isOn:
|
||
|
|
option.state |= QStyle.State_On
|
||
|
|
else:
|
||
|
|
option.state |= QStyle.State_Off
|
||
|
|
self.style().drawControl(QStyle.CE_CheckBox, option, painter)
|
||
|
|
|
||
|
|
def mousePressEvent(self, event):
|
||
|
|
index = self.logicalIndexAt(event.pos())
|
||
|
|
if 0 == index:
|
||
|
|
x = self.sectionPosition(index)
|
||
|
|
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:
|
||
|
|
if self.isOn:
|
||
|
|
self.isOn = False
|
||
|
|
else:
|
||
|
|
self.isOn = True
|
||
|
|
self.clicked.emit(self.isOn)
|
||
|
|
self.update()
|
||
|
|
super(CheckBoxHeader, self).mousePressEvent(event)
|