-
-class DomModel(QAbstractItemModel):
- def __init__(self, document, parent = 0):
- QAbstractItemModel.__init__(self, parent)
- self.domDocument = document
- # one of the children of the rootItem is the 'xml' thing.
- # here I delete it.
- childList = document.childNodes()
- for i in range(childList.count()):
- currElem = childList.item(i)
- if (currElem.nodeType() == QDomNode.ProcessingInstructionNode):
- document.removeChild(currElem)
- break
- self.rootItem = DomItem(document, 0);
-
- def data(self, index, role = Qt.DisplayRole):
- # sometimes it return a QString, sometimes a QVariant. not good.
- 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 = ' <b>%s</b>="<font color="#1e90ff">%s</font>"' % (attr.nodeName(), attr.nodeValue())
- qslist.append(elem)
- return QString('<<b><font color="#b42be2">%s</font></b>%s>'% (node.nodeName(), qslist.join(' ')))
- 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 QVariant()
- #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
- return Qt.ItemIsEnabled | Qt.ItemIsSelectable
-
- def headerData(self, section, orientation, role):
- return QVariant()
-
- def index(self, row, column, parent=None):
- if not parent or 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=None):
- if not parent or 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
-
-
-class DomItem:
- # wrapper around PyQt4.QtXml.QDomNode it keeps an hash of
- # childrens for performance reasons
-
- def __init__(self, node, row, parent = 0):
- # node is of type PyQt4.QtXml.QDomNode
- self.domNode = node
- self.parentItem = parent
- self.rowNumber = row
- self.childItems = {}
-
- def child(self, i):
- if i in self.childItems:
- return self.childItems[i]
- if i >= 0 and i < self.domNode.childNodes().count():
- childNode = self.domNode.childNodes().item(i)
- childItem = DomItem(childNode, i, self)
- self.childItems[i] = childItem
- return childItem
- return None
-
- def parent(self):
- return self.parentItem
-
- def node(self):
- return self.domNode
-
- def row(self):
- return self.rowNumber
-
-
-class RSpecWindow(QDialog):