mirror of
https://github.com/avatao-content/baseimage-tutorial-framework
synced 2025-06-28 13:05:11 +00:00
Trigger first step in an individual event handler
This commit is contained in:
committed by
therealkrispet
parent
1f813a2138
commit
ddb47d696c
@ -11,22 +11,17 @@ LOG = logging.getLogger(__name__)
|
||||
class FSMHandler:
|
||||
keys = ['fsm']
|
||||
|
||||
def __init__(self, *, fsm_type, initial_trigger):
|
||||
def __init__(self, *, fsm_type):
|
||||
self.connector = None
|
||||
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.update' : self.handle_update
|
||||
}
|
||||
|
||||
def start(self):
|
||||
self.connector.subscribe('frontend.ready')
|
||||
|
||||
def handle_event(self, message, _):
|
||||
try:
|
||||
message = self.command_handlers[message['key']](message)
|
||||
@ -37,11 +32,6 @@ 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): # pylint: disable=inconsistent-return-statements
|
||||
if self.fsm.step(message['trigger']):
|
||||
return message
|
||||
|
Reference in New Issue
Block a user