mirror of
https://github.com/avatao-content/baseimage-tutorial-framework
synced 2024-11-05 14:41:20 +00:00
Move FSMAware to a separate file in networking
This commit is contained in:
parent
fe79b598a7
commit
5715c57ebc
@ -3,8 +3,9 @@
|
|||||||
|
|
||||||
from abc import ABC, abstractmethod
|
from abc import ABC, abstractmethod
|
||||||
|
|
||||||
|
from tfw.networking import FSMAware
|
||||||
from tfw.networking.event_handlers import ServerConnector
|
from tfw.networking.event_handlers import ServerConnector
|
||||||
from tfw.crypto import message_checksum, KeyManager, verify_message
|
from tfw.crypto import message_checksum
|
||||||
from tfw.config.logs import logging
|
from tfw.config.logs import logging
|
||||||
|
|
||||||
LOG = logging.getLogger(__name__)
|
LOG = logging.getLogger(__name__)
|
||||||
@ -106,37 +107,6 @@ class EventHandlerBase(ABC):
|
|||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
class FSMAware:
|
|
||||||
def __init__(self):
|
|
||||||
self.fsm_state = None
|
|
||||||
self.fsm_in_accepted_state = False
|
|
||||||
self._auth_key = KeyManager().auth_key
|
|
||||||
|
|
||||||
def update_fsm_data(self, message):
|
|
||||||
if message['key'] == 'fsm_update' and verify_message(self._auth_key, message):
|
|
||||||
self._handle_fsm_update(message)
|
|
||||||
return True
|
|
||||||
return False
|
|
||||||
|
|
||||||
def _handle_fsm_update(self, message):
|
|
||||||
try:
|
|
||||||
new_state = message['data']['current_state']
|
|
||||||
if self.fsm_state != new_state:
|
|
||||||
self.handle_fsm_step(**(message['data']))
|
|
||||||
self.fsm_state = new_state
|
|
||||||
self.fsm_in_accepted_state = message['data']['in_accepted_state']
|
|
||||||
except KeyError:
|
|
||||||
LOG.error('Invalid fsm_update message received!')
|
|
||||||
|
|
||||||
def handle_fsm_step(self, **kwargs):
|
|
||||||
"""
|
|
||||||
Called in case the TFW FSM has stepped.
|
|
||||||
|
|
||||||
:param kwargs: fsm_update 'data' field
|
|
||||||
"""
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
class FSMAwareEventHandler(EventHandlerBase, FSMAware, ABC):
|
class FSMAwareEventHandler(EventHandlerBase, FSMAware, ABC):
|
||||||
# pylint: disable=abstract-method
|
# pylint: disable=abstract-method
|
||||||
"""
|
"""
|
||||||
|
@ -7,3 +7,4 @@ from .zmq_connector_base import ZMQConnectorBase
|
|||||||
from .message_sender import MessageSender
|
from .message_sender import MessageSender
|
||||||
from .event_handlers.server_connector import ServerUplinkConnector as TFWServerConnector
|
from .event_handlers.server_connector import ServerUplinkConnector as TFWServerConnector
|
||||||
from .server.tfw_server import TFWServer
|
from .server.tfw_server import TFWServer
|
||||||
|
from .fsm_aware import FSMAware
|
||||||
|
45
lib/tfw/networking/fsm_aware.py
Normal file
45
lib/tfw/networking/fsm_aware.py
Normal file
@ -0,0 +1,45 @@
|
|||||||
|
# Copyright (C) 2018 Avatao.com Innovative Learning Kft.
|
||||||
|
# All Rights Reserved. See LICENSE file for details.
|
||||||
|
|
||||||
|
from tfw.crypto import KeyManager, verify_message
|
||||||
|
|
||||||
|
from tfw.config.logs import logging
|
||||||
|
|
||||||
|
LOG = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
|
class FSMAware:
|
||||||
|
"""
|
||||||
|
Base class for stuff that has to be aware of the framework FSM.
|
||||||
|
This is done by processing 'fsm_update' messages.
|
||||||
|
"""
|
||||||
|
def __init__(self):
|
||||||
|
self.fsm_state = None
|
||||||
|
self.fsm_in_accepted_state = False
|
||||||
|
self.fsm_last_update = None
|
||||||
|
self._auth_key = KeyManager().auth_key
|
||||||
|
|
||||||
|
def update_fsm_data(self, message):
|
||||||
|
if message['key'] == 'fsm_update' and verify_message(self._auth_key, message):
|
||||||
|
self._handle_fsm_update(message)
|
||||||
|
return True
|
||||||
|
return False
|
||||||
|
|
||||||
|
def _handle_fsm_update(self, message):
|
||||||
|
try:
|
||||||
|
new_state = message['data']['current_state']
|
||||||
|
if self.fsm_state != new_state:
|
||||||
|
self.handle_fsm_step(**(message['data']))
|
||||||
|
self.fsm_state = new_state
|
||||||
|
self.fsm_in_accepted_state = message['data']['in_accepted_state']
|
||||||
|
self.fsm_last_update = message['data']
|
||||||
|
except KeyError:
|
||||||
|
LOG.error('Invalid fsm_update message received!')
|
||||||
|
|
||||||
|
def handle_fsm_step(self, **kwargs):
|
||||||
|
"""
|
||||||
|
Called in case the TFW FSM has stepped.
|
||||||
|
|
||||||
|
:param kwargs: fsm_update 'data' field
|
||||||
|
"""
|
||||||
|
pass
|
Loading…
Reference in New Issue
Block a user