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
|
# Copyright (c) 2016 Ultimaker B.V.
# Cura is released under the terms of the AGPLv3 or higher.
from UM.Logger import Logger
class MachineActionManager:
def __init__(self):
## Dict of all known machine actions
self._machine_actions = {}
## Dict of all required actions by machine reference.
self._required_actions = {}
## Dict of all supported actions by machine reference
self._supported_actions = {}
## Dict of all actions that need to be done when first added by machine reference.
self._first_start_actions = {}
## Add a required action to a machine
# Raises an exception when the action is not recognised.
def addRequiredAction(self, machine, action_key):
if action_key in self._machine_actions:
if machine in self._required_actions:
self._required_actions[machine].append(self._machine_actions[action_key])
else:
self._required_actions[machine] = set(self._machine_actions[action_key])
else:
# Todo: define specific Exception types (instead of general type)
raise Exception("Action %s, which is required for %s is not known." % (action_key, machine.getKey()))
## Add a supported action to a machine.
def addSupportedAction(self, machine, action_key):
if action_key in self._machine_actions:
if machine in self._supported_actions:
self._supported_actions[machine].append(self._machine_actions[action_key])
else:
self._supported_actions[machine] = set(self._machine_actions[action_key])
else:
Logger.log("W", "Unable to add %s to %s, as the action is not recognised", action_key, machine.getKey())
## Add an action to the first start list of a machine.
def addFirstStartAction(self, machine, action_key, index = None):
if action_key in self._machine_actions:
if machine in self._supported_actions and index is not None:
self._supported_actions[machine].insert(index, self._machine_actions[action_key])
else:
self._supported_actions[machine] = [self._machine_actions[action_key]]
else:
Logger.log("W", "Unable to add %s to %s, as the action is not recognised", action_key, machine.getKey())
## Add a (unique) MachineAction
# if the Key of the action is not unique, an exception is raised.
def addMachineAction(self, action):
if action.getKey() not in self._machine_action:
self._machine_action[action.getKey()] = action
else:
# Todo: define specific Exception types (instead of general type)
raise Exception("MachineAction with key %s was already added. Actions must have unique keys.", action.getKey())
## Get all actions supported by given machine
# \param machine The machine you want the supported actions of
# \returns set of supported actions.
def getSupportedActions(self, machine):
if machine in self._supported_actions:
return self._supported_actions[machine]
else:
return set()
## Get all actions required by given machine
# \param machine The machine you want the required actions of
# \returns set of required actions.
def getRequiredActions(self, machine):
if machine in self._required_actions:
return self._required_actions[machine]
else:
return set()
## Remove Machine action from manager
# \param action to remove
def removeMachineAction(self, action):
try:
del self._machine_actions[action.getKey()]
except KeyError:
Logger.log("w", "Trying to remove MachineAction (%s) that was already removed", action.getKey())
## Get MachineAction by key
# \param key String of key to select
# \return Machine action if found, None otherwise
def getMachineAction(self, key):
if key in self._machine_actions:
return self._machine_actions[key]
else:
return None
|