4 from PyQt4.QtCore import *
5 from PyQt4.QtGui import *
6 from PyQt4.QtXml import *
8 from sface.config import config
9 from sface.screens.sfascreen import SfaScreen
12 class RSpecView(QTreeView):
13 def __init__(self, parent):
14 QTreeView.__init__(self, parent)
16 self.setAnimated(True)
17 self.setItemsExpandable(True)
18 self.setRootIsDecorated(True)
19 self.setHeaderHidden(True)
20 self.setAttribute(Qt.WA_MacShowFocusRect, 0)
21 self.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
23 def expandMatchingText(self, txt):
29 def recursiveExpand(index):
30 parent = index.parent()
31 if parent and parent.isValid():
32 recursiveExpand(parent)
36 if index.data().toString() == txt:
37 recursiveExpand(index)
38 self.scrollTo(index, self.PositionAtCenter)
40 rows = model.rowCount(index)
42 child_index = index.child(r, 0)
45 root_rows = model.rowCount()
46 for r in range(root_rows):
47 index = model.index(r, 0)
51 class DomModel(QAbstractItemModel):
52 def __init__(self, document, parent = 0):
53 QAbstractItemModel.__init__(self, parent)
54 self.domDocument = document
55 # one of the children of the rootItem is the 'xml' thing.
57 childList = document.childNodes()
58 for i in range(childList.count()):
59 currElem = childList.item(i)
60 if (currElem.nodeType() == QDomNode.ProcessingInstructionNode):
61 document.removeChild(currElem)
63 self.rootItem = DomItem(document, 0);
65 def data(self, index, role):
66 if not index.isValid():
68 if role != Qt.DisplayRole:
70 node = index.internalPointer().node()
71 attributeMap = node.attributes()
75 if node.nodeType() == QDomNode.ElementNode:
76 qslist = QStringList()
77 for i in range(attributeMap.count()):
78 attr = attributeMap.item(i)
79 elem = '%s="%s"' % (attr.nodeName(), attr.nodeValue())
81 return QString("<%s %s>" % (node.nodeName(), qslist.join(' ')))
82 elif node.nodeType() == QDomNode.AttributeNode:
83 return QString('Whozat?!')
84 elif node.nodeType() == QDomNode.TextNode:
85 return node.nodeValue()
86 elif node.nodeType() == QDomNode.CDATASectionNode:
87 return QString('unsupported node type')
88 elif node.nodeType() == QDomNode.EntityReferenceNode:
89 return QString('unsupported node type')
90 elif node.nodeType() == QDomNode.EntityNode:
91 return QString('unsupported node type')
92 elif node.nodeType() == QDomNode.ProcessingInstructionNode:
94 #return node.nodeName()
95 elif node.nodeType() == QDomNode.CommentNode:
96 return QString('#').append(node.nodeValue())
97 elif node.nodeType() == QDomNode.DocumentNode:
98 return QString('unsupported node type')
99 elif node.nodeType() == QDomNode.DocumentTypeNode:
100 return QString('unsupported node type')
101 elif node.nodeType() == QDomNode.DocumentFragmentNode:
102 return QString('unsupported node type')
103 elif node.nodeType() == QDomNode.NotationNode:
104 return QString('unsupported node type')
105 elif node.nodeType() == QDomNode.BaseNode:
106 return QString('unsupported node type')
107 elif node.nodeType() == QDomNode.CharacterDataNode:
108 return QString('unsupported node type')
114 def flags(self, index):
115 if not index.isValid():
116 return Qt.ItemIsEnabled
117 return Qt.ItemIsEnabled | Qt.ItemIsSelectable
119 def headerData(self, section, orientation, role):
122 def index(self, row, column, parent=None):
123 if not parent or not parent.isValid():
124 parentItem = self.rootItem
126 parentItem = parent.internalPointer()
128 childItem = parentItem.child(row)
129 # childItem would be None to say "false"?
131 return self.createIndex(row, column, childItem)
135 def parent(self, child):
136 if not child.isValid():
138 childItem = child.internalPointer()
139 parentItem = childItem.parent()
141 if not parentItem or parentItem == self.rootItem:
143 return self.createIndex(parentItem.row(), 0, parentItem)
145 def rowCount(self, parent=None):
146 if not parent or not parent.isValid():
147 parentItem = self.rootItem
149 parentItem = parent.internalPointer()
151 return parentItem.node().childNodes().count()
153 def columnCount(self, parent):
154 # just one column we'll print tag name (and attributes) or the
160 # wrapper around PyQt4.QtXml.QDomNode it keeps an hash of
161 # childrens for performance reasons
163 def __init__(self, node, row, parent = 0):
164 # node is of type PyQt4.QtXml.QDomNode
166 self.parentItem = parent
171 if i in self.childItems:
172 return self.childItems[i]
173 if i >= 0 and i < self.domNode.childNodes().count():
174 childNode = self.domNode.childNodes().item(i)
175 childItem = DomItem(childNode, i, self)
176 self.childItems[i] = childItem
181 return self.parentItem
187 return self.rowNumber
190 class RSpecWindow(QDialog):
191 def __init__(self, parent=None):
192 QDialog.__init__(self, parent)
193 self.setWindowTitle("RSPec View")
198 self.view = RSpecView(self)
199 layout = QVBoxLayout()
200 layout.addWidget(self.view)
201 self.setLayout(layout)
209 def showNode(self, hostname):
210 self.view.expandMatchingText(hostname)
212 def updateView(self):
218 rspec_file = config.getSliceRSpecFile()
219 if not os.path.exists(rspec_file):
222 self.document = QDomDocument("RSpec")
223 self.document.setContent(open(rspec_file,'r').read())
224 self.model = DomModel(self.document, self)
226 self.view.setModel(self.model)
227 self.view.expand(self.model.index(0, 0)) #expand first level only