runner: Add coloured logging

Author: WizardCM

runner.py

@@ -8,6 +8,11 @@
 from obsbot import __version__, __codename__
 from obsbot.main import OBSBot
 
+RESET = "\x1b[0m"
+WARNING = "\x1b[33;20m%s" + RESET
+ERROR = "\x1b[31;20m%s" + RESET
+DEBUG = "\x1b[36;20m%s" + RESET
+
 _log_dt_fmt = '%Y-%m-%d %H:%M:%S'
 _log_fmt = '[{asctime}] [{levelname}] {name}: {message}'
 logging.basicConfig(format=_log_fmt, datefmt=_log_dt_fmt, style='{', level=logging.INFO)
@@ -18,6 +23,9 @@
 def setup_logging(logfile=None, debug=False):
     try:
         max_bytes = 32 * 1024 * 1024  # 32 MiB
+        logging.addLevelName(logging.WARNING, WARNING % logging.getLevelName(logging.WARNING))
+        logging.addLevelName(logging.ERROR, ERROR % logging.getLevelName(logging.ERROR))
+        logging.addLevelName(logging.DEBUG, DEBUG % logging.getLevelName(logging.DEBUG))
         logging.getLogger('discord').setLevel(logging.WARNING)
         logging.getLogger('discord.http').setLevel(logging.WARNING)