QTreeView 和 QTableView 的 Qt 模型

问题描述 投票:0回答:2

我正在尝试创建一个可用于 QTableView 和 QTreeView 的模型。举个例子,我的数据是这样的:

身份证 地点 姓名
101 201 苹果
201 厨房
102 201 香蕉
301 地窖
302 301 土豆
202 302 指甲

因此每个条目都有一个位置,该位置本身就是模型中的一个条目。对于 QTableView,我想简单地显示彼此下的所有条目,如上所示,而对于 QTreeView,我想要类似的东西

  • 201:厨房
    • 101:苹果
    • 102:香蕉
  • 301:地窖
    • 302:土豆
      • 202:钉子

我的问题是,我无法弄清楚如何实现 QAbstractProxyModel.maptoSource() 或 mapfromSource() ,因为我丢失了有关 QTableView 中父级的信息。阅读https://www.qtcentre.org/threads/26163-Map-table-to-tree-through-model-view-possible似乎这根本不可能。然而 QAbstractProxyModel 明确表示它是为了在两个视图中显示数据。谁能指出我正确的方向,或者知道是否可以实现这样的模型?特别是在Python中,不幸的是我找不到任何例子。


我真的很喜欢仅使用无缩进的 TreeView 作为一种 TableView 的想法。不幸的是,我在创建模型时仍然遇到问题。目前,仅显示顶部条目。

class MyModel(qtg.QStandardItemModel):
    def __init__(
        self,
        engine
    ):
        self.engine = engine
        self.hierarchy_key = 'location_id'

        
        self.column_names = ['id', 'location_id', 'name', 'quantity']
        super().__init__(0, len(self.fields))

        self.setHorizontalHeaderLabels(self.column_names)
        self.root = self.invisibleRootItem()
        self.build_model()

    def build_model(self):
        def add_children_to_tree(entries, parent_item):
            for entry in entries:
                items = []
                for col in self.column_names:
                    text = getattr(entry, col)
                    item = qtg.QStandardItem(text)
                    items.append(qtg.QStandardItem(text))

                parent_item.appendRow(items)
                item = items[1] #the location_id item
                parent_item.setChild(item.index().row(), item.index().column(), item)

                with session_scope(self.engine) as session:
                    child_entries = (
                        session.query(self.entry_type)
                            .filter(
                            getattr(getattr(self.entry_type, self.hierarchy_key), "is_")(
                                entry.id
                            )
                        )
                            .all()
                    )
                    if child_entries:
                        add_children_to_tree(child_entries, item)

        self.removeRows(0, self.rowCount())
        with session_scope(self.engine) as session:
            root_entries = session.query(self.entry_type).filter(getattr(getattr(self.entry_type, self.hierarchy_key), "is_")(None)).all()
            if not isinstance(root_entries, list):
               root_entries = [root_entries]
            add_children_to_tree(root_entries, self.root)

这个想法是会话查询会产生一个条目列表。每个条目都是数据库中的一条记录,具有“id”、“location_id”等属性。因此,每个属性都是一个项目,项目列表在模型中创建一行。我无法弄清楚如何按照此处显示的方式使该行项目成为另一行的子项: Tree view 我认为 setChild() 函数需要以不同的方式调用?

python qt pyside2 qtableview qtreeview
2个回答
2
投票

由于明显缺乏 python 的示例,我将在这里发布我的 simpletreemodel 的修改版本,这就是最终对我有用的版本。然后按照建议使用 QTreeView 而不是 QTableView,我让表格也按照我想要的方式运行。总的来说,这会创建 MyItem,它是一个包含整行信息的项目,然后,如果层次结构键 (location_id) 的值等于父级的 id,则我使用递归将子级添加到父级。


class MyItem(object):
    def __init__(self, data, parent=None):
        self.parentItem = parent
        self.itemData = data
        self.childItems = []

    def appendChild(self, item):
        self.childItems.append(item)

    def child(self, row):
        return self.childItems[row]

    def childCount(self):
        return len(self.childItems)

    def columnCount(self):
        return len(self.itemData)

    def data(self, column=None):
        try:
            if column == None:
                return [self.itemData[i] for i in range(self.columnCount())]
            return self.itemData[column]
        except IndexError:
            return None

    def parent(self):
        return self.parentItem

    def row(self):
        if self.parentItem:
            return self.parentItem.childItems.index(self)
        return 0


