From 6d7f46fabe530e4cafdffa25d7fced730965e481 Mon Sep 17 00:00:00 2001
From: Numerlor <25886452+Numerlor@users.noreply.github.com>
Date: Tue, 7 Sep 2021 00:15:36 +0200
Subject: Add coloredlogs

---
 bot/log.py | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

(limited to 'bot/log.py')

diff --git a/bot/log.py b/bot/log.py
index 9ae40984..d44e6933 100644
--- a/bot/log.py
+++ b/bot/log.py
@@ -1,7 +1,11 @@
 import logging
 import logging.handlers
+import os
+import sys
 from pathlib import Path
 
+import coloredlogs
+
 from bot.constants import Client
 
 
@@ -23,6 +27,19 @@ def setup() -> None:
     # Console handler prints to terminal
     console_handler = logging.StreamHandler()
 
+    if "COLOREDLOGS_LEVEL_STYLES" not in os.environ:
+        coloredlogs.DEFAULT_LEVEL_STYLES = {
+            **coloredlogs.DEFAULT_LEVEL_STYLES,
+            "trace": {"color": 246},
+            "critical": {"background": "red"},
+            "debug": coloredlogs.DEFAULT_LEVEL_STYLES["info"],
+        }
+
+    if "COLOREDLOGS_LOG_FORMAT" not in os.environ:
+        coloredlogs.DEFAULT_LOG_FORMAT = "%(asctime)s - %(name)s %(levelname)s: %(message)s"
+
+    coloredlogs.install(stream=sys.stdout)
+
     # Silence irrelevant loggers
     logging.getLogger("discord").setLevel(logging.ERROR)
     logging.getLogger("websockets").setLevel(logging.ERROR)
@@ -32,7 +49,6 @@ def setup() -> None:
 
     # Setup new logging configuration
     logging.basicConfig(
-        format="%(asctime)s - %(name)s %(levelname)s: %(message)s",
         datefmt="%D %H:%M:%S",
         level=logging.TRACE if Client.debug else logging.INFO,
         handlers=[console_handler, file_handler],
-- 
cgit v1.2.3