mrtoth/baseimage-tutorial-framework
1
0
Fork 0
mirror of https://github.com/avatao-content/baseimage-tutorial-framework synced 2025-11-04 10:02:55 +00:00
Code Issues Releases Wiki Activity

Implement signed event handler

Browse Source
This commit is contained in:
R. Richard
2019-08-05 13:47:05 +02:00
parent 0a218c8d6d
commit e87a16991c
4 changed files with 24 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tfw/components/fsm/fsm_handler.py
View File

@@ -24,11 +24,10 @@ class FSMHandler:
An 'fsm_update' message is broadcasted after every successful
command.
"""
def __init__(self, *, fsm_type, require_signature=False):
def __init__(self, *, fsm_type):
self.fsm = fsm_type()
self._fsm_updater = FSMUpdater(self.fsm)
self.auth_key = KeyManager().auth_key
self._require_signature = require_signature
self.command_handlers = {
'trigger': self.handle_trigger,
@@ -54,10 +53,6 @@ class FSMHandler:
the action to try triggering in data['value']
"""
trigger = message['data']['value']
if self._require_signature:
if not verify_message(self.auth_key, message):
LOG.error('Ignoring unsigned trigger command: %s', message)
return None
if self.fsm.step(trigger):
return message
return None