mirror of
				https://github.com/avatao-content/baseimage-tutorial-framework
				synced 2025-11-04 10:02:55 +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