X-Git-Url: http://git.onelab.eu/?p=sface.git;a=blobdiff_plain;f=sface%2Frspecwindow.py;h=7c6f061bd20ee7a4cf5837a96882d6ed14e3d759;hp=17bef6a0561b71470c89adbd32f7ed46c727c402;hb=ed2d6eecaf798bdd9f8129ce7186d8f30ec01467;hpb=dd3e0cf249a536798fcbb3aaf788dca1d674ab26 diff --git a/sface/rspecwindow.py b/sface/rspecwindow.py index 17bef6a..7c6f061 100644 --- a/sface/rspecwindow.py +++ b/sface/rspecwindow.py @@ -5,25 +5,21 @@ from PyQt4.QtCore import * from PyQt4.QtGui import * from PyQt4.QtXml import * +from sface.xmlwidget import * from sface.config import config from sface.screens.sfascreen import SfaScreen -class RSpecView(QTreeView): +class RSpecView(XmlView): def __init__(self, parent): - QTreeView.__init__(self, parent) - - self.setAnimated(True) - self.setItemsExpandable(True) - self.setRootIsDecorated(True) - self.setHeaderHidden(True) - self.setAttribute(Qt.WA_MacShowFocusRect, 0) - self.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding) + XmlView.__init__(self, parent) def expandMatchingText(self, txt): self.collapseAll() self.expandToDepth(0) + model = self.model() + def recursiveExpand(index): parent = index.parent() if parent and parent.isValid(): @@ -34,195 +30,33 @@ class RSpecView(QTreeView): if index.data().toString() == txt: recursiveExpand(index) self.scrollTo(index, self.PositionAtCenter) + return rows = model.rowCount(index) for r in range(rows): child_index = index.child(r, 0) search(child_index) - model = self.model() root_rows = model.rowCount() for r in range(root_rows): index = model.index(r, 0) search(index) - -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): - 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 = '%s="%s"' % (attr.nodeName(), attr.nodeValue()) - qslist.append(elem) - return QString("<%s %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): +class RSpecWindow(XmlWindow): def __init__(self, parent=None): - QDialog.__init__(self, parent) - self.setWindowTitle("RSPec View") - - self.document = None - self.model = None - - self.view = RSpecView(self) - layout = QVBoxLayout() - layout.addWidget(self.view) - self.setLayout(layout) + XmlWindow.__init__(self, parent, 'RSpec Window') - self.updateView() + def initView(self): + return RSpecView(self) - def show(self): - self.updateView() - QDialog.show(self) - def showNode(self, hostname): self.view.expandMatchingText(hostname) def updateView(self): - del self.document - del self.model - self.document = None - self.model = None + XmlWindow.updateView(self) rspec_file = config.getSliceRSpecFile() if not os.path.exists(rspec_file): return - self.document = QDomDocument("RSpec") self.document.setContent(open(rspec_file,'r').read()) - self.model = DomModel(self.document, self) - - self.view.setModel(self.model) - self.view.expand(self.model.index(0, 0)) #expand first level only - -