#!/usr/bin/env python
#############################################################################
##
## Copyright (C) 2010 Riverbank Computing Limited.
## Copyright (C) 2010 Nokia Corporation and/or its subsidiary(-ies).
## All rights reserved.
##
## This file is part of the examples of PyQt.
##
## $QT_BEGIN_LICENSE:BSD$
## You may use this file under the terms of the BSD license as follows:
##
## "Redistribution and use in source and binary forms, with or without
## modification, are permitted provided that the following conditions are
## met:
## * Redistributions of source code must retain the above copyright
## notice, this list of conditions and the following disclaimer.
## * Redistributions in binary form must reproduce the above copyright
## notice, this list of conditions and the following disclaimer in
## the documentation and/or other materials provided with the
## distribution.
## * Neither the name of Nokia Corporation and its Subsidiary(-ies) nor
## the names of its contributors may be used to endorse or promote
## products derived from this software without specific prior written
## permission.
##
## THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
## "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
## LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
## A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
## OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
## SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
## LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
## DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
## THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
## (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
## OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE."
## $QT_END_LICENSE$
##
#############################################################################
# This is only needed for Python v2 but is harmless for Python v3.
import sip
sip.setapi('QVariant', 2)
from PyQt4 import QtCore, QtGui, QtXml
class XbelGenerator(object):
def __init__(self, treeWidget):
self.treeWidget = treeWidget
self.out = QtCore.QTextStream()
def write(self, device):
self.out.setDevice(device)
self.out.setCodec('UTF-8')
self.out << '<?xml version="1.0" encoding="UTF-8"?>\n' \
<< '<!DOCTYPE xbel>\n' \
<< '<xbel version="1.0">\n'
for i in range(self.treeWidget.topLevelItemCount()):
self.generateItem(self.treeWidget.topLevelItem(i), 1)
self.out << '</xbel>\n'
return True
def indent(self, depth):
IndentSize = 4
return IndentSize * depth * ' '
def escapedText(self, txt):
result = txt
result.replace('&', '&')
result.replace('<', '<')
result.replace('>', '>')
return result
def escapedAttribute(self, txt):
result = self.escapedText(txt)
result.replace('"', '"')
return '"' + result + '"'
def generateItem(self, item, depth):
tagName = item.data(0, QtCore.Qt.UserRole)
if tagName == 'folder':
folded = (not self.treeWidget.isItemExpanded(item))
self.out << self.indent(depth) << '<folder folded="'
if folded:
self.out << 'yes'
else:
self.out << 'no'
self.out << '">\n' \
<< self.indent(depth + 1) << '<title>' \
<< self.escapedText(item.text(0)) \
<< '</title>\n'
for i in range(item.childCount()):
self.generateItem(item.child(i), depth + 1)
self.out << self.indent(depth) << '</folder>\n'
elif tagName == 'bookmark':
self.out << self.indent(depth) << '<bookmark'
if item.text(1):
self.out << ' href=' \
<< self.escapedAttribute(item.text(1))
self.out << '>\n' \
<< self.indent(depth + 1) << '<title>' \
<< self.escapedText(item.text(0)) \
<< '</title>\n' \
<< self.indent(depth) << '</bookmark>\n'
elif tagName == 'separator':
self.out << self.indent(depth) << '<separator/>\n'
class XbelHandler(QtXml.QXmlDefaultHandler):
def __init__(self, treeWidget):
super(XbelHandler, self).__init__()
self.treeWidget = treeWidget
self.folderIcon = QtGui.QIcon()
self.bookmarkIcon = QtGui.QIcon()
self.currentText = ''
self.errorStr = ''
self.item = None
self.metXbelTag = False
style = self.treeWidget.style()
self.folderIcon.addPixmap(style.standardPixmap(QtGui.QStyle.SP_DirClosedIcon),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.folderIcon.addPixmap(style.standardPixmap(QtGui.QStyle.SP_DirOpenIcon),
QtGui.QIcon.Normal, QtGui.QIcon.On)
self.bookmarkIcon.addPixmap(style.standardPixmap(QtGui.QStyle.SP_FileIcon))
def startElement(self, namespaceURI, localName, qName, attributes):
if not self.metXbelTag and qName != 'xbel':
self.errorStr = "The file is not an XBEL file."
return False
if qName == 'xbel':
version = attributes.value('version')
if version and version != '1.0':
self.errorStr = "The File is not an XBEL version 1.0 file."
return False
self.metXbelTag = True
elif qName == 'folder':
self.item = self.createChildItem(qName)
self.item.setFlags(self.item.flags() | QtCore.Qt.ItemIsEditable)
self.item.setIcon(0, self.folderIcon)
self.item.setText(0, "Folder")
folded = (attributes.value('folded') != 'no')
self.treeWidget.setItemExpanded(self.item, (not folded))
elif qName == 'bookmark':
self.item = self.createChildItem(qName)
self.item.setFlags(self.item.flags() | QtCore.Qt.ItemIsEditable)
self.item.setIcon(0, self.bookmarkIcon)
self.item.setText(0, "Unkown title")
self.item.setText(1, attributes.value('href'))
elif qName == 'separator':
self.item = self.createChildItem(qName)
self.item.setFlags(self.item.flags() & ~QtCore.Qt.ItemIsSelectable)
self.item.setText(0, 30 * "\xb7")
self.currentText = ''
return True
def endElement(self, namespaceURI, localName, qName):
if qName == 'title':
if self.item:
self.item.setText(0, self.currentText)
elif qName in ('folder', 'bookmark', 'separator'):
self.item = self.item.parent()
return True
def characters(self, txt):
self.currentText += txt
return True
def fatalError(self, exception):
QtGui.QMessageBox.information(self.treeWidget.window(),
"SAX Bookmarks",
"Parse error at line %d, column %d:\n%s" % (exception.lineNumber(), exception.columnNumber(), exception.message()))
return False
def errorString(self):
return self.errorStr
def createChildItem(self, tagName):
if self.item:
childItem = QtGui.QTreeWidgetItem(self.item)
else:
childItem = QtGui.QTreeWidgetItem(self.treeWidget)
childItem.setData(0, QtCore.Qt.UserRole, tagName)
return childItem
class MainWindow(QtGui.QMainWindow):
def __init__(self):
super(MainWindow, self).__init__()
self.treeWidget = QtGui.QTreeWidget()
self.treeWidget.header().setResizeMode(QtGui.QHeaderView.Stretch)
self.treeWidget.setHeaderLabels(("Title", "Location"))
self.setCentralWidget(self.treeWidget)
self.createActions()
self.createMenus()
self.statusBar().showMessage("Ready")
self.setWindowTitle("SAX Bookmarks")
self.resize(480, 320)
def open(self):
fileName = QtGui.QFileDialog.getOpenFileName(self,
"Open Bookmark File", QtCore.QDir.currentPath(),
"XBEL Files (*.xbel *.xml)")
if not fileName:
return
self.treeWidget.clear()
handler = XbelHandler(self.treeWidget)
reader = QtXml.QXmlSimpleReader()
reader.setContentHandler(handler)
reader.setErrorHandler(handler)
file = QtCore.QFile(fileName)
if not file.open(QtCore.QFile.ReadOnly | QtCore.QFile.Text):
QtGui.QMessageBox.warning(self, "SAX Bookmarks",
"Cannot read file %s:\n%s." % (fileName, file.errorString()))
return
xmlInputSource = QtXml.QXmlInputSource(file)
if reader.parse(xmlInputSource):
self.statusBar().showMessage("File loaded", 2000)
def saveAs(self):
fileName = QtGui.QFileDialog.getSaveFileName(self,
"Save Bookmark File", QtCore.QDir.currentPath(),
"XBEL Files (*.xbel *.xml)")
if not fileName:
return
file = QtCore.QFile(fileName)
if not file.open(QtCore.QFile.WriteOnly | QtCore.QFile.Text):
QtGui.QMessageBox.warning(self, "SAX Bookmarks",
"Cannot write file %s:\n%s." % (fileName, file.errorString()))
return
generator = XbelGenerator(self.treeWidget)
if generator.write(file):
self.statusBar().showMessage("File saved", 2000)
def about(self):
QtGui.QMessageBox.about(self, "About SAX Bookmarks",
"The <b>SAX Bookmarks</b> example demonstrates how to use "
"Qt's SAX classes to read XML documents and how to generate "
"XML by hand.")
def createActions(self):
self.openAct = QtGui.QAction("&Open...", self, shortcut="Ctrl+O",
triggered=self.open)
self.saveAsAct = QtGui.QAction("&Save As...", self, shortcut="Ctrl+S",
triggered=self.saveAs)
self.exitAct = QtGui.QAction("E&xit", self, shortcut="Ctrl+Q",
triggered=self.close)
self.aboutAct = QtGui.QAction("&About", self, triggered=self.about)
self.aboutQtAct = QtGui.QAction("About &Qt", self,
triggered=QtGui.qApp.aboutQt)
def createMenus(self):
self.fileMenu = self.menuBar().addMenu("&File")
self.fileMenu.addAction(self.openAct)
self.fileMenu.addAction(self.saveAsAct)
self.fileMenu.addAction(self.exitAct)
self.menuBar().addSeparator()
self.helpMenu = self.menuBar().addMenu("&Help")
self.helpMenu.addAction(self.aboutAct)
self.helpMenu.addAction(self.aboutQtAct)
if __name__ == "__main__":
import sys
app = QtGui.QApplication(sys.argv)
mainWin = MainWindow()
mainWin.show()
mainWin.open()
sys.exit(app.exec_())
