python - 如何在 QTableView 列中添加 QTreeView

Question

我是 PyQt 的新手，我正在处理一个包含 QTableView 的项目，其中一列显示系统路径。我想添加一个 QTreeView，以便用户可以单击 + 或 >> 按钮来展开路径下方的内容。

这是我的基本实现:

from PyQt4 import QtGui
from PyQt4 import QtCore

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

        self.resize(600,400)
        self.setWindowTitle("My Basic Treeview")

        self.treeview = QtGui.QTreeView(self)

        self.treeview.model = QtGui.QFileSystemModel()
        self.treeview.model.setRootPath('/opt')
        self.treeview.setModel(self.treeview.model)
        self.treeview.setColumnWidth(0, 200)

        self.setCentralWidget(self.treeview)

if __name__ == '__main__':
    import sys
    app = QtGui.QApplication(sys.argv)
    w = MainWindow()
    w.show()
    sys.exit(app.exec_())

尽管在上述情况下，我得到了所有文件夹，但我只想要 /opt 路径及其下面的文件夹。

import operator
from PyQt4.QtCore import *
from PyQt4.QtGui import *

class MyWindow(QWidget):
    def __init__(self, data_list, header, *args):
        QWidget.__init__(self, *args)
        # setGeometry(x_pos, y_pos, width, height)
        self.setGeometry(300, 200, 570, 450)
        self.setWindowTitle("Click on column title to sort")
        table_model = MyTableModel(self, data_list, header)
        table_view = QTableView()
        table_view.setModel(table_model)
        # set font
        font = QFont("Courier New", 14)
        table_view.setFont(font)
        # set column width to fit contents (set font first!)
        table_view.resizeColumnsToContents()
        # enable sorting
        table_view.setSortingEnabled(True)
        layout = QVBoxLayout(self)
        layout.addWidget(table_view)
        self.setLayout(layout)

class MyTableModel(QAbstractTableModel):
    def __init__(self, parent, mylist, header, *args):
        QAbstractTableModel.__init__(self, parent, *args)
        self.mylist = mylist
        self.header = header

    def rowCount(self, parent):
        return len(self.mylist)

    def columnCount(self, parent):
        return len(self.mylist[0])

    def data(self, index, role):
        if not index.isValid():
            return None
        elif role != Qt.DisplayRole:
            return None
        return self.mylist[index.row()][index.column()]

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

# the solvent data ...
header = ['Name', ' Email', ' Status', ' Path']
# use numbers for numeric data to sort properly
data_list = [
('option_A', 'zyro@email.com', 'Not Copied', '/Opt'),
('option_B', 'zyro@email.com', 'Not Copied', '/Users'),
]
app = QApplication([])
win = MyWindow(data_list, header)
win.show()
app.exec_()

视觉示例:

Answer 1

我认为你的问题可以分为两部分:

1. 如何在 QTreeView 中显示 /opt 路径及其子级，但不显示其兄弟级。换句话说，如何在 QTreeView 中显示根目录；

2. 如何将 QTreeView 添加到 QTableView。

1。如何在 QTreeView 中包含根目录:

QTreeView 的根是在 View 中显示其内容的目录。它在调用方法 setRootIndex 时设置。根据post by wysota on Qt Centre :

You can't display the invisibleRootItem because it is a fake item used only to have an equivalent of empty QModelIndex.

解决方法是将根目录设置为 /opt 的父目录，并使用 QSortFilterProxyModel< 的子类过滤掉 /opt 的兄弟目录/em>。请注意，我还重新实现了 sizeHint 方法，该方法对于调整 QTableView 的行大小是必需的:

from PyQt4 import QtGui, QtCore
import os

