Merge pull request #22 from State-of-The-MLOps/feature/color_logger

Add color to logger
State-of-The-MLOps · Sep 23, 2021 · c48e45a · c48e45a
2 parents 36a465f + 648f72f
commit c48e45a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/logger.py b/logger.py
@@ -1,11 +1,12 @@
 import logging
 import logging.handlers
+from colorlog import ColoredFormatter
 
 L = logging.getLogger('snowdeer_log')
 L.setLevel(logging.DEBUG)
 
-formatter = logging.Formatter(
-    fmt="%(asctime)s [%(levelname)s] [%(filename)s:%(lineno)d -  %(funcName)20s()]\n\t%(message)s",
+formatter = ColoredFormatter(
+    fmt="%(log_color)s [%(levelname)s] %(reset)s %(asctime)s [%(filename)s:%(lineno)d -  %(funcName)20s()]\n\t%(message)s",
     datefmt='%y-%m-%d %H:%M:%S'
 )
 

diff --git a/requirements.txt b/requirements.txt