mirror of
				https://github.com/avatao-content/baseimage-tutorial-framework
				synced 2025-11-03 14:52:55 +00:00 
			
		
		
		
	Rename message keys
This commit is contained in:
		@@ -9,7 +9,7 @@ LOG = logging.getLogger(__name__)
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
class FSMHandler:
 | 
					class FSMHandler:
 | 
				
			||||||
    keys = ['fsm.step', 'fsm.update']
 | 
					    keys = ['fsm.step', 'fsm.announce']
 | 
				
			||||||
    def __init__(self, *, fsm_type):
 | 
					    def __init__(self, *, fsm_type):
 | 
				
			||||||
        self.fsm = fsm_type()
 | 
					        self.fsm = fsm_type()
 | 
				
			||||||
        self._fsm_updater = FSMUpdater(self.fsm)
 | 
					        self._fsm_updater = FSMUpdater(self.fsm)
 | 
				
			||||||
@@ -17,7 +17,7 @@ class FSMHandler:
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
        self.command_handlers = {
 | 
					        self.command_handlers = {
 | 
				
			||||||
            'fsm.step'   : self.handle_step,
 | 
					            'fsm.step'   : self.handle_step,
 | 
				
			||||||
            'fsm.update' : self.handle_update
 | 
					            'fsm.announce' : self.handle_announce
 | 
				
			||||||
        }
 | 
					        }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    def handle_event(self, message, connector):
 | 
					    def handle_event(self, message, connector):
 | 
				
			||||||
@@ -34,6 +34,6 @@ class FSMHandler:
 | 
				
			|||||||
        if self.fsm.step(message['trigger']):
 | 
					        if self.fsm.step(message['trigger']):
 | 
				
			||||||
            return message
 | 
					            return message
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    def handle_update(self, message):
 | 
					    def handle_announce(self, message):
 | 
				
			||||||
        # pylint: disable=no-self-use
 | 
					        # pylint: disable=no-self-use
 | 
				
			||||||
        return message
 | 
					        return message
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -5,7 +5,7 @@ class FSMUpdater:
 | 
				
			|||||||
    @property
 | 
					    @property
 | 
				
			||||||
    def fsm_update(self):
 | 
					    def fsm_update(self):
 | 
				
			||||||
        return {
 | 
					        return {
 | 
				
			||||||
            'key': 'fsm.announce',
 | 
					            'key': 'fsm.update',
 | 
				
			||||||
            **self.fsm_update_data
 | 
					            **self.fsm_update_data
 | 
				
			||||||
        }
 | 
					        }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -36,7 +36,7 @@ class LogInotifyObserver(InotifyObserver, ProcessLogManager):
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
    def on_modified(self, event):
 | 
					    def on_modified(self, event):
 | 
				
			||||||
        self._connector.send_message({
 | 
					        self._connector.send_message({
 | 
				
			||||||
            'key': 'log.new',
 | 
					            'key': 'process.log.new',
 | 
				
			||||||
            'stdout': self.read_stdout(self.process_name, tail=self.log_tail),
 | 
					            'stdout': self.read_stdout(self.process_name, tail=self.log_tail),
 | 
				
			||||||
            'stderr': self.read_stderr(self.process_name, tail=self.log_tail)
 | 
					            'stderr': self.read_stderr(self.process_name, tail=self.log_tail)
 | 
				
			||||||
        }, Scope.BROADCAST)
 | 
					        }, Scope.BROADCAST)
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -6,7 +6,7 @@ LOG = logging.getLogger(__name__)
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
class ProcessLogHandler:
 | 
					class ProcessLogHandler:
 | 
				
			||||||
    keys = ['log.set']
 | 
					    keys = ['process.log.set']
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    def __init__(self, *, process_name, supervisor_uri, log_tail=0):
 | 
					    def __init__(self, *, process_name, supervisor_uri, log_tail=0):
 | 
				
			||||||
        self.connector, self._monitor = None, None
 | 
					        self.connector, self._monitor = None, None
 | 
				
			||||||
@@ -15,7 +15,7 @@ class ProcessLogHandler:
 | 
				
			|||||||
        self._initial_log_tail = log_tail
 | 
					        self._initial_log_tail = log_tail
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        self.command_handlers = {
 | 
					        self.command_handlers = {
 | 
				
			||||||
            'log.set': self.handle_set
 | 
					            'process.log.set': self.handle_set
 | 
				
			||||||
        }
 | 
					        }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    def start(self):
 | 
					    def start(self):
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -6,11 +6,7 @@ LOG = logging.getLogger(__name__)
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
class FSMAware:
 | 
					class FSMAware:
 | 
				
			||||||
    keys = ['fsm.announce']
 | 
					    keys = ['fsm.update']
 | 
				
			||||||
    """
 | 
					 | 
				
			||||||
    Base class for stuff that has to be aware of the framework FSM.
 | 
					 | 
				
			||||||
    This is done by processing 'fsm_update' messages.
 | 
					 | 
				
			||||||
    """
 | 
					 | 
				
			||||||
    def __init__(self):
 | 
					    def __init__(self):
 | 
				
			||||||
        self.fsm_state = None
 | 
					        self.fsm_state = None
 | 
				
			||||||
        self.fsm_in_accepted_state = False
 | 
					        self.fsm_in_accepted_state = False
 | 
				
			||||||
@@ -18,7 +14,7 @@ class FSMAware:
 | 
				
			|||||||
        self._auth_key = KeyManager().auth_key
 | 
					        self._auth_key = KeyManager().auth_key
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    def process_message(self, message):
 | 
					    def process_message(self, message):
 | 
				
			||||||
        if message['key'] == 'fsm.announce':
 | 
					        if message['key'] == 'fsm.update':
 | 
				
			||||||
            if verify_message(self._auth_key, message):
 | 
					            if verify_message(self._auth_key, message):
 | 
				
			||||||
                self._handle_fsm_update(message)
 | 
					                self._handle_fsm_update(message)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 
 | 
				
			|||||||
		Reference in New Issue
	
	Block a user