当QSortFilterProxyModel更改PYQT5时更新QStatusBar中的QWidgets

时间:2018-03-26 11:12:17

标签: python python-3.x pyqt5 qstatusbar

我有一个显示QSortFilterProxyModel的QTableView,我希望有一个汇总行(在QStatusBar中)可以汇总的列。我设法达到这一点,但是,我还想在代理模型更改时更新这些列的总和。我在这一点上陷入困​​境,无法继续前进。

这是我的代码:

from PyQt5.QtGui import *
from PyQt5.QtCore import *
from PyQt5.QtWidgets import *
from pandas import *
import sys


class TBWindow(QMainWindow):
    def __init__(self, parent=None):
        super(TBWindow, self).__init__(parent)

        fList = [['Ajvi', 'Ostatní textil', 'Holesovice', 1, 2],
                 ['Ajvi', 'Sukně', 'Holesovice', 2, 5],
                 ['Ajvi', 'Šaty', 'Holesovice', 2, 0],
                 ['Belusi', 'Brože', 'Fler', 0, 1],
                 ['Belusi', 'Brože', 'Holesovice', 0, 5],
                 ['PP', 'Sukně', 'E-shop', 25, 10],
                 ['PP', 'Sukně', 'Fler', 7, 6],
                 ['PP', 'Sukně', 'Holesovice', 15, 13],
                 ['PP', 'Sukně', 'Other', 6, 2],
                 ['PP', 'Sukně', 'Sashe', 6, 2],
                 ['PP', 'Tašky', 'E-shop', 2, 1],
                 ['PP', 'Tašky', 'Holesovice', 3, 1],
                 ['PP', 'Šaty', 'E-shop', 1, 0]]

        data = DataFrame(fList, columns = ['Seller', 'Section', 'Store', 'Total_pieces: 2017', 'Total_pieces: 2018'])

        self.setWindowTitle('Window')
        self.centralwidget  = QWidget(self)
        self.lineEdit       = QLineEdit(self.centralwidget)
        self.view           = QTableView(self.centralwidget)
        self.comboBox       = QComboBox(self.centralwidget)
        self.label          = QLabel(self.centralwidget)

        self.gridLayout = QGridLayout(self.centralwidget)
        self.gridLayout.addWidget(self.lineEdit, 0, 1, 1, 1)
        self.gridLayout.addWidget(self.view, 1, 0, 1, 3)
        self.gridLayout.addWidget(self.comboBox, 0, 2, 1, 1)
        self.gridLayout.addWidget(self.label, 0, 0, 1, 1)

        self.setCentralWidget(self.centralwidget)
        self.label.setText("Filter")

        self.model = PandasModel(data)
        self.proxy = QSortFilterProxyModel(self)
        self.proxy.setSourceModel(self.model)
        self.view.setModel(self.proxy)

        for column in range(self.view.horizontalHeader().count()):
            self.view.horizontalHeader().setSectionResizeMode(column, QHeaderView.Stretch)

        self.comboBox.addItems(list(data.columns.values))
        self.lineEdit.textChanged.connect(self.on_lineEdit_textChanged)
        self.comboBox.currentIndexChanged.connect(self.on_comboBox_currentIndexChanged)

        lastRow = data.append(data.sum(numeric_only=True), ignore_index=True).tail(1).fillna(0).astype(int).replace(0,'').values
        flat_Row = ['Total: ' + str(item) if item != '' else '' for sublist in lastRow for item in sublist]

        mWdg = QWidget()
        mWdg.setLayout(QHBoxLayout())
        for item in flat_Row:
            mWdg.layout().addWidget(QLabel(item))
        self.statusBar().addPermanentWidget(mWdg, 1)

    @pyqtSlot(str)
    def on_lineEdit_textChanged(self, text):
        search = QRegExp(text, Qt.CaseInsensitive, QRegExp.RegExp)
        self.proxy.setFilterRegExp(search)

    @pyqtSlot(int)
    def on_comboBox_currentIndexChanged(self, index):
        self.proxy.setFilterKeyColumn(index)

class PandasModel(QAbstractTableModel):
    def __init__(self, data, parent=None):
        QAbstractTableModel.__init__(self, parent)
        self._data = data

    def rowCount(self, parent=None):
        return self._data.shape[0]

    def columnCount(self, parent=None):
        return self._data.shape[1]

    def data(self, index, role=Qt.DisplayRole):
        if index.isValid():
            if role == Qt.DisplayRole:
                return str(self._data.iloc[index.row(), index.column()])
        return None

    def headerData(self, col, orientation, role):
        if orientation == Qt.Horizontal and role == Qt.DisplayRole:
            return self._data.columns[col]
        return None

if __name__ == '__main__':

    app = QApplication(sys.argv)
    main = TBWindow()
    main.showMaximized()
    sys.exit(app.exec_())

有任何建议如何实现这一目标?

1 个答案:

答案 0 :(得分:0)

通过以下方式编辑@pyqtSlot和mWdg(到self.mWdg),我设法让它工作:

self

也许,有一种更好(优雅)的方法如何继续而不是删除self.mWdg然后再重新创建它??