adding in style selection to the logging formatter

Author: joseph-flinn

tmuxp/log.py

@@ -31,7 +31,16 @@
 }
 
 
-def default_log_template(self, record):
+def set_style(
+    message, stylized, style_before=None, style_after=None, prefix='', suffix=''
+):
+    if stylized:
+        return prefix + style_before + message + style_after + suffix
+
+    return prefix + message + suffix
+
+
+def default_log_template(self, record, stylized=False):
     """
     Return the prefix for the log message. Template for Formatter.
 
@@ -48,37 +57,34 @@ def default_log_template(self, record):
     """
 
     reset = Style.RESET_ALL
-    levelname = (
-        LEVEL_COLORS.get(record.levelname)
-        + Style.BRIGHT
-        + '(%(levelname)s)'
-        + Style.RESET_ALL
-        + ' '
+    levelname = set_style(
+        '(%(levelname)s)',
+        stylized,
+        style_before=(LEVEL_COLORS.get(record.levelname) + Style.BRIGHT),
+        style_after=Style.RESET_ALL,
+        suffix=' ',
     )
-    asctime = (
-        '['
-        + Fore.BLACK
-        + Style.DIM
-        + Style.BRIGHT
-        + '%(asctime)s'
-        + Fore.RESET
-        + Style.RESET_ALL
-        + ']'
+    asctime = set_style(
+        '%(asctime)s',
+        stylized,
+        style_before=(Fore.BLACK + Style.DIM + Style.BRIGHT),
+        style_after=(Fore.RESET + Style.RESET_ALL),
+        prefix='[',
+        suffix=']',
     )
-    name = (
-        ' '
-        + Fore.WHITE
-        + Style.DIM
-        + Style.BRIGHT
-        + '%(name)s'
-        + Fore.RESET
-        + Style.RESET_ALL
-        + ' '
+    name = set_style(
+        '%(name)s',
+        stylized,
+        style_before=(Fore.WHITE + Style.DIM + Style.BRIGHT),
+        style_after=(Fore.RESET + Style.RESET_ALL),
+        prefix=' ',
+        suffix=' ',
     )
 
-    tpl = reset + levelname + asctime + name + reset
+    if stylized:
+        return reset + levelname + asctime + name + reset
 
-    return tpl
+    return levelname + asctime + name
 
 
 class LogFormatter(logging.Formatter):