--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/Helpviewer/Bookmarks/BookmarksImporters/ChromeImporter.py Sun Mar 18 15:05:38 2012 +0100
@@ -0,0 +1,172 @@
+# -*- coding: utf-8 -*-
+
+# Copyright (c) 2012 Detlev Offenbach <detlev@die-offenbachs.de>
+#
+
+"""
+Module implementing an importer for Chrome bookmarks.
+"""
+
+import os
+import json
+
+from PyQt4.QtCore import QCoreApplication, QDate, Qt
+
+from ..BookmarkNode import BookmarkNode
+
+from .BookmarksImporter import BookmarksImporter
+
+import UI.PixmapCache
+import Globals
+
+
+def getImporterInfo(id):
+ """
+ Module function to get information for the given source id.
+
+ @return tuple with an icon (QPixmap), readable name (string), name of
+ the default bookmarks file (string), an info text (string),
+ a prompt (string) and the default directory of the bookmarks file (string)
+ """
+ if id == "chrome":
+ if Globals.isWindowsPlatform():
+ standardDir = os.path.expandvars(
+ "%USERPROFILE%\\AppData\\Local\\Google\\Chrome\\User Data\\Default")
+ else:
+ standardDir = os.path.expanduser("~/.config/google-chrome/Default")
+ return (
+ UI.PixmapCache.getPixmap("chrome.png"),
+ "Google Chrome",
+ "Bookmarks",
+ QCoreApplication.translate("ChromeImporter",
+ """Google Chrome stores its bookmarks in the <b>Bookmarks</b> """
+ """text file. This file is usually located in"""),
+ QCoreApplication.translate("ChromeImporter",
+ """Please choose the file to begin importing bookmarks."""),
+ standardDir,
+ )
+ elif id == "chromium":
+ if Globals.isWindowsPlatform():
+ standardDir = os.path.expandvars(
+ "%USERPROFILE%\\AppData\\Local\\Google\\Chrome\\User Data\\Default")
+ else:
+ standardDir = os.path.expanduser("~/.config/chromium/Default")
+ return (
+ UI.PixmapCache.getPixmap("chromium.png"),
+ "Chromium",
+ "Bookmarks",
+ QCoreApplication.translate("ChromeImporter",
+ """Chromium stores its bookmarks in the <b>Bookmarks</b> text file. """
+ """This file is usually located in"""),
+ QCoreApplication.translate("ChromeImporter",
+ """Please choose the file to begin importing bookmarks."""),
+ standardDir,
+ )
+ else:
+ raise ValueError("Unsupported browser ID given ({0}).".format(id))
+
+
+class ChromeImporter(BookmarksImporter):
+ """
+ Class implementing the Chrome bookmarks importer.
+ """
+ def __init__(self, id="", parent=None):
+ """
+ Constructor
+
+ @param id source ID (string)
+ @param parent reference to the parent object (QObject)
+ """
+ super().__init__(id, parent)
+
+ self.__fileName = ""
+
+ def setPath(self, path):
+ """
+ Public method to set the path of the bookmarks file or directory.
+
+ @param path bookmarks file or directory (string)
+ """
+ self.__fileName = path
+
+ def open(self):
+ """
+ Public method to open the bookmarks file.
+
+ @return flag indicating success (boolean)
+ """
+ if not os.path.exists(self.__fileName):
+ self._error = True
+ self._errorString = self.trUtf8("File '{0}' does not exist.")\
+ .format(self.__fileName)
+ return False
+ return True
+
+ def importedBookmarks(self):
+ """
+ Public method to get the imported bookmarks.
+
+ @return imported bookmarks (BookmarkNode)
+ """
+ try:
+ f = open(self.__fileName, "r")
+ contents = json.load(f)
+ f.close()
+ except IOError as err:
+ self._error = True
+ self._errorString = self.trUtf8("File '{0}' cannot be read.\nReason: {1}")\
+ .format(self.__fileName, str(err))
+ return None
+
+ importRootNode = BookmarkNode(BookmarkNode.Root)
+ if contents["version"] == 1:
+ self.__processRoots(contents["roots"], importRootNode)
+
+ importRootNode.setType(BookmarkNode.Folder)
+ if self._id == "chrome":
+ importRootNode.title = self.trUtf8("Google Chrome Import")
+ elif self._id == "chromium":
+ importRootNode.title = self.trUtf8("Chromium Import")
+ else:
+ importRootNode.title = self.trUtf8("Imported {0}")\
+ .format(QDate.currentDate().toString(Qt.SystemLocaleShortDate))
+ return importRootNode
+
+ def __processRoots(self, data, rootNode):
+ """
+ Private method to process the bookmark roots.
+
+ @param data dictionary with the bookmarks data (dict)
+ @param rootNode node to add the bookmarks to (BookmarkNode)
+ """
+ for node in data.values():
+ if node["type"] == "folder":
+ self.__generateFolderNode(node, rootNode)
+ elif node["type"] == "url":
+ self.__generateUrlNode(node, rootNode)
+
+ def __generateFolderNode(self, data, rootNode):
+ """
+ Private method to process a bookmarks folder.
+
+ @param data dictionary with the bookmarks data (dict)
+ @param rootNode node to add the bookmarks to (BookmarkNode)
+ """
+ folder = BookmarkNode(BookmarkNode.Folder, rootNode)
+ folder.title = data["name"]
+ for node in data["children"]:
+ if node["type"] == "folder":
+ self.__generateFolderNode(node, folder)
+ elif node["type"] == "url":
+ self.__generateUrlNode(node, folder)
+
+ def __generateUrlNode(self, data, rootNode):
+ """
+ Private method to process a bookmarks node.
+
+ @param data dictionary with the bookmarks data (dict)
+ @param rootNode node to add the bookmarks to (BookmarkNode)
+ """
+ bookmark = BookmarkNode(BookmarkNode.Bookmark, rootNode)
+ bookmark.url = data["url"]
+ bookmark.title = data["name"]
