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
|
# Copyright (c) 2020 Ultimaker B.V.
# Uranium is released under the terms of the LGPLv3 or higher.
from collections import defaultdict
from typing import Dict, Union, List
basic_metadata_type = Union[str, int, float, bool, None]
full_metadata_type = Union[List[basic_metadata_type], basic_metadata_type, Dict[str, basic_metadata_type]]
# modified from Original source: https://github.com/python/mypy/issues/731#issuecomment-539905783
# The recursive type doesn't seem to work for us. MyPy crashes with it. So this limits the type to 3 levels deep nesting.
JSONPrimitive = Union[str, int, bool, None]
JSONTypeLevel1 = Union[JSONPrimitive, Dict[str, JSONPrimitive], List[JSONPrimitive]]
JSONTypeLevel2 = Union[JSONPrimitive, Dict[str, JSONTypeLevel1], List[JSONTypeLevel1]]
JSONType = Union[JSONPrimitive, Dict[str, JSONTypeLevel2], List[JSONTypeLevel2]]
class WorkspaceMetadataStorage:
# The WorkspaceMetadataStorage, as the name implies, allows for plugins to store (and retrieve) extra information
# to a workspace.
def __init__(self) -> None:
# We allow for a set of key value pairs to be stored per plugin.
self._data = defaultdict(dict) # type: Dict[str, Dict[str, JSONType]]
def setEntryToStore(self, plugin_id: str, key: str, data: JSONType) -> None:
self._data[plugin_id][key] = data
def setAllData(self, data: Dict[str, Dict[str, JSONType]]) -> None:
self._data = defaultdict(dict, data)
def getAllData(self) -> Dict[str, Dict[str, JSONType]]:
return self._data
def getPluginMetadata(self, plugin_id: str) -> Dict[str, JSONType]:
return self._data[plugin_id]
def getPluginMetadataEntry(self, plugin_id: str, key: str) -> JSONType:
return self._data[plugin_id].get(key)
def clear(self) -> None:
self._data = defaultdict(dict)
|
