baseimage-tutorial-framework/tfw/components/frontend/frontend_proxy_handler.py

30 lines
973 B
Python
Raw Normal View History

from tfw.internals.networking import Scope
2019-05-26 16:26:33 +00:00
from .message_storage import FrontendMessageStorage
2019-06-27 12:36:18 +00:00
class FrontendProxyHandler:
keys = ['console', 'dashboard', 'frontend.ready', 'message', 'ide.read', 'deploy.finish']
2019-05-26 16:26:33 +00:00
def __init__(self):
self.connector = None
2019-08-08 13:05:37 +00:00
self._frontend_message_storage = FrontendMessageStorage()
2019-05-26 16:26:33 +00:00
def send_message(self, message):
self.connector.send_message(message, scope=Scope.WEBSOCKET)
def handle_event(self, message, _):
2019-05-26 16:26:33 +00:00
self._frontend_message_storage.save_message(message)
2019-08-15 13:36:57 +00:00
if message['key'] == 'frontend.ready':
2019-05-26 16:26:33 +00:00
self.recover_frontend()
2019-08-08 13:05:37 +00:00
if self._filter_message(message):
self.send_message(message)
@staticmethod
def _filter_message(message):
2019-08-23 14:15:27 +00:00
return not message['key'].startswith(('ide', 'message.queue'))
2019-05-26 16:26:33 +00:00
def recover_frontend(self):
for message in self._frontend_message_storage.messages:
self.send_message(message)