import os
+import shlex
import sys
from PyQt4.QtCore import *
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):
elif node.nodeType() == QDomNode.EntityNode:
return QString('unsupported node type')
elif node.nodeType() == QDomNode.ProcessingInstructionNode:
- return QVariant()
+ obj = QObject()
+ obj.setProperty('nodeType', QString('element'))
+ obj.setProperty('content', node.nodeName() + " " + node.nodeValue())
+ return obj
elif node.nodeType() == QDomNode.CommentNode:
obj = QObject()
obj.setProperty('nodeType', QString('comment'))
if not index.isValid():
return Qt.ItemIsEnabled
return Qt.ItemIsEnabled | Qt.ItemIsSelectable
-
+
def headerData(self, section, orientation, role):
return QVariant()
self.childItems[i] = childItem
return childItem
return None
-
+
def parent(self):
return self.parentItem
self.setHeaderHidden(True)
self.setAttribute(Qt.WA_MacShowFocusRect, 0)
self.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
+ self.setHorizontalScrollMode(QAbstractItemView.ScrollPerPixel)
class XmlWindow(QDialog):
def __init__(self, parent=None, title='XML Window'):
self.model = None
self.title = title
- self.view = XmlView(self)
+ self.view = self.initView()
self.delegate = XmlDelegate(self)
self.view.setItemDelegate(self.delegate)
self.delegate.insertNodeDelegate('element', ElemNodeDelegate())
layout.addWidget(self.view)
self.setLayout(layout)
+ self.connect(self.view, SIGNAL('expanded(QModelIndex)'), self.onItemExpanded)
self.updateView()
+ def initView(self):
+ return XmlView(self)
+
def show(self):
self.updateView()
QDialog.show(self)
-
+
+ def readContent(self):
+ raise ValueError("readContent needs to be implemented in the subclass")
+
def updateView(self):
del self.document
del self.model
self.model = DomModel(self.document, self)
self.view.setModel(self.model)
- self.view.expand(self.model.index(0, 0)) #expand first level only
- #move the code below to rspec window
- rspec_file = config.getSliceRSpecFile()
- if not os.path.exists(rspec_file):
- return
+ self.document.setContent(self.readContent())
- self.document.setContent(open(rspec_file,'r').read())
+ if self.document.childNodes().count() == 0:
+ # empty document - do nothing
+ pass
+ elif self.document.childNodes().item(0).nodeType() == QDomNode.ProcessingInstructionNode:
+ # the first item is the <xml> tag, so expand the second
+ self.view.expand(self.model.index(1,0))
+ else:
+ # the document didn't start with an <xml> tag; let's try to expand
+ # the first item on the assumption that it is our root level xml
+ # tag.
+ self.view.expand(self.model.index(0,0))
+ self.view.header().resizeSection(0,0)
+ self.view.resizeColumnToContents(0)
+ def onItemExpanded(self, index):
+ self.view.header().resizeSection(0,0)
+ self.view.resizeColumnToContents(0)
class XmlDelegate(QItemDelegate):
del self.delegates[nodeType]
def paint(self, painter, option, index):
+ if isinstance(index.model().data(index),QVariant):
+ return
nodeType = index.model().data(index).property('nodeType')
delegate = self.delegates.get(str(nodeType.toString()))
#print "TYPE:", str(type(str(nodeType.toString())))
def sizeHint(self, option, index):
fm = option.fontMetrics
- text = index.model().data(index).property('content').toString()
+ if isinstance(index.model().data(index),QVariant):
+ return QSize(0, 0)
+
+ nodeType = index.model().data(index).property('nodeType')
+ delegate = self.delegates.get(str(nodeType.toString()), None)
+
+ # Use the text from the appropriate delegate if we have it, to
+ # compute the size. Should probably finish the sizeHint() methods
+ # of the delegates at some point, and call them instead.
+ if delegate!=None and hasattr(delegate, "getItemText"):
+ text = delegate.getItemText(option, index)
+ else:
+ text = index.model().data(index).property('content').toString()
+
document = QTextDocument()
document.setDefaultFont(option.font)
document.setHtml(text)
+
# the +5 is for margin. The +4 is voodoo;
# fm.height just give it too small.
- return QSize(document.idealWidth() + 5, fm.height() + 4)
+ return QSize(document.idealWidth() + 5, fm.height() + 4)
class ElemNodeDelegate(QAbstractItemDelegate):
- def paint(self, painter, option, index):
+ def getItemText(self, option, index):
text = index.model().data(index)
- palette = QApplication.palette()
- document = QTextDocument()
- document.setDefaultFont(option.font)
nonHighGlobPattern = '<<b><font color="#b42be2">%s</font></b>%s>'
nonHighAttPattern = ' <b>%s</b>="<font color="#1e90ff">%s</font>"'
highGlobPattern = '<<b>%s</b>%s>'
highAttPattern = ' <b>%s</b>="%s"'
+
def getHtmlText(plainText, globPattern, attPattern):
# print "PLAIN TEXT:", plainText
tmp = plainText.split(' ', 1)
AttListHtml = ''
if len(tmp) > 1:
# many elems don't have atts...
- attList = tmp[1].split()
+ # use shlex.split so we can handle quoted strings with spaces
+ # in them, like <link enpoints="foo bar">. Note that there are
+ # documented problems with shlex.split and unicode, so we
+ # convert any potential unicode to a string first.
+ attList = shlex.split(str(tmp[1]))
for att in attList:
- tmp = att.split('=')
- attName = tmp[0]
- attValue = tmp[1][1:-1]
+ tmp = att.split('=',1)
+ if len(tmp)>=2:
+ attName = tmp[0]
+ attValue = tmp[1]
+ else:
+ # this shouldn't happen, but if it does, pretend the
+ # attribute value is blank.
+ attName = tmp[0]
+ attValue = ""
AttListHtml += (nonHighAttPattern % (attName, attValue))
html = (globPattern % (elemName, AttListHtml))
return html
+
def colorize(color, text):
return '<font color=' + color + '>' + text + '</font>'
+
text = str(index.model().data(index).property('content').toString())
# print "TEXT:", text
if option.state & QStyle.State_Selected:
+ palette = QApplication.palette()
htmlText = colorize(palette.highlightedText().color().name(),
getHtmlText(text, highGlobPattern, highAttPattern))
- document.setHtml(QString(htmlText))
else:
htmlText = getHtmlText(text, nonHighGlobPattern, nonHighAttPattern)
- document.setHtml(QString(htmlText))
+
+ return htmlText
+
+ def paint(self, painter, option, index):
+ palette = QApplication.palette()
+ document = QTextDocument()
+ document.setDefaultFont(option.font)
+
+ htmlText = self.getItemText(option, index)
+ document.setHtml(QString(htmlText))
+
color = palette.highlight().color() \
if option.state & QStyle.State_Selected \
else palette.base().color()
+
painter.save()
# print "COLOR:", color.name()
# voodoo: if not highlighted, filling the rect
document.drawContents(painter)
painter.restore()
- def sizeHint(self, option, index):
- sizeHint(self, option, index)
-
class TextNodeDelegate(QAbstractItemDelegate):
def paint(self, painter, option, index):
#print "TEXT DELEG CALLED"
paint(self, painter, option, index)
- def sizeHint(self, option, index):
- sizeHint(self, option, index)
-
class CommentNodeDelegate(QAbstractItemDelegate):
def paint(self, painter, option, index):
#print "TEXT DELEG CALLED"
paint(self, painter, option, index)
-def paint(self, painter, option, index):
- text = index.model().data(index).property('content').toString()
- palette = QApplication.palette()
- document = QTextDocument()
- document.setDefaultFont(option.font)
- if option.state & QStyle.State_Selected:
- rx = QRegExp(QString('<font .*>'))
- rx.setMinimal(True)
- # If selected, I remove the <font color="..."> by hand,
- # and give the highlight color
- document.setHtml(QString("<font color=%1>%2</font>") \
- .arg(palette.highlightedText().color().name())\
- .arg(text.replace(rx, QString('')).
- replace(QString('</font>'),QString(''))))
- else:
- document.setHtml(text)
- color = palette.highlight().color() \
- if option.state & QStyle.State_Selected \
- else palette.base().color()
- painter.save()
- # voodoo: if not highlighted, filling the rect
- # with the base color makes no difference
- painter.fillRect(option.rect, color)
- painter.translate(option.rect.x(), option.rect.y())
- document.drawContents(painter)
- painter.restore()
-
+ def paint(self, painter, option, index):
+ text = index.model().data(index).property('content').toString()
+ palette = QApplication.palette()
+ document = QTextDocument()
+ document.setDefaultFont(option.font)
+ if option.state & QStyle.State_Selected:
+ rx = QRegExp(QString('<font .*>'))
+ rx.setMinimal(True)
+ # If selected, I remove the <font color="..."> by hand,
+ # and give the highlight color
+ document.setHtml(QString("<font color=%1>%2</font>") \
+ .arg(palette.highlightedText().color().name())\
+ .arg(text.replace(rx, QString('')).
+ replace(QString('</font>'),QString(''))))
+ else:
+ document.setHtml(text)
+ color = palette.highlight().color() \
+ if option.state & QStyle.State_Selected \
+ else palette.base().color()
+ painter.save()
+ # voodoo: if not highlighted, filling the rect
+ # with the base color makes no difference
+ painter.fillRect(option.rect, color)
+ painter.translate(option.rect.x(), option.rect.y())
+ document.drawContents(painter)
+ painter.restore()
git://git.onelab.eu / sface.git / blobdiff