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
|
# Copyright (c) 2015 Ultimaker B.V.
# Uranium is released under the terms of the LGPLv3 or higher.
from UM.PluginObject import PluginObject
import collections
from typing import Optional, Any, Callable, List
## Base class for plugins that extend the functionality of Uranium.
# Every extension adds a (sub) menu to the extension menu with one or
# more menu items.
class Extension(PluginObject):
def __init__(self):
super().__init__()
self._menu_function_dict = collections.OrderedDict()
self._menu_name = None # type: Optional[str]
## Add an item to the sub-menu of the extension
# \param name \type{string}
# \param function \type{function}
def addMenuItem(self, name: str, func: Callable[[], Any]):
self._menu_function_dict[name] = func
## Set name of the menu where all menu items are placed in
# \param name \type{string}
def setMenuName(self, name: str):
self._menu_name = name
## Get the name of the menu where all menu items are placed in
# \param menu name \type{string}
def getMenuName(self) -> str:
return self._menu_name
## Call function associated with option
# \param name \type{string}
def activateMenuItem(self, name: str):
if name in self._menu_function_dict:
self._menu_function_dict[name]()
## Get list of all menu item names
# \return \type{list}
def getMenuItemList(self) -> List[str]:
return list(self._menu_function_dict.keys())
|