43 lines
974 B
Python
43 lines
974 B
Python
# iottb/utils/user_interaction.py
|
|
|
|
import click
|
|
from iottb.definitions import TB_ECHO_STYLES
|
|
import sys
|
|
import os
|
|
|
|
|
|
def tb_echo2(msg: str, lvl='i', log=True):
|
|
style = TB_ECHO_STYLES.get(lvl, {})
|
|
click.secho(f'[IOTTB]', **style)
|
|
click.secho(f'[IOTTB] \t {msg}', **style)
|
|
|
|
|
|
last_prefix = None
|
|
|
|
|
|
def tb_echo(msg: str, lvl='i', log=True):
|
|
global last_prefix
|
|
prefix = f'Testbed [{lvl.upper()}]\n'
|
|
|
|
if last_prefix != prefix:
|
|
click.secho(prefix, nl=False, **TB_ECHO_STYLES['header'])
|
|
last_prefix = prefix
|
|
|
|
click.secho(f' {msg}', **TB_ECHO_STYLES[lvl])
|
|
|
|
|
|
def main():
|
|
tb_echo('Info message', 'i')
|
|
tb_echo('Warning message', 'w')
|
|
tb_echo('Error message', 'e')
|
|
tb_echo('Success message', 's')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
# arrrgggg hacky
|
|
current_dir = os.path.dirname(os.path.abspath(__file__))
|
|
project_root = os.path.abspath(os.path.join(current_dir, '../../'))
|
|
sys.path.insert(0, project_root)
|
|
|
|
main()
|