diff options
Diffstat (limited to 'inverter-bot')
-rwxr-xr-x | inverter-bot | 13 |
1 files changed, 10 insertions, 3 deletions
diff --git a/inverter-bot b/inverter-bot index edae9a5..6f8c40b 100755 --- a/inverter-bot +++ b/inverter-bot @@ -90,6 +90,13 @@ def handle_exc(update: Update, e) -> None: elif not isinstance(e, TimedOut): reply(update, 'exception: ' + str(e)) + +def beautify_table(s): + lines = s.split('\n') + lines = list(map(lambda line: re.sub(r'\s+', ' ', line), lines)) + lines = list(map(lambda line: re.sub(r'(.*?): (.*)', r'<b>\1:</b> \2', line), lines)) + return '\n'.join(lines) + # # command/message handlers # @@ -176,7 +183,7 @@ def msg_generation(update: Update, context: CallbackContext) -> None: def msg_gs(update: Update, context: CallbackContext) -> None: try: status = inverter.exec('get-status', format=Format.TABLE) - reply(update, status) + reply(update, beautify_table(status)) except Exception as e: handle_exc(update, e) @@ -184,7 +191,7 @@ def msg_gs(update: Update, context: CallbackContext) -> None: def msg_ri(update: Update, context: CallbackContext) -> None: try: rated = inverter.exec('get-rated', format=Format.TABLE) - reply(update, rated) + reply(update, beautify_table(rated)) except Exception as e: handle_exc(update, e) @@ -192,7 +199,7 @@ def msg_ri(update: Update, context: CallbackContext) -> None: def msg_errors(update: Update, context: CallbackContext) -> None: try: errors = inverter.exec('get-errors', format=Format.TABLE) - reply(update, errors) + reply(update, beautify_table(errors)) except Exception as e: handle_exc(update, e) |