prettify logs with collors

Author: duartecoelhomovai

movai_core_shared/consts.py

@@ -7,6 +7,7 @@
    - Dor Marcous ([email protected]) - 2022
 """
 from os import getpid
+import logging
 
 # system
 PID = getpid()
@@ -148,3 +149,20 @@
 METRICS_QUERY_HANDLER_MSG_TYPE = "metrics_query"
 NOTIFICATIONS_HANDLER_MSG_TYPE = "notifications"
 ALERTS_HANDLER_METRIC_TYPE = "alerts"
+
+CALLBACK_STDOUT_COLORS =  {
+        logging.DEBUG: "\033[36m",
+        logging.INFO: "\u001b[0m",
+        logging.WARNING: "\x1b[33m",
+        logging.ERROR: "\x1b[31;20m",
+        logging.CRITICAL: "\x1b[41;1m",
+        logging.WARN: "\x1b[33m",
+}
+
+SPAWNER_STDOUT_COLORS = {
+        logging.DEBUG: "\x1b[1;34m",
+        logging.INFO: "\x1b[1;37m",
+        logging.WARNING: "\x1b[1;33m",
+        logging.ERROR: "\x1b[31;1m",
+        logging.CRITICAL: "\x1b[41;1m",
+}

movai_core_shared/envvars.py

@@ -1,6 +1,6 @@
 """ Compilation of necessary environment variables to push to the database """
 import os
-from logging import NOTSET, INFO, DEBUG, WARNING, FATAL, CRITICAL, ERROR, getLevelName
+from logging import DEBUG, getLevelName
 import socket
 from .consts import (
     MESSAGE_SERVER_HOST,

movai_core_shared/logger.py

@@ -31,6 +31,8 @@
     PID,
     USER_LOG_TAG,
     CALLBACK_LOGGER,
+    CALLBACK_STDOUT_COLORS,
+    SPAWNER_STDOUT_COLORS
 )
 from movai_core_shared.envvars import (
     DEVICE_NAME,
@@ -70,17 +72,12 @@
 logging.getLogger('rosout').setLevel(MOVAI_CALLBACK_VERBOSITY_LEVEL)
 
 class StdOutHandler(logging.StreamHandler):
-    _COLORS = {
-        logging.DEBUG: "\x1b[30;1m",  # light black (gray)
-        logging.INFO: "",  # default (white)
-        logging.WARNING: "\x1b[33;1m",  # yellow
-        logging.ERROR: "\x1b[31;1m",  # red
-        logging.CRITICAL: "\x1b[41;1m",  # bright red
-    }
+    _COLORS = SPAWNER_STDOUT_COLORS
     _COLOR_RESET = "\u001b[0m"
 
-    def __init__(self, stream=None):
+    def __init__(self, color=SPAWNER_STDOUT_COLORS, stream=None):
         super().__init__(stream)
+        self._COLORS = color
 
     def emit(self, record):
         try:
@@ -119,7 +116,8 @@ def emit(self, record):
             self.flush()
         except Exception:
             self.handleError(record)
-
+logging.getLogger('rosout').addHandler(StdOutHandler(color=CALLBACK_STDOUT_COLORS, stream=sys.stdout))
+logging.getLogger('rosout').propagate = False
 
 class RemoteHandler(logging.StreamHandler):
     """