Added single highlight from args

geirawsm · May 9, 2024 · 245ea12 · 245ea12
1 parent 3479546
commit 245ea12
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/sausage_bot/util/args.py b/sausage_bot/util/args.py
@@ -54,6 +54,11 @@
                           action='store',
                           default=None,
                           dest='log_highlight')
+logging_args.add_argument('--highlight-color', '-hlc',
+                          help='Chose color to Highlight',
+                          action='store',
+                          default='red',
+                          dest='log_highlight_color')
 logging_args.add_argument('--log-all',
                           help='Log all levels (log, verbose, log-print, '
                                'log-database and debug)',

diff --git a/sausage_bot/util/log/log.py b/sausage_bot/util/log/log.py
@@ -53,9 +53,12 @@ def log_function(
             extra_color = eval('Fore.{}'.format(extra_color.upper()))
     function_name = log_func_name()
     if args.log_print:
-        if args.log_highlight is not None and str(args.log_highlight)\
-                in function_name['name']:
-            color = Fore.RED
+        if args.log_highlight is not None:
+            if str(args.log_highlight) in function_name['name'] or\
+                    str(args.log_highlight) in log_in:
+                color = eval('Fore.{}'.format(
+                    args.log_highlight_color.upper()
+                ))
     dt = pendulum.now(config.TIMEZONE)
     _dt_full = dt.format('DD.MM.YYYY HH.mm.ss')
     if args.log_print: