mirror of
https://github.com/avatao-content/baseimage-tutorial-framework
synced 2025-06-28 13:25:12 +00:00
Move initial FSM trigger to backend
This commit is contained in:
@ -11,16 +11,21 @@ LOG = logging.getLogger(__name__)
|
||||
class FSMHandler:
|
||||
keys = ['fsm']
|
||||
|
||||
def __init__(self, *, fsm_type):
|
||||
def __init__(self, *, fsm_type, initial_trigger):
|
||||
self.fsm = fsm_type()
|
||||
self._fsm_updater = FSMUpdater(self.fsm)
|
||||
self.auth_key = KeyManager().auth_key
|
||||
self.initial_trigger = initial_trigger
|
||||
|
||||
self.command_handlers = {
|
||||
'frontend.ready': self.handle_ready,
|
||||
'fsm.step' : self.handle_step,
|
||||
'fsm.announce' : self.handle_announce
|
||||
}
|
||||
|
||||
def start(self):
|
||||
self.connector.subscribe('frontend.ready')
|
||||
|
||||
def handle_event(self, message, connector):
|
||||
try:
|
||||
message = self.command_handlers[message['key']](message)
|
||||
@ -31,6 +36,11 @@ class FSMHandler:
|
||||
except KeyError:
|
||||
LOG.error('IGNORING MESSAGE: Invalid message received: %s', message)
|
||||
|
||||
def handle_ready(self, message):
|
||||
self.fsm.step(self.initial_trigger)
|
||||
self.connector.unsubscribe('frontend.ready')
|
||||
return message
|
||||
|
||||
def handle_step(self, message):
|
||||
if self.fsm.step(message['trigger']):
|
||||
return message
|
||||
|
Reference in New Issue
Block a user