+++ /dev/null
-from PyQt4.QtCore import QAbstractItemModel, QVariant, QString, QStringList
-from PyQt4.QtCore import QModelIndex
-from PyQt4.QtCore import Qt
-from PyQt4.QtXml import QDomNode
-from domItem import domItem
-
-class domModel(QAbstractItemModel):
- def __init__(self, document, parent = 0):
- QAbstractItemModel.__init__(self, parent)
- self.domDocument = document
- self.rootItem = domItem(document, 0);
-
- def data(self, index, role):
- if not index.isValid():
- return QVariant()
- if role != Qt.DisplayRole:
- return QVariant()
- node = index.internalPointer().node()
- attributeMap = node.attributes()
-
- col = index.column()
- if col == 0:
- if node.nodeType() == QDomNode.ElementNode:
- qslist = QStringList()
- for i in range(attributeMap.count()):
- attr = attributeMap.item(i)
- elem = (attr.nodeName()
- .append(QString('="'))
- .append(attr.nodeValue())
- .append(QString('"')))
- qslist.append(elem)
- return (QString('<').
- append(node.nodeName()).
- append(' ').
- append(qslist.join(' ')).
- append('>'))
- elif node.nodeType() == QDomNode.AttributeNode:
- return QString('Whozat?!')
- elif node.nodeType() == QDomNode.TextNode:
- return node.nodeValue()
- elif node.nodeType() == QDomNode.CDATASectionNode:
- return QString('unsupported node type')
- elif node.nodeType() == QDomNode.EntityReferenceNode:
- return QString('unsupported node type')
- elif node.nodeType() == QDomNode.EntityNode:
- return QString('unsupported node type')
- elif node.nodeType() == QDomNode.ProcessingInstructionNode:
- return node.nodeName()
- elif node.nodeType() == QDomNode.CommentNode:
- return QString('#').append(node.nodeValue())
- elif node.nodeType() == QDomNode.DocumentNode:
- return QString('unsupported node type')
- elif node.nodeType() == QDomNode.DocumentTypeNode:
- return QString('unsupported node type')
- elif node.nodeType() == QDomNode.DocumentFragmentNode:
- return QString('unsupported node type')
- elif node.nodeType() == QDomNode.NotationNode:
- return QString('unsupported node type')
- elif node.nodeType() == QDomNode.BaseNode:
- return QString('unsupported node type')
- elif node.nodeType() == QDomNode.CharacterDataNode:
- return QString('unsupported node type')
- else:
- return QVariant()
- else:
- return QVariant()
-
- def flags(self, index):
- if not index.isValid():
- return Qt.ItemIsEnabled
- # does this `|` thing hold?
- return Qt.ItemIsEnabled | Qt.ItemIsSelectable
-
- def headerData(self, section, orientation, role):
- if orientation == Qt.Horizontal and role == Qt.DisplayRole:
- if section == 0:
- return QString("Do I need an header?")
- else:
- return QVariant()
- else:
- return QVariant()
-
- def index(self, row, column, parent):
- if not parent.isValid():
- parentItem = self.rootItem
- else:
- parentItem = parent.internalPointer()
-
- childItem = parentItem.child(row)
- # childItem would be None to say "false"?
- if childItem:
- return self.createIndex(row, column, childItem)
- else:
- return QModelIndex()
-
- def parent(self, child):
- if not child.isValid():
- return QModelIndex()
- childItem = child.internalPointer()
- parentItem = childItem.parent()
-
- if not parentItem or parentItem == self.rootItem:
- return QModelIndex()
- return self.createIndex(parentItem.row(), 0, parentItem)
-
- def rowCount(self, parent):
- if not parent.isValid():
- parentItem = self.rootItem
- else:
- parentItem = parent.internalPointer()
-
- return parentItem.node().childNodes().count()
-
- def columnCount(self, parent):
- # just one column
- # we'll print tag name (and attributes)
- # or the tag content
- return 1