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)
41 rows = model.rowCount(index)
43 child_index = index.child(r, 0)
46 root_rows = model.rowCount()
47 for r in range(root_rows):
48 index = model.index(r, 0)
52 class DomModel(QAbstractItemModel):
53 def __init__(self, document, parent = 0):
54 QAbstractItemModel.__init__(self, parent)
55 self.domDocument = document
56 # one of the children of the rootItem is the 'xml' thing.
58 childList = document.childNodes()
59 for i in range(childList.count()):
60 currElem = childList.item(i)
61 if (currElem.nodeType() == QDomNode.ProcessingInstructionNode):
62 document.removeChild(currElem)
64 self.rootItem = DomItem(document, 0);
66 def data(self, index, role = Qt.DisplayRole):
67 # sometimes it return a QString, sometimes a QVariant. not good.
68 if not index.isValid():
70 if role != Qt.DisplayRole:
72 node = index.internalPointer().node()
73 attributeMap = node.attributes()
77 if node.nodeType() == QDomNode.ElementNode:
78 qslist = QStringList()
79 for i in range(attributeMap.count()):
80 attr = attributeMap.item(i)
81 elem = ' <b>%s</b>="<font color="#1e90ff">%s</font>"' % (attr.nodeName(), attr.nodeValue())
83 return QString('<<b><font color="#b42be2">%s</font></b>%s>'% (node.nodeName(), qslist.join(' ')))
84 elif node.nodeType() == QDomNode.AttributeNode:
85 return QString('Whozat?!')
86 elif node.nodeType() == QDomNode.TextNode:
87 return node.nodeValue()
88 elif node.nodeType() == QDomNode.CDATASectionNode:
89 return QString('unsupported node type')
90 elif node.nodeType() == QDomNode.EntityReferenceNode:
91 return QString('unsupported node type')
92 elif node.nodeType() == QDomNode.EntityNode:
93 return QString('unsupported node type')
94 elif node.nodeType() == QDomNode.ProcessingInstructionNode:
96 #return node.nodeName()
97 elif node.nodeType() == QDomNode.CommentNode:
98 return QString('#').append(node.nodeValue())
99 elif node.nodeType() == QDomNode.DocumentNode:
100 return QString('unsupported node type')
101 elif node.nodeType() == QDomNode.DocumentTypeNode:
102 return QString('unsupported node type')
103 elif node.nodeType() == QDomNode.DocumentFragmentNode:
104 return QString('unsupported node type')
105 elif node.nodeType() == QDomNode.NotationNode:
106 return QString('unsupported node type')
107 elif node.nodeType() == QDomNode.BaseNode:
108 return QString('unsupported node type')
109 elif node.nodeType() == QDomNode.CharacterDataNode:
110 return QString('unsupported node type')
116 def flags(self, index):
117 if not index.isValid():
118 return Qt.ItemIsEnabled
119 return Qt.ItemIsEnabled | Qt.ItemIsSelectable
121 def headerData(self, section, orientation, role):
124 def index(self, row, column, parent=None):
125 if not parent or not parent.isValid():
126 parentItem = self.rootItem
128 parentItem = parent.internalPointer()
130 childItem = parentItem.child(row)
131 # childItem would be None to say "false"?
133 return self.createIndex(row, column, childItem)
137 def parent(self, child):
138 if not child.isValid():
140 childItem = child.internalPointer()
141 parentItem = childItem.parent()
143 if not parentItem or parentItem == self.rootItem:
145 return self.createIndex(parentItem.row(), 0, parentItem)
147 def rowCount(self, parent=None):
148 if not parent or not parent.isValid():
149 parentItem = self.rootItem
151 parentItem = parent.internalPointer()
153 return parentItem.node().childNodes().count()
155 def columnCount(self, parent):
156 # just one column we'll print tag name (and attributes) or the
162 # wrapper around PyQt4.QtXml.QDomNode it keeps an hash of
163 # childrens for performance reasons
165 def __init__(self, node, row, parent = 0):
166 # node is of type PyQt4.QtXml.QDomNode
168 self.parentItem = parent
173 if i in self.childItems:
174 return self.childItems[i]
175 if i >= 0 and i < self.domNode.childNodes().count():
176 childNode = self.domNode.childNodes().item(i)
177 childItem = DomItem(childNode, i, self)
178 self.childItems[i] = childItem
183 return self.parentItem
189 return self.rowNumber
192 class RSpecWindow(QDialog):
193 def __init__(self, parent=None):
194 QDialog.__init__(self, parent)
195 self.setWindowTitle("RSpec View")
200 self.view = RSpecView(self)
201 self.delegate = RSpecDelegate(self)
202 self.view.setItemDelegate(self.delegate)
203 layout = QVBoxLayout()
204 layout.addWidget(self.view)
205 self.setLayout(layout)
213 def showNode(self, hostname):
214 self.view.expandMatchingText(hostname)
216 def updateView(self):
222 rspec_file = config.getSliceRSpecFile()
223 if not os.path.exists(rspec_file):
226 self.document = QDomDocument("RSpec")
227 self.document.setContent(open(rspec_file,'r').read())
228 self.model = DomModel(self.document, self)
230 self.view.setModel(self.model)
231 self.view.expand(self.model.index(0, 0)) #expand first level only
234 class RSpecDelegate(QAbstractItemDelegate):
236 def __init__(self, parent=None):
237 QAbstractItemDelegate.__init__(self, parent)
239 def paint(self, painter, option, index):
240 text = index.model().data(index)
241 palette = QApplication.palette()
242 document = QTextDocument()
243 document.setDefaultFont(option.font)
244 if option.state & QStyle.State_Selected:
245 rx = QRegExp(QString('<font .*>'))
247 # If selected, I remove the <font color="..."> by hand,
248 # and give the highlight color
249 document.setHtml(QString("<font color=%1>%2</font>") \
250 .arg(palette.highlightedText().color().name())\
251 .arg(text.replace(rx, QString('')).
252 replace(QString('</font>'),QString(''))))
254 document.setHtml(text)
255 color = palette.highlight().color() \
256 if option.state & QStyle.State_Selected \
257 else palette.base().color()
259 # voodoo: if not highlighted, filling the rect
260 # with the base color makes no difference
261 painter.fillRect(option.rect, color)
262 painter.translate(option.rect.x(), option.rect.y())
263 document.drawContents(painter)
266 def sizeHint(self, option, index):
267 fm = option.fontMetrics
268 text = index.model().data(index)
269 document = QTextDocument()
270 document.setDefaultFont(option.font)
271 document.setHtml(text)
272 # the +5 is for margin. The +4 is voodoo;
273 # fm.height just give it too small.
274 return QSize(document.idealWidth() + 5, fm.height() + 4)