Fix no color issue

login-securite · Jun 30, 2022 · ec193c0 · ec193c0
1 parent ecd5fda
commit ec193c0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lsassy/logger.py b/lsassy/logger.py
@@ -50,7 +50,7 @@ def highlight(msg):
     :param msg: Message to highlight
     :return: Highlighted message
     """
-    if logging.getFormatter().no_color:
+    if logging.no_color:
         return msg
     return "\033[1;33m{}\033[0m".format(msg)
 
@@ -66,4 +66,5 @@ def init(quiet=False, no_color=False):
 
     logging.addLevelName(25, 'SUCCESS')
     setattr(logging, 'success', lambda message, *args: logging.getLogger()._log(25, message, args))
+    setattr(logging, 'no_color', no_color)
     logging.getLogger().disabled = quiet