class MyQTreeView(QtGui.QTreeView):

    def __init__(self, path, parent=None):
        super(MyQTreeView, self).__init__(parent)

        ppath = os.path.dirname(path) # parent of path
        self.setFrameStyle(0)

        #---- File System Model ----

        sourceModel = QtGui.QFileSystemModel()
        sourceModel.setRootPath(ppath)

        #---- Filter Proxy Model ----

        proxyModel = MyQSortFilterProxyModel(path)
        proxyModel.setSourceModel(sourceModel)

        #---- Filter Proxy Model ----

        self.setModel(proxyModel)
        self.setHeaderHidden(True)
        self.setRootIndex(proxyModel.mapFromSource(sourceModel.index(ppath)))  

        #--- Hide All Header Sections Except First ----

        header = self.header()
        for sec in range(1, header.count()):
            header.setSectionHidden(sec, True)

    def sizeHint(self):
        baseSize = super(MyQTreeView,self).sizeHint()

        #---- get model index of "path" ----
        qindx = self.rootIndex().child(0, 0)

        if self.isExpanded(qindx): # default baseSize height will be used
            pass

        else:  # shrink baseShize height to the height of the row           
            baseSize.setHeight(self.rowHeight(qindx))

        return baseSize


class MyQSortFilterProxyModel(QtGui.QSortFilterProxyModel):    
    def __init__(self, path, parent=None):
        super(MyQSortFilterProxyModel, self).__init__(parent)

        self.path = path

    def filterAcceptsRow(self, row, parent):

        model = self.sourceModel()
        path_dta = model.index(self.path).data()
        ppath_dta = model.index(os.path.dirname(self.path)).data()

        if parent.data() == ppath_dta:
            if parent.child(row, 0).data() == path_dta:                
                return True
            else:
                return False            
        else:
            return True

2。如何将 *QTreeView* 添加到 *QTableView* :

可以使用QItemDelegate 将QTreeView 添加到QTableView。 post by Pavel Strakhov这对我有很大帮助，因为在回答这个问题之前我从未将 QTableView 与委托(delegate)结合使用。我总是使用 QTableWidget 而不是 setCellWidget 方法。

请注意，我在 MyDelegate 类中设置了一个信号，它调用 MyTableView 类中的方法 resizeRowsToContents。这样，行的高度会根据 MyQTreeView 类的 sizeHint 方法的重新实现进行调整。

class MyTableModel(QtCore.QAbstractTableModel):
    def __init__(self, parent, mylist, header, *args):
        super(MyTableModel, self).__init__(parent, *args)
        self.mylist = mylist
        self.header = header

    def rowCount(self, parent=QtCore.QModelIndex()):
        return len(self.mylist)

    def columnCount(self, parent=QtCore.QModelIndex()):
        return len(self.mylist[0])

    def data(self, index, role):
        if not index.isValid():
            return None
        elif role != QtCore.Qt.DisplayRole:
            return None
        return self.mylist[index.row()][index.column()]

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

class MyDelegate(QtGui.QItemDelegate):

    treeViewHeightChanged = QtCore.pyqtSignal(QtGui.QWidget)

    def createEditor(self, parent, option, index):

        editor = MyQTreeView(index.data(), parent)
        editor.collapsed.connect(self.sizeChanged)
        editor.expanded.connect(self.sizeChanged)

        return editor

    def sizeChanged(self):
        self.treeViewHeightChanged.emit(self.sender())

class MyTableView(QtGui.QTableView):
    def __init__(self, data_list, header, *args):
        super(MyTableView, self).__init__(*args)

        #---- set up model ----

        model = MyTableModel(self, data_list, header)
        self.setModel(model)

        #---- set up delegate in last column ----

        delegate = MyDelegate()

        self.setItemDelegateForColumn(3, delegate)
        for row in range(model.rowCount()):
            self.openPersistentEditor(model.index(row, 3))

        #---- set up font and resize calls ----

        self.setFont(QtGui.QFont("Courier New", 14))
        self.resizeColumnsToContents()
        delegate.treeViewHeightChanged.connect(self.resizeRowsToContents)

3。基本应用:

这是一个基于您在 OP 中提供的代码的基本应用程序:

if __name__ == '__main__':

    header = ['Name', ' Email', ' Status', ' Path']
    data_list = [('option_A', 'zyro@email.com', 'Not Copied', '/opt'),
                 ('option_B', 'zyro@email.com', 'Not Copied', '/usr')]

    app = QtGui.QApplication([])
    win = MyTableView(data_list, header)
    win.setGeometry(300, 200, 570, 450)
    win.show()
    app.exec_()

结果是: