Добавить чекбоксы в заголовки столбцов таблицы

from PyQt5.QtWidgets import QApplication, QTableView, QHeaderView, QStyleOptionButton, QStyle
from PyQt5.QtGui import QStandardItemModel
from PyQt5.QtCore import Qt, pyqtSignal, QSize
 
class HeaderView(QHeaderView):
    state_changed = pyqtSignal(list)
    def __init__(self, orientation=Qt.Horizontal, parent=None):
        super().__init__(orientation, parent)
        self.setSectionsClickable(True)
        self.sectionClicked.connect(self.on_clicked)
        self.sectionCountChanged.connect(self.on_count_changed)
        
    def on_count_changed(self):
        self.state = [True for _ in range(self.count())]
        
    def on_clicked(self, idx):
        self.state[idx] ^= True
        self.state_changed.emit(self.state)
        
    def paintSection(self, painter, rect, logicalIndex):
        painter.save()
        QHeaderView.paintSection(self, painter, rect, logicalIndex)
        painter.restore()
        option = QStyleOptionButton()
        option.iconSize = QSize(10,10)
        option.rect = rect
        if self.state[logicalIndex]: option.state |= QStyle.State_On
        else: option.state |= QStyle.State_Off
        self.style().drawPrimitive(QStyle.PE_IndicatorCheckBox, option, painter)
 
 
app = QApplication([])
view = QTableView()
model=  QStandardItemModel(3,4)
view.setModel(model)
view.setHorizontalHeader(HeaderView())
view.show()
app.exec()

view = QTableWidget(3,4)
view.setHorizontalHeader(HeaderView())
view.setHorizontalHeaderLabels(['qwe qwe ','qwer','qwert','qwerty'])

        option.rect = rect

        option.rect = QRect(rect.left(),rect.top(), rect.height(),  rect.height())

    def headerData(self,section,orientation,role):
        if role == Qt.DisplayRole:
            if orientation == Qt.Horizontal:
                return section
            if orientation == Qt.Vertical:
                return section
        if role == Qt.CheckStateRole:
            print(role)# а ничего нет
            return Qt.Checked

if 'Windows' in QStyleFactory.keys():
    app.setStyle(QStyleFactory.create('Windows'))

    def paintSection(self, painter, rect, logicalIndex):
        painter.save()
        QHeaderView.paintSection(self, painter, rect, logicalIndex)
        painter.restore()
        option = QStyleOptionButton()
        option.initFrom(self)
        option.rect = QRect(rect.left(), 0, 20, 20) #!
        if self.state[logicalIndex]: option.state |= QStyle.State_On
        else: option.state |= QStyle.State_Off
        self.style().drawPrimitive(QStyle.PE_IndicatorCheckBox, option, painter)

from PyQt5.QtWidgets import (QApplication, QTableView, QTableWidget,
QHeaderView, QStyleOptionButton, QStyle, QStyleFactory)
from PyQt5.QtGui import QStandardItemModel, QFont
from PyQt5.QtCore import Qt, pyqtSignal, QSize, QRect
 
class HeaderView(QHeaderView):
    state_changed = pyqtSignal(list)
    def __init__(self, orientation=Qt.Horizontal, parent=None):
        super().__init__(orientation, parent)
        self.setSectionsClickable(True)
        self.sectionClicked.connect(self.on_clicked)
        self.sectionCountChanged.connect(self.on_count_changed)
        self.state_changed.connect(self.on_state_changed)
        
    def on_state_changed(self):
        print('selected:',*[i for i in range(self.count()) if self.state[i]])
        
    def on_count_changed(self):
        self.state = [True for _ in range(self.count())]
        
        
    def on_clicked(self, idx):
        self.state[idx] ^= True
        self.state_changed.emit(self.state)
        
    def paintSection(self, painter, rect, logicalIndex):
        painter.save()
        QHeaderView.paintSection(self, painter, rect, logicalIndex)
        painter.restore()
        option = QStyleOptionButton()
        option.initFrom(self)
        option.rect = QRect(rect.left(), rect.top(), 20, 20)
        if self.state[logicalIndex]: option.state |= QStyle.State_On
        else: option.state |= QStyle.State_Off
        self.style().drawPrimitive(QStyle.PE_IndicatorCheckBox, option, painter)
 
app = QApplication([])
print(QStyleFactory.keys())
app.setStyle('Windows')
view = QTableWidget(3,4)
head = HeaderView()
head.setSectionResizeMode(1)
view.setHorizontalHeader(head)
view.setHorizontalHeaderLabels(['один','два','шестнадцать','четыре'])
view.setFont(QFont('times',18))
view.resize(600,300)
view.show()
app.exec()

    def on_state_changed(self, lst):
        print('selected:',*[i for i in range(len(lst)) if lst[i]])