From 933ffd833cdf284cffe1c531417c58cad9d40253 Mon Sep 17 00:00:00 2001
From: InsanePrawn <insane.prawny@gmail.com>
Date: Sat, 7 May 2022 15:27:55 +0200
Subject: [PATCH] logger: add colors

---
 logger.py        | 11 ++++++++---
 requirements.txt |  1 +
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/logger.py b/logger.py
index fc79ac4..b594d87 100644
--- a/logger.py
+++ b/logger.py
@@ -1,15 +1,20 @@
 import click
+import coloredlogs
 import logging
 import sys
 
 
 def setup_logging(verbose: bool):
+    level_colors = coloredlogs.DEFAULT_LEVEL_STYLES | {'info': {'color': 'magenta', 'bright': True}, 'debug': {'color': 'blue', 'bright': True}}
+    field_colors = coloredlogs.DEFAULT_FIELD_STYLES | {'asctime': {'color': 'white', 'faint': True}}
     level = logging.DEBUG if verbose else logging.INFO
-    logging.basicConfig(
+    coloredlogs.install(
         stream=sys.stdout,
-        format='%(asctime)s %(levelname)s: %(message)s',
-        datefmt='%m/%d/%Y %H:%M:%S',
+        fmt='%(asctime)s %(levelname)s: %(message)s',
+        datefmt='%Y-%m-%d %H:%M:%S',
         level=level,
+        level_styles=level_colors,
+        field_styles=field_colors,
     )
     logging.debug('Logging set up.')
 
diff --git a/requirements.txt b/requirements.txt
index 2b3138e..2cd3a8a 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,3 +3,4 @@ appdirs==1.4.4
 joblib==1.0.1
 toml
 typing_extensions
+coloredlogs