mirror of
https://github.com/avatao-content/test-tutorial-framework
synced 2024-11-15 07:17:17 +00:00
102 lines
3.4 KiB
Python
102 lines
3.4 KiB
Python
from ast import literal_eval
|
|
|
|
from tornado.ioloop import IOLoop
|
|
|
|
from tfw.components import IdeEventHandler, TerminalEventHandler, ProcessManagingEventHandler, BashMonitor
|
|
from tfw.components import TerminalCommands, LogMonitoringEventHandler
|
|
from tfw.networking import MessageSender, TFWServerConnector
|
|
from tfw.config import TFWENV
|
|
from tfw.config.logs import logging
|
|
from tao.config import TAOENV
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
|
|
def cenator(history):
|
|
LOG.debug('User executed command: "%s"', history[-1])
|
|
MessageSender().send('JOHN CENA', f'You\'ve executed "{history[-1]}"')
|
|
|
|
|
|
class TestCommands(TerminalCommands):
|
|
"""
|
|
Some example commands useful for debugging.
|
|
Please remove from production code and inherit your own
|
|
class from TerminalCommands if you need to define custom
|
|
commands in your challenge.
|
|
"""
|
|
# pylint: disable=unused-argument, attribute-defined-outside-init, no-self-use
|
|
def command_sendmessage(self, *args):
|
|
"""
|
|
Insert TFW message template as first argument if executed without args.
|
|
|
|
Evaluate first argumen as a dict and send it to the frontend.
|
|
This is useful for playing around with frontend APIs.
|
|
"""
|
|
if not args:
|
|
message_template = """'{"key": "", "data": {"command": ""}}'"""
|
|
TFWServerConnector().send_to_eventhandler({
|
|
'key': 'shell',
|
|
'data': {
|
|
'command': 'write',
|
|
'shellcmd': f'sendmessage {message_template}'
|
|
}
|
|
})
|
|
else:
|
|
TFWServerConnector().send(literal_eval(args[0]))
|
|
|
|
def command_seppuku_tfw(self, *args):
|
|
"""
|
|
Restart tfw_server.py and event_handler_main.py.
|
|
This can speed up development when combined with mounting
|
|
volumes from host to container.
|
|
"""
|
|
seppuku = ('nohup sh -c "supervisorctl restart tfwserver event_handler_main" &> /dev/null & '
|
|
'clear && echo "Committed seppuku! :)" && sleep infinity')
|
|
uplink = TFWServerConnector()
|
|
uplink.send_to_eventhandler({
|
|
'key': 'shell',
|
|
'data': {
|
|
'command': 'write',
|
|
'shellcmd': f'{seppuku}\n'
|
|
}
|
|
})
|
|
uplink.send({
|
|
'key': 'dashboard',
|
|
'data': {
|
|
'command': 'reload_frontend'
|
|
}
|
|
})
|
|
|
|
|
|
if __name__ == '__main__':
|
|
ide = IdeEventHandler( # Web IDE backend
|
|
key='ide', allowed_directories=[TFWENV.IDE_WD, TFWENV.WEBSERVICE_DIR],
|
|
directory=TFWENV.IDE_WD, exclude=['*.pyc'],
|
|
additional_watched_directories=[TFWENV.WEBSERVICE_DIR]
|
|
)
|
|
terminal = TerminalEventHandler( # Web shell backend
|
|
key='shell',
|
|
monitor=BashMonitor(TFWENV.HISTFILE)
|
|
)
|
|
processmanager = ProcessManagingEventHandler( # Handles 'deploy' button clicks
|
|
key='processmanager',
|
|
dirmonitor=ide.monitor,
|
|
log_tail=2000
|
|
)
|
|
logmonitor = LogMonitoringEventHandler( # Sends live logs of webservice process to frontend
|
|
'webservice',
|
|
log_tail=2000
|
|
)
|
|
eventhandlers = {ide, terminal, processmanager, logmonitor}
|
|
|
|
terminal.historymonitor.subscribe_callback(cenator)
|
|
|
|
commands = TestCommands(bashrc=f'/home/{TAOENV.USER}/.bashrc')
|
|
terminal.historymonitor.subscribe_callback(commands.callback)
|
|
|
|
try:
|
|
IOLoop.instance().start()
|
|
finally:
|
|
for eh in eventhandlers:
|
|
eh.cleanup()
|