baseimage-tutorial-framework/tfw/components/frontend/frontend_handler.py
2019-08-09 10:56:38 +02:00

30 lines
918 B
Python

from tfw.internals.networking import Scope
from .message_storage import FrontendMessageStorage
class FrontendHandler:
keys = ['console', 'dashboard', 'message', 'ide.read', 'recover']
def __init__(self):
self.connector = None
self._frontend_message_storage = FrontendMessageStorage()
def send_message(self, message):
self.connector.send_message(message, scope=Scope.WEBSOCKET)
def handle_event(self, message, _):
self._frontend_message_storage.save_message(message)
if message['key'] == 'recover':
self.recover_frontend()
if self._filter_message(message):
self.send_message(message)
@staticmethod
def _filter_message(message):
return not message['key'].startswith('ide')
def recover_frontend(self):
for message in self._frontend_message_storage.messages:
self.send_message(message)