表格每行旁边的添加按钮

时间:2019-08-14 06:12:30

标签: python python-3.x pyqt pyqt5 qtablewidget

当前,我正在将按钮添加到第一列的Table中。但是我要做的是将按钮添加到表的每一行旁边而不是表的列中。

目前这是我的操作方式:

# initialize a table somehow
table = QTableWidget(parent)
table.setRowCount(3)
table.setColumnCount(5)
table.setHorizontalHeaderLabels(['', 'Field 1', 'Field 2', 'Field 3', 'Field N'])
table.verticalHeader().hide()

# create an cell widget
for index in range(table.rowCount()):
    btn = QPushButton(table)
    btn.setText('add')
    table.setCellWidget(index, 0, btn)

enter image description here

我该怎么做(见下图): enter image description here

1 个答案:

答案 0 :(得分:1)

您可以在垂直标题中添加按钮:

from PyQt5 import QtCore, QtGui, QtWidgets


class ButtonHeaderView(QtWidgets.QHeaderView):
    def __init__(self, parent):
        super().__init__(QtCore.Qt.Vertical, parent)
        self.m_buttons = []
        self.sectionResized.connect(self.adjustPositions)
        self.sectionCountChanged.connect(self.onSectionCountChanged)
        self.parent().horizontalScrollBar().valueChanged.connect(self.adjustPositions)

    @QtCore.pyqtSlot()
    def onSectionCountChanged(self):
        while self.m_buttons:
            button = self.m_buttons.pop()
            button.deleteLater()
        for i in range(self.count()):
            button = QtWidgets.QPushButton(self)
            button.setCursor(QtCore.Qt.ArrowCursor)
            self.m_buttons.append(button)
            self.update_data()
            self.adjustPositions()

    def setModel(self, model):
        super().setModel(model)
        if self.model() is not None:
            self.model().headerDataChanged.connect(self.update_data)

    def update_data(self):
        for i, button in enumerate(self.m_buttons):
            text = self.model().headerData(i, self.orientation(), QtCore.Qt.DisplayRole)
            button.setText(str(text))

    def updateGeometries(self):
        super().updateGeometries()
        self.adjustPositions()

    @QtCore.pyqtSlot()
    def adjustPositions(self):
        w = 0
        for index, button in enumerate(self.m_buttons):
            geom = QtCore.QRect(
                0,
                self.sectionViewportPosition(index),
                button.height(),
                self.sectionSize(index),
            )
            w = max(w, button.height())
            geom.adjust(0, 2, 0, -2)
            button.setGeometry(geom)
        self.setFixedWidth(w)


if __name__ == "__main__":
    import sys

    app = QtWidgets.QApplication(sys.argv)
    w = QtWidgets.QTableWidget(3, 4)
    header = ButtonHeaderView(w)
    w.setVerticalHeader(header)
    w.setHorizontalHeaderLabels(["Field 1", "Field 2", "Field 3", "Field N"])

    header_buttons = []
    for i in range(w.columnCount()):
        header_button = "+"
        header_buttons.append(header_button)
    w.setVerticalHeaderLabels(header_buttons)

    w.resize(320, 240)
    w.show()
    sys.exit(app.exec_())