class MyModel(QtCore.QAbstractItemModel):
    def __init__(self, entry_type, engine, hierarchy_key, description_key, parent=None):
        super(ORMModel, self).__init__(parent)

        self.entry_type = entry_type
        if isinstance(self.entry_type, str):
            self.entry_type = getattr(ds, self.entry_type)
        self.engine = engine
        self.hierarchy_key = hierarchy_key
        
        self.column_names = ['id', 'location_id', 'name', 'quantity']

        self.rootItem = MyItem(self.column_names)
        self.setHeaderData(0, Qt.Horizontal, self.rootItem)
        self.initiateModel()

    def root(self):
        return self.rootItem

    def columnCount(self, parent):
        if parent.isValid():
            return parent.internalPointer().columnCount()
        else:
            return self.rootItem.columnCount()

    def data(self, index, role):
        if not index.isValid():
            return None
        item = index.internalPointer()

        if role == Qt.DisplayRole:
            return item.data(index.column())

        return None

    def flags(self, index):
        if not index.isValid():
            return QtCore.Qt.NoItemFlags

        return QtCore.Qt.ItemIsEnabled | QtCore.Qt.ItemIsSelectable

    def headerData(self, section, orientation, role):
        if orientation == QtCore.Qt.Horizontal and role == QtCore.Qt.DisplayRole:
            return self.rootItem.data(section)

        return None

    def index(self, row, column, parent):
        if not self.hasIndex(row, column, parent):
            return QtCore.QModelIndex()

        if not parent.isValid():
            parentItem = self.rootItem
        else:
            parentItem = parent.internalPointer()

        childItem = parentItem.child(row)
        if childItem:
            return self.createIndex(row, column, childItem)
        else:
            return QtCore.QModelIndex()

    def parent(self, index):
        if not index.isValid():
            return QtCore.QModelIndex()

        childItem = index.internalPointer()
        parentItem = childItem.parent()

        if parentItem == self.rootItem:
            return QtCore.QModelIndex()

        return self.createIndex(parentItem.row(), 0, parentItem)

    def rowCount(self, parent):
        if parent.column() > 0:
            return 0

        if not parent.isValid():
            parentItem = self.rootItem
        else:
            parentItem = parent.internalPointer()

        return parentItem.childCount()

    def initiateModel(self):
        def add_children_to_tree(entries, parent_item):
            for entry in entries:
                row = []
                for field in self.fields.keys():
                    val = getattr(entry, field)
                    if isinstance(val, list):
                        text = "; ".join(map(str, val))
                    else:
                        text = str(val)
                    row.append(text)
                    item = ORMItem(row, parent_item)
                parent_item.appendChild(item)

                with session_scope(self.engine) as session:
                    child_entries = (
                        session.query(self.entry_type)
                        .filter(
                            getattr(
                                getattr(self.entry_type, self.hierarchy_key), "is_"
                            )(entry.id)
                        )
                        .all()
                    )
                    if child_entries:
                        add_children_to_tree(child_entries, item)

        with session_scope(self.engine) as session:
            root_entries = (
                session.query(self.entry_type)
                .filter(
                    getattr(getattr(self.entry_type, self.hierarchy_key), "is_")(None)
                )
                .all()
            )
            if not isinstance(root_entries, list):
                root_entries = [root_entries]
            add_children_to_tree(root_entries, self.rootItem)

0
投票

我已经使用

QTreeView
好几年了(通常与
QStandardItemModel
一起使用),但在不理想的情况下,知道什么在实践中有效,但并没有真正深入了解很多知识。

有一个极好的例子,值得仔细研究这里。这是直接来自 Qt Group 的,是对 之前的 C++ 示例在 Python 中的重写(实际上是 PySide,但很容易适应 PyQt)。

它的基本作用是构建类似于

QStandardItemModel
的东西,并展示它如何与
QTreeView
结合使用。

© www.soinside.com 2019 - 2024. All rights reserved.