def eventFilter(o, e):
if (isinstance(o, QAbstractButton) and
e.type() == QEvent.Paint):
# paint by hand (borrowed from QTableCornerButton)
btn = o
opt = QStyleOptionHeader()
opt.initFrom(btn)
state = QStyle.State_None
if btn.isEnabled():
state |= QStyle.State_Enabled
if btn.isActiveWindow():
state |= QStyle.State_Active
if btn.isDown():
state |= QStyle.State_Sunken
opt.state = state
opt.rect = btn.rect()
opt.text = btn.text()
opt.position = QStyleOptionHeader.OnlyOneSection
painter = QStylePainter(btn)
painter.drawControl(QStyle.CE_Header, opt)
return True # eat event
return False
def set_corner_text(self, table, text):
"""Set table corner text."""
# As this is an ugly hack, do everything in
# try - except blocks, as it may stop working in newer Qt.
# pylint: disable=broad-except
if not hasattr(table, "btn") and not hasattr(table, "btnfailed"):
try:
btn = table.findChild(QAbstractButton)
class Efc(QObject):
@staticmethod
def eventFilter(o, e):
if (isinstance(o, QAbstractButton) and
e.type() == QEvent.Paint):
# paint by hand (borrowed from QTableCornerButton)
btn = o
opt = QStyleOptionHeader()
opt.initFrom(btn)
state = QStyle.State_None
if btn.isEnabled():
state |= QStyle.State_Enabled
if btn.isActiveWindow():
state |= QStyle.State_Active
if btn.isDown():
state |= QStyle.State_Sunken
opt.state = state
opt.rect = btn.rect()
opt.text = btn.text()
opt.position = QStyleOptionHeader.OnlyOneSection
painter = QStylePainter(btn)
painter.drawControl(QStyle.CE_Header, opt)
return True # eat event
return False
table.efc = Efc()
# disconnect default handler for clicks and connect a new one, which supports
# both selection and deselection of all data
btn.clicked.disconnect()
btn.installEventFilter(table.efc)
btn.clicked.connect(self._on_select_all)
table.btn = btn
if sys.platform == "darwin":
btn.setAttribute(Qt.WA_MacSmallSize)
except Exception:
table.btnfailed = True
if hasattr(table, "btn"):
try:
btn = table.btn
btn.setText(text)
opt = QStyleOptionHeader()
opt.text = btn.text()
s = btn.style().sizeFromContents(
QStyle.CT_HeaderSection,
opt, QSize(),
btn).expandedTo(QApplication.globalStrut())
if s.isValid():
table.verticalHeader().setMinimumWidth(s.width())
except Exception:
pass
