mirror of
https://github.com/avatao-content/baseimage-tutorial-framework
synced 2024-09-29 12:31:34 +00:00
57 lines
1.5 KiB
Python
57 lines
1.5 KiB
Python
from abc import ABC, abstractmethod
|
|
from contextlib import suppress
|
|
|
|
from .message_sender import MessageSender
|
|
|
|
|
|
class MessageStorage(ABC):
|
|
def __init__(self):
|
|
self._messages = []
|
|
|
|
def save_message(self, message):
|
|
with suppress(KeyError, AttributeError):
|
|
if self._filter_message(message):
|
|
self._messages.extend(self._transform_message(message))
|
|
|
|
@abstractmethod
|
|
def _filter_message(self, message):
|
|
raise NotImplementedError
|
|
|
|
def _transform_message(self, message): # pylint: disable=no-self-use
|
|
yield message
|
|
|
|
def clear(self):
|
|
self._messages.clear()
|
|
|
|
@property
|
|
def messages(self):
|
|
yield from self._messages
|
|
|
|
|
|
class FrontendMessageStorage(MessageStorage):
|
|
def _filter_message(self, message):
|
|
return message['key'].startswith((
|
|
'console.write',
|
|
'dashboard.layout',
|
|
'dashboard.terminalMenuItem',
|
|
'message.config',
|
|
'message.send',
|
|
'message.queue',
|
|
'ide.read'
|
|
))
|
|
|
|
def _transform_message(self, message):
|
|
transformations = {
|
|
'message.queue': MessageSender.generate_messages_from_queue,
|
|
'ide.read': self._delete_ide_content
|
|
}
|
|
if message['key'] in transformations:
|
|
yield from transformations[message['key']](message)
|
|
else:
|
|
yield message
|
|
|
|
@staticmethod
|
|
def _delete_ide_content(message):
|
|
del message['content']
|
|
yield message
|