--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/src/eric7/WebBrowser/Bookmarks/BookmarksImporters/ChromeImporter.py Thu Jul 07 11:23:56 2022 +0200
@@ -0,0 +1,189 @@
+# -*- coding: utf-8 -*-
+
+# Copyright (c) 2012 - 2022 Detlev Offenbach <detlev@die-offenbachs.de>
+#
+
+"""
+Module implementing an importer for Chrome bookmarks.
+"""
+
+import os
+import json
+
+from PyQt6.QtCore import QCoreApplication, QDate, Qt
+
+from .BookmarksImporter import BookmarksImporter
+
+import UI.PixmapCache
+import Globals
+
+
+def getImporterInfo(sourceId):
+ """
+ Module function to get information for the given source id.
+
+ @param sourceId id of the browser ("chrome" or "chromium")
+ @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)
+ @exception ValueError raised to indicate an invalid browser ID
+ """
+ if sourceId not in ("chrome", "chromium"):
+ raise ValueError(
+ "Unsupported browser ID given ({0}).".format(sourceId))
+
+ if sourceId == "chrome":
+ if Globals.isWindowsPlatform():
+ standardDir = os.path.expandvars(
+ "%USERPROFILE%\\AppData\\Local\\Google\\Chrome\\"
+ "User Data\\Default")
+ elif Globals.isMacPlatform():
+ standardDir = os.path.expanduser(
+ "~/Library/Application Support/Google/Chrome/Default")
+ else:
+ standardDir = os.path.expanduser("~/.config/google-chrome/Default")
+ return (
+ UI.PixmapCache.getPixmap("chrome"),
+ "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,
+ )
+ else:
+ 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"),
+ "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,
+ )
+
+
+class ChromeImporter(BookmarksImporter):
+ """
+ Class implementing the Chrome bookmarks importer.
+ """
+ def __init__(self, sourceId="", parent=None):
+ """
+ Constructor
+
+ @param sourceId source ID (string)
+ @param parent reference to the parent object (QObject)
+ """
+ super().__init__(sourceId, 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.tr(
+ "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:
+ with open(self.__fileName, "r", encoding="utf-8") as f:
+ contents = json.load(f)
+ except OSError as err:
+ self._error = True
+ self._errorString = self.tr(
+ "File '{0}' cannot be read.\nReason: {1}"
+ ).format(self.__fileName, str(err))
+ return None
+
+ from ..BookmarkNode import BookmarkNode
+ importRootNode = BookmarkNode(BookmarkNode.Folder)
+ if contents["version"] == 1:
+ self.__processRoots(contents["roots"], importRootNode)
+
+ if self._id == "chrome":
+ importRootNode.title = self.tr("Google Chrome Import")
+ elif self._id == "chromium":
+ importRootNode.title = self.tr("Chromium Import")
+ else:
+ importRootNode.title = self.tr(
+ "Imported {0}"
+ ).format(QDate.currentDate().toString(
+ Qt.DateFormat.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)
+ """
+ from ..BookmarkNode import BookmarkNode
+ folder = BookmarkNode(BookmarkNode.Folder, rootNode)
+ folder.title = data["name"].replace("&", "&&")
+ 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)
+ """
+ from ..BookmarkNode import BookmarkNode
+ bookmark = BookmarkNode(BookmarkNode.Bookmark, rootNode)
+ bookmark.url = data["url"]
+ bookmark.title = data["name"].replace("&", "&&")
