diff -r 67d861d9e492 -r 5ac4fc1dfc20 WebBrowser/Bookmarks/XbelWriter.py
--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/WebBrowser/Bookmarks/XbelWriter.py Thu Feb 11 20:27:07 2016 +0100
@@ -0,0 +1,102 @@
+# -*- coding: utf-8 -*-
+
+# Copyright (c) 2009 - 2016 Detlev Offenbach <detlev@die-offenbachs.de>
+#
+
+"""
+Module implementing a class to write XBEL bookmark files.
+"""
+
+from __future__ import unicode_literals
+
+from PyQt5.QtCore import QXmlStreamWriter, QIODevice, QFile, Qt
+
+from .BookmarkNode import BookmarkNode
+
+
+class XbelWriter(QXmlStreamWriter):
+ """
+ Class implementing a writer object to generate XBEL bookmark files.
+ """
+ def __init__(self):
+ """
+ Constructor
+ """
+ super(XbelWriter, self).__init__()
+
+ self.setAutoFormatting(True)
+
+ def write(self, fileNameOrDevice, root):
+ """
+ Public method to write an XBEL bookmark file.
+
+ @param fileNameOrDevice name of the file to write (string)
+ or device to write to (QIODevice)
+ @param root root node of the bookmark tree (BookmarkNode)
+ @return flag indicating success (boolean)
+ """
+ if isinstance(fileNameOrDevice, QIODevice):
+ f = fileNameOrDevice
+ else:
+ f = QFile(fileNameOrDevice)
+ if root is None or not f.open(QFile.WriteOnly):
+ return False
+
+ self.setDevice(f)
+ return self.__write(root)
+
+ def __write(self, root):
+ """
+ Private method to write an XBEL bookmark file.
+
+ @param root root node of the bookmark tree (BookmarkNode)
+ @return flag indicating success (boolean)
+ """
+ self.writeStartDocument()
+ self.writeDTD("<!DOCTYPE xbel>")
+ self.writeStartElement("xbel")
+ self.writeAttribute("version", "1.0")
+ if root.type() == BookmarkNode.Root:
+ for child in root.children():
+ self.__writeItem(child)
+ else:
+ self.__writeItem(root)
+
+ self.writeEndDocument()
+ return True
+
+ def __writeItem(self, node):
+ """
+ Private method to write an entry for a node.
+
+ @param node reference to the node to be written (BookmarkNode)
+ """
+ if node.type() == BookmarkNode.Folder:
+ self.writeStartElement("folder")
+ if node.added.isValid():
+ self.writeAttribute("added", node.added.toString(Qt.ISODate))
+ self.writeAttribute("folded", node.expanded and "no" or "yes")
+ self.writeTextElement("title", node.title)
+ for child in node.children():
+ self.__writeItem(child)
+ self.writeEndElement()
+ elif node.type() == BookmarkNode.Bookmark:
+ self.writeStartElement("bookmark")
+ if node.url:
+ self.writeAttribute("href", node.url)
+ if node.added.isValid():
+ self.writeAttribute("added", node.added.toString(Qt.ISODate))
+ if node.modified.isValid():
+ self.writeAttribute(
+ "modified", node.modified.toString(Qt.ISODate))
+ if node.visited.isValid():
+ self.writeAttribute(
+ "visited", node.visited.toString(Qt.ISODate))
+ self.writeTextElement("title", node.title)
+ if node.desc:
+ self.writeTextElement("desc", node.desc)
+ self.writeEndElement()
+ elif node.type() == BookmarkNode.Separator:
+ self.writeEmptyElement("separator")
+ if node.added.isValid():
+ self.writeAttribute("added", node.added.toString(Qt.ISODate))
