Colorize shell output based on log level

Author: cburgdorf

setup.py

@@ -44,6 +44,7 @@
         "plyvel==1.0.5",
         "web3==4.4.1",
         "lahja==0.9.0",
+        "termcolor>=1.1.0,<2.0.0",
         "uvloop==0.11.2;platform_system=='Linux' or platform_system=='Darwin'",
         "websockets==5.0.1",
     ],

trinity/utils/logging.py

@@ -28,6 +28,11 @@
     TraceLogger,
 )
 
+from trinity.utils.shellart import (
+    bold_red,
+    bold_yellow,
+)
+
 if TYPE_CHECKING:
     from multiprocessing import Queue  # noqa: F401
 
@@ -42,7 +47,13 @@ def __init__(self, fmt: str, datefmt: str) -> None:
 
     def format(self, record: logging.LogRecord) -> str:
         record.shortname = record.name.split('.')[-1]  # type: ignore
-        return super().format(record)
+
+        if record.levelno >= logging.ERROR:
+            return bold_red(super().format(record))
+        elif record.levelno >= logging.WARNING:
+            return bold_yellow(super().format(record))
+        else:
+            return super().format(record)
 
 
 class HasTraceLogger:

trinity/utils/shellart.py

@@ -0,0 +1,19 @@
+from termcolor import (
+    colored
+)
+
+
+def bold_green(txt: str) -> str:
+    return colored(txt, 'green', attrs=['bold'])
+
+
+def bold_red(txt: str) -> str:
+    return colored(txt, 'red', attrs=['bold'])
+
+
+def bold_yellow(txt: str) -> str:
+    return colored(txt, 'yellow', attrs=['bold'])
+
+
+def bold_white(txt: str) -> str:
+    return colored(txt, 'white', attrs=['bold'])