prettify output for gs, ri and errs requests

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)