视图/模型/控制器:如何设计分层数据变量

时间:2015-01-08 22:11:04

标签: python qt model pyqt qtreeview

在PyQt4模型中查看tutorial作者
设计Node()要与QAbstractItemModel链接的QTreeView一起使用的类:

class Node(object):    
    def __init__(self, name, parent=None):        
        self._name = name
        self._children=[]
        self._parent=parent

        if parent is not None:
            parent.addChild(self)

    def typeInfo(self):
        return "NODE"

    def addChild(self, child):
        self._children.append(child)

    def name(self):
        return self._name

    def setName(self, name):
        self._name = name

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

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

    def parent(self):
        return self._parent

    def row(self):
        if self._parent is not None:
            return self._parent._children.index(self)

然后他继续声明一个数据变量本身。 首先,他声明一个rootNode变量用作所有顶级项目变量的父级:

rootNode = Node("Root")

然后他宣布了两个顶级项目变量A和B:childNodeA0childNodeB0。 对于这两个顶级项变量中的每一个,都声明了两个子项变量 (我重命名变量以使事物看起来更通用):

childNodeA0 = Node("childNodeA0",  rootNode)
childNodeA1 = Node("childNodeA1",  childNodeA0)
childNodeA2 = Node("childNodeA2",  childNodeA1)

childNodeB0 = Node("childNodeB0",  rootNode)
childNodeB1 = Node("childNodeB1",  childNodeB0)
childNodeB3 = Node("childNodeB3",  childNodeB1)

以下是存储在rootNode变量中的数据的示意图:

|------Root
        |------childNodeA0
                |------childNodeA1
                        |------childNodeA2

        |------childNodeB0
                |------childNodeB1
                        |------childNodeB3

现在我宣布一个名为childNodeC0的第三个项变量:

childNodeC0 = Node("childNodeC0",  rootNode)

我将rootNode作为其父级传递。这是原理图:

|------Root
        |------childNodeA0
                |------childNodeA1
                        |------childNodeA2

        |------childNodeB0
                |------childNodeB1
                        |------childNodeB3

        |------childNodeC0

现在使用.addChild()方法我将childNodeC0变量附加为childNodeB0的子项。 因此,childNodeC0是两个变量的子项:它是rootNodechildNodeB0的孩子:

childNodeB0.addChild(childNodeC0)

这是原理图:

|------Root
        |------childNodeA0
                |------childNodeA1
                        |------childNodeA2   


        |------childNodeB0
                |------childNodeB1
                        |------childNodeB3    

                |------childNodeC0    

        |------childNodeC0

一切按预期进行。但是,当此数据变量分配给“真实”模型时,childNodeC0项目变量仅显示为Root的子项而不是childNodeB0的子项。除此之外,childNodeB0 item-variable在TreeView中显示两次:作为Root的子(因为它应该是)。它也显示为自己的孩子!:

enter image description here

问题:

如何使TreeView正确显示存储在根变量中的数据?

from PyQt4 import QtCore, QtGui
import sys

class Node(object):    
    def __init__(self, name, parent=None):        
        self._name = name
        self._children=[]
        self._parent=parent

        if parent is not None:
            parent.addChild(self)

    def typeInfo(self):
        return "NODE"

    def addChild(self, child):
        self._children.append(child)

    def name(self):
        return self._name

    def setName(self, name):
        self._name = name

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

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

    def parent(self):
        return self._parent

    def row(self):
        if self._parent is not None:
            return self._parent._children.index(self)

    def log(self, tabLevel=-1):
        output     = ""
        tabLevel += 1

        for i in range(tabLevel):
            output += "\t"

        output += "|------" + self._name + "\n"

        for child in self._children:
            output += child.log(tabLevel)

        tabLevel -= 1
        output += "\n"

        return output

    def __repr__(self):
        return self.log()

class SceneGraphModel(QtCore.QAbstractItemModel):    
    def __init__(self, root, parent=None):
        super(SceneGraphModel, self).__init__(parent)
        self._rootNode = root

    def rowCount(self, parent):    

        if not parent.isValid():
            parentNode = self._rootNode
        else:
            parentNode = parent.internalPointer()
            # print 'VALID: ', type(parent), parent.row(), parent.column()

        return parentNode.childCount()

    def columnCount(self, parent):   
        return 1

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

        node = index.internalPointer()

        if role == QtCore.Qt.DisplayRole or role == QtCore.Qt.EditRole:
            if index.column() == 0:
                return node.name()            

    def setData(self, index, value, role=QtCore.Qt.EditRole): 
        if index.isValid():            
            if role == QtCore.Qt.EditRole:                
                node = index.internalPointer()
                node.setName(value)                
                return True
        return False    

    def headerData(self, section, orientation, role):   
        if role == QtCore.Qt.DisplayRole:
            if section == 0:
                return "Column #0"         

    def flags(self, index):         
        return QtCore.Qt.ItemIsEnabled | QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEditable    

    def parent(self, index):   

        node = self.getNode(index)
        parentNode = node.parent()

        if parentNode == self._rootNode:
            return QtCore.QModelIndex()

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

    def index(self, row, column, parent):   

        parentNode = self.getNode(parent)
        childItem = parentNode.child(row)

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

    def getNode(self, index):    
        if index.isValid():
            node = index.internalPointer()
            if node:
                return node            
        return self._rootNode    








if __name__ == '__main__':
    app = QtGui.QApplication(sys.argv)

    rootNode   = Node("Root")
    childNodeA0 = Node("childNodeA0",  rootNode)
    childNodeA1 = Node("childNodeA1",  childNodeA0)
    childNodeA2 = Node("childNodeA2",  childNodeA1)

    childNodeB0 = Node("childNodeB0",  rootNode)
    childNodeB1 = Node("childNodeB1",  childNodeB0)
    childNodeB3 = Node("childNodeB3",  childNodeB1)

    childNodeC0 = Node("childNodeC0",  rootNode)
    childNodeB0.addChild(childNodeC0)    

    treeView=QtGui.QTreeView()    

    model = SceneGraphModel(rootNode)
    treeView.setModel(model)
    treeView.show()

    print rootNode

    sys.exit(app.exec_())

1 个答案:

答案 0 :(得分:2)

我认为使用此数据模型,您不能拥有一个拥有2个不同父级的节点。 addChild令人困惑,因为它似乎可以做到。 如果您查看场景图模型,它会询问Node它的父级(parent(self, index)中):

parentNode = node.parent()

Node.parent()仅返回在项目构造中定义的父项:

return self._parent

如果您想在2个不同的位置拥有相同的孩子,您可能需要复制它或改进数据模型。