В PyQt4 Model-View руководстве автор
разрабатывает 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: childNodeA0
и childNodeB0
. И для каждой из этих двух переменных-элементов верхнего уровня объявляются две дочерние переменные-элементы (я переименовал переменные, чтобы они выглядели более общими):
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
является дочерним элементом двух переменных: это дочерний элемент rootNode
и childNodeB0
:
childNodeB0.addChild(childNodeC0)
Вот схемы:
|------Root
|------childNodeA0
|------childNodeA1
|------childNodeA2
|------childNodeB0
|------childNodeB1
|------childNodeB3
|------childNodeC0
|------childNodeC0
Все идет так, как ожидалось. Но когда эта переменная данных назначается «реальной» модели, переменная элемента childNodeC0
отображается только как дочерний элемент Root
, а не как дочерний элемент childNodeB0
. Кроме того, элемент-переменная childNodeB0
показывается в TreeView дважды: как дочерний элемент Root (как и должно быть). И он также показан как дочерний элемент самого себя!:
Вопрос:
Как заставить 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_())