Unit4-Manager/BookmarkManager.py at master · waLLxAck/Unit4-Manager · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
import uuid
from time import time

from util.Initializer import FILE_HANDLER


class BookmarkManager:
    class BookmarkType:
        FOLDER = "folder"
        URL = "url"

    def __init__(self):
        self.chrome_bookmarks = FILE_HANDLER.get_bookmarks_chrome()

    def __get_root_level_bookmarks(self):
        return self.chrome_bookmarks["roots"]["bookmark_bar"]

    def __search_bookmark(self, start_location, bookmark_name, bookmark_type=BookmarkType.URL):
        for bookmark in self.__get_children(start_location):
            if bookmark["name"] == bookmark_name and bookmark["type"] == bookmark_type:
                return bookmark

    @staticmethod
    def __get_children(bookmark) -> list:
        return bookmark["children"]

    def get_unit4_folder(self):
        folder_name = "Unit4"
        unit4_folder = self.__search_bookmark(self.__get_root_level_bookmarks(), folder_name,
                                              self.BookmarkType.FOLDER)
        if not unit4_folder:
            unit4_folder = self.create_folder(self.__get_root_level_bookmarks(), folder_name)
        return unit4_folder

    def get_projects_folder(self):
        projects_path = FILE_HANDLER.get_settings()["projects_path"]
        folder_name = ""
        root = self.__get_root_level_bookmarks()
        for folder in projects_path.split("/"):
            folder_name = folder
            root = self.get_folder(root, folder_name)
        return root

    def get_folder(self, root, folder_name):
        folder = self.__search_bookmark(root, folder_name,
                                              self.BookmarkType.FOLDER)
        if not folder:
            folder = self.create_folder(root, folder_name)
        return folder

    def create_folder(self, bookmark_folder_root, folder_name):
        if self.__search_bookmark(bookmark_folder_root, folder_name, self.BookmarkType.FOLDER):
            return self.__search_bookmark(bookmark_folder_root, folder_name,
                                          self.BookmarkType.FOLDER)
        time_string = str(time())
        folder = {
            "children": [
            ],
            "date_added": time_string,
            "date_modified": time_string,
            "guid": str(uuid.uuid4()),
            "id": "1636",
            "name": folder_name,
            "type": "folder"
        }
        self.__get_children(bookmark_folder_root).append(folder)
        print(f"Folder '{folder_name}' created.")
        return self.__search_bookmark(bookmark_folder_root, folder_name, self.BookmarkType.FOLDER)

    def create_url(self, bookmark_folder_root, bookmark_name: str, url: str):
        if self.__search_bookmark(bookmark_folder_root, bookmark_name):
            return
        time_string = str(time())
        url_bookmark = {
            "date_added": time_string,
            "guid": str(uuid.uuid4()),
            "id": "1635",
            "meta_info": {
                "imageData": "https://sync-thumbnails.operacdn.com/c75a52acc5452f58075c8f31b3077227b3546dddda6f578c6e12b511ee850c68.jpe",
                "imageDataType": "2",
                "imageID": "1795D60FF3E10A3C72E93565696A3F16067CF88426D95776A402B65D36015AFD",
                "imageIdentifier": "23c9aa7c",
                "imageType": "2"
            },
            "name": bookmark_name,
            "type": "url",
            "url": url
        }
        self.__get_children(bookmark_folder_root).append(url_bookmark)
        print(f"URL '{bookmark_name}' created.")

    def commit_changes(self):
        FILE_HANDLER.save_changes(self.chrome_bookmarks)

    def update_bookmarks(self):
        self.chrome_bookmarks = FILE_HANDLER.get_bookmarks_chrome()